From 10a213d8a35482e72df9b9d73d0f37affa5ca9f1 Mon Sep 17 00:00:00 2001
From: Jan Prochazka <jan.prochazka@gmail.com>
Date: Tue, 29 Dec 2020 12:18:07 +0100
Subject: [PATCH] referenco conditions

---
 packages/sqltree/src/utility.ts               | 29 +++++++-
 .../web/src/designer/generateDesignedQuery.ts | 69 ++++++++++++++-----
 2 files changed, 80 insertions(+), 18 deletions(-)

diff --git a/packages/sqltree/src/utility.ts b/packages/sqltree/src/utility.ts
index ba0bc89a..39301a9e 100644
--- a/packages/sqltree/src/utility.ts
+++ b/packages/sqltree/src/utility.ts
@@ -1,5 +1,5 @@
 import { EngineDriver, SqlDumper } from 'dbgate-types';
-import { Command } from './types';
+import { Command, Condition } from './types';
 import { dumpSqlCommand } from './dumpSqlCommand';
 
 export function treeToSql<T>(driver: EngineDriver, object: T, func: (dmp: SqlDumper, obj: T) => void) {
@@ -16,3 +16,30 @@ export function scriptToSql(driver: EngineDriver, script: Command[]): string {
   }
   return dmp.s;
 }
+
+export function mergeConditions(condition1: Condition, condition2: Condition): Condition {
+  if (!condition1) return condition2;
+  if (!condition2) return condition1;
+  if (condition1.conditionType == 'and' && condition2.conditionType == 'and') {
+    return {
+      conditionType: 'and',
+      conditions: [...condition1.conditions, ...condition2.conditions],
+    };
+  }
+  if (condition1.conditionType == 'and') {
+    return {
+      conditionType: 'and',
+      conditions: [...condition1.conditions, condition2],
+    };
+  }
+  if (condition2.conditionType == 'and') {
+    return {
+      conditionType: 'and',
+      conditions: [condition1, ...condition2.conditions],
+    };
+  }
+  return {
+    conditionType: 'and',
+    conditions: [condition1, condition2],
+  };
+}
diff --git a/packages/web/src/designer/generateDesignedQuery.ts b/packages/web/src/designer/generateDesignedQuery.ts
index 5dc4e7f4..d2cdb2ca 100644
--- a/packages/web/src/designer/generateDesignedQuery.ts
+++ b/packages/web/src/designer/generateDesignedQuery.ts
@@ -1,5 +1,5 @@
 import _ from 'lodash';
-import { dumpSqlSelect, Select, JoinType, Condition, Relation } from 'dbgate-sqltree';
+import { dumpSqlSelect, Select, JoinType, Condition, Relation, mergeConditions } from 'dbgate-sqltree';
 import { EngineDriver } from 'dbgate-types';
 import { DesignerInfo, DesignerTableInfo, DesignerReferenceInfo, DesignerJoinType } from './types';
 
@@ -20,6 +20,12 @@ class DesignerComponent {
   get nonPrimaryTablesAndReferences() {
     return _.zip(this.nonPrimaryTables, this.nonPrimaryReferences);
   }
+  get myAndParentTables() {
+    return [...this.parentTables, ...this.tables];
+  }
+  get parentTables() {
+    return this.parentComponent ? this.parentComponent.myAndParentTables : [];
+  }
 }
 
 function referenceIsConnecting(
@@ -40,6 +46,9 @@ function referenceIsJoin(reference) {
 function referenceIsExists(reference) {
   return ['WHERE EXISTS', 'WHERE NOT EXISTS'].includes(reference.joinType);
 }
+function referenceIsCrossJoin(reference) {
+  return !reference.joinType || reference.joinType == 'CROSS JOIN';
+}
 
 function findConnectingReference(
   designer: DesignerInfo,
@@ -113,21 +122,6 @@ class DesignerComponentCreator {
   }
 }
 
-function mergeConditions(condition1: Condition, condition2: Condition): Condition {
-  if (!condition1) return condition2;
-  if (!condition2) return condition1;
-  if (condition1.conditionType == 'and' && condition2.conditionType == 'and') {
-    return {
-      conditionType: 'and',
-      conditions: [...condition1.conditions, ...condition2.conditions],
-    };
-  }
-  return {
-    conditionType: 'and',
-    conditions: [condition1, condition2],
-  };
-}
-
 function mergeSelects(select1: Select, select2: Select): Select {
   return {
     commandType: 'select',
@@ -147,9 +141,12 @@ function mergeSelects(select1: Select, select2: Select): Select {
   };
 }
 class DesignerQueryDumper {
-  // dumpedTables: DesignerTableInfo[];
   constructor(public designer: DesignerInfo, public components: DesignerComponent[]) {}
 
+  get topLevelTables(): DesignerTableInfo[] {
+    return _.flatten(this.components.map((x) => x.tables));
+  }
+
   dumpComponent(component: DesignerComponent) {
     const select: Select = {
       commandType: 'select',
@@ -168,6 +165,33 @@ class DesignerQueryDumper {
         conditions: getReferenceConditions(ref, this.designer),
       });
     }
+
+    for (const subComponent of component.subComponents) {
+      const subQuery = this.dumpComponent(subComponent);
+      subQuery.selectAll = true;
+      select.where = mergeConditions(select.where, {
+        conditionType: subComponent.parentReference.joinType == 'WHERE NOT EXISTS' ? 'notExists' : 'exists',
+        subQuery,
+      });
+    }
+
+    if (component.parentReference) {
+      select.where = mergeConditions(select.where, {
+        conditionType: 'and',
+        conditions: getReferenceConditions(component.parentReference, this.designer),
+      });
+
+      // cross join conditions in subcomponents
+      for (const ref of this.designer.references) {
+        if (referenceIsCrossJoin(ref) && referenceIsConnecting(ref, component.tables, component.myAndParentTables)) {
+          select.where = mergeConditions(select.where, {
+            conditionType: 'and',
+            conditions: getReferenceConditions(ref, this.designer),
+          });
+        }
+      }
+    }
+
     return select;
   }
 
@@ -178,6 +202,17 @@ class DesignerQueryDumper {
       if (res == null) res = select;
       else res = mergeSelects(res, select);
     }
+
+    // top level cross join conditions
+    const topLevelTables = this.topLevelTables;
+    for (const ref of this.designer.references) {
+      if (referenceIsCrossJoin(ref) && referenceIsConnecting(ref, topLevelTables, topLevelTables)) {
+        res.where = mergeConditions(res.where, {
+          conditionType: 'and',
+          conditions: getReferenceConditions(ref, this.designer),
+        });
+      }
+    }
     return res;
   }
 }