mongo-document 0.0.0 → 1.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 9b05671822bad85d7e736d9e5b3749f6ea7222fb
-  data.tar.gz: 9416bc4ae7dbec35fe24ba830172dd7dd9298474
+  metadata.gz: 321421ae3e2f2c78d3c86a04f509894de8ee3365
+  data.tar.gz: aa7d497d6c2e69fa0dd318358fd0667fa29cc956
 SHA512:
-  metadata.gz: c7ec2292c2c669e54b66df768f40917e77ca7a5290ec4fb5f00818cdb228652eec5965952f227fde7af76c4d36b43191005bc267bd3fccaa541459a744efeb64
-  data.tar.gz: c74b2243f05d11c223a9c56d0260271cfd0630af6a34799f67e3f614c1244cd9756023f9d8850f688bd951e1b291ff4faa081d459f9091f2593d1134d7804dc1
+  metadata.gz: 8680a58b6b74cf10a201e0abe7158aa591f402742fc80b33e015142b3d949e97166d06765352cad759952b122d7c6279e27c45b24ea291251cd61b4fcad9ef3d
+  data.tar.gz: 7f5905060f92d6bde8dfa85615afbbd0d2f9c4173811b97d4ccf3eaa9164bb231eaed89c327dd265b5b3748b9fc9072ff9824cbaaca8d4617e5d0d5d3fbb379e

data/lib/mongo/document.rb

@@ -0,0 +1,99 @@
+require 'mongo'
+require 'yaml'
+require 'active_support'
+require 'active_support/core_ext/string/inflections'
+require 'active_support/core_ext/hash/keys'
+
+module Mongo
+  module DocumentClass
+    def client
+      if !@client
+        raise "No connection! You should call Mongo::Document.establish_connection before."
+      end
+      @client
+    end
+    alias connection client
+
+    def collection
+      @collection ||= client[collection_name]
+    end
+
+    def collection_name
+      @collection_name ||= name.underscore.pluralize
+    end
+
+    def collection_name=(name)
+      @collection_name = name
+    end
+
+    # A little optimazation for the most frequently-used method.
+    def find(*args)
+      collection.find(*args)
+    end
+
+    # For other methods on collection(or not), just send to it.
+    def method_missing(name, *args)
+      collection.send(name, *args)
+    end
+
+    def respond_to_missing?(name, include_private = false)
+      collection.respond_to?(name, include_private) || super
+    end
+
+  end
+
+  module Document
+    class << self
+      def included(cls)
+        cls.extend DocumentClass
+        set_connection(cls)
+        descendants << cls
+      end
+
+      def descendants
+        @descendants ||= []
+      end
+
+      def client
+        @client
+      end
+      alias connection client
+
+      def establish_connection(spec)
+        spec = spec.symbolize_keys
+        hosts_or_uri = nil
+        spec.delete_if do |k, v|
+          if k == :hosts || k == :uri
+            hosts_or_uri = v
+          end
+        end
+        @client = Mongo::Client.new(hosts_or_uri, spec)
+
+        descendants.each do |model|
+          set_connection(model)
+        end
+      end
+
+      def database_file=(file)
+        config = YAML.load(File.read(file))
+        env = ENV['RAILS_ENV'] || ENV['RACK_ENV']
+        spec = config[env]
+        establish_connection(spec)
+      end
+
+      private
+
+      def set_connection(cls)
+        cls.instance_variable_set(:@client, @client) if @client
+      end
+
+    end
+
+    # Auto read database config
+    if !ENV["MD_NO_AUTO_CONFIG"] && File.exist?("#{Dir.pwd}/config/database.yml")
+      self.database_file = "#{Dir.pwd}/config/database.yml"
+    end
+
+  end
+
+end

metadata

@@ -1,16 +1,30 @@
 --- !ruby/object:Gem::Specification
 name: mongo-document
 version: !ruby/object:Gem::Version
-  version: 0.0.0
+  version: 1.0.0
 platform: ruby
 authors:
 - Robert Zhang
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-04-10 00:00:00.000000000 Z
-dependencies: []
-description: A slim Mongo document model
+date: 2016-10-04 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: mongo
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.2'
+description: 
 email: louirobert@gmail.com
 executables: []
 extensions: []
@@ -37,7 +51,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.2.2
+rubygems_version: 2.5.1
 signing_key: 
 specification_version: 4
 summary: A slim Mongo document model