class Hayabusa::Objects attr_reader :args, :events, :data, :ids_cache, :ids_cache_should def initialize(args) require "monitor" @callbacks = {} @args = args @args[:col_id] = :id if !@args[:col_id] @args[:class_pre] = "class_" if !@args[:class_pre] @args[:module] = Kernel if !@args[:module] @args[:cache] = :weak if !@args.key?(:cache) @objects = {} @locks = {} @data = {} @lock_require = Monitor.new Knj.gem_require(:Wref, "wref") if @args[:cache] == :weak and !Kernel.const_defined?(:Wref) require "#{@args[:array_enumerator_path]}array_enumerator" if @args[:array_enum] and !Kernel.const_defined?(:Array_enumerator) #Set up various events. @events = Knj::Event_handler.new @events.add_event(:name => :no_html, :connections_max => 1) @events.add_event(:name => :no_name, :connections_max => 1) @events.add_event(:name => :no_date, :connections_max => 1) @events.add_event(:name => :missing_class, :connections_max => 1) @events.add_event(:name => :require_class, :connections_max => 1) raise "No DB given." if !@args[:db] and !@args[:custom] raise "No class path given." if !@args[:class_path] and (@args[:require] or !@args.key?(:require)) if args[:require_all] Knj.gem_require(:Php4r, "php4r") loads = [] Dir.foreach(@args[:class_path]) do |file| next if file == "." or file == ".." or !file.match(/\.rb$/) file_parsed = file file_parsed.gsub!(@args[:class_pre], "") if @args.key?(:class_pre) file_parsed.gsub!(/\.rb$/, "") file_parsed = Php4r.ucwords(file_parsed) loads << file_parsed self.requireclass(file_parsed, {:load => false}) end loads.each do |load_class| self.load_class(load_class) end end #Set up ID-caching. @ids_cache_should = {} if @args[:models] @ids_cache = {} @args[:models].each do |classname, classargs| @ids_cache_should[classname] = true if classargs[:cache_ids] self.cache_ids(classname) end end end #Caches all IDs for a specific classname. def cache_ids(classname) classname = classname.to_sym return nil if !@ids_cache_should or !@ids_cache_should[classname] newcache = {} @args[:db].q("SELECT `#{@args[:col_id]}` FROM `#{classname}` ORDER BY `#{@args[:col_id]}`") do |data| newcache[data[@args[:col_id]].to_i] = true end @ids_cache[classname] = newcache end def init_class(classname) classname = classname.to_sym return false if @objects.key?(classname) if @args[:cache] == :weak @objects[classname] = Wref::Map.new else @objects[classname] = {} end @locks[classname] = Monitor.new end def uninit_class(classname) @objects.delete(classname) @locks.delete(classname) end #Returns a cloned version of the @objects variable. Cloned because iteration on it may crash some of the other methods in Ruby 1.9+ def objects objs_cloned = {} @objects.keys.each do |key| objs_cloned[key] = @objects[key].clone end return objs_cloned end #Returns the database-connection used by this instance of Objects. def db return @args[:db] end #Returns the total count of objects currently held by this instance. def count_objects count = 0 @objects.keys.each do |key| count += @objects[key].length end return count end #This connects a block to an event. When the event is called the block will be executed. def connect(args, &block) raise "No object given." if !args["object"] raise "No signals given." if !args.key?("signal") and !args.key?("signals") args["block"] = block if block_given? object = args["object"].to_sym @callbacks[object] = {} if !@callbacks[object] conn_id = @callbacks[object].length.to_s @callbacks[object][conn_id] = args return conn_id end #Returns true if the given signal is connected to the given object. def connected?(args) raise "No object given." if !args["object"] raise "No signal given." if !args.key?("signal") object = args["object"].to_sym if @callbacks.key?(object) @callbacks[object].clone.each do |ckey, callback| return true if callback.key?("signal") and callback["signal"].to_s == args["signal"].to_s return true if callback.key?("signals") and (callback["signals"].include?(args["signal"].to_s) or callback["signals"].include?(args["signal"].to_sym)) end end return false end #Unconnects a connect by 'object' and 'conn_id'. def unconnect(args) raise ArgumentError, "No object given." if !args["object"] object = args["object"].to_sym raise ArgumentError, "Object doesnt exist: '#{object}'." if !@callbacks.key?(object) if args["conn_id"] conn_ids = [args["conn_id"]] elsif args["conn_ids"] conn_ids = args["conn_ids"] else raise ArgumentError, "Could not figure out connection IDs." end conn_ids.each do |conn_id| raise Errno::ENOENT, "Conn ID doest exist: '#{conn_id}' (#{args})." if !@callbacks[object].key?(conn_id) @callbacks[object].delete(conn_id) end end #This method is used to call the connected callbacks for an event. def call(args, &block) classstr = args["object"].class.classname.to_sym if @callbacks.key?(classstr) @callbacks[classstr].clone.each do |callback_key, callback| docall = false if callback.key?("signal") and args.key?("signal") and callback["signal"].to_s == args["signal"].to_s docall = true elsif callback["signals"] and args["signal"] and (callback["signals"].include?(args["signal"].to_s) or callback["signals"].include?(args["signal"].to_sym)) docall = true end next if !docall if callback["block"] callargs = [] arity = callback["block"].arity if arity <= 0 #do nothing elsif arity == 1 callargs << args["object"] else raise "Unknown number of arguments: #{arity}" end callback["block"].call(*callargs) elsif callback["callback"] require "php4r" if !Kernel.const_defined?(:Php4r) Php4r.call_user_func(callback["callback"], args) else raise "No valid callback given." end end end end def requireclass(classname, args = {}) classname = classname.to_sym return false if @objects.key?(classname) @lock_require.synchronize do #Maybe the classname got required meanwhile the synchronized wait - check again. return false if @objects.key?(classname) if @events.connected?(:require_class) @events.call(:require_class, { :class => classname }) else doreq = false if args[:require] doreq = true elsif args.key?(:require) and !args[:require] doreq = false elsif @args[:require] or !@args.key?(:require) doreq = true end if doreq filename = "#{@args[:class_path]}/#{@args[:class_pre]}#{classname.to_s.downcase}.rb" filename_req = "#{@args[:class_path]}/#{@args[:class_pre]}#{classname.to_s.downcase}" raise "Class file could not be found: #{filename}." if !File.exists?(filename) require filename_req end end if args[:class] classob = args[:class] else begin classob = @args[:module].const_get(classname) rescue NameError => e if @events.connected?(:missing_class) @events.call(:missing_class, { :class => classname }) classob = @args[:module].const_get(classname) else raise e end end end if (classob.respond_to?(:load_columns) or classob.respond_to?(:datarow_init)) and (!args.key?(:load) or args[:load]) self.load_class(classname, args) end self.init_class(classname) end end #Loads a Datarow-class by calling various static methods. def load_class(classname, args = {}) if args[:class] classob = args[:class] else classob = @args[:module].const_get(classname) end pass_arg = Knj::Hash_methods.new(:ob => self, :db => @args[:db]) classob.load_columns(pass_arg) if classob.respond_to?(:load_columns) classob.datarow_init(pass_arg) if classob.respond_to?(:datarow_init) end #Returns the instance of classname, but only if it already exists. def get_if_cached(classname, id) classname = classname.to_sym id = id.to_i if @objects[classname] and obj = Wref::Map.get!(id) return obj end return nil end #Returns true if a row of the given classname and the ID exists. Will use ID-cache if set in arguments and spawned otherwise it will do an actual lookup. #===Examples # print "User 5 exists." if ob.exists?(:User, 5) def exists?(classname, id) #Make sure the given data are in the correct types. classname = classname.to_sym id = id.to_i #Check if ID-cache is enabled for that classname. Avoid SQL-lookup by using that. if @ids_cache_should.key?(classname) if @ids_cache[classname].key?(id) return true else return false end end #If the object currently exists in cache, we dont have to do a lookup either. return true if @objects.key?(classname) and obj = @objects[classname].get!(id) and !obj.deleted? #Okay - no other options than to actually do a real lookup. begin table = @args[:module].const_get(classname).table row = @args[:db].single(table, {@args[:col_id] => id}) if row return true else return false end rescue Errno::ENOENT return false end end #Gets an object from the ID or the full data-hash in the database. #===Examples # inst = ob.get(:User, 5) def get(classname, data, args = nil) classname = classname.to_sym if data.is_a?(Integer) or data.is_a?(String) or data.is_a?(Fixnum) id = data.to_i elsif data.is_a?(Hash) and data.key?(@args[:col_id].to_sym) id = data[@args[:col_id].to_sym].to_i elsif data.is_a?(Hash) and data.key?(@args[:col_id].to_s) id = data[@args[:col_id].to_s].to_i elsif raise ArgumentError, "Unknown data for class '#{classname}': '#{data.class.to_s}' (#{data})." end if @objects.key?(classname) case @args[:cache] when :weak if obj = @objects[classname].get(id) and obj.id.to_i == id return obj end else return @objects[classname][id] if @objects[classname].key?(id) end end self.requireclass(classname) if !@objects.key?(classname) @locks[classname].synchronize do #Maybe the object got spawned while we waited for the lock? If so we shouldnt spawn another instance. if @args[:cache] == :weak and obj = @objects[classname].get(id) and obj.id.to_i == id return obj end #Spawn object. if @args[:datarow] or @args[:custom] obj = @args[:module].const_get(classname).new(data, args) else pass_args = [data] pass_args = pass_args | @args[:extra_args] if @args[:extra_args] obj = @args[:module].const_get(classname).new(*pass_args) end #Save object in cache. case @args[:cache] when :none return obj else @objects[classname][id] = obj return obj end end raise "Unexpected run?" end #Same as normal get but returns false if not found instead of raising error. def get!(*args, &block) begin return self.get(*args, &block) rescue Errno::ENOENT return false end end def object_finalizer(id) classname = @objects_idclass[id] if classname @objects[classname].delete(id) @objects_idclass.delete(id) end end #Returns the first object found from the given arguments. Also automatically limits the results to 1. def get_by(classname, args = {}) classname = classname.to_sym self.requireclass(classname) classob = @args[:module].const_get(classname) raise "list-function has not been implemented for '#{classname}'." if !classob.respond_to?(:list) args["limit"] = 1 self.list(classname, args) do |obj| return obj end return false end def get_try(obj, col_name, obj_name = nil) if !obj_name if match = col_name.to_s.match(/^(.+)_id$/) obj_name = Php4r.ucwords(match[1]).to_sym else raise "Could not figure out objectname for: #{col_name}." end end id_data = obj[col_name].to_i return false if id_data.to_i <= 0 begin return self.get(obj_name, id_data) rescue Errno::ENOENT return false end end #Returns an array-list of objects. If given a block the block will be called for each element and memory will be spared if running weak-link-mode. #===Examples # ob.list(:User) do |user| # print "Username: #{user.name}\n" # end def list(classname, args = {}, &block) args = {} if args == nil classname = classname.to_sym self.requireclass(classname) classob = @args[:module].const_get(classname) raise "list-function has not been implemented for '#{classname}'." if !classob.respond_to?("list") if @args[:datarow] or @args[:custom] ret = classob.list(Knj::Hash_methods.new(:args => args, :ob => self, :db => @args[:db]), &block) else realargs = [args] realargs = realargs | @args[:extra_args] if @args[:extra_args] ret = classob.list(*realargs, &block) end #If 'ret' is an array and a block is given then the list-method didnt return blocks. We emulate it instead with the following code. if block and ret.is_a?(Array) ret.each do |obj| block.call(obj) end return nil elsif block and ret != nil raise "Return should return nil because of block but didnt. It wasnt an array either..." elsif block return nil else return ret end end #Yields every object that is missing certain required objects (based on 'has_many' required-argument). def list_invalid_required(args, &block) enum = Enumerator.new do |yielder| classname = args[:class] classob = @args[:module].const_get(classname) required_data = classob.required_data if required_data and !required_data.empty? required_data.each do |req_data| self.list(args[:class], :cloned_ubuf => true) do |obj| puts "Checking #{obj.classname}(#{obj.id}) for required #{req_data[:class]}." if args[:debug] id = obj[req_data[:col]] begin raise Errno::ENOENT if !id obj_req = self.get(req_data[:class], id) rescue Errno::ENOENT yielder << {:obj => obj, :type => :required, :id => id, :data => req_data} end end end end end return Knj.handle_return(:enum => enum, :block => block) end #Returns select-options-HTML for inserting into a HTML-select-element. def list_opts(classname, args = {}) Knj::ArrayExt.hash_sym(args) classname = classname.to_sym if args[:list_args].is_a?(Hash) list_args = args[:list_args] else list_args = {} end html = "" if args[:addnew] or args[:add] html << "#{_("Add new")}" elsif args[:none] html << "#{_("None")}" end self.list(classname, args[:list_args]) do |object| html << "" rescue => e html << ">[#{object.class.name}: #{e.message}]" end end return html end #Returns a hash which can be used to generate HTML-select-elements. def list_optshash(classname, args = {}) Knj::ArrayExt.hash_sym(args) classname = classname.to_sym if args[:list_args].is_a?(Hash) list_args = args[:list_args] else list_args = {} end if RUBY_VERSION[0..2] == 1.8 and Php4r.class_exists("Dictionary") list = Dictionary.new else list = {} end if args[:addnew] or args[:add] list["0"] = _("Add new") elsif args[:choose] list["0"] = _("Choose") + ":" elsif args[:all] list["0"] = _("All") elsif args[:none] list["0"] = _("None") end self.list(classname, args[:list_args]) do |object| if object.respond_to?(:name) list[object.id] = object.name elsif object.respond_to?(:title) list[object.id] = object.title else raise "Object of class '#{object.class.name}' doesnt support 'name' or 'title." end end return list end #Returns a list of a specific object by running specific SQL against the database. def list_bysql(classname, sql, args = nil, &block) classname = classname.to_sym ret = [] if !block qargs = nil if args args.each do |key, val| case key when :cloned_ubuf qargs = {:cloned_ubuf => true} else raise "Invalid key: '#{key}'." end end end if @args[:array_enum] enum = Enumerator.new do |yielder| @args[:db].q(sql, qargs) do |d_obs| yielder << self.get(classname, d_obs) end end if block enum.each(&block) return nil else return Array_enumerator.new(enum) end else @args[:db].q(sql, qargs) do |d_obs| if block block.call(self.get(classname, d_obs)) else ret << self.get(classname, d_obs) end end if !block return ret else return nil end end end #Add a new object to the database and to the cache. #===Examples # obj = ob.add(:User, {:username => "User 1"}) def add(classname, data = {}, args = nil) raise "data-variable was not a hash: '#{data.class.name}'." if !data.is_a?(Hash) classname = classname.to_sym self.requireclass(classname) if @args[:datarow] classobj = @args[:module].const_get(classname) #Run the class 'add'-method to check various data. classobj.add(Knj::Hash_methods.new(:ob => self, :db => @args[:db], :data => data)) if classobj.respond_to?(:add) #Check if various required data is given. If not then raise an error telling about it. required_data = classobj.required_data required_data.each do |req_data| raise "No '#{req_data[:class]}' given by the data '#{req_data[:col]}'." if !data.key?(req_data[:col]) raise "The '#{req_data[:class]}' by ID '#{data[req_data[:col]]}' could not be found with the data '#{req_data[:col]}'." if !self.exists?(req_data[:class], data[req_data[:col]]) end #If 'skip_ret' is given, then the ID wont be looked up and the object wont be spawned. Be aware the connected events wont be executed either. In return it will go a lot faster. if args and args[:skip_ret] and !@ids_cache_should.key?(classname) ins_args = nil else ins_args = {:return_id => true} end #Insert and (maybe?) get ID. ins_id = @args[:db].insert(classobj.table, data, ins_args).to_i #Add ID to ID-cache if ID-cache is active for that classname. @ids_cache[classname][ins_id] = true if ins_id != 0 and @ids_cache_should.key?(classname) #Skip the rest if we are told not to return result. return nil if args and args[:skip_ret] #Spawn the object. retob = self.get(classname, ins_id, {:skip_reload => true}) elsif @args[:custom] classobj = @args[:module].const_get(classname) retob = classobj.add(Knj::Hash_methods.new( :ob => self, :data => data )) else args = [data] args = args | @args[:extra_args] if @args[:extra_args] retob = @args[:module].const_get(classname).add(*args) end self.call("object" => retob, "signal" => "add") retob.send(:add_after, {}) if retob.respond_to?(:add_after) return retob end #Adds several objects to the database at once. This is faster than adding every single object by itself, since this will do multi-inserts if supported by the database. #===Examples # ob.adds(:User, [{:username => "User 1"}, {:username => "User 2"}) def adds(classname, datas) if !@args[:datarow] datas.each do |data| @args[:module].const_get(classname).add(*args) self.call("object" => retob, "signal" => "add") end else if @args[:module].const_get(classname).respond_to?(:add) datas.each do |data| @args[:module].const_get(classname).add(Knj::Hash_methods.new( :ob => self, :db => self.db, :data => data )) end end db.insert_multi(classname, datas) end self.cache_ids(classname) end #Calls a static method on a class. Passes the d-variable which contains the Objects-object, database-reference and more... def static(class_name, method_name, *args, &block) raise "Only available with datarow enabled." if !@args[:datarow] and !@args[:custom] class_name = class_name method_name = method_name self.requireclass(class_name) class_obj = @args[:module].const_get(class_name) #Sometimes this raises the exception but actually responds to the class? Therefore commented out. - knj #raise "The class '#{class_obj.name}' has no such method: '#{method_name}' (#{class_obj.methods.sort.join(", ")})." if !class_obj.respond_to?(method_name) pass_args = [] if @args[:datarow] pass_args << Knj::Hash_methods.new(:ob => self, :db => self.db) else pass_args << Knj::Hash_methods.new(:ob => self) end args.each do |arg| pass_args << arg end class_obj.send(method_name, *pass_args, &block) end #Unset object. Do this if you are sure, that there are no more references left. This will be done automatically when deleting it. def unset(object) if object.is_a?(Array) object.each do |obj| unset(obj) end return nil end classname = object.class.name if @args[:module] classname = classname.gsub(@args[:module].name + "::", "") end classname = classname.to_sym @objects[classname].delete(object.id.to_i) end def unset_class(classname) if classname.is_a?(Array) classname.each do |classn| self.unset_class(classn) end return false end classname = classname.to_sym return false if !@objects.key?(classname) @objects.delete(classname) end #Delete an object. Both from the database and from the cache. #===Examples # user = ob.get(:User, 1) # ob.delete(user) def delete(object, args = nil) #Return false if the object has already been deleted. return false if object.deleted? classname = object.class.classname.to_sym self.call("object" => object, "signal" => "delete_before") self.unset(object) obj_id = object.id object.delete if object.respond_to?(:delete) if @args[:datarow] #If autodelete is set by 'has_many'-method, go through it and delete the various objects first. if autodelete_data = object.class.autodelete_data autodelete_data.each do |adel_data| self.list(adel_data[:classname], {adel_data[:colname].to_s => object.id}) do |obj_del| self.delete(obj_del, args) end end end #If depend is set by 'has_many'-method, check if any objects exists and raise error if so. if dep_datas = object.class.depending_data dep_datas.each do |dep_data| if obj = self.get_by(dep_data[:classname], {dep_data[:colname].to_s => object.id}) raise "Cannot delete <#{object.class.name}:#{object.id}> because <#{obj.class.name}:#{obj.id}> depends on it." end end end #If autozero is set by 'has_many'-method, check if any objects exists and set the ID to zero. if autozero_datas = object.class.autozero_data autozero_datas.each do |zero_data| self.list(zero_data[:classname], {zero_data[:colname].to_s => object.id}) do |obj_zero| obj_zero[zero_data[:colname].to_sym] = 0 end end end #Delete any translations that has been set on the object by 'has_translation'-method. if object.class.translations begin _hb.trans_del(object) rescue NameError _kas.trans_del(object) end end #If a buffer is given in arguments, then use that to delete the object. if args and buffer = args[:db_buffer] buffer.delete(object.table, {:id => obj_id}) else @args[:db].delete(object.table, {:id => obj_id}) end end @ids_cache[classname].delete(obj_id.to_i) if @ids_cache_should.key?(classname) self.call("object" => object, "signal" => "delete") object.destroy return nil end #Deletes several objects as one. If running datarow-mode it checks all objects before it starts to actually delete them. Its faster than deleting every single object by itself... def deletes(objs) if !@args[:datarow] objs.each do |obj| self.delete(obj) end else tables = {} begin objs.each do |obj| next if obj.deleted? tablen = obj.table if !tables.key?(tablen) tables[tablen] = [] end tables[tablen] << obj.id obj.delete if obj.respond_to?(:delete) #Remove from ID-cache. classname = obj.class.classname.to_sym @ids_cache[classname].delete(obj.id.to_i) if @ids_cache_should.key?(classname) #Unset any data on the object, so it seems deleted. obj.destroy end ensure #An exception may occur, and we should make sure, that objects that has gotten 'delete' called also are deleted from their tables. tables.each do |table, ids| ids.each_slice(1000) do |ids_slice| @args[:db].delete(table, {:id => ids_slice}) end end end end end #Deletes all objects with the given IDs 500 at a time to prevent memory exhaustion or timeout. #===Examples # ob.delete_ids(:class => :Person, :ids => [1, 3, 5, 6, 7, 8, 9]) def delete_ids(args) while !args[:ids].empty? and ids = args[:ids].shift(500) objs = self.list(args[:class], "id" => ids) self.deletes(objs) end return nil end #Try to clean up objects by unsetting everything, start the garbagecollector, get all the remaining objects via ObjectSpace and set them again. Some (if not all) should be cleaned up and our cache should still be safe... dirty but works. def clean(classn) if classn.is_a?(Array) classn.each do |realclassn| self.clean(realclassn) end return nil end if @args[:cache] == :weak @objects[classn].clean elsif @args[:cache] == :none return false else return false if !@objects.key?(classn) @objects[classn] = {} GC.start @objects.keys.each do |classn| data = @objects[classn] classobj = @args[:module].const_get(classn) ObjectSpace.each_object(classobj) do |obj| begin data[obj.id.to_i] = obj rescue => e if e.message == "No data on object." #Object has been unset - skip it. next end raise e end end end end end #Erases the whole cache and regenerates it from ObjectSpace if not running weak-link-caching. If running weaklink-caching then it will only removes the dead links. def clean_all self.clean(@objects.keys) end def classes_loaded return @objects.keys end #This method helps build SQL from Objects-instances list-method. It should not be called directly but only through Objects.list. def sqlhelper(list_args, args_def) args = args_def if args[:db] db = args[:db] else db = @args[:db] end if args[:table] table_def = "`#{db.escape_table(args[:table])}`." else table_def = "" end sql_joins = "" sql_where = "" sql_order = "" sql_limit = "" sql_groupby = "" do_joins = {} limit_from = nil limit_to = nil if list_args.key?("orderby") orders = [] orderstr = list_args["orderby"] list_args["orderby"] = [list_args["orderby"]] if list_args["orderby"].is_a?(Hash) if list_args["orderby"].is_a?(String) found = false found = true if args[:cols].key?(orderstr) if found sql_order << " ORDER BY " ordermode = " ASC" if list_args.key?("ordermode") if list_args["ordermode"] == "desc" ordermode = " DESC" elsif list_args["ordermode"] == "asc" ordermode = " ASC" raise "Unknown ordermode: #{list_args["ordermode"]}" end list_args.delete("ordermode") end sql_order << "#{table_def}`#{db.escape_column(list_args["orderby"])}`#{ordermode}" list_args.delete("orderby") end elsif list_args["orderby"].is_a?(Array) sql_order << " ORDER BY " list_args["orderby"].each do |val| ordermode = nil orderstr = nil found = false if val.is_a?(Array) if val[1] == "asc" ordermode = " ASC" elsif val[1] == "desc" ordermode = " DESC" end if val[0].is_a?(Array) if args[:joined_tables] args[:joined_tables].each do |table_name, table_data| next if table_name.to_s != val[0][0].to_s do_joins[table_name] = true orders << "`#{db.escape_table(table_name)}`.`#{db.escape_column(val[0][1])}`#{ordermode}" found = true break end end raise "Could not find joined table for ordering: '#{val[0][0]}'." if !found else orderstr = val[0] end elsif val.is_a?(String) orderstr = val ordermode = " ASC" elsif val.is_a?(Hash) and val[:type] == :sql orders << val[:sql] found = true elsif val.is_a?(Hash) and val[:type] == :case caseorder = " CASE" val[:case].each do |key, caseval| col = key.first isval = key.last col_str = nil if col.is_a?(Array) raise "No joined tables for '#{args[:table]}'." if !args[:joined_tables] found = false args[:joined_tables].each do |table_name, table_data| if table_name == col.first do_joins[table_name] = true col_str = "`#{db.escape_table(table_name)}`.`#{db.escape_column(col.last)}`" found = true break end end raise "No such joined table on '#{args[:table]}': '#{col.first}' (#{col.first.class.name}) with the following joined table:\n#{Php4r.print_r(args[:joined_tables], true)}" if !found elsif col.is_a?(String) or col.is_a?(Symbol) col_str = "#{table_def}`#{col}`" found = true else raise "Unknown type for case-ordering: '#{col.class.name}'." end raise "'colstr' was not set." if !col_str caseorder << " WHEN #{col_str} = '#{db.esc(isval)}' THEN '#{db.esc(caseval)}'" end if val[:else] caseorder << " ELSE '#{db.esc(val[:else])}'" end caseorder << " END" orders << caseorder elsif val.is_a?(Hash) raise "No joined tables." if !args.key?(:joined_tables) if val[:mode] == "asc" ordermode = " ASC" elsif val[:mode] == "desc" ordermode = " DESC" end if args[:joined_tables] args[:joined_tables].each do |table_name, table_data| if table_data[:parent_table] table_name_real = table_name elsif table_data[:datarow] table_name_real = self.datarow_from_datarow_argument(table_data[:datarow]).classname else table_name_real = @args[:module].const_get(table_name).classname end if table_name.to_s == val[:table].to_s do_joins[table_name] = true if val[:sql] orders << val[:sql] elsif val[:col] orders << "`#{db.escape_table(table_name_real)}`.`#{db.escape_column(val[:col])}`#{ordermode}" else raise "Couldnt figure out how to order based on keys: '#{val.keys.sort}'." end found = true break end end end else raise "Unknown object: #{val.class.name}" end found = true if args[:cols].key?(orderstr) if !found raise "Column not found for ordering: #{orderstr}." end orders << "#{table_def}`#{db.escape_column(orderstr)}`#{ordermode}" if orderstr end sql_order << orders.join(", ") list_args.delete("orderby") else raise "Unknown orderby object: #{list_args["orderby"].class.name}." end end list_args.each do |realkey, val| found = false if realkey.is_a?(Array) if !args[:joins_skip] datarow_obj = self.datarow_obj_from_args(args_def, list_args, realkey[0]) args = datarow_obj.columns_sqlhelper_args raise "Couldnt get arguments from SQLHelper." if !args else datarow_obj = @args[:module].const_get(realkey[0]) args = args_def end table_sym = realkey[0].to_sym do_joins[table_sym] = true list_table_name_real = table_sym table = "`#{db.escape_table(list_table_name_real)}`." key = realkey[1] else table = table_def args = args_def key = realkey end if args.key?(:cols_bools) and args[:cols_bools].index(key) != nil val_s = val.to_s if val_s == "1" or val_s == "true" realval = "1" elsif val_s == "0" or val_s == "false" realval = "0" else raise "Could not make real value out of class: #{val.class.name} => #{val}." end sql_where << " AND #{table}`#{db.escape_column(key)}` = '#{db.esc(realval)}'" found = true elsif args[:cols].key?(key) if val.is_a?(Array) if val.empty? and db.opts[:type].to_s == "mysql" sql_where << " AND false" else escape_sql = Knj::ArrayExt.join( :arr => val, :callback => proc{|value| db.escape(value) }, :sep => ",", :surr => "'" ) sql_where << " AND #{table}`#{db.escape_column(key)}` IN (#{escape_sql})" end elsif val.is_a?(Hash) and val[:type].to_sym == :col raise "No table was given for join: '#{val}', key: '#{key}' on table #{table}." if !val.key?(:table) do_joins[val[:table].to_sym] = true sql_where << " AND #{table}`#{db.escape_column(key)}` = `#{db.escape_table(val[:table])}`.`#{db.escape_column(val[:name])}`" elsif val.is_a?(Hash) and val[:type] == :sqlval and val[:val] == :null sql_where << " AND #{table}`#{db.escape_column(key)}` IS NULL" elsif val.is_a?(Proc) call_args = Knj::Hash_methods.new(:ob => self, :db => db) sql_where << " AND #{table}`#{db.escape_column(key)}` = '#{db.esc(val.call(call_args))}'" else sql_where << " AND #{table}`#{db.escape_column(key)}` = '#{db.esc(val)}'" end found = true elsif key.to_s == "limit_from" limit_from = val.to_i found = true elsif key.to_s == "limit_to" limit_to = val.to_i found = true elsif key.to_s == "limit" limit_from = 0 limit_to = val.to_i found = true elsif args.key?(:cols_dbrows) and args[:cols_dbrows].index("#{key.to_s}_id") != nil if val == false sql_where << " AND #{table}`#{db.escape_column(key.to_s + "_id")}` = '0'" elsif val.is_a?(Array) if val.empty? sql_where << " AND false" else sql_where << " AND #{table}`#{db.escape_column("#{key}_id")}` IN (#{Knj::ArrayExt.join(:arr => val, :sep => ",", :surr => "'", :callback => proc{|obj| obj.id.sql})})" end else sql_where << " AND #{table}`#{db.escape_column(key.to_s + "_id")}` = '#{db.esc(val.id)}'" end found = true elsif match = key.match(/^([A-z_\d]+)_(search|has)$/) and args[:cols].key?(match[1]) != nil if match[2] == "search" Knj::Strings.searchstring(val).each do |str| sql_where << " AND #{table}`#{db.escape_column(match[1])}` LIKE '%#{db.esc(str)}%'" end elsif match[2] == "has" if val sql_where << " AND #{table}`#{db.escape_column(match[1])}` != ''" else sql_where << " AND #{table}`#{db.escape_column(match[1])}` = ''" end end found = true elsif match = key.match(/^([A-z_\d]+)_(not|lower)$/) and args[:cols].key?(match[1]) if match[2] == "not" if val.is_a?(Array) if val.empty? #ignore. else escape_sql = Knj::ArrayExt.join( :arr => val, :callback => proc{|value| db.escape(value) }, :sep => ",", :surr => "'" ) sql_where << " AND #{table}`#{db.escape_column(match[1])}` NOT IN (#{escape_sql})" end else sql_where << " AND #{table}`#{db.escape_column(match[1])}` != '#{db.esc(val)}'" end elsif match[2] == "lower" sql_where << " AND LOWER(#{table}`#{db.escape_column(match[1])}`) = LOWER('#{db.esc(val)}')" else raise "Unknown mode: '#{match[2]}'." end found = true elsif args.key?(:cols_date) and match = key.match(/^(.+)_(day|week|month|year|from|to|below|above)(|_(not))$/) and args[:cols_date].index(match[1]) != nil not_v = match[4] val = Datet.in(val) if val.is_a?(Time) if match[2] == "day" if val.is_a?(Array) sql_where << " AND (" first = true val.each do |realval| if first first = false else sql_where << " OR " end sql_where << "#{db.sqlspecs.strftime("%d %m %Y", "#{table}`#{db.escape_column(match[1])}`")} #{self.not(not_v, "!")}= #{db.sqlspecs.strftime("%d %m %Y", "'#{db.esc(realval.dbstr)}'")}" end sql_where << ")" else sql_where << " AND #{db.sqlspecs.strftime("%d %m %Y", "#{table}`#{db.escape_column(match[1])}`")} #{self.not(not_v, "!")}= #{db.sqlspecs.strftime("%d %m %Y", "'#{db.esc(val.dbstr)}'")}" end elsif match[2] == "week" sql_where << " AND #{db.sqlspecs.strftime("%W %Y", "#{table}`#{db.escape_column(match[1])}`")} #{self.not(not_v, "!")}= #{db.sqlspecs.strftime("%W %Y", "'#{db.esc(val.dbstr)}'")}" elsif match[2] == "month" sql_where << " AND #{db.sqlspecs.strftime("%m %Y", "#{table}`#{db.escape_column(match[1])}`")} #{self.not(not_v, "!")}= #{db.sqlspecs.strftime("%m %Y", "'#{db.esc(val.dbstr)}'")}" elsif match[2] == "year" sql_where << " AND #{db.sqlspecs.strftime("%Y", "#{table}`#{db.escape_column(match[1])}`")} #{self.not(not_v, "!")}= #{db.sqlspecs.strftime("%Y", "'#{db.esc(val.dbstr)}'")}" elsif match[2] == "from" or match[2] == "above" sql_where << " AND #{table}`#{db.escape_column(match[1])}` >= '#{db.esc(val.dbstr)}'" elsif match[2] == "to" or match[2] == "below" sql_where << " AND #{table}`#{db.escape_column(match[1])}` <= '#{db.esc(val.dbstr)}'" else raise "Unknown date-key: #{match[2]}." end found = true elsif args.key?(:cols_num) and match = key.match(/^(.+)_(from|to|above|below|numeric)$/) and args[:cols_num].index(match[1]) != nil if match[2] == "from" sql_where << " AND #{table}`#{db.escape_column(match[1])}` >= '#{db.esc(val)}'" elsif match[2] == "to" sql_where << " AND #{table}`#{db.escape_column(match[1])}` <= '#{db.esc(val)}'" elsif match[2] == "above" sql_where << " AND #{table}`#{db.escape_column(match[1])}` > '#{db.esc(val)}'" elsif match[2] == "below" sql_where << " AND #{table}`#{db.escape_column(match[1])}` < '#{db.esc(val)}'" else raise "Unknown method of treating cols-num-argument: #{match[2]}." end found = true elsif match = key.match(/^(.+)_lookup$/) and args[:cols].key?("#{match[1]}_id") and args[:cols].key?("#{match[1]}_class") sql_where << " AND #{table}`#{db.escape_column("#{match[1]}_class")}` = '#{db.esc(val.table)}'" sql_where << " AND #{table}`#{db.escape_column("#{match[1]}_id")}` = '#{db.esc(val.id)}'" found = true elsif realkey == "groupby" found = true if val.is_a?(Array) val.each do |col_name| raise "Column '#{val}' not found on table '#{table}'." if !args[:cols].key?(col_name) sql_groupby << ", " if sql_groupby.length > 0 sql_groupby << "#{table}`#{db.escape_column(col_name)}`" end elsif val.is_a?(String) sql_groupby << ", " if sql_groupby.length > 0 sql_groupby << "#{table}`#{db.escape_column(val)}`" else raise "Unknown class given for 'groupby': '#{val.class.name}'." end end list_args.delete(realkey) if found end args = args_def if !args[:joins_skip] raise "No joins defined on '#{args[:table]}' for: '#{args[:table]}'." if !do_joins.empty? and !args[:joined_tables] do_joins.each do |table_name, temp_val| raise "No join defined on table '#{args[:table]}' for table '#{table_name}'." if !args[:joined_tables].key?(table_name) table_data = args[:joined_tables][table_name] if table_data.key?(:parent_table) join_table_name_real = table_name sql_joins << " LEFT JOIN `#{table_data[:parent_table]}` AS `#{table_name}` ON 1=1" else const = @args[:module].const_get(table_name) join_table_name_real = const.classname sql_joins << " LEFT JOIN `#{const.table}` AS `#{const.classname}` ON 1=1" end if table_data[:ob] ob = table_data[:ob] else ob = self end class_name = args[:table].to_sym if table_data[:datarow] datarow = self.datarow_from_datarow_argument(table_data[:datarow]) else self.requireclass(class_name) if @objects.key?(class_name) datarow = @args[:module].const_get(class_name) end if !datarow.columns_sqlhelper_args ob.requireclass(datarow.table.to_sym) raise "No SQL-helper-args on class '#{datarow.table}' ???" if !datarow.columns_sqlhelper_args end newargs = datarow.columns_sqlhelper_args.clone newargs[:table] = join_table_name_real newargs[:joins_skip] = true #Clone the where-arguments and run them against another sqlhelper to sub-join. join_args = table_data[:where].clone ret = self.sqlhelper(join_args, newargs) sql_joins << ret[:sql_where] #If any of the join-arguments are left, then we should throw an error. join_args.each do |key, val| raise "Invalid key '#{key}' when trying to join table '#{table_name}' on table '#{args_def[:table]}'." end end end #If limit arguments has been given then add them. if limit_from and limit_to sql_limit = " LIMIT #{limit_from}, #{limit_to}" end sql_groupby = nil if sql_groupby.length <= 0 return { :sql_joins => sql_joins, :sql_where => sql_where, :sql_limit => sql_limit, :sql_order => sql_order, :sql_groupby => sql_groupby } end #Used by sqlhelper-method to look up datarow-classes and automatically load them if they arent loaded already. def datarow_obj_from_args(args, list_args, class_name) class_name = class_name.to_sym if !args.key?(:joined_tables) raise "No joined tables on '#{args[:table]}' to find datarow for: '#{class_name}'." end args[:joined_tables].each do |table_name, table_data| next if table_name.to_sym != class_name return self.datarow_from_datarow_argument(table_data[:datarow]) if table_data[:datarow] self.requireclass(class_name) if @objects.key?(class_name) return @args[:module].const_get(class_name) end raise "Could not figure out datarow for: '#{class_name}'." end def datarow_from_datarow_argument(datarow_argument) if datarow_argument.is_a?(String) const = Knj::Strings.const_get_full(datarow_argument) else const = datarow_argument end self.load_class(datarow_argument.to_s.split("::").last) if !const.initialized? #Make sure the class is initialized. return const end def not(not_v, val) if not_v == "not" or not_v == "not_" return val end return "" end end require "#{$knjpath}objects/objects_sqlhelper"