From ed946c16119a22cdc7c90492ddeba56d81566400 Mon Sep 17 00:00:00 2001
From: nichenqin <nichenqin@hotmail.com>
Date: Fri, 27 Sep 2024 17:30:33 +0800
Subject: [PATCH] feat: allow to create template reference symmetric field

---
 .../src/modules/template/template.module.ts   | 17 +++-
 .../grid-view/grid-view-data-table.svelte     |  5 +-
 .../blocks/template/template-card.svelte      |  1 +
 .../blocks/template/template-grid-view.svelte | 12 ++-
 packages/env/src/index.ts                     |  1 +
 .../persistence/src/template/template-data.ts | 27 +++++--
 .../reference-field/reference-field.vo.ts     | 41 +++++++---
 packages/table/src/table.factory.ts           | 22 +++++-
 packages/template/src/templates/index.ts      |  2 +
 .../template/src/templates/test.base.json     | 77 +++++++++++++++++++
 10 files changed, 177 insertions(+), 28 deletions(-)
 create mode 100644 packages/template/src/templates/test.base.json

diff --git a/apps/backend/src/modules/template/template.module.ts b/apps/backend/src/modules/template/template.module.ts
index f16390717..8e66fe952 100644
--- a/apps/backend/src/modules/template/template.module.ts
+++ b/apps/backend/src/modules/template/template.module.ts
@@ -1,12 +1,21 @@
 import { singleton } from "@undb/di"
-import { baseTemplateSchema } from "@undb/template"
+import { None } from "@undb/domain"
+import { baseTemplateSchema, injectTemplateQueryRepository, type ITemplateQueryRepository } from "@undb/template"
 import Elysia from "elysia"
 
 @singleton()
 export class TemplateModule {
+  constructor(
+    @injectTemplateQueryRepository()
+    private readonly templateRepo: ITemplateQueryRepository,
+  ) {}
   route() {
-    return new Elysia().get("/api/template/base/schema.json", () => {
-      return baseTemplateSchema
-    })
+    return new Elysia()
+      .get("/api/template/base/schema.json", () => {
+        return baseTemplateSchema
+      })
+      .get("/api/templates", () => {
+        return this.templateRepo.find(None)
+      })
   }
 }
diff --git a/apps/frontend/src/lib/components/blocks/grid-view/grid-view-data-table.svelte b/apps/frontend/src/lib/components/blocks/grid-view/grid-view-data-table.svelte
index fe419e593..2ae13a776 100644
--- a/apps/frontend/src/lib/components/blocks/grid-view/grid-view-data-table.svelte
+++ b/apps/frontend/src/lib/components/blocks/grid-view/grid-view-data-table.svelte
@@ -40,6 +40,7 @@
   export let currentPage: Writable<number | null>
   export let isLoading = false
   export let total: number
+  export let hidePagination = false
 
   const t = getTable()
 
@@ -350,6 +351,7 @@
     </table>
   </ScrollArea>
 
