@andymic/pigeon 2.0.1 → 2.1.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@andymic/pigeon",
-  "version": "2.0.1",
+  "version": "2.1.0",
   "author": "Andreas Michael <ateasm03@gmail.com>",
   "description": "Pigeon is a TypeScript-based tool for generating TypeScript classes and methods from PostgreSQL database schemas.",
   "keywords": [
@@ -35,5 +35,5 @@
     "pg": "^8.16.0",
     "prompt-sync": "^4.2.0"
   },
-  "gitHead": "565fcc64621e0c076f77933662e6f82969ef62e6"
+  "gitHead": "029bee420b59d84dfb269ec7eb3190aee486de10"
 }

package/src/index.js

@@ -341,7 +341,7 @@ function createIndex(db) {
     text += "import pg from \"pg\";\n";
     text += "const { Client, Pool } = pg;\n";
     text += "\n";
-    text += "const pool = new Pool({\n";
+    text += "export const pool = new Pool({\n";
     text += "\thost: \"" + db.host + "\",\n";
     text += "\tport: " + db.port + ",\n";
     text += "\tdatabase: \"" + db.db + "\",\n";
@@ -352,7 +352,7 @@ function createIndex(db) {
     text += "\tconnectionTimeoutMillis: 2000\n";
     text += "});\n";
     text += "\n";
-    text += "const client = new Client({\n";
+    text += "export const client = new Client({\n";
     text += "\thost: \"" + db.host + "\",\n";
     text += "\tport: " + db.port + ",\n";
     text += "\tdatabase: \"" + db.db + "\",\n";
@@ -360,12 +360,9 @@ function createIndex(db) {
     text += "\tpassword: \"" + db.pass + "\"\n";
     text += "});\n";
     text += "\n";
-    text += "/*\n";
-    text += "export default (sql: string, params: any[]) => pool.query(sql, params);\n";
-    text += "*/\n";
+    text += "const poolQuery = (sql: string, params: any[]) => pool.query(sql, params);\n";
     text += "\n";
-    text += "/*\n";
-    text += "export default async (sql: string, params: any[]) => {\n";
+    text += "const clientQuery = async (sql: string, params: any[]) => {\n";
     text += "\ttry {\n";
     text += "\t\tawait client.connect();\n";
     text += "\t\treturn await client.query(sql, params);\n";
@@ -375,7 +372,10 @@ function createIndex(db) {
     text += "\t\tawait client.end();\n";
     text += "\t}\n";
     text += "}\n";
-    text += "*/";
+    text += "\n";
+    text += "//export default poolQuery;\n";
+    text += "\n";
+    text += "//export default clientQuery;\n";
     return text;
 }
 function createClass(tableName, columns) {
@@ -475,9 +475,10 @@ function createGet(tableSchema, tableName, columns) {
     text += " * @param {Partial<" + className + ">} criteria - An object containing properties of the " + className + " to filter by.\n";
     text += " * Only " + nameBeautifier(tableName).toLowerCase() + " matching all provided criteria will be returned.\n";
     text += " * An empty object or undefined criteria will fetch all " + className + " objects.\n";
+    text += " * @param {Function} [runQuery=query] - The function that executes the database query. This is optional and defaults to the production `query` function. It can be used for transactions when using pools.\n";
     text += " * @returns {Promise<" + className + "[]>} - A Promise object returning an array of " + nameBeautifier(tableName) + " matching the criteria.\n";
     text += " */\n";
-    text += "export async function get" + nameBeautifier(tableName).replaceAll(" ", "") + "(criteria: Partial<" + className + ">): Promise<" + className + "[]> {\n";
+    text += "export async function get" + nameBeautifier(tableName).replaceAll(" ", "") + "(criteria: Partial<" + className + ">, runQuery: Function = query): Promise<" + className + "[]> {\n";
     text += "\tconst whereClauses: string[] = [];\n";
     text += "\tconst values: any[] = [];\n";
     text += "\tlet paramIndex = 1;\n";
@@ -495,7 +496,7 @@ function createGet(tableSchema, tableName, columns) {
     text += "\t\tsql += \" WHERE \" + whereClauses.join(\" AND \");\n";
     text += "\tsql += \";\";\n";
     text += "\n";
-    text += "\tconst " + varName + "Query = await query(sql, values);\n";
+    text += "\tconst " + varName + "Query = await runQuery(sql, values);\n";
     text += arrayMaker(1, varName, className, columns) + "\n";
     text += "\treturn " + varName + ";\n";
     text += "}";
@@ -539,6 +540,7 @@ function createAdd(tableSchema, tableName, required, optional, autoGenerated, ha
         text += " * - The field" + (optional.length > 1 ? "s" : "") + " `" + optional.map((column) => {
             return column.name;
         }).join("`, `") + "` " + (optional.length > 1 ? "are" : "is") + " optional.\n";
+    text += " * @param {Function} [runQuery=query] - The function that executes the database query. This is optional and defaults to the production `query` function. It can be used for transactions when using pools.\n";
     text += " * @returns {Promise<" + className + ">} A Promise returning the newly created " + className + " object.\n";
     if (hasForeign) {
         text += " * @throws {Error} An exception in the case of the ";
@@ -549,7 +551,7 @@ function createAdd(tableSchema, tableName, required, optional, autoGenerated, ha
         text += " not existing in their table, or if other pre-insertion validation fails.\n";
     }
     text += " */\n";
-    text += "export async function " + functionName + "(" + paramName + ": " + dataType.replaceAll("'", "\"") + "): Promise<" + className + "> {\n";
+    text += "export async function " + functionName + "(" + paramName + ": " + dataType.replaceAll("'", "\"") + ", runQuery: Function = query): Promise<" + className + "> {\n";
     if (hasForeign) {
         for (const column of editableColumns) {
             if (column.isForeign && column.foreignColumn && column.foreignTable && column.foreignSchema) {
@@ -590,7 +592,7 @@ function createAdd(tableSchema, tableName, required, optional, autoGenerated, ha
     text += "\t\tthrow new Error(\"No data provided for " + paramName + " creation.\");\n";
     text += "\n";
     text += "\tconst sql = \"INSERT INTO " + tableSchema + "." + tableName + " (\" + intoClauses.join(\", \") + \") VALUES (\" + valuesClauses.join(\", \") + \") RETURNING *;\";\n";
-    text += "\tconst insertQuery = await query(sql, values);\n";
+    text += "\tconst insertQuery = await runQuery(sql, values);\n";
     text += "\treturn new " + className + "(\n";
     let columns = editableColumns.concat(autoGenerated);
     columns = [...new Set(columns)];
@@ -620,11 +622,12 @@ function createUpdate(tableSchema, tableName, columns, autoGenerated, hasForeign
     text += " * An empty object would typically mean updating ALL " + nameBeautifier(tableName) + ", which should be handled with caution or disallowed.\n";
     text += " * @param {" + updatesType + "} updates - An object containing the new values for the fields to be updated.\n";
     text += " * Only fields present in this object will be updated.\n";
+    text += " * @param {Function} [runQuery=query] - The function that executes the database query. This is optional and defaults to the production `query` function. It can be used for transactions when using pools.\n";
     text += " * @returns {Promise<" + className + "[]>} A promise that resolves to an array of the updated " + className + " objects.\n";
     text += " * Returns an empty array if no records matched the criteria or if no rows were updated.\n";
     text += " * @throws {Error} If updates object is empty" + (hasForeign ? ", or if foreign key checks fail for provided update values" : "") + ".\n";
     text += " */\n";
-    text += "export async function update" + nameBeautifier(tableName).replaceAll(" ", "") + "(criteria: " + criteriaType.replaceAll("'", "\"") + ", updates: " + updatesType.replaceAll("'", "\"") + "): Promise<" + className + "[]> {\n";
+    text += "export async function update" + nameBeautifier(tableName).replaceAll(" ", "") + "(criteria: " + criteriaType.replaceAll("'", "\"") + ", updates: " + updatesType.replaceAll("'", "\"") + ", runQuery: Function = query): Promise<" + className + "[]> {\n";
     text += "\tif (Object.keys(updates).length === 0)\n";
     text += "\t\tthrow new Error(\"No update data provided.\");\n";
     text += "\n";
@@ -674,7 +677,7 @@ function createUpdate(tableSchema, tableName, columns, autoGenerated, hasForeign
     text += "\n";
     text += "\tlet sql = \"UPDATE " + tableSchema + "." + tableName + " SET \" + setClauses.join(\", \") + (whereClauses.length !== 0 ? \" WHERE \" + whereClauses.join(\" AND \") : \"\") + \" RETURNING *;\";\n";
     text += "\n";
-    text += "\tconst updateQuery = await query(sql, values);\n";
+    text += "\tconst updateQuery = await runQuery(sql, values);\n";
     text += arrayMaker(1, "update", className, columns) + "\n";
     text += "\treturn update;\n";
     text += "}";
@@ -692,9 +695,10 @@ function createDelete(tableSchema, tableName, columns) {
     text += " * @param {Partial<" + className + ">} criteria - An object containing properties of the " + className + " to filter by for deletion.\n";
     text += " * An empty object would typically mean deleting ALL " + nameBeautifier(tableName) + ", which should be handled with caution or disallowed.\n";
     text += " * Only accounts matching all provided criteria will be deleted.\n";
+    text += " * @param {Function} [runQuery=query] - The function that executes the database query. This is optional and defaults to the production `query` function. It can be used for transactions when using pools.\n";
     text += " * @returns {Promise<" + className + "[]>} - A Promise object returning the deleted " + nameBeautifier(tableName) + ".\n";
     text += " */\n";
-    text += "export async function " + functionName + "(criteria : Partial<" + className + ">): Promise<" + className + "[]> {\n";
+    text += "export async function " + functionName + "(criteria : Partial<" + className + ">, runQuery: Function = query): Promise<" + className + "[]> {\n";
     text += "\tconst whereClauses: string[] = [];\n";
     text += "\tconst values: any[] = [];\n";
     text += "\tlet paramIndex = 1;\n";
@@ -711,7 +715,7 @@ function createDelete(tableSchema, tableName, columns) {
     text += "\n";
     text += "\tlet sql = \"DELETE FROM " + tableSchema + "." + tableName + "\" + (whereClauses.length !== 0 ? \" WHERE \" + whereClauses.join(\" AND \") : \"\") + \" RETURNING *;\";\n";
     text += "\n";
-    text += "\tconst removeQuery = await query(sql, values);\n";
+    text += "\tconst removeQuery = await runQuery(sql, values);\n";
     text += arrayMaker(1, "remove", className, columns) + "\n";
     text += "\treturn remove;\n";
     text += "}";