#! -*- encoding : UTF-8 -*-
require 'fileutils'
require 'yaml'
require 'active_record'
require 'sqlite3'

module Arxutils
  module Dbutil
    DB_DIR = 'db'
    MIGRATE_DIR = 'migrate'
    DBCONFIG_SQLITE3 = 'sqlite3'
    DBCONFIG_MYSQL = 'mysql'
    CONFIG_DIR = 'config'
    DATABASELOG = 'database.log'
    
    class Dbinit
      attr_accessor :dbconfig_dest_path , :dbconfig_src_path , :dbconfig_src_fname , :dbconfig_dest_fname , :migrate_dir
      
      def initialize( db_dir , migrate_base_dir , src_config_dir , dbconfig , env, log_fname, forced = false )
        @db_dir = db_dir
        @src_config_dir  = src_config_dir
        @dest_config_dir  = "config"
        @dbconfig_dest_fname = "#{dbconfig}.yaml"
        @dbconfig_src_fname = "#{dbconfig}.tmpl"
        @dbconfig_dest_path = File.join( @dest_config_dir , @dbconfig_dest_fname)
        @dbconfig_src_path = File.join(@src_config_dir  , @dbconfig_src_fname)
        @env = env
        @log_fname = log_fname
        
        if @db_dir and @log_fname
          @log_path = File.join( @db_dir , @log_fname )
          @migrate_dir = File.join( @db_dir , migrate_base_dir )
        end
        FileUtils.mkdir_p( @db_dir ) if @db_dir
        FileUtils.mkdir_p( @migrate_dir ) if @migrate_dir
        FileUtils.mkdir_p( @dest_config_dir )
        if forced
          FileUtils.rm( Dir.glob( File.join( @migrate_dir , "*"))) if @migrate_dir
          FileUtils.rm( Dir.glob( File.join( @dest_config_dir  , "*")))
        end
      end
      
      def setup
        dbconfig = YAML.load( File.read( @dbconfig_dest_path ) )
        ActiveRecord::Base.establish_connection(dbconfig[@env])
        ActiveRecord::Base.logger = Logger.new( @log_path )
      end
    end
  end
end