feat: generation gin and gist indexes for pg SQL DDL

Closes: #5 PR-URL: #19
metarhia · Sep 11, 2020 · ad96ef2 · ad96ef2
1 parent 13365e0
commit ad96ef2
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/lib/ddl.js b/lib/ddl.js
@@ -35,11 +35,12 @@ const createIndex = (entityName, { name, unique, index }) => {
   const prefix = unique ? 'ak' : 'idx';
   let fields = unique || index;
   const type = typeof fields;
-  if (type === 'boolean') fields = [name];
-  else if (type === 'string') fields = [fields];
+  if (type === 'boolean') fields = `("${name}")`;
+  else if (type !== 'string') fields = `("${fields.join('", "')}")`;
+  else if (fields.includes('(')) fields = `USING ${fields}`;
+  else fields = `("${fields}")`;
   const idxName = `"${prefix}${entityName}${toUpperCamel(name)}"`;
-  const idxOn = `"${entityName}" ("${fields.join('", "')}")`;
-  return `CREATE ${uni}INDEX ${idxName} ON ${idxOn};`;
+  return `CREATE ${uni}INDEX ${idxName} ON "${entityName}" ${fields};`;
 };
 
 const foreignKey = (entityName, { name, type }) => {