@fadhilp/stateql 0.5.4 → 0.6.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, 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 {
@@ -211,7 +212,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 +236,7 @@ export declare class StateStore {
         operations: Array<{
             id: string;
             affectedRows: number;
+            outcome?: MongoWriteOutcome;
         }>;
     }): string;
     savePlan(input: {
@@ -260,6 +262,7 @@ export declare class StateStore {
         operationId: string;
         connectionId: string;
         affectedRows: number;
+        outcome?: MongoWriteOutcome;
     }): {
         operation: OperationRecord;
         stateVersion: string;

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 = ?`)
@@ -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,5 @@
-export type Driver = "sqlite" | "postgres" | "mysql";
+export type SqlDriver = "sqlite" | "postgres" | "mysql";
+export type Driver = SqlDriver | "mongodb";
 export type CredentialAccess = "read" | "write";
 export type CredentialOperation = "connect" | "query" | "inspect" | "plan" | "exec" | "apply" | "transaction.commit";
 export interface CredentialRequest {
@@ -108,6 +109,77 @@ 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;
@@ -133,6 +205,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 +218,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 +243,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 +259,7 @@ export interface BatchCommand {
     kind?: string;
     table?: string;
     params?: SqlParameters;
+    mongo?: MongoReadCommand | MongoWriteCommand;
     cache?: "auto" | "bypass" | "require";
     read_only?: boolean;
     secret_env?: string;
@@ -355,6 +441,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 +508,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.6.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"