import { TracingSearchType } from "../../../interfaces/search";

const regexIndefiniteCharacters = "%";

export const clickhouseSearchCondition = (
  query?: string,
  searchType?: TracingSearchType[],
  tablePrefix?: string,
  useTracesAmtCompatMode: boolean = false,
) => {
  const prefix = tablePrefix ? `${tablePrefix}.` : "";

  // We use a hard-coded prefix for user_id as it only occurs in the trace context.
  const conditions = [
    !searchType || searchType.includes("id")
      ? `${prefix}id ILIKE {searchString: String} OR t.user_id ILIKE {searchString: String} OR ${prefix}name ILIKE {searchString: String}`
      : null,
    searchType && searchType.includes("content") && !useTracesAmtCompatMode
      ? `${prefix}input ILIKE {searchString: String} OR ${prefix}output ILIKE {searchString: String}`
      : searchType && searchType.includes("content") && useTracesAmtCompatMode
        ? `finalizeAggregation(${prefix}input) ILIKE {searchString: String} OR finalizeAggregation(${prefix}output) ILIKE {searchString: String}`
        : null,
  ].filter(Boolean);

  return {
    query: query ? `AND (${conditions.join(" OR ")})` : "",
    params: query
      ? {
          searchString: `${regexIndefiniteCharacters}${query}${regexIndefiniteCharacters}`,
        }
      : {},
  };
};