meta-record 1.0.1

data/lib/metarecord/generators/comet/edit_generator.rb

@@ -0,0 +1,153 @@
+require 'metarecord/generators/crails/edit_generator'
+
+class CometEditGenerator < CrailsEditGenerator
+  def _append_macro str
+    @src ||= ""
+    @src += str + "\n"
+  end
+
+  def generate_json_methods object
+    super
+    @rendering_from_json = true
+    _append_macro "#ifdef #{CometDataGenerator.client_define}"
+    _append "void #{@klassname}::from_json(Data data)"
+    _append "{"
+    @indent += 1
+    _append "id = data[\"id\"].defaults_to<#{id_type}>(#{null_id});"
+    _append "edit(data);"
+    @indent -= 1
+    _append "}"
+    _append_macro "#endif"
+    @rendering_from_json = false
+  end
+
+  def validation type, name, data
+    if data[:uniqueness] == true
+      _append_macro "#ifndef #{CometDataGenerator.client_define}"
+      _append validate_uniqueness type, name
+      _append_macro "#endif"
+      data[:uniqueness] = false
+    end
+    super type, name, data
+  end
+
+  def has_one_getter type, name, options
+    type = get_type type
+    tptr = ptr_type type
+    _append_macro "#ifndef #{CometDataGenerator.client_define}"
+    super
+    _append_macro "#else"
+    _append "#{tptr} #{@klassname}::get_#{name}() const"
+    _append "{"
+    _append "  #{tptr} model = std::make_shared<#{type}>();"
+    _append "  model->set_id(get_#{name}_id());"
+    _append "  model->fetch();"
+    _append "  return model;"
+    _append "}"
+    _append_macro "#endif"
+  end
+
+  def joined_has_one_edit type, name, options
+    data_id     = "data[\"#{name}_id\"]"
+    inline_data = "data[\"#{name}\"]"
+    _append_macro "#ifndef #{CometDataGenerator.client_define}"
+    super
+    _append_macro "#else"
+    _append "{"
+    _append "  if (#{data_id} == 0)"
+    _append "    set_#{name}(nullptr);"
+    _append "  else if (!get_#{name}() || #{data_id} != get_#{name}()->get_id())"
+    _append "  {"
+    _append "    auto linked_resource = std::make_shared<#{type}>();"
+    _append "    linked_resource->set_id(#{data_id}.as<#{id_type}>());"
+    _append "    set_#{name}(linked_resource);"
+    _append "  }"
+    _append "}"
+    _append "else if (#{inline_data}.exists())"
+    _append "{"
+    _append "  auto linked_resource = std::make_shared<#{type}>();"
+    _append "  linked_resource->from_json(#{inline_data});"
+    _append "  set_#{name}(linked_resource);"
+    _append "}"
+    _append_macro "#endif"
+  end
+
+  def has_many_fetch type, name, options
+    tptr = ptr_type type
+    singular_name = get_singular_name name
+
+    _append_macro "#ifndef #{CometDataGenerator.client_define}"
+    super type, name, options
+    _append_macro "#else"
+    _append "Comet::Promise #{@klassname}::fetch_#{name}()"
+    _append "{"
+    @indent += 1
+    _append "std::vector<Comet::Promise> promises;\n"
+    _append "for (auto id : #{singular_name}_ids)"
+    _append "{"
+    @indent += 1
+    _append "#{tptr} model;\n"
+    _append "model->set_id(id);"
+    _append "promises.push_back(model->fetch());"
+    _append "#{name}.push_back(model);"
+    _append "#{name}_fetched = true;"
+    @indent -= 1
+    _append "}"
+    _append "return Comet::Promise::all(promises);"
+    @indent -= 1
+    _append "}"
+    _append_macro "#endif"
+  end
+
+  def property type, name, options = {}
+    if options[:ready_only] == true && rendering_edit?
+      options[:read_only] = false
+      _append_macro "#ifdef #{CometDataGenerator.client_define}"
+      super
+      _append_macro "#endif"
+    else
+      super
+    end
+  end
+
+  def has_many type, name, options = {}
+    if options[:read_only] == true && rendering_edit?
+      options[:read_only] = false
+      _append_macro "#ifdef #{CometDataGenerator.client_define}"
+      super
+      _append_macro "#endif"
+    else
+      super
+    end
+  end
+
+  def has_one type, name, options = {}
+    if options[:read_only] == true && rendering_edit?
+      options[:read_only] = false
+      _append_macro "#ifdef #{CometDataGenerator.client_define}"
+      super
+      _append_macro "#endif"
+    else
+      super
+    end
+  end
+
+  class << self
+    def generate_includes
+<<CPP
+#ifndef #{CometDataGenerator.client_define}
+#{super}
+#else
+# include <crails/comet/mvc/helpers.hpp>
+#endif
+CPP
+    end
+
+    def sourcefile_to_destfile sourcefile
+      base      = super sourcefile
+      basepath  = Pathname.new base
+      parentdir = basepath.dirname.to_s + "/shared"
+      "#{parentdir}/#{basepath.basename}"
+    end
+  end
+end

