fix remove tree

[pettanr/pettanr.git] / lib / manifest / model.rb

diff --git a/lib/manifest/model.rb b/lib/manifest/model.rb
index 6c76ea4..c225c38 100644 (file)
--- a/lib/manifest/model.rb
+++ b/lib/manifest/model.rb
@@ -1,64 +1,103 @@
-require_dependency "manifest/model/attribute"
-require_dependency "manifest/model/association"
-require_dependency "manifest/model/list"
+ManifestBase.require_modules "manifest/model/", 
+  %w|attribute association list tree|
 
 module Manifest
-  class Model
+  class Model < ManifestBase::Base
+    include ModelModule
     
-    def self.manager manifest, my_manifests
-      models = {}
-      my_manifests.each {|model_name, model_manifest|
-        models[model_name] = self.new(manifest, model_name, model_manifest)
+    attr :attributes, :associations, :tree, :list, :extend_column_name
+    
+    def set_default
+      super
+      @json['attributes'] ||= {}
+      @json['associations'] ||= {}
+      @json['tree'] ||= {}
+      @json['list'] ||= {}
+      @json['extend_column_name'] ||= 'classname'
+      @json['peta'] ||= {}
+      @json['attributes']['id'] = {
+        'type' => 'number',
+        'primary_key' => 1,
+        'rules' => {
+          'number' => true
+        }
+      }
+      @json['attributes']['created_at'] = {
+        'type' => 'datetime',
+      }
+      @json['attributes']['updated_at'] = {
+        'type' => 'datetime',
+      }
+    end
+    
+    def init
+      super
+      @attributes = ManifestBase.load_name_values self, @json, 'attributes', Attribute
+      @associations = ManifestBase.load_value self, @json, 'associations', Association
+      @tree = {}
+      @json['tree'].each {|tree_name, parent_model_name|
+        @tree[tree_name] = Tree.new(self, tree_name, parent_model_name)
       }
-      models
+      @extend_column_name = @json['extend_column_name']
+      @list = ManifestBase.load_value self, @json, 'list', List
     end
     
-    attr :model_name, :model_manifest, 
-      :attributes, :associations, :tree, :lists
-    def initialize manifest, model_name, model_manifest
-      @manifest = manifest
-      @model_name = model_name
-      @model_manifest = model_manifest
-      self.set_default
-      self.init
+    def model_name
+      @name
     end
     
-    def set_default
-      @model_manifest['attributes'] ||= {}
-      @model_manifest['associations'] ||= {}
-      @model_manifest['tree'] ||= {}
-      @model_manifest['lists'] ||= {}
+    def classify
+      ::Manifest.item_name_to_model @name
     end
     
-    def init
-      @attributes = {}
-      @model_manifest['attributes'].each {|attribute_name, attribute_manifest|
-        @attributes[attribute_name] = ModelModule::Attribute.new(self, attribute_name, attribute_manifest)
+    def table_name
+      self.classify.table_name
+    end
+    
+    def valid_encode_columns
+      r = []
+      @attributes.each {|attribute_name, attribute|
+        next unless attribute.type == 'text'
+        r << attribute_name
+      }
+      r
+    end
+    
+    def owner_type
+      return :author if @attributes['author_id']
+      return :artist if @attributes['artist_id']
+      false
+    end
+    
+    def content_model
+      return true if self.owner_type
+      false
+    end
+    
+    def child_model_manifests tree_name
+      r = []
+      ::Manifest.manifest.models.each {|child_model_name, child_model_manifest|
+        next unless child_model_manifest.tree[tree_name]
+        next unless child_model_manifest.tree[tree_name].parent_model_name == @model_name
+        r << child_model_manifest
       }
-      @associations = ModelModule::Association.new(self, @model_manifest['associations'])
-      @tree = @model_manifest['tree']
-      @lists = {}
-      @model_manifest['lists'].each {|list_name, list_manifest|
-        @lists[list_name] = ModelModule::ListFactory.factory(self, list_name, list_manifest)
+      r
+    end
+    
+    def child_models tree_name
+      self.child_model_manifests(tree_name).map {|child_model_manifest|
+        child_model_manifest.classify
       }
     end
     
-    def classify
-      ::Manifest.item_name_to_model @model_name
+    def child_element_names tree_name
+      self.child_models(tree_name).map {|child_model|
+        self.associations.child_element_name(child_model)
+      }
     end
     
-    def tree_model_manifest tree_name
-      if tree_name
-        if @tree[tree_name]
-          r = ::Manifest.manifest.models[@tree[tree_name]]
-          raise "undefined tree_model for models > #{@model_name} > tree > #{tree_name}\n" unless r
-          r
-        else
-          nil
-        end
-      else
-        nil
-      end
+    def child_element_name item_name
+      self.associations.child_element_name(child_model_manifest)
     end
     
   end