Merge model from ninoxe gem

1 files changed, 94 insertions, 0 deletions

diff --git a/app/models/chouette/command.rb b/app/models/chouette/command.rb
new file mode 100644
index 000000000..d2995a000
--- /dev/null
+++ b/app/models/chouette/command.rb
@@ -0,0 +1,94 @@
+require 'tmpdir'
+
+#if RUBY_PLATFORM == "java"
+  # FIXME disable remove_entry_secure because incompatible with jruby ?! 
+  # See http://jira.codehaus.org/browse/JRUBY-4082
+  module FileUtils
+    def self.remove_entry_secure(*args)
+      self.remove_entry *args
+    end
+  end
+#end
+
+class Chouette::Command
+
+  include Chouette::CommandLineSupport
+
+  @@command = "chouette"
+  cattr_accessor :command
+
+  attr_accessor :database, :schema, :host, :user, :password, :port
+
+  def initialize(options = {})
+    database_options_from_active_record.merge(options).each do |k,v|
+      send "#{k}=", v
+    end
+  end
+
+  def database_options_from_active_record
+    config = Chouette::ActiveRecord.connection_pool.spec.config
+    { 
+      :database => config[:database], 
+      :user => config[:username],
+      :password => config[:password],
+      :port => config[:port],
+      :host => (config[:host] or "localhost")
+    }
+  end
+
+
+  def run!(options = {})
+    Dir.mktmpdir do |config_dir|
+      chouette_properties = File.join(config_dir, "chouette.properties")
+      open(chouette_properties, "w") do |f|
+        f.puts "database.name = #{database}"
+        f.puts "database.schema = #{schema}"
+        #f.puts "database.showsql = true"
+        f.puts "hibernate.username = #{user}"
+        f.puts "hibernate.password = #{password}"
+        f.puts "jdbc.url=jdbc:postgresql://#{host}:#{port}/#{database}"
+        f.puts "jdbc.username = #{user}"
+        f.puts "jdbc.password = #{password}"
+        #f.puts "database.hbm2ddl.auto=update"
+      end
+
+      logger.debug "Chouette properties: #{File.readlines(chouette_properties).collect(&:strip).join(', ')}"
+
+      command_line = "#{command} -classpath #{config_dir} #{command_options(options)}"
+      logger.debug "Execute '#{command_line}'"
+
+      execute! command_line
+    end
+  end
+
+  class Option
+
+    attr_accessor :key, :value
+
+    def initialize(key, value)
+      @key, @value = key.to_s, value
+    end
+
+    def command_key
+      key.camelize(:lower) 
+    end
+
+    def to_s
+      unless value == true
+        "-#{command_key} #{value}"
+      else
+        "-#{command_key}"
+      end
+    end
+
+  end
+
+  def command_options(options)
+    options.collect do |key, value|
+      Option.new(key, value)
+    end.sort_by(&:key).join(' ')
+  end
+
+  
+
+end