136 lines
3.7 KiB
Ruby
136 lines
3.7 KiB
Ruby
module CouchRest
|
|
module Model
|
|
module ClassProxy
|
|
|
|
def self.included(base)
|
|
base.extend(ClassMethods)
|
|
end
|
|
|
|
module ClassMethods
|
|
|
|
# Return a proxy object which represents a model class on a
|
|
# chosen database instance. This allows you to DRY operations
|
|
# where a database is chosen dynamically.
|
|
#
|
|
# ==== Example:
|
|
#
|
|
# db = CouchRest::Database.new(...)
|
|
# articles = Article.on(db)
|
|
#
|
|
# articles.all { ... }
|
|
# articles.by_title { ... }
|
|
#
|
|
# u = articles.get("someid")
|
|
#
|
|
# u = articles.new(:title => "I like plankton")
|
|
# u.save # saved on the correct database
|
|
|
|
def on(database)
|
|
Proxy.new(self, database)
|
|
end
|
|
end
|
|
|
|
class Proxy #:nodoc:
|
|
def initialize(klass, database)
|
|
@klass = klass
|
|
@database = database
|
|
end
|
|
|
|
# Base
|
|
|
|
def new(*args)
|
|
doc = @klass.new(*args)
|
|
doc.database = @database
|
|
doc
|
|
end
|
|
|
|
def method_missing(m, *args, &block)
|
|
if has_view?(m)
|
|
query = args.shift || {}
|
|
return view(m, query, *args, &block)
|
|
elsif m.to_s =~ /^find_(by_.+)/
|
|
view_name = $1
|
|
if has_view?(view_name)
|
|
return first_from_view(view_name, *args)
|
|
end
|
|
end
|
|
super
|
|
end
|
|
|
|
# DocumentQueries
|
|
|
|
def all(opts = {}, &block)
|
|
docs = @klass.all({:database => @database}.merge(opts), &block)
|
|
docs.each { |doc| doc.database = @database if doc.respond_to?(:database) } if docs
|
|
docs
|
|
end
|
|
|
|
def count(opts = {}, &block)
|
|
@klass.all({:database => @database, :raw => true, :limit => 0}.merge(opts), &block)['total_rows']
|
|
end
|
|
|
|
def first(opts = {})
|
|
doc = @klass.first({:database => @database}.merge(opts))
|
|
doc.database = @database if doc && doc.respond_to?(:database)
|
|
doc
|
|
end
|
|
|
|
def last(opts = {})
|
|
doc = @klass.last({:database => @database}.merge(opts))
|
|
doc.database = @database if doc && doc.respond_to?(:database)
|
|
doc
|
|
end
|
|
|
|
def get(id)
|
|
doc = @klass.get(id, @database)
|
|
doc.database = @database if doc && doc.respond_to?(:database)
|
|
doc
|
|
end
|
|
alias :find :get
|
|
|
|
def get!(id)
|
|
doc = @klass.get!(id, @database)
|
|
doc.database = @database if doc && doc.respond_to?(:database)
|
|
doc
|
|
end
|
|
alias :find! :get!
|
|
|
|
# Views
|
|
|
|
def has_view?(view)
|
|
@klass.has_view?(view)
|
|
end
|
|
|
|
def view(name, query={}, &block)
|
|
docs = @klass.view(name, {:database => @database}.merge(query), &block)
|
|
docs.each { |doc| doc.database = @database if doc.respond_to?(:database) } if docs
|
|
docs
|
|
end
|
|
|
|
def first_from_view(name, *args)
|
|
# add to first hash available, or add to end
|
|
(args.last.is_a?(Hash) ? args.last : (args << {}).last)[:database] = @database
|
|
doc = @klass.first_from_view(name, *args)
|
|
doc.database = @database if doc && doc.respond_to?(:database)
|
|
doc
|
|
end
|
|
|
|
# DesignDoc
|
|
|
|
def design_doc
|
|
@klass.design_doc
|
|
end
|
|
|
|
def refresh_design_doc
|
|
@klass.refresh_design_doc(@database)
|
|
end
|
|
|
|
def save_design_doc
|
|
@klass.save_design_doc(@database)
|
|
end
|
|
|
|
end
|
|
end
|
|
end
|
|
end
|