module SequelRails
  module Storage
    class Jdbc < Abstract

      def _is_mysql?
        config['adapter'].match(/^jdbc:mysql/)
      end

      def _is_postgres?
        config['adapter'].match(/^jdbc:postgresql/)
      end

      def _is_sqlite?
        config['adapter'].match(/^jdbc:sqlite/)
      end

      def _root_url
        config['url'].scan(/^jdbc:mysql:\/\/[\w\.]*:?\d*/).first
      end

      def db_name
        config['database']
      end

      def _params
        config['url'].scan(/\?.*$/).first
      end

      def _create
        if _is_sqlite?
          return if in_memory?
          ::Sequel.connect config['url']
        elsif _is_mysql?
          ::Sequel.connect("#{_root_url}#{_params}") do |db|
            db.execute("CREATE DATABASE IF NOT EXISTS `#{db_name}` DEFAULT CHARACTER SET #{charset} DEFAULT COLLATE #{collation}")
          end
        elsif _is_postgres?
          adapter = ::SequelRails::Storage::Postgres.new(config)
          adapter._create
        end
      end

      def _drop
        if _is_sqlite?
          return if in_memory?
          FileUtils.rm db_name if File.exists? db_name
        elsif _is_mysql?
          ::Sequel.connect("#{_root_url}#{_params}") do |db|
            db.execute("DROP DATABASE IF EXISTS `#{db_name}`")
          end
        elsif _is_postgres?
          adapter = ::SequelRails::Storage::Postgres.new(config)
          adapter._drop
        end
      end

      def _dump(filename)
        if _is_postgres?
          adapter = ::SequelRails::Storage::Postgres.new(config)
          adapter._dump(filename)
        else
          raise NotImplementedError
        end
      end

      def _load(filename)
        if _is_postgres?
          adapter = ::SequelRails::Storage::Postgres.new(config)
          adapter._load(filename)
        else
          raise NotImplementedError
        end
      end

      private

      def collation
        @collation ||= super || "utf8_unicode_ci"
      end

      def in_memory?
        return false unless _is_sqlite?
        database == ":memory:"
      end

    end
  end
end