From 34359e7a6033a39cd8f7de7148240ac37a5eda57 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Carlos=20Mart=C3=ADn?= <cmartins@fdi.ucm.es>
Date: Tue, 18 Oct 2011 12:18:11 +0200
Subject: [PATCH] Bug #847 - #901: onedb migrator adds USER TEMPLATE &
 AUTH_DRIVER, IMAGE FSTYPE & PATH.

---
 src/onedb/3.0.0_to_3.1.0.rb | 56 +++++++++++++++++++++++++++++++++++++
 1 file changed, 56 insertions(+)

diff --git a/src/onedb/3.0.0_to_3.1.0.rb b/src/onedb/3.0.0_to_3.1.0.rb
index b2700da390..0eac37f7b3 100644
--- a/src/onedb/3.0.0_to_3.1.0.rb
+++ b/src/onedb/3.0.0_to_3.1.0.rb
@@ -45,6 +45,62 @@ module Migrator
             end
         }
 
+        @db.run "ALTER TABLE image_pool RENAME TO old_image_pool;"
+        @db.run "CREATE TABLE image_pool (oid INTEGER PRIMARY KEY, name VARCHAR(128), body TEXT, uid INTEGER, gid INTEGER, public INTEGER, UNIQUE(name,uid) );"
+
+        @db.fetch("SELECT * FROM old_image_pool") do |row|
+            doc = Document.new(row[:body])
+
+            fstype = ""
+            doc.root.each_element("TEMPLATE/FSTYPE") { |e|
+                fstype = e.text
+            }
+
+            fstype_elem      = doc.root.add_element("FSTYPE")
+            fstype_elem.text = fstype
+
+            path = ""
+            doc.root.each_element("TEMPLATE/PATH") { |e|
+                path = e.text
+            }
+
+            path_elem       = doc.root.add_element("PATH")
+            path_elem.text  = path
+
+            @db[:image_pool].insert(
+                :oid        => row[:oid],
+                :name       => row[:name],
+                :body       => doc.root.to_s,
+                :uid        => row[:uid],
+                :gid        => row[:gid],
+                :public     => row[:public])
+        end
+
+        @db.run "DROP TABLE old_image_pool;"
+
+
+        @db.run "ALTER TABLE user_pool RENAME TO old_user_pool;"
+        @db.run "CREATE TABLE user_pool (oid INTEGER PRIMARY KEY, name VARCHAR(128), body TEXT, UNIQUE(name));"
+
+        @db.fetch("SELECT * FROM old_user_pool") do |row|
+            doc = Document.new(row[:body])
+
+            # TODO: Try to guess if the password contains a DN and set the
+            # driver to 'x509', or assume ssh if the password is not hex
+            auth_elem      = doc.root.add_element("AUTH_DRIVER")
+            auth_elem.text = "core"
+
+            doc.root.add_element("TEMPLATE")
+
+            @db[:user_pool].insert(
+                :oid        => row[:oid],
+                :name       => row[:name],
+                :body       => doc.root.to_s)
+        end
+
+        @db.run "DROP TABLE old_user_pool;"
+
+
         return true
     end