+  {#if !hidePagination}
   <div class="flex items-center justify-center border-t px-4 py-2">
     <div class="flex flex-1 flex-row items-center">
       <ViewPagination {perPage} bind:currentPage={$currentPage} count={total} />
@@ -378,6 +380,7 @@
           {total} Rows
         </span>
       </div>
+      </div>
     </div>
-  </div>
+  {/if}
 </div>
diff --git a/apps/frontend/src/lib/components/blocks/template/template-card.svelte b/apps/frontend/src/lib/components/blocks/template/template-card.svelte
index 12707d52a..71c4f3352 100644
--- a/apps/frontend/src/lib/components/blocks/template/template-card.svelte
+++ b/apps/frontend/src/lib/components/blocks/template/template-card.svelte
@@ -80,6 +80,7 @@
           </div>
           <div class="mr-10 space-y-2">
             <Button
+              class="w-44"
               disabled={$createFromTemplate.isPending}
               on:click={() => $createFromTemplate.mutate({ id: template.id, includeData })}
             >
diff --git a/apps/frontend/src/lib/components/blocks/template/template-grid-view.svelte b/apps/frontend/src/lib/components/blocks/template/template-grid-view.svelte
index 68f85af25..aa4a23cfb 100644
--- a/apps/frontend/src/lib/components/blocks/template/template-grid-view.svelte
+++ b/apps/frontend/src/lib/components/blocks/template/template-grid-view.svelte
@@ -22,5 +22,15 @@
 </script>
 
 {#if store}
-  <GridViewDataTable {viewId} readonly {perPage} {currentPage} isLoading={false} total={records.length} />
+  {#key $viewId}
+    <GridViewDataTable
+      {viewId}
+      readonly
+      {perPage}
+      {currentPage}
+      isLoading={false}
+      total={records.length}
+      hidePagination
+    />
+  {/key}
 {/if}
diff --git a/packages/env/src/index.ts b/packages/env/src/index.ts
index 0e729e303..46bab5132 100644
--- a/packages/env/src/index.ts
+++ b/packages/env/src/index.ts
@@ -148,6 +148,7 @@ export const env = createEnv({
   clientPrefix: "UNDB_PUBLIC_",
   client: {},
   server: {
+    NODE_ENV: z.enum(["development", "production", "test"]).default("development"),
     UNDB_BASE_URL: z.string().optional().default("http://localhost:3721"),
     UNDB_COOKIE_DOMAIN: z.string().optional(),
   },
diff --git a/packages/persistence/src/template/template-data.ts b/packages/persistence/src/template/template-data.ts
index 57cd8f726..ad8ba3426 100644
--- a/packages/persistence/src/template/template-data.ts
+++ b/packages/persistence/src/template/template-data.ts
@@ -1,3 +1,4 @@
+import { env } from "@undb/env"
 import { templates, type IBaseTemplateDTO, type ITemplateDTO } from "@undb/template"
 
 export const templateData: ITemplateDTO[] = [
@@ -13,6 +14,17 @@ export const templateData: ITemplateDTO[] = [
       template: templates.projectManagement as IBaseTemplateDTO,
     },
   },
+  {
+    id: "6ba7b814-9dad-11d1-80b4-00c04fd430c8",
+    icon: "💼",
+    name: "CRM",
+    category: "sales",
+    description: "A template for managing customer relationships, deals, and activities.",
+    template: {
+      type: "base",
+      template: templates.crm as IBaseTemplateDTO,
+    },
+  },
   {
     id: "6ba7b810-9dad-11d1-80b4-00c04fd430c8",
     icon: "🎉",
@@ -46,15 +58,18 @@ export const templateData: ITemplateDTO[] = [
       template: templates.officeInventoryManagement as IBaseTemplateDTO,
     },
   },
-  {
+]
+
+if (env.NODE_ENV === "development") {
+  templateData.unshift({
     id: "6ba7b814-9dad-11d1-80b4-00c04fd430c8",
     icon: "💼",
-    name: "CRM",
+    name: "Test",
     category: "sales",
-    description: "A template for managing customer relationships, deals, and activities.",
+    description: "A template for testing",
     template: {
       type: "base",
-      template: templates.crm as IBaseTemplateDTO,
+      template: templates.test as IBaseTemplateDTO,
     },
-  },
-]
+  })
+}
diff --git a/packages/table/src/modules/schema/fields/variants/reference-field/reference-field.vo.ts b/packages/table/src/modules/schema/fields/variants/reference-field/reference-field.vo.ts
index 8f84a142c..d682d5c71 100644
--- a/packages/table/src/modules/schema/fields/variants/reference-field/reference-field.vo.ts
+++ b/packages/table/src/modules/schema/fields/variants/reference-field/reference-field.vo.ts
@@ -51,24 +51,29 @@ export const createReferenceFieldDTO = createBaseFieldDTO
   })
   .omit({ display: true })
 
+const createOwnerReferenceFieldOption = z.object({
+  foreignTable: z.object({
+    baseName: baseNameSchema.optional(),
+    tableName: tableName,
+  }),
+})
+const createSymmetricReferenceFieldOption = z.object({
+  symmetricFieldId: fieldId,
+})
+
 const createTablesReferenceOption = createReferenceFieldOption
   .omit({
     foreignTableId: true,
   })
-  .merge(
-    z.object({
-      foreignTable: z.object({
-        baseName: baseNameSchema.optional(),
-        tableName: tableName,
-      }),
-    }),
-  )
+  .merge(createOwnerReferenceFieldOption.partial())
+  .merge(createSymmetricReferenceFieldOption.partial())
 
 export const createTablesReferenceFieldDTO = createReferenceFieldDTO.merge(
   z.object({
     option: createTablesReferenceOption,
   }),
 )
+export type ICreateTablesReferenceFieldDTO = z.infer<typeof createTablesReferenceFieldDTO>
 
 export type ICreateReferenceFieldDTO = z.infer<typeof createReferenceFieldDTO>
 export const updateReferenceFieldDTO = createReferenceFieldDTO
@@ -121,15 +126,27 @@ export class ReferenceField extends AbstractField<
       name: dto.name,
       option: { foreignTableId, condition, isOwner: true },
       id: FieldIdVo.fromStringOrCreate(dto.id).value,
+      constraint: dto.constraint,
     })
   }
 
