Started to write data migration code.

bricks/zena/zena/migrate/03_zerox1_data.rb

@@ -12,26 +12,175 @@ def self.up
       puts "=>  No legacy data to migrate."
       return
     end
-    puts "Plop!"
     # ============================================ contact_contents
     # migrate content to properties
+    class ContactContent < ActiveRecord::Base
+      TRANSLATE_KEYS = {
+        'name' => 'last_name',
+        'zip'  => 'postal_code',
+        'city' => 'locality',
+      }
+      def migrate(version)
+        %w{first_name name address zip city telephone mobile email birthday country}.each do |key|
+          value = self[key]
+          next if value.blank?
+          version.prop[TRANSLATE_KEYS[key] || key] = value
+        end
+      end
+    end
 
     # ============================================ document_contents
     # migrate page ref to attachments
+    class AttachmentMig < Versions::SharedAttachment
+      set_table_name :attachments
+
+      def filepath(format=nil)
+        "#{SITES_ROOT}#{$size.data_path}/full_new/#{super()}"
+      end
+    end
+
+    add_column :document_contents, :attachment_id, :integer
+
     # migrate content to properties
+    class DocumentContent < ActiveRecord::Base
+      TRANSLATE_KEYS = {
+        'exif_json' => 'exif'
+      }
+
+      def migrate_attachment(version)
+        if !self.attachment_id
+          # old path
+          digest = Digest::SHA1.hexdigest(self[:id].to_s)
+          # make sure name is not corrupted
+          fname = name.gsub(/[^a-zA-Z\-_0-9]/,'')
+
+          old_path = "#{SITES_ROOT}#{$site.data_path}/full/#{digest[0..0]}/#{digest[1..1]}/#{digest[2..2]}/#{fname}"
+
+          atta = AttachmentMig.new
+          atta['user_id'] = version.user_id
+          atta['site_id'] = version.site_id
+          atta.file = File.new(old_path)
+          atta.save
+          self.attachment_id = atta.id
+          self.save
+        end
+        version.attachment_id = self.attachment_id
+      end
+
+      def migrate(version)
+        exif = self['exif_json']
+        if !exif.blank?
+          value.prop['exif'] = ExifData.json_create('data' => exif)
+        end
+        %w{size content_type ext width height}.each do |key|
+          value = self[key]
+          next if value.blank?
+          version.prop[TRANSLATE_KEYS[key] || key] = value
+        end
+        migrate_attachment(version)
+      end
+    end
+
+    # ============================================ dyn_attributes
+    class DynAttribute < ActiveRecord::Base
+      set_table_name :dyn_attributes
+
+      def migrate(version)
+        version.prop[key] = value
+      end
+    end
+
+    # ============================================ template_contents
+    # migrate content to properties
+    # rebuild index ==> should recreate idx_templates content
+    class TemplateContent < ActiveRecord::Base
+      TRANSLATE_KEYS = {
+        'klass' => 'target_klass'
+      }
+      def migrate(version)
+        version['skin_id'] = version.node[:section_id]
+
+        %w{format tkpath mode klass}.each do |key|
+          value = self[key]
+          next if value.blank?
+          version.prop[TRANSLATE_KEYS[key] || key] = value
+        end
+      end
+    end
+
+    class NodeMig < ActiveRecord::Base
+      set_table_name :nodes
+      has_many :versions, :class_name => 'VersionMig', :foreign_key => 'node_id'
+    end
 
     # ============================================ versions
+    class VersionMig < ActiveRecord::Base
+      set_table_name :versions
+      belongs_to :node, :class_name => 'NodeMig', :foreign_key => 'node_id'
+      has_one :contact_content, :foreign_key => 'version_id'
+      has_one :document_content, :foreign_key => 'version_id'
+      has_one :template_content, :foreign_key => 'node_id', :primary_key => 'node_id'
+
+      include Property
+      # Should be the same serialization as in Version and Site
+      include Property::Serialization::JSON
+
+      def migrate!
+        migrate_contact
+        migrate_document
+        migrate_template
+        migrate_dyn_attributes
+        save
+      end
+
+      def migrate_contact
+        if contact_content
+          contact_content.migrate(self)
+        end
+      end
+
+      def migrate_document
+        if document_content
+          document_content.migrate(self)
+        end
+      end
+
+      def migrate_template
+        if template_content
+          template_content.migrate(self)
+        end
+      end
+
+      def migrate_dyn_attributes
+        DynAttribute.find(:all, :conditions => ["owner_table = 'versions' AND owner_id = ?", self.id]).each do |dyn|
+          dyn.migrate(self)
+        end
+      end
+    end
+
     # === dyn_attributes
-    # migrate content to verisons properties
+    # migrate content to version properties
     # === idx_text_high => title
     # === idx_text_medium => summary
     # === idx_text_low => text
 
-    # migrate to properties
+    class SiteMig < ActiveRecord::Base
+      set_table_name :sites
 
-    # ============================================ template_contents
-    # migrate content to properties
-    # rebuild index ==> should recreate idx_templates content
+      def data_path
+        "/#{self[:host]}/data"
+      end
+    end
+
+    # migrate to properties
+    SiteMig.all.each do |s|
+      $site = s
+      NodeMig.find(:all, :conditions => ['site_id = ?', s.id]).each do |n|
+        n.versions.each do |v|
+          v.migrate!
+        end
+      end
+    end
 
     # ============================================ nodes
     # 1. Set skin_id from skin name