@slates-integrations/postgresql 0.2.0-rc.6 → 0.2.0-rc.9

package/src/lib/errors.ts

@@ -1,55 +0,0 @@
-import { ServiceError, badRequestError } from '@lowerdeck/error';
-
-export let postgresServiceError = (message: string) =>
-  new ServiceError(badRequestError({ message }));
-
-export let postgresUpstreamError = (
-  message: string,
-  options: {
-    reason?: string;
-    code?: string;
-    parent?: unknown;
-  } = {}
-) => {
-  let error = postgresServiceError(message);
-  error.data.reason = options.reason ?? 'postgresql_error';
-  error.data.upstreamCode = options.code;
-
-  if (options.parent instanceof Error) {
-    error.setParent(options.parent);
-  }
-
-  return error;
-};
-
-export let toPostgresServiceError = (error: unknown, fallbackMessage: string) => {
-  if (error instanceof ServiceError) {
-    return error;
-  }
-
-  if (error instanceof Error) {
-    return postgresUpstreamError(`${fallbackMessage}: ${error.message}`, {
-      parent: error
-    });
-  }
-
-  return postgresUpstreamError(fallbackMessage);
-};
-
-export let postgresFieldsError = (
-  fields: Record<string, string>,
-  fallbackMessage = 'Unknown error'
-) => {
-  let message = fields['message'] || fallbackMessage;
-  if (fields['detail']) message += ` - ${fields['detail']}`;
-  if (fields['hint']) message += ` (Hint: ${fields['hint']})`;
-  if (fields['position']) message += ` at position ${fields['position']}`;
-
-  return postgresUpstreamError(
-    `PostgreSQL error [${fields['code'] || 'UNKNOWN'}]: ${message}`,
-    {
-      reason: 'postgresql_server_error',
-      code: fields['code']
-    }
-  );
-};

package/src/lib/helpers.ts

