image.rb in wakame-vdc-agents-11.06.0

- old
+ new

@@ -12,21 +12,23 @@
     inheritable_schema do
       Fixnum :boot_dev_type, :null=>false, :default=>BOOT_DEV_SAN
       Text :source, :null=>false
       String :arch, :size=>10, :null=>false
       Text :description
+      Boolean :is_public, :null=>false, :default=>false
       #Fixnum :parent_image_id
+      String :format, :null=>false, :default=>'raw'
 
       String :state, :size=>20, :null=>false, :default=>:init.to_s
+      index :is_public
     end
 
     # serialize plugin must be defined at the bottom of all class
     # method calls.
     # Possible source column data:
-    # vdc volume:
-    # {:type=>:vdcvol, :account_id=>'a-xxxxx', :snapshot_id=>'snap-xxxxxx'}
-    # {:type=>:http, :uri=>'http://localhost/xxx/xxx'}
+    # {:snapshot_id=>'snap-xxxxxx'}
+    # {:uri=>'http://localhost/xxx/xxx'}
     plugin :serialization
     serialize_attributes :yaml, :source
     
     def validate
       unless [BOOT_DEV_SAN, BOOT_DEV_LOCAL].member?(self.boot_dev_type)
@@ -34,13 +36,39 @@
       end
       
       unless HostPool::SUPPORTED_ARCH.member?(self.arch)
         errors.add(:arch, "Unsupported arch type: #{self.arch}")
       end
+      
+      # validate source
+      md = self.source
+      case self.boot_dev_type
+      when BOOT_DEV_LOCAL
+        errors.add(:source, "Unknown image URI") if md[:uri].nil? || md[:uri] == ''
+      when BOOT_DEV_SAN
+        errors.add(:source, "Unknown snapshot ID") if md[:snapshot_id].nil? || md[:snapshot_id] == '' || VolumeSnapshot[md[:snapshot_id]].nil?
+      end
     end
 
     def to_hash
       super.merge({:source=>self.source.dup, :description=>description.to_s})
+    end
+
+    # note on "lookup_account_id":
+    # the source column sometime contains the information which
+    # should not be shown to other accounts. so that the method takes
+    # an argument who is looking into then filters the data in source
+    # column accordingly.
+    def to_api_document(lookup_account_id)
+      h = to_hash
+      if self.account_id == lookup_account_id
+      else
+        if h[:source][:type] == :http
+          # do not show URI for non-owner accounts.
+          h[:source][:uri] = nil
+        end
+      end
+      h
     end
     
   end
 end