data/lib/metarecord/generators/crails/data_generator.rb

@@ -0,0 +1,289 @@
+require 'metarecord/model'
+require 'metarecord/generator_base'
+
+class CrailsDataGenerator < GeneratorBase
+  attr_accessor :forward_declarations
+
+  def reset
+    super
+    @forward_declarations = []
+    @src = {}
+  end
+
+  def _append str, opts = {}
+    @src[@current_visibility] ||= ""
+    @src[@current_visibility] += " " * (@indent * @tab_size)
+    @src[@current_visibility] += str + "\n" unless opts[:no_line_return]
+  end
+
+  def _append_macro str
+    @src[@current_visibility] ||= ""
+    @src[@current_visibility] += str + "\n"
+  end
+
+  def get_headers
+    forward_source = ""
+    @forward_declarations.each do |type|
+      parts = type.split('::').reject {|part| part.empty?}
+      last_part = parts.last
+      if parts.length > 1
+        parts[0...parts.size-1].each do |part|
+          forward_source += "namespace #{part} { "
+        end
+        forward_source += "class #{last_part};"
+        (parts.size - 1).times do forward_source += " } " end
+        forward_source += "\n"
+      else
+        forward_source += "class #{last_part};\n"
+      end
+    end
+    forward_source
+  end
+
+  def generate_class_head object
+    _append "#pragma db object abstract"
+    _append "struct #{object[:name]}"
+  end
+
+  def generate_for object
+    reset
+    @forward_declarations << object[:classname] unless object[:classname].nil?
+    @current_visibility = :_preblock
+    @indent += 1
+    generate_class_head object
+    _append "{"
+    @indent += 1
+    _append "friend class odb::access;"
+    _append "#pragma db transient"
+    _append "DataTree errors;"
+    @current_visibility = :public
+    edit_resource_declaration
+    prepare_order_by
+    self.instance_eval &object[:block]
+    @current_visibility = :_postblock
+    @indent -= 1
+    _append "};"
+    @indent -= 1
+    render
+  end
+
+  def edit_resource_declaration
+    _append "virtual std::vector<std::string> find_missing_parameters(Data) const;"
+    _append "virtual void                     edit(Data);"
+    _append "virtual std::string              to_json() const;"
+    _append "virtual bool                     is_valid();"
+    _append "virtual void                     on_dependent_destroy(#{id_type});"
+    _append ""
+  end
+
+  def prepare_order_by
+    _append "template<typename QUERY>"
+    _append "static QUERY default_order_by(QUERY query)"
+    _append "{"
+    _append "  return query; // order_by"
+    _append "}\n"
+  end
+
+  def render
+    @current_visibility = :_result
+    result = @src[:_preblock]
+    [:public, :protected, :private].each do |key|
+      unless @src[key].nil?
+        result += "  #{key}:\n"
+        result += @src[key]
+      end
+    end
+    result += @src[:_postblock]
+    result
+  end
+
+  def visibility value
+    @current_visibility = value
+  end
+
+  def with_visibility value, &block
+    old_visibility = @current_visibility
+    visibility value
+    self.instance_eval &block
+    visibility old_visibility
+  end
+
+  def resource_name name
+    visibility :public
+    _append "static const std::string scope;"
+    _append "static const std::string plural_scope;"
+    _append "static const std::string view;"
+  end
+
+  def order_by name, flow = nil
+    src = "return query + \"ORDER BY\" + QUERY::#{name}"
+    src += " + \"#{flow.to_s.upcase}\"" unless flow.nil?
+    src += ";"
+    @src[:public] = @src[:public].gsub /^(\s*).*\/\/ order_by$/, '\1' + src
+  end
+
+  def datatree_property name, options
+    with_visibility :public do
+      virt = if options[:allow_override] == true then "virtual " else "" end
+      _append "#{virt}DataTree& get_#{name}() { return #{name}; }"
+      _append "#{virt}const DataTree& get_#{name}() const { return #{name}; }"
+      _append "#{virt}void set_#{name}(Data value) { this->#{name}.clear(); this->#{name}.as_data().merge(value); }"
+    end
+  end
+
+  def property type, name, options = {}
+    if type == 'DataTree'
+      datatree_property name, options
+    else
+       with_visibility :public do
+       virt = if options[:allow_override] == true then "virtual " else "" end
+        _append "#{virt}#{type} get_#{name}() const { return #{name}; }"
+        _append "#{virt}void set_#{name}(#{type} value) { this->#{name} = value; }"
+      end
+    end
+    make_pragma_db options[:db] unless options[:db].nil?
+    if options[:default].nil?
+      _append "#{type} #{name};"
+    else
+      _append "#{type} #{name} = #{get_value options[:default]};"
+    end
+  end
+
+  def make_pragma_db options, authorized_options = []
+    is_option_available = lambda {|a|
+      (not options[a].nil?) && (authorized_options.size == 0 || authorized_options.include?(a))
+    }
+    src  = ""
+    src += "transient " if is_option_available.call(:transient) && options[:transient] == true
+    src += "type(\"#{options[:type]}\") "          if is_option_available.call :type
+    src += "column(\"#{options[:column]}\") "      if is_option_available.call :column
+    src += "default(#{get_value options[:default]}) " if is_option_available.call :default
+    src += "null " if is_option_available.call(:null) && options[:null] == true
+    _append "#pragma db #{src}" if src.size > 0
+  end
+
+  def has_one type, name, options = {}
+    @forward_declarations << type
+    type = get_type type
+    tptr = ptr_type type
+    virt = if options[:allow_override] == true then "virtual " else "" end
+    if options[:joined] != false
+      with_visibility :public do
+        _append "#{virt}#{tptr} get_#{name}() const { return #{name}; }"
+        _append "#{virt}void set_#{name}(#{tptr} v) { this->#{name} = v; }"
+        _append "#{virt}#{id_type} get_#{name}_id() const;"
+      end
+      _append "#{tptr} #{name};"
+    else
+      with_visibility :public do
+        _append "#{virt}#{tptr} get_#{name}() const;"
+        _append "#{virt}void set_#{name}(#{tptr} v);"
+        _append "#{id_type} get_#{name}_id() const { return #{name}_id; }"
+        _append "void set_#{name}_id(#{id_type} v) { #{name}_id = v; }"
+      end
+      make_pragma_db options[:db] unless options[:db].nil?
+      _append "#{id_type} #{name}_id = #{null_id};"
+    end
+    with_visibility :public do
+      _append <<CPP
+
+    template<typename ARRAY>
+    static void collect_#{name}(ARRAY& array, std::map<#{id_type}, #{tptr} >& results)
+    {
+      for (auto model : array)
+      {
+        if (results.find(model.get_#{name}_id()) == results.end())
+          results[model.get_#{name}_id()] = model.get_#{name}();
+      }
+    }
+
+CPP
+    end
+  end
+
+  def has_many type, name, options = {}
+    @forward_declarations << type
+    type = get_type type
+    tptr = ptr_type type
+    list_type = "std::list<#{tptr} >"
+    singular_name = get_singular_name name
+    with_visibility :public do
+      virt = if options[:allow_override] == true then "virtual " else "" end
+      _append "#{virt}bool update_#{name}(Data);"
+      _append "#{virt}void add_#{singular_name}(#{tptr});"
+      _append "#{virt}void remove_#{singular_name}(const #{type}&);"
+      _append "#{virt}void collect_#{name}(std::map<#{id_type}, #{tptr} >&);"
+    end
+    if options[:joined] != false
+      _join_based_has_many list_type, name, options
+    else
+      _id_based_has_many list_type, name, options
+    end
+  end
+
+  def _join_based_has_many list_type, name, options
+    singular_name = get_singular_name name
+    with_visibility :public do
+      _append "const #{list_type}& get_#{name}() const { return #{name}; }"
+      _append "std::vector<#{id_type}> get_#{singular_name}_ids() const;"
+    end
+    _append "#{list_type} #{name};"
+  end
+
+  def has_many_fetch list_type, name, options
+    with_visibility :public do
+      _append "void fetch_#{name}();"
+    end
+  end
+
+  def _id_based_has_many list_type, name, options
+    singular_name = get_singular_name name
+    store_type = "std::vector<#{id_type}>"
+    with_visibility :public do
+      _append "const #{store_type}& get_#{singular_name}_ids() const { return #{singular_name}_ids; }"
+      _append "const #{list_type}& get_#{name}();"
+      _append "void set_#{singular_name}_ids(const #{store_type}& val) { #{singular_name}_ids = val; #{name}_fetched = false; }"
+    end
+    has_many_fetch list_type, name, options
+    options[:db]        ||= {}
+    options[:db][:type] ||= "INTEGER[]"
+    make_pragma_db options[:db], [:type, :column, :null]
+    _append "#{store_type} #{singular_name}_ids;"
+    make_pragma_db transient: true
+    _append "#{list_type} #{name};"
+    make_pragma_db transient: true
+    _append "bool #{name}_fetched = false;"
+  end
+
+  class << self
+    def extension ; ".hpp" ; end
+
+    def generate_includes
+<<CPP
+# include <vector>
+# include <list>
+# include <map>
+# include <string>
+# include <crails/datatree.hpp>
+# include <crails/odb/id_type.hpp>
+CPP
+    end
+
+    def make_file filename, data
+      file_define = "_#{filename[0...-3].upcase.gsub "/", "_"}_HPP"
+      source = <<CPP
+#ifndef  #{file_define}
+# define #{file_define}
+#{generate_includes}
+#{collect_includes_for(filename, true).join "\n"}
+namespace odb { class access; }
+#{data[:headers].join ""}
+namespace #{METARECORD_NAMESPACE}
+{
+#{data[:bodies].join "\n"}}
+
+#endif
+CPP
+    end
+  end
+end

data/lib/metarecord/generators/crails/destroy_generator.rb

@@ -0,0 +1,178 @@
+
+require 'metarecord/model'
+require 'metarecord/generator_base'
+
+class CrailsDestroyPreparator < GeneratorBase
+  attr_accessor :data
+
+  def prepare
+    @data = {}
+    Model.list.each do |model|
+      @model = model
+      self.instance_eval &model[:block]
+    end
+  end
+
+  def has_many type, name, options = {}
+    unless options[:dependent].nil?
+      @data[type] ||= []
+      @data[type] << ({
+        type: "has_many", class: @model[:classname],
+	field: name, options: options
+      })
+    end
+  end
+
+  def has_one type, name, options = {}
+    unless options[:dependent].nil?
+      @data[type] ||= []
+      @data[type] << ({
+        type: "has_one", class: @model[:classname],
+	field: name, options: options
+      })
+    end
+  end
+end
+
+class CrailsDestroyGenerator < GeneratorBase
+  class << self
+    attr_accessor :destroy_data
+
+    def my_prepare
+      destroy_preparator = CrailsDestroyPreparator.new
+      destroy_preparator.prepare
+      @destroy_data = destroy_preparator.data
+    end
+
+    def extension ; ".destroy.cpp" ; end
+
+    def make_file filename, data
+      include = "lib/" + filename[0...-2] + "hpp"
+      source  = "#include \"#{include}\"\n"
+      source += "#include \"lib/#{filename[0...-2]}queries.hpp\"\n"
+      source += "#include <crails/odb/helpers.hpp>\n"
+      source += "#include <crails/odb/any.hpp>\n"
+      source += "#include <#{GeneratorBase.odb_connection[:include]}>\n"
+      source += "#include \"lib/odb/application-odb.hxx\"\n"
+      source += (collect_includes_for filename).join "\n"
+      source += "\n" + (data[:bodies].join "\n")
+    end
+  end
+
+  def generate_for object
+    reset
+    CrailsDestroyGenerator.my_prepare if CrailsDestroyGenerator.destroy_data.nil?
+    @finalclass = object[:classname]
+    @klassname = get_classname(object)
+    _append "void #{@klassname}::on_dependent_destroy(#{id_type} self_id)"
+    _append "{"
+    unless object[:classname].nil?
+      @indent += 1
+      depended_by = CrailsDestroyGenerator.destroy_data[object[:classname]]
+      if depended_by.class == Array
+        _append "auto& database = *#{GeneratorBase.odb_connection[:object]}::instance;"
+        depended_by.each do |relation|
+          _append "// #{relation[:class]}"
+          _append "{"
+          @indent += 1
+          if relation[:type] == "has_one"
+            one_to_many relation
+          elsif relation[:type] == "has_many"
+            many_to_many relation
+          end
+          @indent -= 1
+          _append "}\n"
+        end
+      end
+      @indent -= 1
+    end
+    _append "}"
+    @src
+  end
+
+  def one_to_many relation
+    _append "typedef odb::query<::#{relation[:class]}> Query;"
+    _append "odb::result<::#{relation[:class]}> models;"
+    _append "Query query;\n"
+
+    if relation[:options][:joined] != false
+      _append "query = Query::#{relation[:field]}->id == self_id;"
+    else
+      _append "query = Query::#{relation[:field]}_id == self_id;"
+    end
+    _append "database.find<::#{relation[:class]}>(models, query);"
+    _append "for (auto model : odb::to_vector<::#{relation[:class]}>(models))"
+    if relation[:options][:dependent] == :destroy
+      _append "  database.destroy(model);"
+    elsif relation[:options][:dependent] == :unlink
+      _append "{"
+      if relation[:options][:joined] != false
+        _append "  model.set_#{relation[:field]}(nullptr);"
+      else
+        _append "  model.set_#{relation[:field]}_id(0);"
+      end
+      _append "  database.save(model);"
+      _append "}"
+    end
+  end
+
+  def many_to_many relation
+    if relation[:options][:joined] != false
+      outer_name = (relation[:field].split(/-|_|\s/).each do |i| i.capitalize! end).join
+      view_klass = "#{relation[:class].split("::").join}By#{outer_name}"
+      klassname  = @klassname.split("::").last
+      singular_name = get_singular_name relation[:field]
+      _append "typedef odb::query<#{view_klass}> Query;"
+      _append "odb::result<#{view_klass}> models;"
+      _append "Query query;\n"
+      _append "query = Query::#{klassname}::id == self_id;"
+      _append "database.find<#{view_klass}>(models, query);"
+      _append "for (auto model : odb::to_vector<#{relation[:class]}, #{view_klass}>(models))"
+      _append "{"
+      @indent += 1
+      if relation[:options][:dependent] == :destroy
+	_append "if (model.get_#{relation[:field]}().size() == 1)"
+	_append "  database.destroy(model);"
+	_append "else"
+	_append "{"
+        _append "  model.remove_#{singular_name}(*static_cast<#{@finalclass}*>(this));"
+	_append "  database.save(model);"
+	_append "}"
+      else relation[:options][:dependent] == :unlink
+        _append "model.remove_#{singular_name}(*static_cast<#{@finalclass}*>(this));"
+	_append "database.save(model);"
+      end
+      @indent -= 1
+      _append "}"
+    else
+      _append "typedef odb::query<#{relation[:class]}> Query;"
+      _append "std::vector<#{id_type}> ids = { self_id };"
+      _append "odb::result<#{relation[:class]}> models;"
+      _append "Query query;\n"
+      id_field = "#{get_singular_name relation[:field]}_ids"
+      _append "query = Query::#{id_field} + \"@>\" + ODB::array_to_string(ids, \"int\");"
+      _append "database.find<#{relation[:class]}>(models, query);"
+      _append "for (auto model : odb::to_vector<#{relation[:class]}>(models))"
+      _append "{"
+      @indent += 1
+      if relation[:options][:dependent] == :destroy
+        _append "auto id_list = model.get_#{id_field}();\n"
+        _append "if (id_list.size() == 1)"
+        _append "  database.destroy(model);"
+        _append "else"
+        _append "{"
+        _append "  id_list.erase(std::find(id_list.begin(), id_list.end(), self_id));"
+        _append "  model.set_#{id_field}(id_list);"
+        _append "  database.save(model);"
+        _append "}"
+      elsif relation[:options][:dependent] == :unlink
+        _append "auto id_list = model.get_#{id_field}();\n"
+        _append "id_list.erase(std::find(id_list.begin(), id_list.end(), self_id));"
+        _append "model.set_#{id_field}(id_list);"
+        _append "database.save(model);"
+      end
+      @indent -= 1
+      _append "}"
+    end
+  end
+end