lstoll-bumble 0.0.1

data/lib/bumble.rb

@@ -0,0 +1 @@
+require File.dirname(__FILE__) + '/bumble/bumble'

data/lib/bumble/bumble.rb

@@ -0,0 +1,205 @@
+require 'java'
+require 'time'
+
+module Bumble
+  module DS
+    import com.google.appengine.api.datastore.DatastoreServiceFactory
+    import com.google.appengine.api.datastore.Entity
+    import com.google.appengine.api.datastore.FetchOptions
+    import com.google.appengine.api.datastore.KeyFactory
+    import com.google.appengine.api.datastore.Key
+    import com.google.appengine.api.datastore.EntityNotFoundException
+    import com.google.appengine.api.datastore.Query
+    import com.google.appengine.api.datastore.Text
+    Service = DatastoreServiceFactory.datastore_service
+  end
+
+  module InstanceMethods
+    def initialize attrs = {}, *args
+      super *args
+      @__entity = DS::Entity.new self.class.name
+      @new_record = true
+      attrs.each do |k,v|
+        self.send "#{k}=", v
+      end
+    end
+
+    def key
+      __ds_key.get_id
+    end
+    
+    def to_param
+      self.key
+    end
+    
+    def save!
+      if @new_record
+        self.created_at = Time.now if respond_to? :created_at=
+        @new_record = false
+      end
+      self.updated_at = Time.now if respond_to? :updated_at=
+      DS::Service.put(@__entity)
+    end
+    
+    def delete!
+      self.class.delete key
+    end
+
+    def properties
+      res = {'key' => key}
+      @__entity.get_properties.each do |k,v|
+        res[k] = v.is_a?(DS::Text) ? v.value : v
+      end
+      res
+    end
+
+    def == object
+      object.respond_to?(:properties) && object.properties == properties
+    end
+
+    def __ds_key
+      @__entity.key
+    end
+    
+    def __ds_get name
+      name = name.to_s
+      if @__entity.has_property(name)
+        ret = @__entity.get_property(name)
+        return ret.value if ret.is_a? DS::Text
+        return Time.parse ret.to_s if ret.is_a? Java::JavaUtil::Date
+        ret
+      end
+    end
+    
+    def __ds_set name, value
+      value = DS::Text.new value if value.is_a?(String) && value.length > 499
+      @__entity.set_property name.to_s, value
+    end
+  end
+
+  module ClassMethods
+    def has_many(attr, type, key = nil, options = {})
+      key ||= "#{self.to_s.downcase}_id"
+      type_name = (type.is_a?(Symbol) || type.is_a?(String)) ? type : type.name
+      self.class_eval <<DEF
+  def #{attr}
+    #{type_name}.all({#{key.inspect} => self.key}, #{options.inspect})
+  end
+DEF
+    end
+    
+    def belongs_to attr, type = nil
+      type ||= attr.to_s.capitalize
+      self.ds "#{attr}_id"
+      type_name = (type.is_a?(Symbol) || type.is_a?(String)) ? type : type.name
+      self.class_eval <<DEF
+  def #{attr}
+    if defined?(@#{attr})
+      @#{attr}
+    else 
+      @#{attr} = if self.#{attr}_id
+          #{type_name}.get(self.#{attr}_id)
+        else
+          nil
+        end
+    end
+  end
+
+  def #{attr}=(value)
+    self.#{attr}_id = value.key
+    @#{attr} = value
+  end
+DEF
+    end
+    
+    # defines zero or more data store attributes - will create attribute accessors for these
+    def ds(*names)
+      names.each do |name|
+        self.class_eval <<DEF
+  def #{name}
+    if defined?(@#{name})
+      @#{name}
+    else 
+      @#{name} = __ds_get(#{name.inspect})
+    end
+  end
+
+  def #{name}=(value)
+    __ds_set(#{name.inspect}, value)
+    @#{name} = value
+  end
+DEF
+      end
+    end
+
+    def get key
+      if key.is_a? Array
+        create_from_entities DS::Service.get(key.map{|k| DS::KeyFactory.create_key self.name, k.to_i}).values
+      else
+        create_from_entity(DS::Service.get(DS::KeyFactory.create_key(self.name, key.to_i)))
+      end
+    end
+    alias :[] :get
+    
+    def delete key
+      DS::Service.delete([DS::KeyFactory.create_key(self.name, key.to_i)].to_java(DS::Key))
+    end
+    
+    def first conditions = {}
+      query = build_query_from_conditions conditions
+      result = DS::Service.prepare(query).as_single_entity
+      create_from_entity(result) if result
+    end
+
+    def create attrs = {}
+      val = new(attrs)
+      val.save!
+      val
+    end
+    
+    def all conditions = {}, options = {}
+      q = build_query_from_conditions conditions
+      fo = DS::FetchOptions::Builder.with_limit options[:limit] if options[:limit]
+      if options[:offset]
+        fo ? (fo.offset options[:offset]) : (fo = DS::FetchOptions::Builder.with_offset options[:offset])
+      end
+      q.add_sort options[:order].to_s if options[:order]
+      q.add_sort options[:iorder].to_s, DS::Query::SortDirection::DESCENDING if options[:iorder]
+      iter = DS::Service.prepare q
+      iter = fo ? (iter.as_iterable fo) : iter.as_iterable
+      create_from_entities iter
+    end
+    
+    def count
+      q = DS::Query.new self.name
+      DS::Service.prepare(q).count_entities
+    end
+    
+    private
+    def create_from_entity ent
+      obj = self.new
+      obj.instance_variable_set :@__entity, ent
+      obj.instance_variable_set :@new_record, false
+      obj
+    end
+    
+    def create_from_entities collection
+      collection.map do |ent|
+        create_from_entity ent
+      end
+    end
+    
+    def build_query_from_conditions conditions
+      query = DS::Query.new self.name
+      conditions.each do |k, v|
+        query.add_filter k.to_s, DS::Query::FilterOperator::EQUAL, v
+      end
+      query
+    end
+  end
+  
+  def self.included base
+    base.send :include, InstanceMethods
+    base.send :extend,  ClassMethods
+  end
+end

metadata

@@ -0,0 +1,61 @@
+--- !ruby/object:Gem::Specification 
+name: lstoll-bumble
+version: !ruby/object:Gem::Version 
+  version: 0.0.1
+platform: ruby
+authors: 
+- Ola Bini
+- Ivan Schneider
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-05-06 00:00:00 -07:00
+default_executable: 
+dependencies: []
+
+description: Wrapper for google app engine's data store
+email: lstoll@lstoll.net
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/bumble.rb
+- lib/bumble/bumble.rb
+has_rdoc: true
+homepage: http://github.com/olabini/buble
+post_install_message: 
+rdoc_options: 
+- --inline-source
+- --charset=UTF-8
+- --main
+- README.rdoc
+- --title
+- Ruby Goole AppEngine Support
+- lib
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.2.0
+signing_key: 
+specification_version: 2
+summary: Wrapper for google app engine's data store
+test_files: []
+