Include external models in project snapshots

src/db/repositories/project-state-restore-command.repository.ts

@@ -27,6 +27,10 @@ import { projectChangeSets } from "../schema/project-change-sets.js";
 import { projects } from "../schema/projects.js";
 import { rooms } from "../schema/rooms.js";
 import { externalCsvConfigurations } from "../schema/external-csv-configurations.js";
+import { externalImportBatches } from "../schema/external-import-batches.js";
+import { externalModelObjects } from "../schema/external-model-objects.js";
+import { externalModelSources } from "../schema/external-model-sources.js";
+import { externalRoomMappings } from "../schema/external-room-mappings.js";
 import { executeProjectCommandTransaction } from "./project-command-transaction.persistence.js";
 import {
   readProjectStateSnapshot,
@@ -163,6 +167,12 @@ function canonicalProjectState(state: ProjectStateSnapshot) {
     floors: byId(state.floors),
     rooms: byId(state.rooms),
     externalCsvConfiguration: state.externalCsvConfiguration,
+    externalModel: {
+      source: state.externalModel.source,
+      importBatches: byId(state.externalModel.importBatches),
+      roomMappings: byId(state.externalModel.roomMappings),
+      objects: byId(state.externalModel.objects),
+    },
   });
 }
 
@@ -185,6 +195,10 @@ function replaceProjectState(
   database: AppDatabase,
   state: ProjectStateSnapshot
 ) {
+  database
+    .delete(externalModelSources)
+    .where(eq(externalModelSources.projectId, state.project.id))
+    .run();
   database
     .delete(externalCsvConfigurations)
     .where(eq(externalCsvConfigurations.projectId, state.project.id))
@@ -263,6 +277,21 @@ function replaceProjectState(
     circuitDeviceRows,
     state.circuits.flatMap((circuit) => circuit.deviceRows)
   );
+  if (state.externalModel.source !== null) {
+    database.insert(externalModelSources).values(state.externalModel.source).run();
+  }
+  insertMany(
+    database,
+    externalImportBatches,
+    state.externalModel.importBatches.map(
+      ({ originalContentBase64, ...batch }) => ({
+        ...batch,
+        originalBytes: Buffer.from(originalContentBase64, "base64"),
+      })
+    )
+  );
+  insertMany(database, externalRoomMappings, state.externalModel.roomMappings);
+  insertMany(database, externalModelObjects, state.externalModel.objects);
 }
 
 function insertMany<TTable extends Parameters<AppDatabase["insert"]>[0]>(

src/db/repositories/project-state-snapshot.persistence.ts

@@ -20,6 +20,7 @@ import { projectDevices } from "../schema/project-devices.js";
 import { projects } from "../schema/projects.js";
 import { rooms } from "../schema/rooms.js";
 import { externalCsvConfigurations } from "../schema/external-csv-configurations.js";
+import { ExternalModelStateRepository } from "./external-model-state.repository.js";
 
 export interface PersistedProjectStateSnapshot {
   currentRevision: number;
@@ -65,6 +66,8 @@ export function readProjectStateSnapshot(
     .from(externalCsvConfigurations)
     .where(eq(externalCsvConfigurations.projectId, projectId))
     .get() ?? null;
+  const externalModel = new ExternalModelStateRepository(database)
+    .getByProject(projectId).state;
   const listRows = database
     .select()
     .from(circuitLists)
@@ -248,6 +251,7 @@ export function readProjectStateSnapshot(
     },
     distributionBoards: boardRows,
     externalCsvConfiguration,
+    externalModel,
     circuitLists: listRows,
     circuitSections: sectionRows,
     distributionBoardComponents: componentRows,

src/db/repositories/project-transfer.repository.ts

@@ -21,6 +21,10 @@ import { projectDevices } from "../schema/project-devices.js";
 import { projects } from "../schema/projects.js";
 import { rooms } from "../schema/rooms.js";
 import { externalCsvConfigurations } from "../schema/external-csv-configurations.js";
+import { externalImportBatches } from "../schema/external-import-batches.js";
+import { externalModelObjects } from "../schema/external-model-objects.js";
+import { externalModelSources } from "../schema/external-model-sources.js";
+import { externalRoomMappings } from "../schema/external-room-mappings.js";
 import {
   hashProjectStatePayload,
   readProjectStateSnapshot,
@@ -157,6 +161,21 @@ function insertProjectState(
     circuitDeviceRows,
     state.circuits.flatMap((circuit) => circuit.deviceRows)
   );
+  if (state.externalModel.source !== null) {
+    database.insert(externalModelSources).values(state.externalModel.source).run();
+  }
+  insertMany(
+    database,
+    externalImportBatches,
+    state.externalModel.importBatches.map(
+      ({ originalContentBase64, ...batch }) => ({
+        ...batch,
+        originalBytes: Buffer.from(originalContentBase64, "base64"),
+      })
+    )
+  );
+  insertMany(database, externalRoomMappings, state.externalModel.roomMappings);
+  insertMany(database, externalModelObjects, state.externalModel.objects);
 }
 
 function insertMany<TTable extends Parameters<AppDatabase["insert"]>[0]>(