driftsql 1.0.1 → 1.0.3

package/dist/index.d.mts

@@ -2,6 +2,9 @@ import { PoolConfig } from 'pg';
 import { Config } from '@libsql/client';
 import mysql from 'mysql2/promise';
 
+type Drivers = ClientOptions['drivers'];
+declare const inspectDB: (drivers: Drivers) => Promise<never>;
+
 type UnifiedQueryResult<T extends Record<string, any>> = {
     rows: T[];
     rowCount: number;
@@ -30,8 +33,10 @@ declare class DriftSQLClient<DT> {
     private libsqlClient?;
     private neonClient?;
     private postgresClient?;
+    private drivers;
     constructor(options: ClientOptions);
     private convertLibsqlResult;
+    readonly inspect: () => Promise<void>;
     query<T extends Record<string, any>>(query: string, args?: (string | number | boolean | null)[]): Promise<UnifiedQueryResult<T>>;
     status(): Promise<{
         ok: boolean;
@@ -46,5 +51,5 @@ declare class DriftSQLClient<DT> {
     close(): Promise<void>;
 }
 
-export { DriftSQLClient };
+export { DriftSQLClient, inspectDB };
 export type { ClientOptions };

package/dist/index.d.ts

@@ -2,6 +2,9 @@ import { PoolConfig } from 'pg';
 import { Config } from '@libsql/client';
 import mysql from 'mysql2/promise';
 
+type Drivers = ClientOptions['drivers'];
+declare const inspectDB: (drivers: Drivers) => Promise<never>;
+
 type UnifiedQueryResult<T extends Record<string, any>> = {
     rows: T[];
     rowCount: number;
@@ -30,8 +33,10 @@ declare class DriftSQLClient<DT> {
     private libsqlClient?;
     private neonClient?;
     private postgresClient?;
+    private drivers;
     constructor(options: ClientOptions);
     private convertLibsqlResult;
+    readonly inspect: () => Promise<void>;
     query<T extends Record<string, any>>(query: string, args?: (string | number | boolean | null)[]): Promise<UnifiedQueryResult<T>>;
     status(): Promise<{
         ok: boolean;
@@ -46,5 +51,5 @@ declare class DriftSQLClient<DT> {
     close(): Promise<void>;
 }
 
-export { DriftSQLClient };
+export { DriftSQLClient, inspectDB };
 export type { ClientOptions };

package/dist/index.mjs

@@ -3,6 +3,140 @@ import ky from 'ky';
 import { Pool } from 'pg';
 import { createClient } from '@libsql/client';
 import mysql from 'mysql2/promise';
+import fs from 'node:fs/promises';
+
+const supportedDrivers = ["postgres"];
+const mapPostgresToTypeScript = (postgresType, isNullable = false) => {
+  const nullable = isNullable ? " | null" : "";
+  switch (postgresType.toLowerCase()) {
+    case "uuid": {
+      return `string${nullable}`;
+    }
+    case "character varying":
+    case "varchar":
+    case "text":
+    case "char":
+    case "character": {
+      return `string${nullable}`;
+    }
+    case "integer":
+    case "int":
+    case "int4":
+    case "smallint":
+    case "int2":
+    case "bigint":
+    case "int8":
+    case "serial":
+    case "bigserial":
+    case "numeric":
+    case "decimal":
+    case "real":
+    case "float4":
+    case "double precision":
+    case "float8": {
+      return `number${nullable}`;
+    }
+    case "boolean":
+    case "bool": {
+      return `boolean${nullable}`;
+    }
+    case "timestamp":
+    case "timestamp with time zone":
+    case "timestamp without time zone":
+    case "timestamptz":
+    case "date":
+    case "time":
+    case "time with time zone":
+    case "time without time zone":
+    case "timetz":
+    case "interval": {
+      return `Date${nullable}`;
+    }
+    case "json":
+    case "jsonb": {
+      return `any${nullable}`;
+    }
+    case "array": {
+      return `any[]${nullable}`;
+    }
+    case "bytea": {
+      return `Buffer${nullable}`;
+    }
+    default: {
+      console.warn(`Unknown PostgreSQL type: ${postgresType}, defaulting to 'any'`);
+      return `any${nullable}`;
+    }
+  }
+};
+const inspectDB = async (drivers) => {
+  if (!drivers) throw new Error("No drivers provided for inspection");
+  const configuredDrivers = Object.keys(drivers).filter((key) => drivers[key] !== void 0);
+  if (configuredDrivers.length === 0) {
+    throw new Error("No drivers are configured");
+  }
+  const supportedConfiguredDrivers = configuredDrivers.filter((driver) => supportedDrivers.includes(driver));
+  if (supportedConfiguredDrivers.length === 0) {
+    throw new Error(`No supported drivers found. Configured: ${configuredDrivers.join(", ")}. Supported: ${supportedDrivers.join(", ")}`);
+  }
+  console.log(`Found supported drivers: ${supportedConfiguredDrivers.join(", ")}`);
+  let generatedTypes = "";
+  const client = new DriftSQLClient({ drivers });
+  const tables = await client.query(
+    `SELECT table_name 
+     FROM information_schema.tables 
+     WHERE table_schema = 'public' 
+     AND table_type = 'BASE TABLE'
+     ORDER BY table_name`
+  );
+  console.log("Tables in the database:", tables.rows.map((t) => t.table_name).join(", "));
+  for (const table of tables.rows) {
+    const tableName = table.table_name;
+    console.log(`Inspecting table: ${tableName}`);
+    const columns = await client.query(
+      `
+      SELECT 
+        column_name, 
+        data_type, 
+        is_nullable,
+        column_default
+      FROM information_schema.columns 
+      WHERE table_name = $1 
+      AND table_schema = 'public'
+      ORDER BY ordinal_position
+    `,
+      [tableName]
+    );
+    if (columns.rows.length === 0) {
+      console.log(`No columns found for table: ${tableName}`);
+      continue;
+    }
+    console.log(`Columns in ${tableName}:`, columns.rows.map((c) => `${c.column_name} (${c.data_type}${c.is_nullable === "YES" ? ", nullable" : ""})`).join(", "));
+    const uniqueColumns = /* @__PURE__ */ new Map();
+    columns.rows.forEach((col) => {
+      if (!uniqueColumns.has(col.column_name)) {
+        uniqueColumns.set(col.column_name, col);
+      }
+    });
+    generatedTypes += `export interface ${tableName.charAt(0).toUpperCase() + tableName.slice(1)} {
+`;
+    for (const col of uniqueColumns.values()) {
+      const tsType = mapPostgresToTypeScript(col.data_type, col.is_nullable === "YES");
+      generatedTypes += `  ${col.column_name}: ${tsType};
+`;
+    }
+    generatedTypes += "}\n\n";
+  }
+  generatedTypes += "export interface Database {\n";
+  for (const table of tables.rows) {
+    const tableName = table.table_name.charAt(0).toUpperCase() + table.table_name.slice(1);
+    generatedTypes += `  ${tableName}: ${tableName};
+`;
+  }
+  generatedTypes += "}\n\n";
+  await fs.writeFile("db-types.ts", generatedTypes, "utf8");
+  console.log("TypeScript types written to db-types.ts");
+  process.exit(0);
+};
 
 class DriftSQLClient {
   client;
@@ -11,6 +145,7 @@ class DriftSQLClient {
   libsqlClient;
   neonClient;
   postgresClient;
+  drivers;
   constructor(options) {
     this.client = ky.create({
       prefixUrl: options.url,
@@ -33,6 +168,7 @@ class DriftSQLClient {
     this.pool = options.drivers?.postgres ? new Pool(options.drivers.postgres) : void 0;
     this.libsqlClient = options.drivers?.libsql ? createClient(options.drivers.libsql) : void 0;
     this.mysqlClient = options.drivers?.mysql ? mysql.createConnection(options.drivers.mysql) : void 0;
+    this.drivers = options.drivers || {};
   }
   convertLibsqlResult(result) {
     const rows = result.rows.map((row) => {
@@ -49,6 +185,9 @@ class DriftSQLClient {
       fields: result.columns.map((col) => ({ name: col, dataTypeID: 0 }))
     };
   }
+  inspect = async () => {
+    return inspectDB(this.drivers);
+  };
   async query(query, args) {
     if (this.pool) {
       try {
@@ -222,7 +361,13 @@ class DriftSQLClient {
     if (this.libsqlClient) {
       this.libsqlClient.close();
     }
+    if (this.mysqlClient) {
+      await (await this.mysqlClient).end();
+    }
+    if (this.postgresClient) {
+      await this.postgresClient.end();
+    }
   }
 }
 
-export { DriftSQLClient };
+export { DriftSQLClient, inspectDB };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "driftsql",
-  "version": "1.0.1",
+  "version": "1.0.3",
   "author": "lasse vestergaard",
   "description": "A lightweight SQL client for TypeScript, supporting multiple databases like PostgreSQL, MySQL, and LibSQL.",
   "repository": "lassejlv/driftsql",