@graffy/pg 0.16.2 → 0.16.3-alpha.2

package/index.cjs

@@ -206,24 +206,32 @@ const getJsonBuildValue = (value) => {
     return sql`${value}::text`;
   return sql`${JSON.stringify(stripAttributes(value))}::jsonb`;
 };
-const lookup = (prop) => {
+const lookup = (prop, options) => {
   const [prefix, ...suffix] = prop.split(".");
-  return suffix.length ? sql`"${raw(prefix)}" #> ${suffix}` : sql`"${raw(prefix)}"`;
+  if (!suffix.length)
+    return sql`"${raw(prefix)}"`;
+  const { types: types2 } = options.schema;
+  if (types2[prefix] === "jsonb") {
+    return sql`"${raw(prefix)}" #> ${suffix}`;
+  } else if (types2[prefix] === "cube" && suffix.length === 1) {
+    return sql`"${raw(prefix)}" ~> ${parseInt(suffix[0])}`;
+  } else {
+    throw Error(`pg.cannot_lookup ${prop}`);
+  }
 };
 const lookupNumeric = (prop) => {
   const [prefix, ...suffix] = prop.split(".");
-  return suffix.length ? sql`CASE WHEN "${raw(prefix)}" #> ${suffix} = 'null'::jsonb THEN 0 ELSE ("${raw(
-    prefix
-  )}" #> ${suffix})::numeric END` : sql`"${raw(prefix)}"`;
+  return suffix.length ? sql`CASE WHEN "${raw(prefix)}" #> ${suffix} = 'null'::jsonb
+      THEN 0 ELSE ("${raw(prefix)}" #> ${suffix})::numeric END` : sql`"${raw(prefix)}"`;
 };
 const aggSql = {
   $sum: (prop) => sql`sum((${lookupNumeric(prop)})::numeric)`,
-  $card: (prop) => sql`count(distinct(${lookup(prop)}))`,
+  $card: (prop, options) => sql`count(distinct(${lookup(prop, options)}))`,
   $avg: (prop) => sql`avg((${lookupNumeric(prop)})::numeric)`,
   $max: (prop) => sql`max((${lookupNumeric(prop)})::numeric)`,
   $min: (prop) => sql`min((${lookupNumeric(prop)})::numeric)`
 };
-const getSelectCols = (table, projection = null) => {
+const getSelectCols = (options, projection = null) => {
   if (!projection)
     return sql`*`;
   const sqls = [];
@@ -233,7 +241,7 @@ const getSelectCols = (table, projection = null) => {
     } else if (aggSql[key]) {
       const subSqls = [];
       for (const prop in projection[key]) {
-        subSqls.push(sql`${prop}::text, ${aggSql[key](prop)}`);
+        subSqls.push(sql`${prop}::text, ${aggSql[key](prop, options)}`);
       }
       sqls.push(
         sql`jsonb_build_object(${join(subSqls, ", ")}) AS "${raw(key)}"`
@@ -317,14 +325,17 @@ function stripAttributes(object) {
   if (Array.isArray(object)) {
     return object.map((item) => stripAttributes(item));
   }
-  return Object.entries(object).reduce((out, [key, val]) => {
-    if (key === "$put" || val === null)
+  return Object.entries(object).reduce(
+    (out, [key, val]) => {
+      if (key === "$put" || val === null)
+        return out;
+      if (out === null)
+        out = {};
+      out[key] = stripAttributes(val);
       return out;
-    if (out === null)
-      out = {};
-    out[key] = stripAttributes(val);
-    return out;
-  }, null);
+    },
+    null
+  );
 }
 const opSql = {
   $and: "AND",
@@ -432,10 +443,10 @@ function getArgSql({ $first, $last, $after, $before, $since, $until, $all, $curs
     where.push(getSql(filter, options));
   if (!hasRangeArg)
     return { meta: meta(baseKey), where, limit: 1 };
-  const groupCols = Array.isArray($group) && $group.length && $group.map(lookup);
+  const groupCols = Array.isArray($group) && $group.length && $group.map((prop) => lookup(prop, options));
   const group = groupCols ? join(groupCols, ", ") : void 0;
   const orderCols = ($order || [idCol]).map(
-    (orderItem) => orderItem[0] === "!" ? sql`-(${lookup(orderItem.slice(1))})::float8` : lookup(orderItem)
+    (orderItem) => orderItem[0] === "!" ? sql`-(${lookup(orderItem.slice(1), options)})::float8` : lookup(orderItem, options)
   );
   Object.entries({ $after, $before, $since, $until }).forEach(
     ([name, value]) => {
@@ -445,7 +456,7 @@ function getArgSql({ $first, $last, $after, $before, $since, $until, $all, $curs
   );
   const order = !$group && join(
     ($order || [idCol]).map(
-      (orderItem) => orderItem[0] === "!" ? sql`${lookup(orderItem.slice(1))} ${$last ? sql`ASC` : sql`DESC`}` : sql`${lookup(orderItem)} ${$last ? sql`DESC` : sql`ASC`}`
+      (orderItem) => orderItem[0] === "!" ? sql`${lookup(orderItem.slice(1), options)} ${$last ? sql`ASC` : sql`DESC`}` : sql`${lookup(orderItem, options)} ${$last ? sql`DESC` : sql`ASC`}`
     ),
     ", "
   );
@@ -497,7 +508,7 @@ function selectByArgs(args, projection, options) {
   const clampedLimit = Math.min(MAX_LIMIT, limit || MAX_LIMIT);
   return sql`
     SELECT
-    ${getSelectCols(table, projection)}, ${meta}
+    ${getSelectCols(options, projection)}, ${meta}
     FROM "${raw(table)}"
     ${where.length ? sql`WHERE ${join(where, " AND ")}` : empty}
     ${group ? sql`GROUP BY ${group}` : empty}
@@ -509,7 +520,7 @@ function selectByIds(ids, projection, options) {
   const { table, idCol } = options;
   return sql`
     SELECT
-    ${getSelectCols(table, projection)}, ${getIdMeta(options)}
+    ${getSelectCols(options, projection)}, ${getIdMeta(options)}
     FROM "${raw(table)}"
     WHERE "${raw(idCol)}" IN (${join(ids)})
   `;
@@ -542,7 +553,7 @@ function patch(object, arg, options) {
   return sql`
     UPDATE "${raw(table)}" SET ${getUpdates(row, options)}
     WHERE ${where}
-    RETURNING ${getSelectCols()}, ${meta}`;
+    RETURNING ${getSelectCols(options)}, ${meta}`;
 }
 function put(object, arg, options) {
   const { idCol, table } = options;
@@ -560,7 +571,7 @@ function put(object, arg, options) {
   return sql`
     INSERT INTO "${raw(table)}" (${cols}) VALUES (${vals})
     ON CONFLICT (${conflictTarget}) DO UPDATE SET (${cols}) = (${vals})
-    RETURNING ${getSelectCols()}, ${meta}`;
+    RETURNING ${getSelectCols(options)}, ${meta}`;
 }
 function del(arg, options) {
   const { table } = options;

package/index.mjs

@@ -204,24 +204,32 @@ const getJsonBuildValue = (value) => {
     return sql`${value}::text`;
   return sql`${JSON.stringify(stripAttributes(value))}::jsonb`;
 };
-const lookup = (prop) => {
+const lookup = (prop, options) => {
   const [prefix, ...suffix] = prop.split(".");
-  return suffix.length ? sql`"${raw(prefix)}" #> ${suffix}` : sql`"${raw(prefix)}"`;
+  if (!suffix.length)
+    return sql`"${raw(prefix)}"`;
+  const { types: types2 } = options.schema;
+  if (types2[prefix] === "jsonb") {
+    return sql`"${raw(prefix)}" #> ${suffix}`;
+  } else if (types2[prefix] === "cube" && suffix.length === 1) {
+    return sql`"${raw(prefix)}" ~> ${parseInt(suffix[0])}`;
+  } else {
+    throw Error(`pg.cannot_lookup ${prop}`);
+  }
 };
 const lookupNumeric = (prop) => {
   const [prefix, ...suffix] = prop.split(".");
-  return suffix.length ? sql`CASE WHEN "${raw(prefix)}" #> ${suffix} = 'null'::jsonb THEN 0 ELSE ("${raw(
-    prefix
-  )}" #> ${suffix})::numeric END` : sql`"${raw(prefix)}"`;
+  return suffix.length ? sql`CASE WHEN "${raw(prefix)}" #> ${suffix} = 'null'::jsonb
+      THEN 0 ELSE ("${raw(prefix)}" #> ${suffix})::numeric END` : sql`"${raw(prefix)}"`;
 };
 const aggSql = {
   $sum: (prop) => sql`sum((${lookupNumeric(prop)})::numeric)`,
-  $card: (prop) => sql`count(distinct(${lookup(prop)}))`,
+  $card: (prop, options) => sql`count(distinct(${lookup(prop, options)}))`,
   $avg: (prop) => sql`avg((${lookupNumeric(prop)})::numeric)`,
   $max: (prop) => sql`max((${lookupNumeric(prop)})::numeric)`,
   $min: (prop) => sql`min((${lookupNumeric(prop)})::numeric)`
 };
-const getSelectCols = (table, projection = null) => {
+const getSelectCols = (options, projection = null) => {
   if (!projection)
     return sql`*`;
   const sqls = [];
@@ -231,7 +239,7 @@ const getSelectCols = (table, projection = null) => {
     } else if (aggSql[key]) {
       const subSqls = [];
       for (const prop in projection[key]) {
-        subSqls.push(sql`${prop}::text, ${aggSql[key](prop)}`);
+        subSqls.push(sql`${prop}::text, ${aggSql[key](prop, options)}`);
       }
       sqls.push(
         sql`jsonb_build_object(${join(subSqls, ", ")}) AS "${raw(key)}"`
@@ -315,14 +323,17 @@ function stripAttributes(object) {
   if (Array.isArray(object)) {
     return object.map((item) => stripAttributes(item));
   }
-  return Object.entries(object).reduce((out, [key, val]) => {
-    if (key === "$put" || val === null)
+  return Object.entries(object).reduce(
+    (out, [key, val]) => {
+      if (key === "$put" || val === null)
+        return out;
+      if (out === null)
+        out = {};
+      out[key] = stripAttributes(val);
       return out;
-    if (out === null)
-      out = {};
-    out[key] = stripAttributes(val);
-    return out;
-  }, null);
+    },
+    null
+  );
 }
 const opSql = {
   $and: "AND",
@@ -430,10 +441,10 @@ function getArgSql({ $first, $last, $after, $before, $since, $until, $all, $curs
     where.push(getSql(filter, options));
   if (!hasRangeArg)
     return { meta: meta(baseKey), where, limit: 1 };
-  const groupCols = Array.isArray($group) && $group.length && $group.map(lookup);
+  const groupCols = Array.isArray($group) && $group.length && $group.map((prop) => lookup(prop, options));
   const group = groupCols ? join(groupCols, ", ") : void 0;
   const orderCols = ($order || [idCol]).map(
-    (orderItem) => orderItem[0] === "!" ? sql`-(${lookup(orderItem.slice(1))})::float8` : lookup(orderItem)
+    (orderItem) => orderItem[0] === "!" ? sql`-(${lookup(orderItem.slice(1), options)})::float8` : lookup(orderItem, options)
   );
   Object.entries({ $after, $before, $since, $until }).forEach(
     ([name, value]) => {
@@ -443,7 +454,7 @@ function getArgSql({ $first, $last, $after, $before, $since, $until, $all, $curs
   );
   const order = !$group && join(
     ($order || [idCol]).map(
-      (orderItem) => orderItem[0] === "!" ? sql`${lookup(orderItem.slice(1))} ${$last ? sql`ASC` : sql`DESC`}` : sql`${lookup(orderItem)} ${$last ? sql`DESC` : sql`ASC`}`
+      (orderItem) => orderItem[0] === "!" ? sql`${lookup(orderItem.slice(1), options)} ${$last ? sql`ASC` : sql`DESC`}` : sql`${lookup(orderItem, options)} ${$last ? sql`DESC` : sql`ASC`}`
     ),
     ", "
   );
@@ -495,7 +506,7 @@ function selectByArgs(args, projection, options) {
   const clampedLimit = Math.min(MAX_LIMIT, limit || MAX_LIMIT);
   return sql`
     SELECT
-    ${getSelectCols(table, projection)}, ${meta}
+    ${getSelectCols(options, projection)}, ${meta}
     FROM "${raw(table)}"
     ${where.length ? sql`WHERE ${join(where, " AND ")}` : empty}
     ${group ? sql`GROUP BY ${group}` : empty}
@@ -507,7 +518,7 @@ function selectByIds(ids, projection, options) {
   const { table, idCol } = options;
   return sql`
     SELECT
-    ${getSelectCols(table, projection)}, ${getIdMeta(options)}
+    ${getSelectCols(options, projection)}, ${getIdMeta(options)}
     FROM "${raw(table)}"
     WHERE "${raw(idCol)}" IN (${join(ids)})
   `;
@@ -540,7 +551,7 @@ function patch(object, arg, options) {
   return sql`
     UPDATE "${raw(table)}" SET ${getUpdates(row, options)}
     WHERE ${where}
-    RETURNING ${getSelectCols()}, ${meta}`;
+    RETURNING ${getSelectCols(options)}, ${meta}`;
 }
 function put(object, arg, options) {
   const { idCol, table } = options;
@@ -558,7 +569,7 @@ function put(object, arg, options) {
   return sql`
     INSERT INTO "${raw(table)}" (${cols}) VALUES (${vals})
     ON CONFLICT (${conflictTarget}) DO UPDATE SET (${cols}) = (${vals})
-    RETURNING ${getSelectCols()}, ${meta}`;
+    RETURNING ${getSelectCols(options)}, ${meta}`;
 }
 function del(arg, options) {
   const { table } = options;

package/package.json

@@ -2,7 +2,7 @@
   "name": "@graffy/pg",
   "description": "The standard Postgres module for Graffy. Each instance this module mounts a Postgres table as a Graffy subtree.",
   "author": "aravind (https://github.com/aravindet)",
-  "version": "0.16.2",
+  "version": "0.16.3-alpha.2",
   "main": "./index.cjs",
   "exports": {
     "import": "./index.mjs",
@@ -16,7 +16,7 @@
   },
   "license": "Apache-2.0",
   "dependencies": {
-    "@graffy/common": "0.16.2",
+    "@graffy/common": "0.16.3-alpha.2",
     "debug": "^4.3.3"
   },
   "peerDependencies": {

package/types/sql/clauses.d.ts

@@ -1,8 +1,8 @@
 export function cubeLiteralSql(value: any): Sql;
 export function getJsonBuildTrusted(variadic: any): Sql;
-export function lookup(prop: any): Sql;
+export function lookup(prop: any, options: any): Sql;
 export function lookupNumeric(prop: any): Sql;
-export function getSelectCols(table: any, projection?: any): Sql;
+export function getSelectCols(options: any, projection?: any): Sql;
 export function getInsert(row: any, options: any): {
     cols: Sql;
     vals: Sql;