module Dag
  module CLI
    class Import
      def initialize(client, db, table, files, format, params)
        @client = client
        @db = db
        @table = table
        @files = files
        @params = params
        @format = format
      end

      def execute
        begin
          @client.database(@db)
        rescue Dag::Client::DatabaseNotFound
          @client.create_database(@db)
        end

        begin
          @client.database(@db).table(@table)
        rescue Dag::Client::TableNotFound
          @client.create_table(@db, @table, format: @format)
        end

        @client.import(@db, @table, @files, @params)
      end
    end
  end
end