lib/arxutils/migrate.rb in arxutils-0.1.31 vs lib/arxutils/migrate.rb in arxutils-0.1.35

@@ -1,40 +1,73 @@
 # -*- coding: utf-8 -*-
 require 'arxutils'
 
 require 'fileutils'
+require 'active_support'
 require 'active_record'
+require 'pp'
 
 module Arxutils
   class Migrate
     attr_accessor :dbinit , :dbconfig_dest_path , :dbconfig_dest_fname , :dbconfig_src_path , :dbconfig_src_fname
     
-    def Migrate.migrate( data_ary , idx , dbconfig , forced )
+    def Migrate.migrate( data_ary , relation_def_fpath , module_name, count_classname_downcase , count_field , dbconfig , forced )
       src_config_dir = Arxutils.configdir
-      mig = Migrate.new(Dbutil::DB_DIR, Dbutil::MIGRATE_DIR , src_config_dir , dbconfig, Dbutil::DATABASELOG, forced )
-      mig.make_dbconfig( data_ary[idx] )
+      mig = Migrate.new( Dbutil::MIGRATE_DIR , src_config_dir , dbconfig, Dbutil::DATABASELOG, forced )
+      # dbconfigのテンプレートは内容が固定である。convertを呼び出し、Arxのインスタンスを作成するときに、適切なdata_aryの要素を与える必要がある（ただしテンプレートへの埋め込みには用いられない
+      mig.make_dbconfig( dbconfig )
       
-      data_ary.reduce(0) do |next_num , x| 
+      data_ary.reduce(0) { |next_num , x| 
         mig.make( next_num , x )
-      end
+      }
 
+      content_array = data_ary.map { |x| 
+        mig.make_relation( x , "count", "end_count_id" )
+      }.select{ |x| x.size > 0 }
+      need_count_class_plural = content_array.reduce([]){ |s,x|
+        s << x[:need_count_class_plural] if x[:need_count_class_plural] != nil
+        s
+      }
+      if content_array.size > 0
+        data_count = {count_classname: "Count" ,
+                      count_field: count_field,
+                      need_count_class_plural: need_count_class_plural,
+                     }
+        ary = content_array.collect{|x| x[:content] }.flatten
+        count_content = mig.convert_count_class_relation( data_count , "relation_count.tmpl" )
+        ary.unshift( count_content )
+        content_array = ary
+      end
+      File.open( relation_def_fpath , 'w' , {:encoding => Encoding::UTF_8}){ |f|
+        f.puts("module #{module_name}")
+        content_array.map{ |content|
+          f.puts( content )
+          f.puts( "\n" )
+        }
+        f.puts("end")
+      }
+      
       Dbutil::DbMgr.setup( mig.dbinit )
 
       mig.migrate
     end
     
-    def initialize( db_dir , migrate_dir , config_dir , dbconfig, log_fname, forced = false )
-      @dbinit = Dbutil::Dbinit.new( db_dir , migrate_dir , config_dir , dbconfig, log_fname, forced )
+    def initialize( migrate_base_dir , config_dir , dbconfig, log_fname, forced = false )
+      @dbinit = Dbutil::Dbinit.new( migrate_base_dir , config_dir , dbconfig, log_fname, forced )
       @dbconfig_dest_path = @dbinit.dbconfig_dest_path
       @dbconfig_src_path = @dbinit.dbconfig_src_path
       @dbconfig_src_fname = @dbinit.dbconfig_src_fname
 
-      @migrate_dir = migrate_dir
+      @migrate_dir = @dbinit.migrate_dir
       @src_path = Arxutils.templatedir
       @src_config_path = Arxutils.configdir
     end
 
+    def convert_count_class_relation( data , src_fname )
+      convert( data , @src_path , src_fname )
+    end
+
     def convert( data , src_dir , src_fname )
       arx = Arx.new( data , File.join( src_dir, src_fname ) )
       arx.create
     end
     
@@ -42,10 +75,32 @@
       content = convert( data , @src_config_path , @dbconfig_src_fname )
       File.open( @dbconfig_dest_path , 'w' , {:encoding => Encoding::UTF_8}){ |f|
         f.puts( content )
       }
     end
+
+    def make_relation( data , count_classname_downcase , count_field )
+      if data[:classname_downcase] != count_classname_downcase
+        data[:flist].reduce( { content: [], need_count_class: nil } ){ |s, x|
+          case x
+          when "base" , "noitem"
+            name_base = "relation"
+            data[:relation] = [] unless data[:relation]
+          else
+            data[:count_classname_downcase] = count_classname_downcase
+            data[:count_field] = count_field
+            name_base = "relation_#{x}"
+            s[:need_count_class_plural] ||= data[:plural]
+          end
+          content = convert( data , @src_path , "#{name_base}.tmpl" )
+          s[:content] << content
+          s
+        }
+      else
+        {}
+      end
+    end
     
     def make( next_num , data )
       data[:flist].reduce(next_num) do |idy , x|
         idy += 10
         content = convert( data , @src_path , "#{x}.tmpl" )
@@ -54,10 +109,9 @@
           additional = ""
         else
           additional = x
         end
         fname = File.join( @migrate_dir , sprintf("%03d_create_%s%s.rb" , idy , additional , data[:classname_downcase]) )
-        p fname
         File.open( fname , 'w' , {:encoding => Encoding::UTF_8}){ |f|
           f.puts( content )
         }
         idy
       end