@@ -1,48 +0,0 @@
-import { PostgresClient, type ConnectionConfig } from './client';
-
-export interface AuthOutput {
-  host: string;
-  port: number;
-  database: string;
-  username: string;
-  password: string;
-  sslMode: 'disable' | 'require' | 'verify-ca' | 'verify-full';
-  connectionString: string;
-}
-
-export interface ConfigOutput {
-  defaultSchema: string;
-  queryTimeout: number;
-  maxRows: number;
-}
-
-export let createClient = (auth: AuthOutput, config: ConfigOutput): PostgresClient => {
-  let connectionConfig: ConnectionConfig = {
-    host: auth.host,
-    port: auth.port,
-    database: auth.database,
-    username: auth.username,
-    password: auth.password,
-    sslMode: auth.sslMode,
-    queryTimeout: config.queryTimeout
-  };
-  return new PostgresClient(connectionConfig);
-};
-
-// Escape an identifier (table name, column name, etc.) for safe use in SQL
-export let escapeIdentifier = (name: string): string => {
-  return '"' + name.replace(/"/g, '""') + '"';
-};
-
-// Escape a literal value for safe use in SQL
-export let escapeLiteral = (value: string): string => {
-  return "'" + value.replace(/'/g, "''") + "'";
-};
-
-// Build a qualified table name with optional schema
-export let qualifiedTableName = (tableName: string, schemaName?: string): string => {
-  if (schemaName) {
-    return `${escapeIdentifier(schemaName)}.${escapeIdentifier(tableName)}`;
-  }
-  return escapeIdentifier(tableName);
-};

package/src/lib/protocol.ts

@@ -1,336 +0,0 @@
-// PostgreSQL Wire Protocol helpers
-// Implements the minimal subset of the v3 protocol needed for simple query execution
-
-export let MessageTypes = {
-  // Frontend (client -> server)
-  STARTUP: 0,
-  PASSWORD: 'p'.charCodeAt(0),
-  QUERY: 'Q'.charCodeAt(0),
-  TERMINATE: 'X'.charCodeAt(0),
-  // Backend (server -> client)
-  AUTHENTICATION: 'R'.charCodeAt(0),
-  ERROR_RESPONSE: 'E'.charCodeAt(0),
-  ROW_DESCRIPTION: 'T'.charCodeAt(0),
-  DATA_ROW: 'D'.charCodeAt(0),
-  COMMAND_COMPLETE: 'C'.charCodeAt(0),
-  READY_FOR_QUERY: 'Z'.charCodeAt(0),
-  NOTICE_RESPONSE: 'N'.charCodeAt(0),
-  PARAMETER_STATUS: 'S'.charCodeAt(0),
-  BACKEND_KEY_DATA: 'K'.charCodeAt(0),
-  EMPTY_QUERY_RESPONSE: 'I'.charCodeAt(0)
-} as const;
-
-export let AuthenticationTypes = {
-  OK: 0,
-  CLEARTEXT_PASSWORD: 3,
-  MD5_PASSWORD: 5,
-  SASL: 10,
-  SASL_CONTINUE: 11,
-  SASL_FINAL: 12
-} as const;
-
-// Simple byte buffer writer
-export class MessageWriter {
-  private parts: Uint8Array[] = [];
-  private length = 0;
-
-  writeInt32(val: number): this {
-    let buf = new Uint8Array(4);
-    let view = new DataView(buf.buffer);
-    view.setInt32(0, val, false);
-    this.parts.push(buf);
-    this.length += 4;
-    return this;
-  }
-
-  writeInt16(val: number): this {
-    let buf = new Uint8Array(2);
-    let view = new DataView(buf.buffer);
-    view.setInt16(0, val, false);
-    this.parts.push(buf);
-    this.length += 2;
-    return this;
-  }
-
-  writeByte(val: number): this {
-    let buf = new Uint8Array([val]);
-    this.parts.push(buf);
-    this.length += 1;
-    return this;
-  }
-
-  writeCString(val: string): this {
-    let encoder = new TextEncoder();
-    let encoded = encoder.encode(val);
-    let buf = new Uint8Array(encoded.length + 1);
-    buf.set(encoded);
-    buf[encoded.length] = 0;
-    this.parts.push(buf);
-    this.length += buf.length;
-    return this;
-  }
-
-  writeBytes(val: Uint8Array): this {
-    this.parts.push(val);
-    this.length += val.length;
-    return this;
-  }
-
-  toBuffer(): Uint8Array {
-    let result = new Uint8Array(this.length);
-    let offset = 0;
-    for (let part of this.parts) {
-      result.set(part, offset);
-      offset += part.length;
-    }
-    return result;
-  }
-
-  getLength(): number {
-    return this.length;
-  }
-}
-
-// Build startup message (no message type byte, just length + protocol version + params)
-export let buildStartupMessage = (user: string, database: string): Uint8Array => {
-  let params = new MessageWriter();
-  params.writeInt32(196608); // Protocol version 3.0
-  params.writeCString('user');
-  params.writeCString(user);
-  params.writeCString('database');
-  params.writeCString(database);
-  params.writeCString('client_encoding');
-  params.writeCString('UTF8');
-  params.writeByte(0); // Terminator
-
-  let body = params.toBuffer();
-  let msg = new MessageWriter();
-  msg.writeInt32(body.length + 4); // length includes itself
-  msg.writeBytes(body);
-  return msg.toBuffer();
-};
-
-// Build a typed message (type byte + length + body)
-export let buildMessage = (type: number, body: Uint8Array): Uint8Array => {
-  let msg = new MessageWriter();
-  msg.writeByte(type);
-  msg.writeInt32(body.length + 4); // length includes itself
-  msg.writeBytes(body);
-  return msg.toBuffer();
-};
-
-export let buildPasswordMessage = (password: string): Uint8Array => {
-  let writer = new MessageWriter();
-  writer.writeCString(password);
-  return buildMessage(MessageTypes.PASSWORD, writer.toBuffer());
-};
-
-export let buildQueryMessage = (query: string): Uint8Array => {
-  let writer = new MessageWriter();
-  writer.writeCString(query);
-  return buildMessage(MessageTypes.QUERY, writer.toBuffer());
-};
-
-export let buildTerminateMessage = (): Uint8Array => {
-  return buildMessage(MessageTypes.TERMINATE, new Uint8Array(0));
-};
-
-// Parse incoming messages from a buffer
-export interface ParsedMessage {
-  type: number;
-  length: number;
-  body: Uint8Array;
-}
-
-export let parseMessages = (
-  buffer: Uint8Array
-): { messages: ParsedMessage[]; remaining: Uint8Array } => {
-  let messages: ParsedMessage[] = [];
-  let offset = 0;
-  let view = new DataView(buffer.buffer, buffer.byteOffset, buffer.byteLength);
-
-  while (offset + 5 <= buffer.length) {
-    let type = buffer[offset]!;
-    let length = view.getInt32(offset + 1, false);
-
-    if (offset + 1 + length > buffer.length) {
-      break; // Incomplete message
-    }
-
-    let body = buffer.slice(offset + 5, offset + 1 + length);
-    messages.push({ type, length, body });
-    offset += 1 + length;
-  }
-
-  return {
-    messages,
-    remaining: buffer.slice(offset)
-  };
-};
-
-// Parse error/notice response fields
-export let parseErrorFields = (body: Uint8Array): Record<string, string> => {
-  let fields: Record<string, string> = {};
-  let decoder = new TextDecoder();
-  let offset = 0;
-
-  while (offset < body.length) {
-    let fieldType = body[offset]!;
-    offset++;
-    if (fieldType === 0) break;
-
-    let end = body.indexOf(0, offset);
-    if (end === -1) break;
-
-    let value = decoder.decode(body.slice(offset, end));
-    let key = String.fromCharCode(fieldType);
-
-    // Map common field types
-    if (key === 'S') fields['severity'] = value;
-    else if (key === 'V') fields['severityNonLocalized'] = value;
-    else if (key === 'C') fields['code'] = value;
-    else if (key === 'M') fields['message'] = value;
-    else if (key === 'D') fields['detail'] = value;
-    else if (key === 'H') fields['hint'] = value;
-    else if (key === 'P') fields['position'] = value;
-    else if (key === 'q') fields['internalPosition'] = value;
-    else if (key === 'W') fields['where'] = value;
-    else if (key === 's') fields['schema'] = value;
-    else if (key === 't') fields['table'] = value;
-    else if (key === 'c') fields['column'] = value;
-    else if (key === 'd') fields['dataType'] = value;
-    else if (key === 'n') fields['constraint'] = value;
-    else fields[key] = value;
-
-    offset = end + 1;
-  }
-
-  return fields;
-};
-
-// Parse RowDescription message
-export interface ColumnDescription {
-  name: string;
-  tableOid: number;
-  columnIndex: number;
-  typeOid: number;
-  typeLength: number;
-  typeModifier: number;
-  formatCode: number;
-}
-
-export let parseRowDescription = (body: Uint8Array): ColumnDescription[] => {
-  let view = new DataView(body.buffer, body.byteOffset, body.byteLength);
-  let decoder = new TextDecoder();
-  let fieldCount = view.getInt16(0, false);
-  let columns: ColumnDescription[] = [];
-  let offset = 2;
-
-  for (let i = 0; i < fieldCount; i++) {
-    let nameEnd = body.indexOf(0, offset);
-    let name = decoder.decode(body.slice(offset, nameEnd));
-    offset = nameEnd + 1;
-
-    let tableOid = view.getInt32(offset, false);
-    offset += 4;
-    let columnIndex = view.getInt16(offset, false);
-    offset += 2;
-    let typeOid = view.getInt32(offset, false);
-    offset += 4;
-    let typeLength = view.getInt16(offset, false);
-    offset += 2;
-    let typeModifier = view.getInt32(offset, false);
-    offset += 4;
-    let formatCode = view.getInt16(offset, false);
-    offset += 2;
-
-    columns.push({
-      name,
-      tableOid,
-      columnIndex,
-      typeOid,
-      typeLength,
-      typeModifier,
-      formatCode
-    });
-  }
-
-  return columns;
-};
-
-// Parse DataRow message
-export let parseDataRow = (body: Uint8Array): (string | null)[] => {
-  let view = new DataView(body.buffer, body.byteOffset, body.byteLength);
-  let decoder = new TextDecoder();
-  let fieldCount = view.getInt16(0, false);
-  let values: (string | null)[] = [];
-  let offset = 2;
-
-  for (let i = 0; i < fieldCount; i++) {
-    let length = view.getInt32(offset, false);
-    offset += 4;
-
-    if (length === -1) {
-      values.push(null);
-    } else {
-      let value = decoder.decode(body.slice(offset, offset + length));
-      values.push(value);
-      offset += length;
-    }
-  }
-
-  return values;
-};
-
-// Parse CommandComplete message to extract affected row count
-export let parseCommandComplete = (
-  body: Uint8Array
-): { command: string; rowCount: number | null } => {
-  let decoder = new TextDecoder();
-  let tag = decoder.decode(body.slice(0, body.indexOf(0)));
-  let parts = tag.split(' ');
-  let command = parts[0] || tag;
-  let lastPart = parts[parts.length - 1];
-  let rowCount = lastPart && /^\d+$/.test(lastPart) ? parseInt(lastPart, 10) : null;
-  return { command, rowCount };
-};
-
-// Map PostgreSQL OIDs to human-readable type names
-export let oidToTypeName = (oid: number): string => {
-  let typeMap: Record<number, string> = {
-    16: 'boolean',
-    17: 'bytea',
-    18: 'char',
-    19: 'name',
-    20: 'bigint',
-    21: 'smallint',
-    23: 'integer',
-    25: 'text',
-    26: 'oid',
-    114: 'json',
-    142: 'xml',
-    600: 'point',
-    700: 'real',
-    701: 'double precision',
-    790: 'money',
-    869: 'inet',
-    1042: 'character',
-    1043: 'varchar',
-    1082: 'date',
-    1083: 'time',
-    1114: 'timestamp',
-    1184: 'timestamptz',
-    1186: 'interval',
-    1266: 'timetz',
-    1700: 'numeric',
-    2950: 'uuid',
-    3802: 'jsonb',
-    3904: 'int4range',
-    3906: 'numrange',
-    3908: 'tsrange',
-    3910: 'tstzrange',
-    3912: 'daterange',
-    3926: 'int8range'
-  };
-  return typeMap[oid] || `oid:${oid}`;
-};

package/src/spec.ts

@@ -1,13 +0,0 @@
-import { SlateSpecification } from '@slates/provider';
-import { auth } from './auth';
-import { config } from './config';
-
-export let spec = SlateSpecification.create({
-  key: 'postgresql',
-  name: 'PostgreSQL',
-  description:
-    'Connect to PostgreSQL databases to query, insert, update, and delete data, manage schemas and tables, and monitor table changes.',
-  metadata: {},
-  config,
-  auth
-});

package/src/tools/delete-rows.ts

@@ -1,84 +0,0 @@
-import { SlateTool } from '@slates/provider';
-import { spec } from '../spec';
-import { createClient, qualifiedTableName } from '../lib/helpers';
-import { postgresServiceError } from '../lib/errors';
-import { z } from 'zod';
-
-export let deleteRows = SlateTool.create(spec, {
-  name: 'Delete Rows',
-  key: 'delete_rows',
-  description: `Delete rows from a PostgreSQL table based on a WHERE condition.
-Supports returning the deleted rows and requires explicit confirmation for full-table deletes.`,
-  instructions: [
-    'Always provide a WHERE condition to target specific rows.',
-    'Set confirmDeleteAll to true to delete all rows when no WHERE condition is provided.'
-  ],
-  tags: {
-    destructive: true
-  }
-})
-  .input(
-    z.object({
-      tableName: z.string().describe('Name of the table to delete from'),
-      schemaName: z.string().optional().describe('Schema containing the table'),
-      where: z
-        .string()
-        .optional()
-        .describe(
-          'SQL WHERE condition (without the WHERE keyword). Example: "id = 1 AND status = \'inactive\'"'
-        ),
-      confirmDeleteAll: z
-        .boolean()
-        .optional()
-        .default(false)
-        .describe('Must be true to delete all rows when no WHERE condition is specified'),
-      returning: z
-        .boolean()
-        .optional()
-        .default(true)
-        .describe('Whether to return the deleted rows using RETURNING *')
-    })
-  )
-  .output(
-    z.object({
-      deletedCount: z.number().describe('Number of rows deleted'),
-      returnedRows: z
-        .array(z.record(z.string(), z.any()))
-        .describe('Deleted rows (if returning was enabled)')
-    })
-  )
-  .handleInvocation(async ctx => {
-    let client = createClient(ctx.auth, ctx.config);
-    let schema = ctx.input.schemaName || ctx.config.defaultSchema;
-    let fullTableName = qualifiedTableName(ctx.input.tableName, schema);
-
-    if (!ctx.input.where && !ctx.input.confirmDeleteAll) {
-      throw postgresServiceError(
-        'No WHERE condition specified. Set confirmDeleteAll to true to delete all rows from the table.'
-      );
-    }
-
-    let sql = `DELETE FROM ${fullTableName}`;
-
-    if (ctx.input.where) {
-      sql += ` WHERE ${ctx.input.where}`;
-    }
-
-    if (ctx.input.returning) {
-      sql += ' RETURNING *';
-    }
-
-    ctx.info(
-      `Deleting rows from ${fullTableName}${ctx.input.where ? ` where ${ctx.input.where}` : ' (all rows)'}`
-    );
-    let result = await client.query(sql, ctx.config.queryTimeout);
-
-    return {
-      output: {
-        deletedCount: result.rowCount ?? 0,
-        returnedRows: result.rows
-      },
-      message: `Deleted **${result.rowCount ?? 0}** row(s) from \`${ctx.input.tableName}\`.`
-    };
-  })
-  .build();