dbgate/packages/sqltree/src/dumpSqlExpression.ts

import _ from 'lodash';
import type { SqlDumper } from 'dbgate-types';
import { Expression, ColumnRefExpression } from './types';
import { dumpSqlSourceRef } from './dumpSqlSource';
import { dumpSqlSelect } from './dumpSqlCommand';

export function dumpSqlExpression(dmp: SqlDumper, expr: Expression) {
  switch (expr.exprType) {
    case 'column':
      {
        if (expr.source) {
          if (dumpSqlSourceRef(dmp, expr.source)) {
            dmp.put('.');
          }
        }
        dmp.put('%i', expr.columnName);
      }
      break;

    case 'placeholder':
      dmp.putRaw('{PLACEHOLDER}');
      break;

    case 'value':
      if (expr.dataType) {
        dmp.put('%V', {
          value: expr.value,
          dataType: expr.dataType,
        });
      } else {
        dmp.put('%v', expr.value);
      }
      break;

    case 'raw':
      dmp.put('%s', expr.sql);
      break;

    case 'unaryRaw':
      if (expr.beforeSql) dmp.putRaw(expr.beforeSql);
      dumpSqlExpression(dmp, expr.expr);
      if (expr.afterSql) dmp.putRaw(expr.afterSql);
      break;

    case 'call':
      dmp.put('%s(', expr.func);
      if (expr.argsPrefix) dmp.put('%s ', expr.argsPrefix);
      dmp.putCollection(',', expr.args, x => dumpSqlExpression(dmp, x));
      dmp.put(')');
      break;

    case 'methodCall':
      dumpSqlExpression(dmp, expr.thisObject);
      dmp.put('.%s(', expr.method);
      dmp.putCollection(',', expr.args, x => dumpSqlExpression(dmp, x));
      dmp.put(')');
      break;

    case 'transform':
      dmp.transform(expr.transform, () => dumpSqlExpression(dmp, expr.expr));
      break;

    case 'rowNumber':
      dmp.put(' ^row_number() ^over (^order ^by ');
      dmp.putCollection(', ', expr.orderBy, x => {
        dumpSqlExpression(dmp, x);
        dmp.put(' %k', x.direction);
      });
      dmp.put(')');
      break;

    case 'select':
      dmp.put('(');
      dumpSqlSelect(dmp, expr.select);
      dmp.put(')');
      break;
  }
}