import {
  IContainsTable,
  IPrimitiveValue,
  ISql,
  ISqlAdapter,
  sql,
} from "@trong-orm/sql";

import { IBaseToken, isToken, TokenType } from "../types";
import { isBinaryOperator } from "./binary";

export const wrapParentheses = <
  T extends IBaseToken | ISqlAdapter | IPrimitiveValue | IContainsTable
>(
  val: T
): T | ISql =>
  (isBinaryOperator(val) && val._operator === "OR") ||
  (isToken(val) &&
    (val.type === TokenType.Select || val.type === TokenType.Values))
    ? sql`(${val})`
    : val;