fix: complex custom promary key generation

Refs: #52

PR-URL: #56

lib/schema-pg.js

@@ -83,8 +83,20 @@ const foreignKey = (entityName, def) => {
   );
 };
 
-const primaryCustom = (entityName, fields) => {
-  const fieldNames = `"${fields.map(name => name + 'Id').join('", "')}"`;
+const primaryCustom = (entityName, fields, entity) => {
+  const idx = [...fields];
+  if (entity) {
+    for (let i = 0; i < idx.length; i++) {
+      const field = idx[i];
+      const def = entity[field];
+      if (!def) continue;
+      const { type } = def;
+      if (!type) continue;
+      const ref = isUpperCamel(type);
+      idx[i] = ref ? field + 'Id' : field;
+    }
+  }
+  const fieldNames = `"${idx.join('", "')}"`;
   const constraint = `"pk${entityName}" PRIMARY KEY (${fieldNames})`;
   return `ALTER TABLE "${entityName}" ADD CONSTRAINT ${constraint};`;
 };
@@ -147,7 +159,7 @@ class PgSchema extends DatabaseSchema {
       }
       if (def.unique || def.index) idx.push(createIndex(name, def));
       if (def.primary) {
-        idx[0] = primaryCustom(name, def.primary);
+        idx[0] = primaryCustom(name, def.primary, entity);
         sql.splice(1, 1);
       }
       if (def.many) idx.push(await createMany(name, def, this));