-  static createSymmetricField(foreignTable: TableDo, table: TableDo, field: ReferenceField) {
+  static createSymmetricField(
+    foreignTable: TableDo,
+    table: TableDo,
+    field: ReferenceField,
+    dto?: ICreateTablesReferenceFieldDTO,
+  ) {
     const symmetricField = new ReferenceField({
       type: "reference",
-      name: table.schema.getNextFieldName(foreignTable.name.value),
-      option: { isOwner: false, foreignTableId: foreignTable.id.value, symmetricFieldId: field.id.value },
-      id: FieldIdVo.create().value,
+      id: dto?.id || FieldIdVo.create().value,
+      name: table.schema.getNextFieldName(dto?.name || foreignTable.name.value),
+      option: {
+        isOwner: false,
+        foreignTableId: foreignTable.id.value,
+        symmetricFieldId: field.id.value,
+        condition: dto?.option.condition,
+      },
+      constraint: dto?.constraint,
     })
 
     symmetricField.connect(field)
diff --git a/packages/table/src/table.factory.ts b/packages/table/src/table.factory.ts
index a64008ea0..8b28f1dc9 100644
--- a/packages/table/src/table.factory.ts
+++ b/packages/table/src/table.factory.ts
@@ -10,6 +10,7 @@ import {
   type ICreateFieldDTO,
   type ICreateReferenceFieldDTO,
   type ICreateSchemaDTO,
+  type ICreateTablesReferenceFieldDTO,
 } from "./modules"
 import { FieldNameShouldBeUnique } from "./modules/schema/rules"
 import { TableIdVo } from "./table-id.vo"
@@ -82,15 +83,25 @@ export class TableFactory {
     const baseName = getNextName(baseNames, base.name.value)
 
     const referenceIds = new Set<string>()
+    const symmetricFields: ICreateTablesReferenceFieldDTO[] = []
     const tables = dtos.map((dto) => {
       const schema = dto.schema
-        .filter((f) => f.type !== "rollup")
         .map((field) => {
+          if (field.type === "rollup") {
+            return null
+          }
           if (field.type === "reference") {
-            const foreignTableId = ids.mustGet(baseName, field.option.foreignTable.tableName)
             if (!field.id) {
               field.id = FieldIdVo.create().value
             }
+
+            // Symmetric field
+            if (!field.option.foreignTable) {
+              symmetricFields.push(field)
+              return null
+            }
+
+            const foreignTableId = ids.mustGet(baseName, field.option.foreignTable.tableName)
             if (field.option.createSymmetricField) {
               referenceIds.add(field.id!)
             }
@@ -103,7 +114,8 @@ export class TableFactory {
             } as ICreateReferenceFieldDTO
           }
           return field as ICreateFieldDTO
-        }) as ICreateSchemaDTO
+        })
+        .filter((v) => !!v) as ICreateSchemaDTO
 
       const id = ids.mustGet(baseName, dto.name)
       const table = this.create({ ...dto, id, schema })
@@ -118,7 +130,9 @@ export class TableFactory {
         if (!foreignTable) {
           throw new Error("Foreign table not found")
         }
-        const symmetricField = ReferenceField.createSymmetricField(table, foreignTable, referenceField)
+        const dto = symmetricFields.find((f) => f.option.symmetricFieldId === referenceField.id.value)
+        console.log(dto)
+        const symmetricField = ReferenceField.createSymmetricField(table, foreignTable, referenceField, dto)
         foreignTable.$createFieldSpec(symmetricField)
       }
     }
diff --git a/packages/template/src/templates/index.ts b/packages/template/src/templates/index.ts
index 60eb249f0..f69f66843 100644
--- a/packages/template/src/templates/index.ts
+++ b/packages/template/src/templates/index.ts
@@ -2,9 +2,11 @@ import { default as crm } from "./crm.base.json"
 import { default as eventPlaningList } from "./eventPlaning.base.json"
 import { default as officeInventoryManagement } from "./officeInventoryManagement.base.json"
 import { default as projectManagement } from "./projectManagement.base.json"
+import { default as test } from "./test.base.json"
 import { default as todoList } from "./todoList.base.json"
 
 const templates = {
+  test,
   todoList,
   projectManagement,
   officeInventoryManagement,
diff --git a/packages/template/src/templates/test.base.json b/packages/template/src/templates/test.base.json
new file mode 100644
index 000000000..8ee491014
--- /dev/null
+++ b/packages/template/src/templates/test.base.json
@@ -0,0 +1,77 @@
+{
+  "Test": {
+    "tables": {
+      "Table1": {
+        "schema": {
+          "Title": {
+            "id": "title",
+            "type": "string"
+          },
+          "Ref2": {
+            "id": "ref2",
+            "type": "reference",
+            "option": {
+              "createSymmetricField": true,
+              "foreignTable": {
+                "tableName": "Table2"
+              }
+            }
+          },
+          "Roll1": {
+            "id": "roll1",
+            "type": "rollup",
+            "option": {
+              "referenceFieldId": "ref2",
+              "rollupFieldId": "title",
+              "fn": "lookup"
+            }
+          }
+        },
+        "records": [
+          {
+            "Title": "1-1",
+            "Ref2": ["1", "2"]
+          },
+          {
+            "Title": "1-2",
+            "Ref2": ["2"]
+          }
+        ]
+      },
+      "Table2": {
+        "schema": {
+          "Title": {
+            "id": "title",
+            "type": "string"
+          },
+          "Ref1": {
+            "id": "ref1",
+            "type": "reference",
+            "option": {
+              "symmetricFieldId": "ref2"
+            }
+          },
+          "T1": {
+            "id": "t1",
+            "type": "rollup",
+            "option": {
+              "referenceFieldId": "ref1",
+              "rollupFieldId": "title",
+              "fn": "lookup"
+            }
+          }
+        },
+        "records": [
+          {
+            "id": "1",
+            "Title": "2-1"
+          },
+          {
+            "id": "2",
+            "Title": "2-2"
+          }
+        ]
+      }
+    }
+  }
+}