const { DBSQLClient } = require('../');

const client = new DBSQLClient();

const host = '****.databricks.com';
const path = '/sql/2.0/warehouses/****';
const token = 'dapi********************************';

client
  .connect({ host, path, token })
  .then(async (client) => {
    const session = await client.openSession();
    await createTables(session);

    const typeInfo = await session.getTypeInfo().then(handleOperation);
    console.log(typeInfo);
    console.log();

    const catalogs = await session.getCatalogs().then(handleOperation);
    console.log(catalogs);
    console.log();

    const schemas = await session.getSchemas({}).then(handleOperation);
    console.log(schemas);
    console.log();

    const tables = await session.getTables({}).then(handleOperation);
    console.log(tables);
    console.log();

    const tableTypes = await session.getTableTypes({}).then(handleOperation);
    console.log(tableTypes);
    console.log();

    const columns = await session.getColumns({}).then(handleOperation);
    console.log(columns);
    console.log();

    await session.close();
    await client.close();
  })
  .catch((error) => {
    console.log(error);
  });

async function handleOperation(operation) {
  const result = await operation.fetchAll({
    progress: true,
    callback: (stateResponse) => {
      console.log(stateResponse.taskStatus);
    },
  });
  await operation.close();
  return result;
}

const createTables = async (session) => {
  await session
    .executeStatement('CREATE TABLE IF NOT EXISTS table1 ( id STRING, value INTEGER )')
    .then(handleOperation);
  await session
    .executeStatement('CREATE TABLE IF NOT EXISTS table2 ( id STRING, table1_fk INTEGER )')
    .then(handleOperation);
};