fix(schema-compiler): Fix BigQuery DATE_ADD push down template for years/quarters/months

Author: KSDaemon

packages/cubejs-schema-compiler/src/adapter/BigqueryQuery.ts

@@ -186,11 +186,21 @@ export class BigqueryQuery extends BaseQuery {
   }
 
   public subtractTimestampInterval(date, interval) {
-    return `TIMESTAMP_SUB(${date}, INTERVAL ${this.formatInterval(interval)[0]})`;
+    const [intervalFormatted, timeUnit] = this.formatInterval(interval);
+    if (['YEAR', 'MONTH', 'QUARTER'].includes(timeUnit)) {
+      return this.timeStampCast(`DATETIME_SUB(DATETIME(${date}), INTERVAL ${intervalFormatted})`);
+    }
+
+    return `TIMESTAMP_SUB(${date}, INTERVAL ${intervalFormatted})`;
   }
 
   public addTimestampInterval(date, interval) {
-    return `TIMESTAMP_ADD(${date}, INTERVAL ${this.formatInterval(interval)[0]})`;
+    const [intervalFormatted, timeUnit] = this.formatInterval(interval);
+    if (['YEAR', 'MONTH', 'QUARTER'].includes(timeUnit)) {
+      return this.timeStampCast(`DATETIME_ADD(DATETIME(${date}), INTERVAL ${intervalFormatted})`);
+    }
+
+    return `TIMESTAMP_ADD(${date}, INTERVAL ${intervalFormatted})`;
   }
 
   public nowTimestampSql() {
@@ -242,7 +252,7 @@ export class BigqueryQuery extends BaseQuery {
     templates.functions.STRPOS = 'STRPOS({{ args_concat }})';
     templates.functions.DATEDIFF = 'DATETIME_DIFF(CAST({{ args[2] }} AS DATETIME), CAST({{ args[1] }} AS DATETIME), {{ date_part }})';
     // DATEADD is being rewritten to DATE_ADD
-    // templates.functions.DATEADD = 'DATETIME_ADD(CAST({{ args[2] }} AS DATETTIME), INTERVAL {{ interval }} {{ date_part }})';
+    templates.functions.DATE_ADD = '{% if date_part|upper in [\'YEAR\', \'MONTH\', \'QUARTER\'] %}TIMESTAMP(DATETIME_ADD(DATETIME({{ args[2] }}), INTERVAL {{ interval }} {{ date_part }})){% else %}TIMESTAMP_ADD({{ args[2] }}, INTERVAL {{ interval }} {{ date_part }}){% endif %}';
     templates.functions.CURRENTDATE = 'CURRENT_DATE';
     delete templates.functions.TO_CHAR;
     templates.expressions.binary = '{% if op == \'%\' %}MOD({{ left }}, {{ right }}){% else %}({{ left }} {{ op }} {{ right }}){% endif %}';