@fadhilp/stateql 0.5.4 → 0.7.0

package/dist/src/store.d.ts

@@ -1,5 +1,5 @@
 import { DatabaseSync } from "node:sqlite";
-import type { Column, Driver, Row, SqlParameters, StateConfidence } from "./types.js";
+import type { Column, CommandOrigin, Driver, MongoWriteOutcome, Row, SqlParameters, StateConfidence } from "./types.js";
 export interface SessionRecord {
     id: string;
     name: string;
@@ -61,6 +61,7 @@ export interface OperationRecord {
     idempotency_key: string | null;
     state_version_before: string;
     state_version_after: string | null;
+    outcome_json: string | null;
     created_at: string;
 }
 export interface TransactionRecord {
@@ -97,6 +98,7 @@ export interface HistoryRecord {
     timestamp: string;
     session_id: string;
     actor_id: string;
+    origin: CommandOrigin;
     command: string;
     sql: string | null;
     handle: string | null;
@@ -211,7 +213,7 @@ export declare class StateStore {
         previous?: OperationRecord;
         denied?: "membership" | "transaction";
     };
-    finishOperation(operationId: string, affectedRows: number, stateVersion: string): OperationRecord;
+    finishOperation(operationId: string, affectedRows: number, stateVersion: string, outcome?: MongoWriteOutcome): OperationRecord;
     failOperation(operationId: string): void;
     markOperationOutcomeUnknown(operationId: string): void;
     createTransaction(input: {
@@ -235,6 +237,7 @@ export declare class StateStore {
         operations: Array<{
             id: string;
             affectedRows: number;
+            outcome?: MongoWriteOutcome;
         }>;
     }): string;
     savePlan(input: {
@@ -260,6 +263,7 @@ export declare class StateStore {
         operationId: string;
         connectionId: string;
         affectedRows: number;
+        outcome?: MongoWriteOutcome;
     }): {
         operation: OperationRecord;
         stateVersion: string;
@@ -267,6 +271,7 @@ export declare class StateStore {
     addHistory(input: {
         sessionId: string;
         actorId: string;
+        origin?: CommandOrigin;
         command: string;
         sql?: string;
         handle?: string;
@@ -276,7 +281,7 @@ export declare class StateStore {
         errorCode?: string;
         id?: string;
     }): HistoryRecord;
-    history(sessionId: string, limit: number): HistoryRecord[];
+    history(sessionId: string, limit: number, origin?: CommandOrigin): HistoryRecord[];
     recentOperations(sessionId: string, limit: number): OperationRecord[];
     knownResults(sessionId: string, limit: number): Array<ResultRecord & {
         alias: string | null;

package/dist/src/store.js

@@ -480,12 +480,13 @@ export class StateStore {
             throw error;
         }
     }
-    finishOperation(operationId, affectedRows, stateVersion) {
+    finishOperation(operationId, affectedRows, stateVersion, outcome) {
         this.db
             .prepare(`UPDATE operations
-         SET status = 'committed', affected_rows = ?, state_version_after = ?
+         SET status = 'committed', affected_rows = ?, state_version_after = ?,
+             outcome_json = ?
          WHERE id = ?`)
-            .run(affectedRows, stateVersion, operationId);
+            .run(affectedRows, stateVersion, outcomeJson(outcome), operationId);
         return this.getOperation(operationId);
     }
     failOperation(operationId) {
@@ -695,9 +696,10 @@ export class StateStore {
                 stateVersion = `sv_${row.version}`;
                 this.db
                     .prepare(`UPDATE operations
-             SET status = 'committed', affected_rows = ?, state_version_after = ?
+             SET status = 'committed', affected_rows = ?, state_version_after = ?,
+                 outcome_json = ?
              WHERE id = ?`)
-                    .run(operation.affectedRows, stateVersion, operation.id);
+                    .run(operation.affectedRows, stateVersion, outcomeJson(operation.outcome), operation.id);
             }
             this.db
                 .prepare(`UPDATE transactions
@@ -772,9 +774,10 @@ export class StateStore {
             const stateVersion = `sv_${row.version}`;
             this.db
                 .prepare(`UPDATE operations
-           SET status = 'committed', affected_rows = ?, state_version_after = ?
+           SET status = 'committed', affected_rows = ?, state_version_after = ?,
+               outcome_json = ?
            WHERE id = ?`)
-                .run(input.affectedRows, stateVersion, input.operationId);
+                .run(input.affectedRows, stateVersion, outcomeJson(input.outcome), input.operationId);
             this.db
                 .prepare(`UPDATE plans SET applied_operation_id = ?, claim_token = NULL
            WHERE id = ? AND claim_token = ?`)
@@ -792,10 +795,10 @@ export class StateStore {
         const id = input.id ?? this.nextId("cmd");
         this.db
             .prepare(`INSERT INTO history
-          (id, timestamp, session_id, actor_id, command, sql, handle, executed,
-           cached, success, error_code)
-         VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`)
-            .run(id, this.now().toISOString(), input.sessionId, input.actorId, input.command, boundedHistorySql(input.sql), input.handle ?? null, input.executed ? 1 : 0, input.cached ? 1 : 0, input.success ? 1 : 0, input.errorCode ?? null);
+          (id, timestamp, session_id, actor_id, origin, command, sql, handle,
+           executed, cached, success, error_code)
+         VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`)
+            .run(id, this.now().toISOString(), input.sessionId, input.actorId, input.origin ?? "legacy", input.command, boundedHistorySql(input.sql), input.handle ?? null, input.executed ? 1 : 0, input.cached ? 1 : 0, input.success ? 1 : 0, input.errorCode ?? null);
         this.db
             .prepare(`DELETE FROM history
          WHERE rowid IN (
@@ -809,13 +812,13 @@ export class StateStore {
             .prepare("SELECT * FROM history WHERE id = ?")
             .get(id);
     }
-    history(sessionId, limit) {
+    history(sessionId, limit, origin) {
         return this.db
             .prepare(`SELECT * FROM history
-         WHERE session_id = ?
+         WHERE session_id = ?${origin === undefined ? "" : " AND origin = ?"}
          ORDER BY rowid DESC
          LIMIT ?`)
-            .all(sessionId, limit);
+            .all(...(origin === undefined ? [sessionId, limit] : [sessionId, origin, limit]));
     }
     recentOperations(sessionId, limit) {
         return this.db
@@ -1011,6 +1014,9 @@ export class StateStore {
         }
     }
 }
+function outcomeJson(outcome) {
+    return outcome === undefined ? null : JSON.stringify(toJsonSafe(outcome));
+}
 function boundedHistorySql(sql) {
     if (sql === undefined)
         return null;

package/dist/src/types.d.ts

@@ -1,4 +1,11 @@
-export type Driver = "sqlite" | "postgres" | "mysql";
+export type SqlDriver = "sqlite" | "postgres" | "mysql";
+export type Driver = SqlDriver | "mongodb";
+export type CommandOrigin = "legacy" | "user" | "model" | "system" | "api";
+/** Trusted host metadata for one executeCommand call; never part of BatchCommand input. */
+export interface CommandExecutionContext {
+    signal?: AbortSignal;
+    origin?: CommandOrigin;
+}
 export type CredentialAccess = "read" | "write";
 export type CredentialOperation = "connect" | "query" | "inspect" | "plan" | "exec" | "apply" | "transaction.commit";
 export interface CredentialRequest {
@@ -63,6 +70,7 @@ export interface HistoryEntry {
     timestamp: string;
     session_id: string;
     actor_id: string;
+    origin: CommandOrigin;
     command: string;
     sql: string | null;
     handle: string | null;
@@ -108,10 +116,84 @@ export interface StateQLSnapshot {
     history: HistoryEntry[];
 }
 export type SqlParameters = unknown[] | Record<string, unknown>;
+export type MongoDocument = Record<string, unknown>;
+export interface MongoFindOptions {
+    projection?: MongoDocument;
+    sort?: MongoDocument | Array<[string, 1 | -1]>;
+    skip?: number;
+    limit?: number;
+    hint?: string | MongoDocument;
+    collation?: MongoDocument;
+}
+export interface MongoAggregateOptions {
+    allowDiskUse?: boolean;
+    hint?: string | MongoDocument;
+    collation?: MongoDocument;
+}
+export interface MongoMutationOptions {
+    upsert?: boolean;
+    collation?: MongoDocument;
+    hint?: string | MongoDocument;
+}
+export type MongoReadCommand = {
+    operation: "find";
+    collection: string;
+    filter?: MongoDocument;
+    options?: MongoFindOptions;
+} | {
+    operation: "aggregate";
+    collection: string;
+    pipeline: MongoDocument[];
+    options?: MongoAggregateOptions;
+};
+export type MongoWriteCommand = {
+    operation: "insertOne";
+    collection: string;
+    document: MongoDocument;
+    options?: Record<string, never>;
+} | {
+    operation: "insertMany";
+    collection: string;
+    documents: MongoDocument[];
+    options?: {
+        ordered?: boolean;
+    };
+} | {
+    operation: "updateOne" | "updateMany";
+    collection: string;
+    filter: MongoDocument;
+    update: MongoDocument | MongoDocument[];
+    options?: MongoMutationOptions;
+} | {
+    operation: "replaceOne";
+    collection: string;
+    filter: MongoDocument;
+    replacement: MongoDocument;
+    options?: MongoMutationOptions;
+} | {
+    operation: "deleteOne" | "deleteMany";
+    collection: string;
+    filter: MongoDocument;
+    options?: Omit<MongoMutationOptions, "upsert">;
+};
+export interface MongoWriteOutcome {
+    acknowledged: boolean;
+    inserted_id?: unknown;
+    inserted_ids?: unknown[];
+    inserted_count?: number;
+    matched_count?: number;
+    modified_count?: number;
+    upserted_count?: number;
+    upserted_id?: unknown;
+    deleted_count?: number;
+}
 export interface ExecutionOptions {
     timeoutMs?: number;
     signal?: AbortSignal;
 }
+export interface HistoryOptions {
+    origin?: CommandOrigin;
+}
 export interface StateQLOptions extends ExecutionOptions {
     home?: string;
     session?: string;
@@ -133,6 +215,9 @@ export interface QueryOptions extends ExecutionOptions {
     params?: SqlParameters;
     cache?: "auto" | "bypass" | "require";
 }
+export interface MongoQueryOptions extends ExecutionOptions {
+    cache?: "auto" | "bypass" | "require";
+}
 export interface FilterOptions {
     params?: SqlParameters;
 }
@@ -143,6 +228,12 @@ export interface ExecOptions extends ExecutionOptions {
     allowUnbounded?: boolean;
     allowDestructive?: boolean;
 }
+export interface MongoExecOptions extends ExecutionOptions {
+    replay?: boolean;
+    idempotencyKey?: string;
+    allowUnbounded?: boolean;
+    allowDestructive?: boolean;
+}
 export interface ConnectOptions extends ExecutionOptions {
     name?: string;
     readOnly?: boolean;
@@ -162,7 +253,11 @@ export interface PlanOptions extends ExecutionOptions {
     allowUnbounded?: boolean;
     allowDestructive?: boolean;
 }
-export type BatchCommandName = "connect" | "disconnect" | "status" | "profile.add" | "profile.list" | "profile.show" | "profile.remove" | "session.start" | "session.list" | "session.show" | "session.summary" | "session.close" | "query" | "filter" | "exec" | "show" | "rows" | "count" | "columns" | "alias.set" | "inspect" | "transaction.begin" | "transaction.status" | "transaction.commit" | "transaction.rollback" | "plan" | "apply" | "history" | "receipt" | "doctor" | "purge" | "capabilities";
+export interface MongoPlanOptions extends ExecutionOptions {
+    allowUnbounded?: boolean;
+    allowDestructive?: boolean;
+}
+export type BatchCommandName = "connect" | "disconnect" | "status" | "profile.add" | "profile.list" | "profile.show" | "profile.remove" | "session.start" | "session.list" | "session.show" | "session.summary" | "session.close" | "query" | "filter" | "exec" | "show" | "rows" | "count" | "columns" | "alias.set" | "inspect" | "transaction.begin" | "transaction.status" | "transaction.commit" | "transaction.rollback" | "plan" | "mongo.query" | "mongo.exec" | "mongo.plan" | "apply" | "history" | "receipt" | "doctor" | "purge" | "capabilities";
 export interface BatchCommand {
     command: BatchCommandName;
     target?: string;
@@ -174,6 +269,7 @@ export interface BatchCommand {
     kind?: string;
     table?: string;
     params?: SqlParameters;
+    mongo?: MongoReadCommand | MongoWriteCommand;
     cache?: "auto" | "bypass" | "require";
     read_only?: boolean;
     secret_env?: string;
@@ -186,11 +282,14 @@ export interface BatchCommand {
     limit?: number;
     isolation?: string;
     timeout_ms?: number;
+    /** Retrieval filter for the history command; does not attribute this command. */
+    history_origin?: CommandOrigin;
     scope?: "expired" | "results" | "history" | "all";
 }
 export interface BatchOptions {
     continueOnError?: boolean;
     maxCommands?: number;
+    executionContext?: CommandExecutionContext;
 }
 export interface Column {
     name: string;
@@ -355,6 +454,7 @@ export interface OperationData {
     state_version_before: string;
     state_version_after: string | null;
     replay_of?: string;
+    outcome?: MongoWriteOutcome;
 }
 export interface ExecData extends OperationData {
     duplicate?: boolean;
@@ -421,6 +521,7 @@ export interface PurgeData {
 export interface CapabilitiesData {
     drivers: Driver[];
     features: Record<string, boolean>;
+    driver_features?: Partial<Record<Driver, Record<string, boolean>>>;
 }
 export interface RemovedProfileData {
     profile: string;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@fadhilp/stateql",
-  "version": "0.5.4",
+  "version": "0.7.0",
   "description": "Stateful, agent-oriented database CLI for safe result reuse",
   "repository": {
     "type": "git",
@@ -25,7 +25,7 @@
   ],
   "scripts": {
     "build": "tsc -p tsconfig.json",
-    "test": "npm run build && node --test dist/test/cli.test.js dist/test/credential.test.js dist/test/mysql.test.js dist/test/postgres.test.js dist/test/query.test.js dist/test/sqlite.test.js dist/test/store.test.js dist/test/terminal.test.js dist/test/transaction.test.js dist/test/write.test.js",
+    "test": "npm run build && node --test dist/test/cli.test.js dist/test/credential.test.js dist/test/mongodb.test.js dist/test/mysql.test.js dist/test/postgres.test.js dist/test/query.test.js dist/test/sqlite.test.js dist/test/store.test.js dist/test/terminal.test.js dist/test/transaction.test.js dist/test/write.test.js",
     "release": "npm version",
     "version": "npm install",
     "prepack": "npm test"
@@ -44,6 +44,7 @@
     "node": ">=22.16"
   },
   "dependencies": {
+    "mongodb": "^6.21.0",
     "mysql2": "^3.23.1",
     "node-sql-parser": "^5.4.0",
     "pg": "^8.16.3"