@utaba/deep-memory-storage-sqlserver 0.1.0 → 0.3.0

package/dist/index.d.cts

@@ -1,6 +1,7 @@
-import sql from 'mssql';
-import { StorageProvider, SearchProvider, SearchableEntity } from '@utaba/deep-memory/providers';
+import * as _utaba_deep_memory_types from '@utaba/deep-memory/types';
 import { StorageRepositoryConfig, StoredRepository, RepositoryFilter, PaginatedResult, StoredRepositorySummary, RepositoryUpdate, RepositoryStats, MemoryVocabulary, PaginationOptions, VocabularyChangeRecord, StoredEntity, StoredEntityUpdate, StorageFindQuery, StoredRelationship, RelationshipQueryOptions, StorageExploreOptions, StorageNeighbourhood, StoragePathOptions, StoragePathResult, StorageTimelineOptions, StorageTimelineResult, ExportChunk, ImportChunk, BulkImportResult, SearchOptions, SearchHit } from '@utaba/deep-memory/types';
+import sql from 'mssql';
+import { StorageProvider, EnsureSchemaResult, SearchProvider, SearchableEntity } from '@utaba/deep-memory/providers';
 
 /** Configuration for SqlServerStorageProvider */
 interface SqlServerStorageProviderConfig {
@@ -19,12 +20,24 @@ declare class SqlServerStorageProvider implements StorageProvider {
     private getPool;
     initialise(): Promise<void>;
     dispose(): Promise<void>;
-    ensureSchema(): Promise<void>;
+    /**
+     * Ensure the target database exists. If constructed with a sql.config and the
+     * database does not yet exist, creates it via a temporary connection to master,
+     * then (re)connects the main pool to the newly created database.
+     *
+     * @returns true if the database was created, false if it already existed.
+     */
+    private ensureDatabase;
+    ensureSchema(): Promise<EnsureSchemaResult>;
     createRepository(config: StorageRepositoryConfig): Promise<StoredRepository>;
     getRepository(repositoryId: string): Promise<StoredRepository | null>;
     listRepositories(filter?: RepositoryFilter): Promise<PaginatedResult<StoredRepositorySummary>>;
     updateRepository(repositoryId: string, updates: RepositoryUpdate): Promise<StoredRepository>;
-    deleteRepository(repositoryId: string): Promise<void>;
+    deleteRepository(repositoryId: string, _onProgress?: _utaba_deep_memory_types.DeleteProgressCallback): Promise<void>;
+    deleteAllContents(repositoryId: string, _onProgress?: _utaba_deep_memory_types.DeleteProgressCallback): Promise<{
+        deletedEntities: number;
+        deletedRelationships: number;
+    }>;
     getRepositoryStats(repositoryId: string): Promise<RepositoryStats>;
     getVocabulary(repositoryId: string): Promise<MemoryVocabulary>;
     saveVocabulary(repositoryId: string, vocabulary: MemoryVocabulary): Promise<void>;
@@ -56,7 +69,7 @@ declare class SqlServerStorageProvider implements StorageProvider {
     findPaths(repositoryId: string, sourceId: string, targetId: string, options: StoragePathOptions): Promise<StoragePathResult>;
     getTimeline(repositoryId: string, entityId: string, options: StorageTimelineOptions): Promise<StorageTimelineResult>;
     exportAll(repositoryId: string): AsyncIterable<ExportChunk>;
-    importBulk(repositoryId: string, data: ImportChunk[]): Promise<BulkImportResult>;
+    importBulk(repositoryId: string, data: ImportChunk[], _options?: _utaba_deep_memory_types.BulkImportOptions): Promise<BulkImportResult>;
     /** Creates a TVP table from an array of string IDs (eliminates plan cache bloat from IN clauses). */
     private createIdListTvp;
     /** Returns a StoredEntity without the embedding column (lighter I/O). */

package/dist/index.d.ts

@@ -1,6 +1,7 @@
-import sql from 'mssql';
-import { StorageProvider, SearchProvider, SearchableEntity } from '@utaba/deep-memory/providers';
+import * as _utaba_deep_memory_types from '@utaba/deep-memory/types';
 import { StorageRepositoryConfig, StoredRepository, RepositoryFilter, PaginatedResult, StoredRepositorySummary, RepositoryUpdate, RepositoryStats, MemoryVocabulary, PaginationOptions, VocabularyChangeRecord, StoredEntity, StoredEntityUpdate, StorageFindQuery, StoredRelationship, RelationshipQueryOptions, StorageExploreOptions, StorageNeighbourhood, StoragePathOptions, StoragePathResult, StorageTimelineOptions, StorageTimelineResult, ExportChunk, ImportChunk, BulkImportResult, SearchOptions, SearchHit } from '@utaba/deep-memory/types';
+import sql from 'mssql';
+import { StorageProvider, EnsureSchemaResult, SearchProvider, SearchableEntity } from '@utaba/deep-memory/providers';
 
 /** Configuration for SqlServerStorageProvider */
 interface SqlServerStorageProviderConfig {
@@ -19,12 +20,24 @@ declare class SqlServerStorageProvider implements StorageProvider {
     private getPool;
     initialise(): Promise<void>;
     dispose(): Promise<void>;
-    ensureSchema(): Promise<void>;
+    /**
+     * Ensure the target database exists. If constructed with a sql.config and the
+     * database does not yet exist, creates it via a temporary connection to master,
+     * then (re)connects the main pool to the newly created database.
+     *
+     * @returns true if the database was created, false if it already existed.
+     */
+    private ensureDatabase;
+    ensureSchema(): Promise<EnsureSchemaResult>;
     createRepository(config: StorageRepositoryConfig): Promise<StoredRepository>;
     getRepository(repositoryId: string): Promise<StoredRepository | null>;
     listRepositories(filter?: RepositoryFilter): Promise<PaginatedResult<StoredRepositorySummary>>;
     updateRepository(repositoryId: string, updates: RepositoryUpdate): Promise<StoredRepository>;
-    deleteRepository(repositoryId: string): Promise<void>;
+    deleteRepository(repositoryId: string, _onProgress?: _utaba_deep_memory_types.DeleteProgressCallback): Promise<void>;
+    deleteAllContents(repositoryId: string, _onProgress?: _utaba_deep_memory_types.DeleteProgressCallback): Promise<{
+        deletedEntities: number;
+        deletedRelationships: number;
+    }>;
     getRepositoryStats(repositoryId: string): Promise<RepositoryStats>;
     getVocabulary(repositoryId: string): Promise<MemoryVocabulary>;
     saveVocabulary(repositoryId: string, vocabulary: MemoryVocabulary): Promise<void>;
@@ -56,7 +69,7 @@ declare class SqlServerStorageProvider implements StorageProvider {
     findPaths(repositoryId: string, sourceId: string, targetId: string, options: StoragePathOptions): Promise<StoragePathResult>;
     getTimeline(repositoryId: string, entityId: string, options: StorageTimelineOptions): Promise<StorageTimelineResult>;
     exportAll(repositoryId: string): AsyncIterable<ExportChunk>;
-    importBulk(repositoryId: string, data: ImportChunk[]): Promise<BulkImportResult>;
+    importBulk(repositoryId: string, data: ImportChunk[], _options?: _utaba_deep_memory_types.BulkImportOptions): Promise<BulkImportResult>;
     /** Creates a TVP table from an array of string IDs (eliminates plan cache bloat from IN clauses). */
     private createIdListTvp;
     /** Returns a StoredEntity without the embedding column (lighter I/O). */

package/dist/index.js

@@ -431,7 +431,51 @@ var SqlServerStorageProvider = class {
     }
     this.pool = null;
   }
+  /**
+   * Ensure the target database exists. If constructed with a sql.config and the
+   * database does not yet exist, creates it via a temporary connection to master,
+   * then (re)connects the main pool to the newly created database.
+   *
+   * @returns true if the database was created, false if it already existed.
+   */
+  async ensureDatabase() {
+    if (this.config.connection instanceof sql.ConnectionPool) {
+      return false;
+    }
+    const cfg = this.config.connection;
+    const dbName = cfg.database;
+    if (!dbName) {
+      return false;
+    }
+    let created = false;
+    const masterPool = new sql.ConnectionPool({ ...cfg, database: "master" });
+    try {
+      await masterPool.connect();
+      const result = await masterPool.request().input("dbName", sql.NVarChar, dbName).query(
+        `SELECT COUNT(*) AS cnt FROM sys.databases WHERE name = @dbName`
+      );
+      const exists = (result.recordset[0]?.cnt ?? 0) > 0;
+      if (!exists) {
+        if (!/^[A-Za-z0-9_-]+$/.test(dbName)) {
+          throw new ProviderError(
+            `Invalid database name '${dbName}'. Only alphanumeric characters, hyphens, and underscores are allowed.`
+          );
+        }
+        await masterPool.request().query(`CREATE DATABASE [${dbName}]`);
+        created = true;
+      }
+    } finally {
+      await masterPool.close();
+    }
+    if (!this.pool || !this.pool.connected) {
+      this.pool = new sql.ConnectionPool(cfg);
+      this.ownsPool = true;
+      await this.pool.connect();
+    }
+    return created;
+  }
   async ensureSchema() {
+    const databaseCreated = await this.ensureDatabase();
     const pool = this.getPool();
     const metaCheck = await pool.request().query(
       `SELECT COUNT(*) AS cnt FROM sys.tables WHERE name = 'dm_meta' AND schema_id = SCHEMA_ID('${this.schema}')`
@@ -448,7 +492,12 @@ var SqlServerStorageProvider = class {
         );
       }
       if (currentVersion === SCHEMA_VERSION) {
-        return;
+        return {
+          databaseCreated,
+          schemaCreated: false,
+          alreadyUpToDate: !databaseCreated,
+          schemaVersion: SCHEMA_VERSION
+        };
       }
     }
     const ddl = getSchemaSQL(this.schema);
@@ -463,6 +512,12 @@ var SqlServerStorageProvider = class {
         }
       }
     }
+    return {
+      databaseCreated,
+      schemaCreated: true,
+      alreadyUpToDate: false,
+      schemaVersion: SCHEMA_VERSION
+    };
   }
   // ─── Repository ──────────────────────────────────────────────────
   async createRepository(config) {
@@ -612,13 +667,23 @@ var SqlServerStorageProvider = class {
     if (!updated) throw new RepositoryNotFoundError(repositoryId);
     return updated;
   }
-  async deleteRepository(repositoryId) {
+  async deleteRepository(repositoryId, _onProgress) {
     const pool = this.getPool();
     const result = await pool.request().input("id", sql.UniqueIdentifier, repositoryId).query(`DELETE FROM ${this.t("dm_repositories")} WHERE [repository_id] = @id`);
     if (result.rowsAffected[0] === 0) {
       throw new RepositoryNotFoundError(repositoryId);
     }
   }
+  async deleteAllContents(repositoryId, _onProgress) {
+    await this.assertRepository(repositoryId);
+    const pool = this.getPool();
+    const relResult = await pool.request().input("id", sql.UniqueIdentifier, repositoryId).query(`DELETE FROM ${this.t("dm_relationships")} WHERE [repository_id] = @id`);
+    const entityResult = await pool.request().input("id", sql.UniqueIdentifier, repositoryId).query(`DELETE FROM ${this.t("dm_entities")} WHERE [repository_id] = @id`);
+    return {
+      deletedEntities: entityResult.rowsAffected[0] ?? 0,
+      deletedRelationships: relResult.rowsAffected[0] ?? 0
+    };
+  }
   async getRepositoryStats(repositoryId) {
     await this.assertRepository(repositoryId);
     const pool = this.getPool();
@@ -1308,7 +1373,7 @@ var SqlServerStorageProvider = class {
       }
     }
   }
-  async importBulk(repositoryId, data) {
+  async importBulk(repositoryId, data, _options) {
     await this.assertRepository(repositoryId);
     const pool = this.getPool();
     let entitiesImported = 0;