import type { AnyObject, UpdateParams } from "prostgles-types";
import type { ParsedTableRule } from "../../PublishParser/publishTypesAndUtils";
import type { Filter, LocalParams } from "../DboBuilderTypes";
import { getErrorAsObject, getSerializedClientErrorFromPGError } from "../dboBuilderUtils";
import type { TableHandler } from "./TableHandler";

export const upsert = async function (
  this: TableHandler,
  filter: Filter,
  newData: AnyObject,
  params?: UpdateParams,
  table_rules?: ParsedTableRule,
  localParams?: LocalParams,
): Promise<any> {
  const start = Date.now();
  try {
    const _upsert = async function (tblH: TableHandler) {
      return tblH
        .find(filter, { select: "", limit: 1 }, undefined, table_rules, localParams)
        .then((exists) => {
          if (exists.length) {
            return tblH.update(filter, newData, params, table_rules, localParams);
          } else {
            return tblH.insert(
              { ...newData, ...filter },
              params,
              undefined,
              table_rules,
              localParams,
            );
          }
        });
    };

    /* Do it within a transaction to ensure consisency */
    if (!this.tx) {
      return this.dboBuilder.getTX((dbTX) => _upsert(dbTX[this.name] as TableHandler));
    }
    const result = await _upsert(this);
    await this._log({
      command: "upsert",
      localParams,
      data: { filter, newData, params },
      duration: Date.now() - start,
    });
    return result;
  } catch (e) {
    await this._log({
      command: "upsert",
      localParams,
      data: { filter, newData, params },
      duration: Date.now() - start,
      error: getErrorAsObject(e),
    });
    throw getSerializedClientErrorFromPGError(e, {
      type: "tableMethod",
      localParams,
      view: this,
      prostgles: this.dboBuilder.prostgles,
    });
  }
};