patram 0.1.1 → 0.3.0

package/lib/layout-stored-queries.js

@@ -97,7 +97,7 @@ function createStoredQueryPhrases(where_clause) {
 }
 
 /**
- * @param {{ is_negated: boolean, term: { kind: 'field', field_name: 'id' | 'kind' | 'path' | 'status' | 'title', operator: '=' | '^=' | '~', value: string } | { kind: 'relation', relation_name: string } | { kind: 'relation_target', relation_name: string, target_id: string } }} clause
+ * @param {import('./parse-where-clause.types.ts').ParsedClause} clause
  * @param {boolean} should_prefix_and
  * @returns {StoredQuerySegment[]}
  */
@@ -121,10 +121,14 @@ function createClausePhrase(clause, should_prefix_and) {
 }
 
 /**
- * @param {{ kind: 'field', field_name: 'id' | 'kind' | 'path' | 'status' | 'title', operator: '=' | '^=' | '~', value: string } | { kind: 'relation', relation_name: string } | { kind: 'relation_target', relation_name: string, target_id: string }} term
+ * @param {import('./parse-where-clause.types.ts').ParsedTerm} term
  * @returns {StoredQuerySegment[]}
  */
 function createTermSegments(term) {
+  if (term.kind === 'aggregate') {
+    return createAggregateSegments(term);
+  }
+
   if (term.kind === 'field') {
     return [
       { kind: 'field_name', text: term.field_name },
@@ -133,6 +137,10 @@ function createTermSegments(term) {
     ];
   }
 
+  if (term.kind === 'field_set') {
+    return createFieldSetSegments(term);
+  }
+
   if (term.kind === 'relation_target') {
     return [
       { kind: 'field_name', text: term.relation_name },
@@ -147,6 +155,92 @@ function createTermSegments(term) {
   ];
 }
 
+/**
+ * @param {import('./parse-where-clause.types.ts').ParsedFieldSetTerm} term
+ * @returns {StoredQuerySegment[]}
+ */
+function createFieldSetSegments(term) {
+  return [
+    { kind: 'field_name', text: term.field_name },
+    { kind: 'plain', text: ' ' },
+    { kind: 'operator', text: term.operator },
+    { kind: 'plain', text: ' ' },
+    { kind: 'operator', text: '[' },
+    ...createListSegments(term.values),
+    { kind: 'operator', text: ']' },
+  ];
+}
+
+/**
+ * @param {import('./parse-where-clause.types.ts').ParsedAggregateTerm} term
+ * @returns {StoredQuerySegment[]}
+ */
+function createAggregateSegments(term) {
+  /** @type {StoredQuerySegment[]} */
+  const segments = [
+    { kind: 'field_name', text: term.aggregate_name },
+    { kind: 'operator', text: '(' },
+    ...createTraversalSegments(term.traversal),
+    { kind: 'operator', text: ', ' },
+    ...createNestedClauseSegments(term.clauses),
+    { kind: 'operator', text: ')' },
+  ];
+
+  if (term.aggregate_name === 'count') {
+    segments.push({ kind: 'plain', text: ' ' });
+    segments.push({ kind: 'operator', text: term.comparison ?? '=' });
+    segments.push({ kind: 'plain', text: ' ' });
+    segments.push({ kind: 'literal', text: String(term.value ?? 0) });
+  }
+
+  return segments;
+}
+
+/**
+ * @param {{ direction: 'in' | 'out', relation_name: string }} traversal
+ * @returns {StoredQuerySegment[]}
+ */
+function createTraversalSegments(traversal) {
+  return [
+    { kind: 'field_name', text: traversal.direction },
+    { kind: 'operator', text: ':' },
+    { kind: 'field_name', text: traversal.relation_name },
+  ];
+}
+
+/**
+ * @param {import('./parse-where-clause.types.ts').ParsedClause[]} clauses
+ * @returns {StoredQuerySegment[]}
+ */
+function createNestedClauseSegments(clauses) {
+  return clauses.flatMap((clause, clause_index) => {
+    const clause_phrase = createClausePhrase(clause, clause_index > 0);
+
+    if (clause_index === 0) {
+      return clause_phrase;
+    }
+
+    return [{ kind: 'plain', text: ' ' }, ...clause_phrase];
+  });
+}
+
+/**
+ * @param {string[]} values
+ * @returns {StoredQuerySegment[]}
+ */
+function createListSegments(values) {
+  return values.flatMap((value, value_index) => {
+    if (value_index === 0) {
+      return [{ kind: 'literal', text: value }];
+    }
+
+    return [
+      { kind: 'operator', text: ', ' },
+      { kind: 'literal', text: value },
+    ];
+  });
+}
+
 /**
  * @param {string} where_clause
  * @returns {StoredQuerySegment[][]}