proto.io 0.0.158 → 0.0.159

package/dist/adapters/storage/progres.mjs

@@ -7,7 +7,7 @@ import Decimal from 'decimal.js';
 import { escapeLiteral, escapeIdentifier } from 'pg/lib/utils';
 import { a as QueryCoditionalSelector, b as QueryFieldSelector, c as QueryExpressionSelector, d as QueryDistanceExpression, e as QueryCoditionalExpression, f as QueryComparisonExpression, g as QueryNotExpression, h as QueryArrayExpression, i as QueryValueExpression, j as QueryKeyExpression, Q as QuerySelector, F as FieldSelectorExpression } from '../../internals/index-tU-lsQqj.mjs';
 import '@o2ter/crypto-js';
-import { _ as _resolveColumn, g as generateId, Q as QueryValidator } from '../../internals/random-DLFngtjg.mjs';
+import { a as resolveColumn, r as resolveDataType, g as generateId, Q as QueryValidator } from '../../internals/random-Dst9-WVo.mjs';
 import { P as PVK } from '../../internals/private-BUpLAMZi.mjs';
 
 //
@@ -240,7 +240,7 @@ class QueryCompiler {
         const names = {};
         const populates = {};
         for (const include of includes) {
-            const { paths: [colname, ...subpath], dataType } = _resolveColumn(this.schema, className, include);
+            const { paths: [colname, ...subpath], dataType } = resolveColumn(this.schema, className, include);
             names[colname] = dataType;
             if (isPointer(dataType) || isRelation(dataType)) {
                 if (_.isEmpty(subpath))
@@ -257,13 +257,12 @@ class QueryCompiler {
                     colname,
                 };
                 if (isRelation(dataType) && dataType.foreignField) {
-                    const targetType = this.schema[dataType.target].fields[dataType.foreignField];
+                    const targetType = resolveDataType(this.schema, dataType.target, dataType.foreignField);
+                    if (_.isNil(targetType))
+                        throw Error(`Invalid path: ${include}`);
                     if (!isPointer(targetType) && !isRelation(targetType))
                         throw Error(`Invalid path: ${include}`);
-                    populates[colname].foreignField = {
-                        colname: dataType.foreignField,
-                        type: targetType.type,
-                    };
+                    populates[colname].foreignField = dataType.foreignField;
                 }
                 populates[colname].subpaths.push(subpath.join('.'));
             }
@@ -358,7 +357,7 @@ class QueryCompiler {
     _encodeUpdateAttrs(className, attrs) {
         const updates = [];
         for (const [path, op] of _.toPairs(attrs)) {
-            const { paths: [column, ...subpath], dataType } = _resolveColumn(this.schema, className, path);
+            const { paths: [column, ...subpath], dataType } = resolveColumn(this.schema, className, path);
             if (isShape(dataType)) {
                 const [_op, value] = decodeUpdateOp(op);
                 if (_op !== '$set')
@@ -378,7 +377,7 @@ class QueryCompiler {
     _encodeObjectAttrs(className, attrs) {
         const result = {};
         for (const [key, value] of _.toPairs(attrs)) {
-            const { paths: [column, ...subpath], dataType } = _resolveColumn(this.schema, className, key);
+            const { paths: [column, ...subpath], dataType } = resolveColumn(this.schema, className, key);
             if (!_.isEmpty(subpath))
                 throw Error(`Invalid insert key: ${key}`);
             if (isShape(dataType)) {
@@ -784,6 +783,34 @@ class SqlStorage {
 //  OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 //  THE SOFTWARE.
 //
+const foreignFieldType = (schema, className, path) => {
+    let fields = schema[className].fields;
+    let last;
+    let result = false;
+    for (const key of _.toPath(path)) {
+        const dataType = fields[key];
+        if (_.isNil(dataType))
+            return;
+        if (isPrimitive(dataType) || isVector(dataType))
+            return;
+        if (isShape(dataType)) {
+            fields = dataType.shape;
+            continue;
+        }
+        if (_.isNil(schema[dataType.target]))
+            return;
+        if (dataType.type === 'relation')
+            result = true;
+        fields = schema[dataType.target].fields;
+        last = dataType;
+    }
+    if (!last)
+        return;
+    return {
+        target: last.target,
+        type: result ? 'relation' : last.type,
+    };
+};
 const _fetchElement = (parent, colname, subpath, dataType) => {
     const element = sql `${{ identifier: parent.name }}.${{ identifier: parent.name.startsWith('_expr_$') ? '$' : colname }}`;
     if (!parent.className) {
@@ -824,7 +851,7 @@ const _fetchElement = (parent, colname, subpath, dataType) => {
     return { element, json: false };
 };
 const resolvePaths = (compiler, className, paths) => {
-    const { paths: [colname, ...subpath], dataType } = _resolveColumn(compiler.schema, className, paths.join('.'));
+    const { paths: [colname, ...subpath], dataType } = resolveColumn(compiler.schema, className, paths.join('.'));
     if (!_.isEmpty(subpath) && isVector(dataType)) {
         if (subpath.length !== 1)
             throw Error(`Invalid key: ${paths.join('.')}`);
@@ -850,14 +877,38 @@ const resolvePaths = (compiler, className, paths) => {
 const fetchElement = (compiler, parent, field) => {
     if (parent.className) {
         const { dataType, colname, subpath } = resolvePaths(compiler, parent.className, _.toPath(field));
+        const { element, json } = _fetchElement(parent, colname, subpath, dataType);
+        if (!_.isEmpty(subpath)) {
+            const foreignField = foreignFieldType(compiler.schema, parent.className, field);
+            if (foreignField) {
+                return {
+                    element,
+                    dataType: foreignField,
+                    relation: {
+                        target: foreignField.target,
+                        sql: (callback) => sql `SELECT 
+              ${callback(sql `UNNEST`)} 
+            FROM UNNEST(${{ identifier: parent.name }}.${{ identifier: colname }})`,
+                    },
+                };
+            }
+        }
         if (isPointer(dataType))
             return { element: sql `${{ identifier: parent.name }}.${{ identifier: `${colname}._id` }}`, dataType };
-        const { element, json } = _fetchElement(parent, colname, subpath, dataType);
-        return { element, json, dataType: json ? null : dataType, relation: isRelation(dataType) ? dataType : null };
+        return {
+            element,
+            dataType: json ? null : dataType,
+            relation: isRelation(dataType) ? {
+                target: dataType.target,
+                sql: (callback) => sql `SELECT
+          ${callback(sql `${json ? sql `VALUE` : sql `UNNEST`}`)}
+        FROM ${json ? sql `jsonb_array_elements(${element})` : sql `UNNEST(${element})`}`,
+            } : null,
+        };
     }
     const [colname, ...subpath] = _.toPath(field);
-    const { element, json } = _fetchElement(parent, colname, subpath);
-    return { element, json, dataType: null, relation: null };
+    const { element } = _fetchElement(parent, colname, subpath);
+    return { element, dataType: null, relation: null };
 };
 
 //
@@ -1803,7 +1854,7 @@ const encodeQueryExpression = (compiler, parent, expr) => {
 //
 const encodeFieldExpression = (compiler, context, parent, field, expr) => {
     const [colname] = _.toPath(field);
-    const { element, json, dataType, relation } = fetchElement(compiler, parent, field);
+    const { element, dataType, relation } = fetchElement(compiler, parent, field);
     const encodeValue = (value) => dataType ? encodeType(colname, dataType, value) : _encodeJsonValue(_encodeValue(value));
     switch (expr.type) {
         case '$eq':
@@ -1963,10 +2014,10 @@ const encodeFieldExpression = (compiler, context, parent, field, expr) => {
                 if (dataType === 'array' || (!_.isString(dataType) && dataType?.type === 'array')) {
                     return sql `${element} <@ ${{ value: _encodeValue(expr.value) }}`;
                 }
-                else if (!_.isString(dataType) && dataType?.type === 'relation') {
+                else if (relation) {
                     if (!_.every(expr.value, x => x instanceof TObject && x.objectId))
                         break;
-                    return sql `ARRAY(SELECT (UNNEST ->> '_id') FROM UNNEST(${element})) <@ ARRAY[${_.map(expr.value, (x) => sql `${{ value: x.objectId }}`)}]`;
+                    return sql `ARRAY(${relation.sql((v) => sql `${v} ->> '_id'`)}) <@ ARRAY[${_.map(expr.value, (x) => sql `${{ value: x.objectId }}`)}]`;
                 }
                 else if (!dataType) {
                     return sql `jsonb_typeof(${element}) ${nullSafeEqual()} 'array' AND ${element} <@ ${_encodeJsonValue(_encodeValue(expr.value))}`;
@@ -1981,10 +2032,10 @@ const encodeFieldExpression = (compiler, context, parent, field, expr) => {
                 if (dataType === 'array' || (!_.isString(dataType) && dataType?.type === 'array')) {
                     return sql `${element} @> ${{ value: _encodeValue(expr.value) }}`;
                 }
-                else if (!_.isString(dataType) && dataType?.type === 'relation') {
+                else if (relation) {
                     if (!_.every(expr.value, x => x instanceof TObject && x.objectId))
                         break;
-                    return sql `ARRAY(SELECT (UNNEST ->> '_id') FROM UNNEST(${element})) @> ARRAY[${_.map(expr.value, (x) => sql `${{ value: x.objectId }}`)}]`;
+                    return sql `ARRAY(${relation.sql((v) => sql `${v} ->> '_id'`)}) @> ARRAY[${_.map(expr.value, (x) => sql `${{ value: x.objectId }}`)}]`;
                 }
                 else if (!dataType) {
                     return sql `jsonb_typeof(${element}) ${nullSafeEqual()} 'array' AND ${element} @> ${_encodeJsonValue(_encodeValue(expr.value))}`;
@@ -1999,10 +2050,10 @@ const encodeFieldExpression = (compiler, context, parent, field, expr) => {
                 if (dataType === 'array' || (!_.isString(dataType) && dataType?.type === 'array')) {
                     return sql `NOT ${element} && ${{ value: _encodeValue(expr.value) }}`;
                 }
-                else if (!_.isString(dataType) && dataType?.type === 'relation') {
+                else if (relation) {
                     if (!_.every(expr.value, x => x instanceof TObject && x.objectId))
                         break;
-                    return sql `NOT ARRAY(SELECT (UNNEST ->> '_id') FROM UNNEST(${element})) && ARRAY[${_.map(expr.value, (x) => sql `${{ value: x.objectId }}`)}]`;
+                    return sql `NOT ARRAY(${relation.sql((v) => sql `${v} ->> '_id'`)}) && ARRAY[${_.map(expr.value, (x) => sql `${{ value: x.objectId }}`)}]`;
                 }
                 else if (!dataType) {
                     return sql `jsonb_typeof(${element}) ${nullSafeEqual()} 'array' AND NOT ${element} && ${_encodeJsonValue(_encodeValue(expr.value))}`;
@@ -2017,10 +2068,10 @@ const encodeFieldExpression = (compiler, context, parent, field, expr) => {
                 if (dataType === 'array' || (!_.isString(dataType) && dataType?.type === 'array')) {
                     return sql `${element} && ${{ value: _encodeValue(expr.value) }}`;
                 }
-                else if (!_.isString(dataType) && dataType?.type === 'relation') {
+                else if (relation) {
                     if (!_.every(expr.value, x => x instanceof TObject && x.objectId))
                         break;
-                    return sql `ARRAY(SELECT (UNNEST ->> '_id') FROM UNNEST(${element})) && ARRAY[${_.map(expr.value, (x) => sql `${{ value: x.objectId }}`)}]`;
+                    return sql `ARRAY(${relation.sql((v) => sql `${v} ->> '_id'`)}) && ARRAY[${_.map(expr.value, (x) => sql `${{ value: x.objectId }}`)}]`;
                 }
                 else if (!dataType) {
                     return sql `jsonb_typeof(${element}) ${nullSafeEqual()} 'array' AND ${element} && ${_encodeJsonValue(_encodeValue(expr.value))}`;
@@ -2127,11 +2178,7 @@ const encodeFieldExpression = (compiler, context, parent, field, expr) => {
                     break;
                 if (relation) {
                     return sql `NOT EXISTS(
-            SELECT * FROM (
-              SELECT
-                ${json ? sql `VALUE` : sql `UNNEST`} AS "$"
-              FROM ${json ? sql `jsonb_array_elements(${element})` : sql `UNNEST(${element})`}
-            ) AS ${{ identifier: tempName }}
+            SELECT * FROM (${relation.sql((v) => sql `${v} AS "$"`)}) AS ${{ identifier: tempName }}
             WHERE NOT (${filter})
           )`;
                 }
@@ -2158,11 +2205,7 @@ const encodeFieldExpression = (compiler, context, parent, field, expr) => {
                     break;
                 if (relation) {
                     return sql `EXISTS(
-            SELECT * FROM (
-              SELECT
-                ${json ? sql `VALUE` : sql `UNNEST`} AS "$"
-              FROM ${json ? sql `jsonb_array_elements(${element})` : sql `UNNEST(${element})`}
-            ) AS ${{ identifier: tempName }}
+            SELECT * FROM (${relation.sql((v) => sql `${v} AS "$"`)}) AS ${{ identifier: tempName }}
             WHERE ${filter}
           )`;
                 }
@@ -2225,6 +2268,28 @@ const resolveSubpaths = (compiler, populate) => {
     }
     return subpaths;
 };
+const _isPointer = (schema, className, path) => {
+    let fields = schema[className].fields;
+    let last;
+    for (const key of _.toPath(path)) {
+        const dataType = fields[key];
+        if (_.isNil(dataType))
+            throw Error(`Invalid path: ${path}`);
+        if (isPrimitive(dataType) || isVector(dataType))
+            throw Error(`Invalid path: ${path}`);
+        if (isShape(dataType)) {
+            fields = dataType.shape;
+            continue;
+        }
+        if (dataType.type !== 'pointer')
+            return false;
+        if (_.isNil(schema[dataType.target]))
+            throw Error(`Invalid path: ${path}`);
+        fields = schema[dataType.target].fields;
+        last = dataType;
+    }
+    return last?.type === 'pointer';
+};
 const selectPopulate = (compiler, parent, populate, field) => {
     const _local = (field) => sql `${{ identifier: parent.name }}.${{ identifier: field }}`;
     const _foreign = (field) => sql `${{ identifier: populate.name }}.${{ identifier: field }}`;
@@ -2242,11 +2307,11 @@ const selectPopulate = (compiler, parent, populate, field) => {
     if (_.isNil(populate.foreignField)) {
         cond = sql `${sql `(${{ quote: populate.className + '$' }} || ${_foreign('_id')})`} = ANY(${_local(field)})`;
     }
-    else if (populate.foreignField.type === 'pointer') {
-        cond = sql `${sql `(${{ quote: parent.className + '$' }} || ${_local('_id')})`} = ${_foreign(populate.foreignField.colname)}`;
+    else if (_isPointer(compiler.schema, populate.className, populate.foreignField)) {
+        cond = sql `${sql `(${{ quote: parent.className + '$' }} || ${_local('_id')})`} = ${_foreign(populate.colname)}`;
     }
     else {
-        cond = sql `${sql `(${{ quote: parent.className + '$' }} || ${_local('_id')})`} = ANY(${_foreign(populate.foreignField.colname)})`;
+        cond = sql `${sql `(${{ quote: parent.className + '$' }} || ${_local('_id')})`} = ANY(${_foreign(populate.colname)})`;
     }
     return {
         columns: [sql `
@@ -2263,11 +2328,80 @@ const selectPopulate = (compiler, parent, populate, field) => {
     `],
     };
 };
+const encodeRemix = (parent, remix) => sql `${remix?.className === parent.className ? sql `
+  (SELECT * FROM ${{ identifier: remix.name }} UNION SELECT * FROM ${{ identifier: parent.className }})
+` : { identifier: parent.className }}`;
+const encodeForeignField = (compiler, context, parent, foreignField, remix) => {
+    const { paths: [colname, ...subpath], dataType } = resolveColumn(compiler.schema, parent.className, foreignField);
+    const tempName = `_populate_$${compiler.nextIdx()}`;
+    const _local = (field) => sql `${{ identifier: parent.name }}.${{ identifier: field }}`;
+    const _foreign = (field) => sql `${{ identifier: tempName }}.${{ identifier: field }}`;
+    if (_.isEmpty(subpath) && isRelation(dataType) && dataType.foreignField) {
+        const { joins, field, rows, array } = encodeForeignField(compiler, context, { className: dataType.target, name: tempName }, dataType.foreignField, remix);
+        return {
+            joins: [],
+            field: sql `(
+        SELECT ${sql `(${{ quote: parent.className + '$' }} || ${_foreign('_id')})`}
+        FROM ${encodeRemix(parent, remix)} AS ${{ identifier: tempName }}
+        ${!_.isEmpty(joins) ? { literal: joins, separator: '\n' } : sql ``}
+        WHERE ${sql `(${{ quote: parent.className + '$' }} || ${_local('_id')})`} = ${array || rows ? sql `ANY(${field})` : field}
+      )`,
+            array: false,
+            rows: true,
+        };
+    }
+    if (_.isEmpty(subpath)) {
+        return {
+            joins: [],
+            field: sql `${{ identifier: parent.name }}.${{ identifier: foreignField }}`,
+            array: isRelation(dataType),
+            rows: false,
+        };
+    }
+    if (!isPointer(dataType) && !isRelation(dataType))
+        throw Error(`Invalid path: ${foreignField}`);
+    const { joins, field, rows, array } = encodeForeignField(compiler, context, { className: dataType.target, name: tempName }, subpath.join('.'), remix);
+    if (isPointer(dataType)) {
+        return {
+            joins: [sql `
+        LEFT JOIN ${encodeRemix(parent, remix)} AS ${{ identifier: tempName }}
+        ON ${sql `(${{ quote: dataType.target + '$' }} || ${_foreign('_id')})`} = ${_local(colname)}
+      `, ...joins],
+            field,
+            array,
+            rows,
+        };
+    }
+    let cond;
+    if (_.isNil(dataType.foreignField)) {
+        cond = sql `${sql `(${{ quote: dataType.target + '$' }} || ${_foreign('_id')})`} = ANY(${_local(colname)})`;
+    }
+    else if (_isPointer(compiler.schema, dataType.target, dataType.foreignField)) {
+        cond = sql `${sql `(${{ quote: parent.className + '$' }} || ${_local('_id')})`} = ${_foreign(dataType.foreignField)}`;
+    }
+    else {
+        cond = sql `${sql `(${{ quote: parent.className + '$' }} || ${_local('_id')})`} = ANY(${_foreign(dataType.foreignField)})`;
+    }
+    return {
+        joins: [],
+        field: sql `(
+      SELECT ${array ? sql `UNNEST(${field})` : field} FROM ${encodeRemix(parent, remix)} AS ${{ identifier: tempName }}
+      ${!_.isEmpty(joins) ? { literal: joins, separator: '\n' } : sql ``}
+      WHERE ${cond}
+    )`,
+        array: false,
+        rows: true,
+    };
+};
 const encodePopulate = (compiler, context, parent, remix) => {
     const _filter = compiler._encodeFilter(context, parent, parent.filter);
     const _populates = _.map(parent.populates, (populate, field) => selectPopulate(compiler, parent, populate, field));
     const _joins = _.compact(_.map(_populates, ({ join }) => join));
     const _includes = _.pickBy(parent.includes, v => isPrimitive(v));
+    const { joins: _joins2 = [], field: _foreignField = undefined, rows = false, } = parent.foreignField ? encodeForeignField(compiler, context, {
+        className: parent.className,
+        name: parent.name,
+    }, parent.foreignField, remix) : {};
     return _.reduce(parent.populates, (acc, populate) => ({
         ...encodePopulate(compiler, context, populate, remix),
         ...acc,
@@ -2278,14 +2412,12 @@ const encodePopulate = (compiler, context, parent, remix) => {
         ${{
             literal: [
                 ..._.map(_.keys(_includes), colname => sql `${{ identifier: parent.name }}.${{ identifier: colname }}`),
-                ...parent.foreignField ? [sql `${{ identifier: parent.name }}.${{ identifier: parent.foreignField.colname }}`] : [],
                 ..._.flatMap(_populates, ({ columns: column }) => column),
+                ..._foreignField ? [sql `${rows ? sql `ARRAY(${_foreignField})` : _foreignField} AS ${{ identifier: parent.colname }}`] : [],
             ], separator: ',\n'
         }}
-        FROM ${remix?.className === parent.className ? sql `
-        (SELECT * FROM ${{ identifier: remix.name }} UNION SELECT * FROM ${{ identifier: parent.className }})
-        ` : { identifier: parent.className }} AS ${{ identifier: parent.name }}
-        ${!_.isEmpty(_joins) ? { literal: _joins, separator: '\n' } : sql ``}
+        FROM ${encodeRemix(parent, remix)} AS ${{ identifier: parent.name }}
+        ${!_.isEmpty(_joins) || !_.isEmpty(_joins2) ? { literal: [..._joins, ..._joins2], separator: '\n' } : sql ``}
       ) AS ${{ identifier: parent.name }}
       ${_filter ? sql `WHERE ${_filter}` : sql ``}
     `,
@@ -2320,13 +2452,13 @@ const encodeRelation = (compiler, context, parent, relatedBy) => {
     const name = `_relation_$${relatedBy.className.toLowerCase()}`;
     const _local = (field) => sql `${{ identifier: parent.name }}.${{ identifier: field }}`;
     const _foreign = (field) => sql `${{ identifier: name }}.${{ identifier: field }}`;
-    return sql `
-  EXISTS (
+    const { joins, field } = encodeForeignField(compiler, context, { className: relatedBy.className, name }, relatedBy.key);
+    return sql `EXISTS (
     SELECT 1
     FROM ${{ identifier: relatedBy.className }} AS ${{ identifier: name }}
-    WHERE ${_foreign('_id')} = ${{ value: relatedBy.objectId }} AND ${sql `(${{ quote: parent.className + '$' }} || ${_local('_id')})`} = ANY(${_foreign(relatedBy.key)})
-  )
-  `;
+    ${!_.isEmpty(joins) ? { literal: joins, separator: '\n' } : sql ``}
+    WHERE ${_foreign('_id')} = ${{ value: relatedBy.objectId }} AND ${sql `(${{ quote: parent.className + '$' }} || ${_local('_id')})`} = ANY(${field})
+  )`;
 };
 
 //