pqb 0.7.0 → 0.7.2

package/dist/index.js

@@ -1312,6 +1312,45 @@ const pushColumnInfoSql = (ctx, table, query) => {
   }
 };
 
+const pushCopySql = (ctx, model, query, quotedAs) => {
+  const { sql } = ctx;
+  const { copy } = query;
+  const columns = copy.columns ? `(${copy.columns.map(q).join(", ")})` : "";
+  const target = "from" in copy ? copy.from : copy.to;
+  sql.push(
+    `COPY ${q(model.table)}${columns} ${"from" in copy ? "FROM" : "TO"} ${typeof target === "string" ? quote(target) : `PROGRAM ${quote(target.program)}`}`
+  );
+  if (Object.keys(copy).length > (copy.columns ? 2 : 1)) {
+    const options = [];
+    if (copy.format)
+      options.push(`FORMAT ${copy.format}`);
+    if (copy.freeze)
+      options.push(`FREEZE ${copy.freeze}`);
+    if (copy.delimiter)
+      options.push(`DELIMITER ${quote(copy.delimiter)}`);
+    if (copy.null)
+      options.push(`NULL ${quote(copy.null)}`);
+    if (copy.header)
+      options.push(`HEADER ${copy.header}`);
+    if (copy.quote)
+      options.push(`QUOTE ${quote(copy.quote)}`);
+    if (copy.escape)
+      options.push(`ESCAPE ${quote(copy.escape)}`);
+    if (copy.forceQuote)
+      options.push(
+        `FORCE_QUOTE ${copy.forceQuote === "*" ? "*" : `(${copy.forceQuote.map(q).join(", ")})`}`
+      );
+    if (copy.forceNotNull)
+      options.push(`FORCE_NOT_NULL (${copy.forceNotNull.map(q).join(", ")})`);
+    if (copy.forceNull)
+      options.push(`FORCE_NULL (${copy.forceNull.map(q).join(", ")})`);
+    if (copy.encoding)
+      options.push(`ENCODING ${quote(copy.encoding)}`);
+    sql.push(`WITH (${options.join(", ")})`);
+  }
+  pushWhereStatementSql(ctx, model, query, quotedAs);
+};
+
 const toSqlCacheKey = Symbol("toSqlCache");
 const toSql = (model, options) => {
   return !(options == null ? void 0 : options.clearCache) && model.query[toSqlCacheKey] || (model.query[toSqlCacheKey] = makeSql(model, options));
@@ -1356,6 +1395,10 @@ const makeSql = (model, { values = [] } = {}) => {
       pushDeleteSql(ctx, model, query, quotedAs2);
       return { text: sql.join(" "), values };
     }
+    if (query.type === "copy") {
+      pushCopySql(ctx, model, query, quotedAs2);
+      return { text: sql.join(" "), values };
+    }
   }
   const quotedAs = model.table && q(query.as || model.table);
   sql.push("SELECT");
@@ -1566,7 +1609,7 @@ const assignMethodsToClass = (klass, methods) => {
       writable: true,
       value(...args) {
         const cloned = cloneInstance(this);
-        return methods[name].call(cloned, args);
+        return methods[name].apply(cloned, args);
       }
     });
   }
@@ -5466,6 +5509,19 @@ class RawMethods {
   }
 }
 
+class CopyMethods {
+  copy(arg) {
+    return this.clone()._copy(arg);
+  }
+  _copy(arg) {
+    Object.assign(this.query, {
+      type: "copy",
+      copy: arg
+    });
+    return this;
+  }
+}
+
 class QueryMethods {
   all() {
     return this.clone()._all();
@@ -5653,7 +5709,8 @@ applyMixins(QueryMethods, [
   QueryUpsert,
   QueryGet,
   MergeQueryMethods,
-  RawMethods
+  RawMethods,
+  CopyMethods
 ]);
 
 var __defProp = Object.defineProperty;
@@ -5694,7 +5751,7 @@ class Db {
     this.columnTypes = columnTypes;
     this.whereQueryBuilder = WhereQueryBuilder;
     this.onQueryBuilder = OnQueryBuilder;
-    var _a;
+    var _a, _b;
     this.__model = this;
     const logger = options.logger || console;
     this.query = {
@@ -5710,8 +5767,17 @@ class Db {
     this.primaryKeys = Object.keys(shape).filter(
       (key) => shape[key].isPrimaryKey
     );
+    const primaryKeysFromData = (_b = getTableData().primaryKey) == null ? void 0 : _b.columns;
+    if (primaryKeysFromData)
+      this.primaryKeys.push(...primaryKeysFromData);
     if (this.primaryKeys.length === 1) {
       this.singlePrimaryKey = this.primaryKeys[0];
+    } else if (this.primaryKeys.length === 0 && shape !== anyShape && options.noPrimaryKey !== "ignore") {
+      const message = `Table ${table} has no primary key`;
+      if (options.noPrimaryKey === "error")
+        throw new Error(message);
+      else
+        logger.warn(message);
     }
     const columns = Object.keys(
       shape
@@ -5761,18 +5827,19 @@ const createDb = (_a) => {
     "logger",
     "columnTypes"
   ]);
-  var _a2;
+  var _a2, _b2;
   const adapter = "adapter" in options ? options.adapter : new Adapter(options);
   const commonOptions = {
     log,
     logger,
-    autoPreparedStatements: (_a2 = options.autoPreparedStatements) != null ? _a2 : false
+    autoPreparedStatements: (_a2 = options.autoPreparedStatements) != null ? _a2 : false,
+    noPrimaryKey: (_b2 = options.noPrimaryKey) != null ? _b2 : "error"
   };
   const qb = new Db(
     adapter,
     void 0,
     void 0,
-    {},
+    anyShape,
     ct,
     commonOptions
   );