@query-doctor/core 0.2.5 → 0.3.0

package/dist/index.js

@@ -780,6 +780,9 @@ var PostgresQueryBuilder = class _PostgresQueryBuilder {
     __publicField(this, "isIntrospection", false);
     __publicField(this, "explainFlags", []);
     __publicField(this, "_preamble", 0);
+    __publicField(this, "parameters", {});
+    // substitution for `limit $1` -> `limit 5`
+    __publicField(this, "limitSubstitution");
   }
   get preamble() {
     return this._preamble;
@@ -813,6 +816,14 @@ var PostgresQueryBuilder = class _PostgresQueryBuilder {
     this.explainFlags = flags;
     return this;
   }
+  parameterize(parameters) {
+    Object.assign(this.parameters, parameters);
+    return this;
+  }
+  replaceLimit(limit) {
+    this.limitSubstitution = limit;
+    return this;
+  }
   build() {
     let commands = this.generateSetCommands();
     commands += this.generateExplain().query;
@@ -841,14 +852,28 @@ var PostgresQueryBuilder = class _PostgresQueryBuilder {
     return commands;
   }
   generateExplain() {
-    let query = "";
+    let finalQuery = "";
     if (this.explainFlags.length > 0) {
-      query += `explain (${this.explainFlags.join(", ")}) `;
+      finalQuery += `explain (${this.explainFlags.join(", ")}) `;
+    }
+    const query = this.substituteQuery();
+    const semicolon = query.endsWith(";") ? "" : ";";
+    const preamble = finalQuery.length;
+    finalQuery += `${query}${semicolon}`;
+    return { query: finalQuery, preamble };
+  }
+  substituteQuery() {
+    let query = this.query;
+    if (this.limitSubstitution !== void 0) {
+      query = query.replace(
+        /limit\s+\$\d+/g,
+        `limit ${this.limitSubstitution}`
+      );
+    }
+    for (const [key, value] of Object.entries(this.parameters)) {
+      query = query.replaceAll(`\\$${key}`, value.toString());
     }
-    const semicolon = this.query.endsWith(";") ? "" : ";";
-    const preamble = query.length;
-    query += `${this.query}${semicolon}`;
-    return { query, preamble };
+    return query;
   }
 };