Store annotation_map

seamapi · Dec 27, 2023 · daf746a · daf746a
1 parent 4bc5dd1
commit daf746a
Show file tree

Hide file tree

Showing 6 changed files with 17 additions and 3 deletions.
diff --git a/src/lib/database/seed-from-api.ts b/src/lib/database/seed-from-api.ts
@@ -28,9 +28,13 @@ export const seedDatabaseFromApi = async (
 
     if (hasManufacturer) continue
 
-    const { logo, ...rest } = manufacturer
+    const { logo, annotations, ...rest } = manufacturer
     db.addManufacturer({
       ...rest,
+      annotation_map: annotations.reduce(
+        (acc, cur) => ({ ...acc, [cur.annotation_code]: cur }),
+        {},
+      ),
       logo:
         logo != null
           ? {

diff --git a/src/lib/database/seed.ts b/src/lib/database/seed.ts
@@ -19,6 +19,12 @@ export const seedDatabase = (db: Database) => {
     integration_support_level: "stable",
     is_connect_webview_supported: true,
     requires_seam_support_to_add_account: false,
+    annotation_map: {
+      subscription_required: {
+        annotation_code: "subscription_required",
+        message: "Need to subscribe",
+      },
+    },
   })
 
   const device_model = db.addDeviceModel({

diff --git a/src/lib/models.ts b/src/lib/models.ts
@@ -13,7 +13,7 @@ export const stored_manufacturer = schemas.manufacturer
   .merge(
     z.object({
       logo: stored_image_reference.optional(),
-      annotations: z
+      annotation_map: z
         .record(
           schemas.manufacturer_annotation_code,
           schemas.manufacturer_annotation,

diff --git a/src/lib/public-mappings/manufacturer.ts b/src/lib/public-mappings/manufacturer.ts
@@ -18,7 +18,7 @@ export const publicMapManufacturer = ({
 }: PublicMapManufacturerOptions): Manufacturer => ({
   manufacturer_id: manufacturer.manufacturer_id,
   display_name: manufacturer.display_name,
-  annotations: Object.values(manufacturer.annotations ?? []),
+  annotations: Object.values(manufacturer.annotation_map ?? {}),
   logo: manufacturer.logo
     ? publicMapImageReference({
         image: manufacturer.logo,

diff --git a/test/api/v1/manufacturers/get.test.ts b/test/api/v1/manufacturers/get.test.ts
@@ -16,4 +16,7 @@ test("GET /v1/manufacturers/get", async (t: ExecutionContext) => {
 
   t.is(data.manufacturer.manufacturer_id, manufacturer.manufacturer_id)
   t.is(data.manufacturer.device_model_count, 1)
+  t.is(data.manufacturer.annotations.length, 1)
+  const annotation_message = data.manufacturer.annotations[0]?.message ?? ""
+  t.true(annotation_message.length > 0)
 })
diff --git a/test/seed.test.ts b/test/seed.test.ts
@@ -33,6 +33,7 @@ test("seed database from api", async (t) => {
     integration_support_level: "stable",
     is_connect_webview_supported: true,
     requires_seam_support_to_add_account: false,
+    annotation_map: {},
   })
 
   const device_model = apiDb.addDeviceModel({