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

package/dist/index.cjs

@@ -462,7 +462,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 import_mssql.default.ConnectionPool) {
+      return false;
+    }
+    const cfg = this.config.connection;
+    const dbName = cfg.database;
+    if (!dbName) {
+      return false;
+    }
+    let created = false;
+    const masterPool = new import_mssql.default.ConnectionPool({ ...cfg, database: "master" });
+    try {
+      await masterPool.connect();
+      const result = await masterPool.request().input("dbName", import_mssql.default.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 import_deep_memory.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 import_mssql.default.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}')`
@@ -479,7 +523,12 @@ var SqlServerStorageProvider = class {
         );
       }
       if (currentVersion === SCHEMA_VERSION) {
-        return;
+        return {
+          databaseCreated,
+          schemaCreated: false,
+          alreadyUpToDate: !databaseCreated,
+          schemaVersion: SCHEMA_VERSION
+        };
       }
     }
     const ddl = getSchemaSQL(this.schema);
@@ -494,6 +543,12 @@ var SqlServerStorageProvider = class {
         }
       }
     }
+    return {
+      databaseCreated,
+      schemaCreated: true,
+      alreadyUpToDate: false,
+      schemaVersion: SCHEMA_VERSION
+    };
   }
   // ─── Repository ──────────────────────────────────────────────────
   async createRepository(config) {
@@ -643,13 +698,23 @@ var SqlServerStorageProvider = class {
     if (!updated) throw new import_deep_memory.RepositoryNotFoundError(repositoryId);
     return updated;
   }
-  async deleteRepository(repositoryId) {
+  async deleteRepository(repositoryId, _onProgress) {
     const pool = this.getPool();
     const result = await pool.request().input("id", import_mssql.default.UniqueIdentifier, repositoryId).query(`DELETE FROM ${this.t("dm_repositories")} WHERE [repository_id] = @id`);
     if (result.rowsAffected[0] === 0) {
       throw new import_deep_memory.RepositoryNotFoundError(repositoryId);
     }
   }
+  async deleteAllContents(repositoryId, _onProgress) {
+    await this.assertRepository(repositoryId);
+    const pool = this.getPool();
+    const relResult = await pool.request().input("id", import_mssql.default.UniqueIdentifier, repositoryId).query(`DELETE FROM ${this.t("dm_relationships")} WHERE [repository_id] = @id`);
+    const entityResult = await pool.request().input("id", import_mssql.default.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();
@@ -1339,7 +1404,7 @@ var SqlServerStorageProvider = class {
       }
     }
   }
-  async importBulk(repositoryId, data) {
+  async importBulk(repositoryId, data, _options) {
     await this.assertRepository(repositoryId);
     const pool = this.getPool();
     let entitiesImported = 0;