require "fileutils"

module Merb
  module Orms
    module Sequel

      class << self

        def config_file() Merb.root / "config" / "database.yml" end
        def sample_dest() Merb.root / "config" / "database.yml.sample" end
        def sample_source() File.dirname(__FILE__) / "database.yml.sample" end
      
        def copy_sample_config
          FileUtils.cp sample_source, sample_dest unless File.exists?(sample_dest)
        end
      
        def config

          @config ||=
          begin
            # Convert string keys to symbols
            full_config = Erubis.load_yaml_file(config_file)
            config = (Merb::Plugins.config[:merb_sequel] = {})
            (full_config[Merb.environment.to_sym] || full_config[Merb.environment]).each do |key, value|
              config[key.to_sym] = value
            end
            config
          end

        end
      
        # Database connects as soon as the gem is loaded
        def connect
          
          require "sequel"

          if File.exists?(config_file)
            Merb.logger.info!("Connecting to the '#{config[:database]}' database on '#{config[:host]}' using '#{config[:adapter]}' ...")
            connection = ::Sequel.connect(config_options(config))
            Merb.logger.error!("Connection Error: #{e}") unless connection
            connection
          else
            copy_sample_config
            Merb.logger.set_log(STDERR)
            Merb.logger.error! "No database.yml file found in #{Merb.root}/config."
            Merb.logger.error! "A sample file was created called config/database.sample.yml for you to copy and edit."
            exit(1)
          end
          
        end
        
        def config_options(config = {})
          
          options = {}
          options[:adapter]  = (config[:adapter]  || "sqlite")
          options[:host]     = (config[:host]     || "localhost")
          options[:user]     = (config[:username] || config[:user] || "root")
          options[:password] = config[:password] if config[:password]
          if (config[:encoding] || config[:charset])
            options[:encoding] = (config[:encoding] || config[:charset])
          end
          options[:database] = config[:database]  if config[:database]
          options[:logger]   = Merb.logger
          options
        end
        
        # Registering this ORM lets the user choose sequel as a session store
        # in merb.yml's session_store: option.
        def register_session_type
          Merb.register_session_type(
          "sequel",
          "merb/session/sequel_session",
          "Using Sequel database sessions"
          )
        end

      end
      
    end
    
  end

end