@graphenedata/cli 0.0.6 → 0.0.8

package/cli.ts

@@ -5,18 +5,17 @@ import {printDiagnostics, printTable} from './printer.ts'
 import {analyze, getDiagnostics, loadWorkspace, toSql, type Query} from '../lang/core.ts'
 import fs from 'fs-extra'
 import path from 'path'
-import {loadConfig} from '../lang/config.ts'
+import dotenv from 'dotenv'
+import {config, loadConfig} from '../lang/config.ts'
 import {runServeInBackground, stopGrapheneIfRunning} from './background.ts'
 import {check} from './check.ts'
-import {runQuery} from './connections/index.ts'
+import {getConnection, runQuery} from './connections/index.ts'
 import {loginPkce} from './auth.ts'
 
+dotenv.config({quiet: true})
 const program = new Command()
 
-program
-  .name('graphene')
-  .description('Graphene CLI')
-  .version('1.0.0')
+program.name('graphene').description('Graphene CLI').version('1.0.0')
 
 program.hook('preAction', async () => {
   if (process.env.CLI_DELAY) { // useful if you want to attach a debugger
@@ -51,6 +50,40 @@ program
     printTable(res.rows)
   })
 
+program.command('schema')
+  .description('Inspect database tables or describe a table')
+  .argument('[schema | table]', 'Optional schema or table name to describe')
+  .action(async (tableArg: string) => {
+    let connection = await getConnection()
+    let datasets = await connection.listDatasets()
+
+    // if there's no arg and more than one dataset, just list the datasets
+    if (!tableArg && datasets.length > 1) {
+      return console.log(`Datasets available:\n${datasets.join('\n')}`)
+    }
+
+    // figure out if you're wanting to list tables in a schema/dataset
+    let dsToList: string | null = null
+    let parts = tableArg ? tableArg.split('.') : []
+    if (datasets.includes(tableArg)) dsToList = tableArg // you gave the name of a dataset
+    else if (!tableArg && datasets.length == 1) dsToList = datasets[0] // only one dataset, and no args
+    else if (!tableArg && config.namespace) dsToList = config.namespace // default namespace configured
+    else if (!tableArg && config.dialect == 'duckdb') dsToList = '<default>'
+    else if (tableArg && config.dialect == 'snowflake' && parts.length == 2) dsToList = tableArg
+
+    if (dsToList) {
+      let tables = await connection.listTables(dsToList)
+      return console.log(`Tables${dsToList ? ` in ${dsToList}` : ''}:\n${tables.join('\n')}`)
+    }
+
+    // otherwise, assume you're wanting to see tables
+    let cols = await connection.describeTable(tableArg)
+    if (!cols.length) return console.log(`Table ${tableArg} not found`)
+    console.log(`table ${tableArg} (`)
+    cols.forEach(col => console.log(`  ${col.name} ${col.dataType}`))
+    console.log(')')
+  })
+
 program
   .command('serve')
   .description('Run the local server')