@reifydb/shell 0.9.0 → 0.9.2

package/dist/index.js

@@ -27,6 +27,29 @@ var defaultTheme = {
   brightCyan: "#94e2d5",
   brightWhite: "#a6adc8"
 };
+var reifydbDarkTheme = {
+  background: "#18181b",
+  foreground: "#e4e4e7",
+  cursor: "#7e85f2",
+  cursorAccent: "#18181b",
+  selectionBackground: "rgba(126, 133, 242, 0.30)",
+  black: "#27272a",
+  red: "#f87171",
+  green: "#4ade80",
+  yellow: "#fbbf24",
+  blue: "#7e85f2",
+  magenta: "#a78bfa",
+  cyan: "#7e85f2",
+  white: "#e4e4e7",
+  brightBlack: "#71717a",
+  brightRed: "#fca5a5",
+  brightGreen: "#86efac",
+  brightYellow: "#fcd34d",
+  brightBlue: "#a5b4fc",
+  brightMagenta: "#c4b5fd",
+  brightCyan: "#a5b4fc",
+  brightWhite: "#fafafa"
+};
 var COLORS = {
   reset: "\x1B[0m",
   bold: "\x1B[1m",
@@ -290,9 +313,10 @@ var LineEditor = class {
     this.cursorPos = newPos;
   }
   // Called by shell to render the line with prompt
-  render(prompt) {
+  // `display` carries zero-width ANSI codes only, so cursor math stays on the raw buffer
+  render(prompt, display) {
     this.terminal.write(
-      "\r" + TerminalAdapter.clearToEndOfLine() + prompt + this.buffer
+      "\r" + TerminalAdapter.clearToEndOfLine() + prompt + (display ?? this.buffer)
     );
     const cursorOffset = this.buffer.length - this.cursorPos;
     if (cursorOffset > 0) {
@@ -343,8 +367,8 @@ var CommandHistory = class {
   position = -1;
   savedInput = "";
   storage;
-  constructor(storage, history_key) {
-    this.storage = storage ?? new LocalStorageHistoryStorage(history_key);
+  constructor(storage, historyKey) {
+    this.storage = storage ?? new LocalStorageHistoryStorage(historyKey);
     this.entries = this.storage.load();
   }
   add(command) {
@@ -563,125 +587,177 @@ async function showShape(args, context) {
     terminal.writeln(`${C.red}Usage: .shape <table_name>${C.reset}`);
     return;
   }
-  const table_name = args[0];
-  const shape = await context.executor.getShape(table_name);
+  const tableName = args[0];
+  const shape = await context.executor.getShape(tableName);
   if (shape) {
     terminal.writeln("");
     terminal.writeln(shape);
     terminal.writeln("");
   } else {
-    terminal.writeln(`${C.red}Table not found: ${table_name}${C.reset}`);
+    terminal.writeln(`${C.red}Table not found: ${tableName}${C.reset}`);
   }
 }
 
 // src/output/table.ts
+import {
+  NONE_PRESENTATION,
+  cellText,
+  inferColumns,
+  padToWidth,
+  planTable,
+  valueRole
+} from "@reifydb/core";
+
+// src/ansi.ts
+var RESET = "\x1B[0m";
+function sgr(style) {
+  const parameters = [];
+  if (style.bold) parameters.push("1");
+  if (style.dim) parameters.push("2");
+  if (style.italic) parameters.push("3");
+  if (style.color) {
+    const hex = style.color.replace("#", "");
+    const value = parseInt(hex, 16);
+    parameters.push(`38;2;${value >> 16 & 255};${value >> 8 & 255};${value & 255}`);
+  }
+  return parameters.length === 0 ? "" : `\x1B[${parameters.join(";")}m`;
+}
+function paint(text, prefix) {
+  return prefix === "" ? text : prefix + text + RESET;
+}
+
+// src/output/table.ts
+var BORDERS = {
+  unicode: {
+    vertical: "\u2502",
+    horizontal: "\u2500",
+    top: ["\u250C", "\u252C", "\u2510"],
+    middle: ["\u251C", "\u253C", "\u2524"],
+    bottom: ["\u2514", "\u2534", "\u2518"],
+    padding: 1,
+    columnOverhead: 3,
+    tableOverhead: 1
+  },
+  ascii: {
+    vertical: "|",
+    horizontal: "-",
+    top: ["+", "+", "+"],
+    middle: ["+", "+", "+"],
+    bottom: ["+", "+", "+"],
+    padding: 1,
+    columnOverhead: 3,
+    tableOverhead: 1
+  },
+  none: {
+    vertical: "",
+    horizontal: "\u2500",
+    top: ["", "", ""],
+    middle: ["", "", ""],
+    bottom: ["", "", ""],
+    padding: 1,
+    columnOverhead: 2,
+    tableOverhead: 0
+  }
+};
+var reifydbDarkTableTheme = {
+  border: { color: "#3f3f46" },
+  header: { color: "#818cf8", bold: true },
+  values: {
+    none: {
+      color: NONE_PRESENTATION.muted ? "#52525b" : void 0,
+      italic: NONE_PRESENTATION.italic
+    },
+    boolean: { color: "#fbbf24" },
+    number: { color: "#a5b4fc" },
+    temporal: { color: "#5eead4" },
+    uuid: { color: "#c4b5fd" },
+    string: { color: "#e4e4e7" },
+    blob: { color: "#a1a1aa" }
+  }
+};
 var TableRenderer = class {
   data;
-  columns;
-  maxWidth;
-  truncate;
+  glyphs;
+  plan;
+  borderPrefix;
+  headerPrefix;
+  valuePrefixes;
   constructor(data, options = {}) {
     this.data = data;
-    this.maxWidth = options.maxWidth ?? 120;
-    this.truncate = options.truncate ?? false;
-    this.columns = this.calculateColumns();
-  }
-  calculateColumns() {
-    if (this.data.length === 0) return [];
-    const columns = /* @__PURE__ */ new Map();
-    for (const row of this.data) {
-      for (const [key, value] of Object.entries(row)) {
-        const valueStr = this.formatValue(value);
-        const currentMax = columns.get(key) ?? key.length;
-        columns.set(key, Math.max(currentMax, valueStr.length));
-      }
-    }
-    return Array.from(columns.entries()).map(([name, width]) => ({
-      name,
-      width: Math.max(width, name.length)
-    }));
-  }
-  formatValue(value) {
-    if (value === null || value === void 0) {
-      return "undefined";
-    }
-    if (value && typeof value === "object" && "type" in value && typeof value.toString === "function") {
-      return value.toString();
-    }
-    if (typeof value === "object") {
-      return JSON.stringify(value);
+    this.glyphs = BORDERS[options.border ?? "unicode"];
+    const theme = options.theme ?? reifydbDarkTableTheme;
+    this.borderPrefix = theme.border ? sgr(theme.border) : "";
+    this.headerPrefix = theme.header ? sgr(theme.header) : "";
+    this.valuePrefixes = /* @__PURE__ */ new Map();
+    for (const [role, style] of Object.entries(theme.values ?? {})) {
+      if (style) this.valuePrefixes.set(role, sgr(style));
     }
-    return String(value);
+    const columns = options.columns ?? inferColumns(data);
+    this.plan = planTable(data, columns, {
+      width: options.truncate ? options.maxWidth : void 0,
+      columnOverhead: this.glyphs.columnOverhead,
+      tableOverhead: this.glyphs.tableOverhead
+    });
+  }
+  get columns() {
+    return this.plan.columns.map((entry) => ({ name: entry.column.name, width: entry.width }));
   }
-  truncateString(str, maxLen) {
-    if (str.length <= maxLen) return str;
-    if (maxLen <= 3) return str.slice(0, maxLen);
-    return str.slice(0, maxLen - 3) + "...";
+  get dropped() {
+    return this.plan.dropped.map((column) => column.name);
   }
   render() {
-    if (this.data.length === 0 || this.columns.length === 0) {
+    if (this.data.length === 0 || this.plan.columns.length === 0) {
       return ["(no results)"];
     }
+    const bordered = this.glyphs.vertical !== "";
     const lines = [];
-    let columnsToShow = this.columns;
-    let widths = this.columns.map((c) => c.width);
-    if (this.truncate) {
-      const result = this.fitColumns(this.maxWidth);
-      columnsToShow = result.columns;
-      widths = result.widths;
-    }
-    const separator = "+" + widths.map((w) => "-".repeat(w + 2)).join("+") + "+";
-    lines.push(separator);
-    const headerCells = columnsToShow.map(
-      (col, i) => this.padCenter(col.name, widths[i])
-    );
-    lines.push("| " + headerCells.join(" | ") + " |");
-    lines.push(separator);
+    if (bordered) lines.push(this.rule(this.glyphs.top));
+    lines.push(this.headerRow());
+    lines.push(this.rule(this.glyphs.middle));
     for (const row of this.data) {
-      const cells = columnsToShow.map((col, i) => {
-        const value = this.formatValue(row[col.name]);
-        const truncated = this.truncate ? this.truncateString(value, widths[i]) : value;
-        return this.padRight(truncated, widths[i]);
-      });
-      lines.push("| " + cells.join(" | ") + " |");
+      lines.push(this.dataRow(row));
     }
-    lines.push(separator);
+    if (bordered) lines.push(this.rule(this.glyphs.bottom));
     return lines;
   }
-  fitColumns(maxWidth) {
-    const columns = [];
-    const widths = [];
-    let currentWidth = 1;
-    for (const col of this.columns) {
-      const colWidth = Math.min(col.width, 40);
-      const totalColWidth = colWidth + 3;
-      if (currentWidth + totalColWidth <= maxWidth) {
-        columns.push(col);
-        widths.push(colWidth);
-        currentWidth += totalColWidth;
-      } else {
-        break;
-      }
-    }
-    if (columns.length === 0 && this.columns.length > 0) {
-      const firstCol = this.columns[0];
-      const availableWidth = maxWidth - 4;
-      columns.push(firstCol);
-      widths.push(Math.max(availableWidth, 10));
+  rule(corners) {
+    const pad = this.glyphs.padding * 2;
+    if (this.glyphs.vertical === "") {
+      return paint(this.glyphs.horizontal.repeat(this.width()), this.borderPrefix);
     }
-    return { columns, widths };
+    const segments = this.plan.columns.map(
+      (entry) => this.glyphs.horizontal.repeat(entry.width + pad)
+    );
+    const [left, joint, right] = corners;
+    return paint(left + segments.join(joint) + right, this.borderPrefix);
   }
-  padCenter(str, width) {
-    const padding = width - str.length;
-    if (padding <= 0) return str.slice(0, width);
-    const left = Math.floor(padding / 2);
-    const right = padding - left;
-    return " ".repeat(left) + str + " ".repeat(right);
+  width() {
+    const gaps = this.glyphs.columnOverhead * (this.plan.columns.length - 1);
+    const body = this.plan.columns.reduce((total, entry) => total + entry.width, 0);
+    return this.glyphs.padding + body + gaps;
   }
-  padRight(str, width) {
-    const padding = width - str.length;
-    if (padding <= 0) return str.slice(0, width);
-    return str + " ".repeat(padding);
+  headerRow() {
+    const cells = this.plan.columns.map(
+      (entry) => paint(padToWidth(entry.column.name, entry.width, entry.column.align), this.headerPrefix)
+    );
+    return this.joinCells(cells);
+  }
+  dataRow(row) {
+    const cells = this.plan.columns.map((entry) => {
+      const value = row[entry.column.name];
+      const text = padToWidth(cellText(value, entry.width), entry.width, entry.column.align);
+      return paint(text, this.valuePrefixes.get(valueRole(value)) ?? "");
+    });
+    return this.joinCells(cells);
+  }
+  joinCells(cells) {
+    const pad = " ".repeat(this.glyphs.padding);
+    if (this.glyphs.vertical === "") {
+      const gap = " ".repeat(this.glyphs.columnOverhead);
+      return pad + cells.join(gap);
+    }
+    const bar = paint(this.glyphs.vertical, this.borderPrefix);
+    return bar + pad + cells.join(pad + bar + pad) + pad + bar;
   }
 };
 
@@ -690,49 +766,62 @@ var C2 = TerminalAdapter.COLORS;
 var OutputFormatter = class {
   terminal;
   displayMode;
-  constructor(terminal, displayMode = "full") {
+  border;
+  theme;
+  constructor(terminal, displayMode = "full", options = {}) {
     this.terminal = terminal;
     this.displayMode = displayMode;
+    this.border = options.border;
+    this.theme = options.theme;
   }
   setDisplayMode(mode) {
     this.displayMode = mode;
   }
   formatResult(result) {
     if (!result.success) {
-      this.formatError(result.error ?? "Unknown error", result.execution_time);
+      this.formatError(result.error ?? "Unknown error", result.executionTime);
       return;
     }
     if (!result.data || result.data.length === 0) {
       this.terminal.writeln("");
       this.terminal.writeln(`${C2.dim}Query executed successfully. No rows returned.${C2.reset}`);
-      this.formatExecutionTime(result.execution_time);
+      this.formatExecutionTime(result.executionTime);
       return;
     }
-    this.formatTable(result.data, result.execution_time);
+    this.formatTable(result, result.executionTime);
   }
-  formatTable(data, execution_time) {
+  formatTable(result, executionTime) {
+    const data = result.data ?? [];
     const renderer = new TableRenderer(data, {
       maxWidth: this.displayMode === "truncate" ? this.terminal.cols - 2 : void 0,
-      truncate: this.displayMode === "truncate"
+      truncate: this.displayMode === "truncate",
+      border: this.border,
+      theme: this.theme,
+      columns: result.columns
     });
-    const lines = renderer.render();
     this.terminal.writeln("");
-    for (const line of lines) {
+    for (const line of renderer.render()) {
       this.terminal.writeln(line);
     }
-    const row_count = data.length;
+    const dropped = renderer.dropped;
+    if (dropped.length > 0) {
+      this.terminal.writeln(
+        `${C2.dim}${dropped.length} more column${dropped.length !== 1 ? "s" : ""}: ${dropped.join(", ")}${C2.reset}`
+      );
+    }
+    const rowCount = data.length;
     this.terminal.writeln("");
     this.terminal.write(
-      `${C2.green}${row_count} row${row_count !== 1 ? "s" : ""}${C2.reset}`
+      `${C2.green}${rowCount} row${rowCount !== 1 ? "s" : ""}${C2.reset}`
     );
-    this.formatExecutionTime(execution_time);
+    this.formatExecutionTime(executionTime);
   }
-  formatError(error, execution_time) {
+  formatError(error, executionTime) {
     this.terminal.writeln("");
     for (const line of error.split("\n")) {
       this.terminal.writeln(line);
     }
-    this.formatExecutionTime(execution_time);
+    this.formatExecutionTime(executionTime);
   }
   formatExecutionTime(ms) {
     this.terminal.writeln(` ${C2.dim}(${ms}ms)${C2.reset}`);
@@ -760,12 +849,16 @@ var Shell = class {
   continuationPrompt;
   continuationPromptLen;
   welcomeMessage;
+  initialInput;
+  highlight;
   onExit;
   onFullscreenChange;
   constructor(container, options) {
     this.executor = options.executor;
     this.displayMode = options.displayMode ?? "full";
     this.welcomeMessage = options.welcomeMessage;
+    this.initialInput = options.initialInput;
+    this.highlight = options.highlight;
     this.onExit = options.onExit;
     this.onFullscreenChange = options.onFullscreenChange;
     this.primaryPrompt = options.prompt ?? DEFAULT_PRIMARY_PROMPT;
@@ -774,14 +867,21 @@ var Shell = class {
     this.continuationPromptLen = options.continuationPromptLength ?? DEFAULT_CONTINUATION_PROMPT_LEN;
     this.terminal = new TerminalAdapter(container, options.theme);
     this.lineEditor = new LineEditor(this.terminal);
-    this.history = new CommandHistory(options.historyStorage, options.history_key);
+    this.history = new CommandHistory(options.historyStorage, options.historyKey);
     this.multiline = new MultilineBuffer();
-    this.formatter = new OutputFormatter(this.terminal, this.displayMode);
+    this.formatter = new OutputFormatter(this.terminal, this.displayMode, {
+      border: options.tableBorder,
+      theme: options.tableTheme
+    });
     this.setupKeyHandler();
   }
   start() {
     this.showWelcomeBanner();
     this.showPrompt();
+    if (this.initialInput) {
+      this.lineEditor.setValue(this.initialInput);
+      this.redrawLine();
+    }
     this.terminal.focus();
   }
   dispose() {
@@ -929,7 +1029,8 @@ var Shell = class {
   }
   redrawLine() {
     const prompt = this.multiline.isEmpty ? this.primaryPrompt : this.continuationPrompt;
-    this.lineEditor.render(prompt);
+    const context = this.multiline.isEmpty ? void 0 : this.multiline.content + " ";
+    this.lineEditor.render(prompt, this.highlight?.(this.lineEditor.value, context));
   }
   navigateHistory(direction) {
     let entry;
@@ -986,12 +1087,47 @@ var Shell = class {
   }
   clearScreen() {
     this.terminal.write(TerminalAdapter.clearScreen());
-    this.showPrompt();
-    this.terminal.write(this.lineEditor.value);
+    this.redrawLine();
   }
 };
 
+// src/input/highlight.ts
+import { tokenizeRql } from "@reifydb/core";
+var reifydbDarkPalette = {
+  keyword: { color: "#818cf8", bold: true },
+  string: { color: "#a5b4fc" },
+  number: { color: "#a5b4fc", bold: true },
+  comment: { color: "#71717a", italic: true },
+  operator: { color: "#a1a1aa" },
+  identifier: { color: "#e4e4e7" },
+  key: { color: "#818cf8", bold: true },
+  namespace: { color: "#5eead4" },
+  entity: { color: "#99f6e4" },
+  function: { color: "#fbbf24" }
+};
+function rqlHighlighter(palette = reifydbDarkPalette) {
+  const prefixes = /* @__PURE__ */ new Map();
+  for (const [kind, style] of Object.entries(palette)) {
+    if (style) prefixes.set(kind, sgr(style));
+  }
+  return (line, context) => {
+    const offset = context?.length ?? 0;
+    const end = offset + line.length;
+    let painted = "";
+    for (const token of tokenizeRql((context ?? "") + line)) {
+      const start = Math.max(token.start, offset);
+      const stop = Math.min(token.end, end);
+      if (start >= stop) continue;
+      const text = line.slice(start - offset, stop - offset);
+      const prefix = prefixes.get(token.kind);
+      painted += prefix ? prefix + text + RESET : text;
+    }
+    return painted;
+  };
+}
+
 // src/executors/wasm-executor.ts
+import { inferColumns as inferColumns2 } from "@reifydb/core";
 var WasmExecutor = class {
   db;
   constructor(db) {
@@ -1004,24 +1140,26 @@ var WasmExecutor = class {
       return {
         success: true,
         data: [],
-        execution_time: 0
+        executionTime: 0
       };
     }
-    const start_time = performance.now();
+    const startTime = performance.now();
     try {
       const results = await this.db.admin(query);
       const endTime = performance.now();
+      const data = Array.isArray(results) ? results : [];
       return {
         success: true,
-        data: Array.isArray(results) ? results : [],
-        execution_time: Math.round(endTime - start_time)
+        data,
+        columns: inferColumns2(data),
+        executionTime: Math.round(endTime - startTime)
       };
     } catch (error) {
       const endTime = performance.now();
       return {
         success: false,
         error: error instanceof Error ? error.message : String(error),
-        execution_time: Math.round(endTime - start_time)
+        executionTime: Math.round(endTime - startTime)
       };
     }
   }
@@ -1040,16 +1178,16 @@ var WasmExecutor = class {
       return [];
     }
   }
-  async getShape(table_name) {
+  async getShape(tableName) {
     try {
       const result = await this.db.admin(
-        `FROM system::columns FILTER table = "${table_name}" MAP { name, type }`
+        `FROM system::columns FILTER table = "${tableName}" MAP { name, type }`
       );
       if (Array.isArray(result) && result.length > 0) {
         const columns = result.map(
           (row) => `  ${row.name}: ${row.type}`
         ).join(",\n");
-        return `${table_name} {
+        return `${tableName} {
 ${columns}
 }`;
       }
@@ -1061,6 +1199,7 @@ ${columns}
 };
 
 // src/executors/ws-executor.ts
+import { inferColumns as inferColumns3 } from "@reifydb/core";
 var WsExecutor = class {
   client;
   constructor(client) {
@@ -1073,10 +1212,10 @@ var WsExecutor = class {
       return {
         success: true,
         data: [],
-        execution_time: 0
+        executionTime: 0
       };
     }
-    const start_time = performance.now();
+    const startTime = performance.now();
     try {
       const frames = await this.client.admin(query, null, []);
       const endTime = performance.now();
@@ -1094,23 +1233,24 @@ var WsExecutor = class {
       return {
         success: true,
         data,
-        execution_time: Math.round(endTime - start_time)
+        columns: inferColumns3(data),
+        executionTime: Math.round(endTime - startTime)
       };
     } catch (error) {
       const endTime = performance.now();
-      let error_message;
+      let errorMessage;
       if (error && typeof error === "object" && "diagnostic" in error) {
         const diagnostic = error.diagnostic;
-        error_message = diagnostic.message;
+        errorMessage = diagnostic.message;
       } else if (error instanceof Error) {
-        error_message = error.message;
+        errorMessage = error.message;
       } else {
-        error_message = String(error);
+        errorMessage = String(error);
       }
       return {
         success: false,
-        error: error_message,
-        execution_time: Math.round(endTime - start_time)
+        error: errorMessage,
+        executionTime: Math.round(endTime - startTime)
       };
     }
   }
@@ -1132,10 +1272,10 @@ var WsExecutor = class {
       return [];
     }
   }
-  async getShape(table_name) {
+  async getShape(tableName) {
     try {
       const frames = await this.client.admin(
-        `FROM system::columns FILTER table = "${table_name}" MAP { name, type }`,
+        `FROM system::columns FILTER table = "${tableName}" MAP { name, type }`,
         null,
         []
       );
@@ -1145,7 +1285,7 @@ var WsExecutor = class {
           const r = row;
           return `  ${this.extractValue(r.name)}: ${this.extractValue(r.type)}`;
         }).join(",\n");
-        return `${table_name} {
+        return `${tableName} {
 ${columns}
 }`;
       }
@@ -1169,12 +1309,19 @@ export {
   MemoryHistoryStorage,
   MultilineBuffer,
   OutputFormatter,
+  RESET,
   Shell,
   TableRenderer,
   TerminalAdapter,
   WasmExecutor,
   WsExecutor,
   defaultTheme,
-  handleDotCommand
+  handleDotCommand,
+  paint,
+  reifydbDarkPalette,
+  reifydbDarkTableTheme,
+  reifydbDarkTheme,
+  rqlHighlighter,
+  sgr
 };
 //# sourceMappingURL=index.js.map