type ModelDescriptor = {
  modelId?: unknown;
  supportsVision?: boolean | null;
  capabilities?: unknown;
  defaultParams?: unknown;
};

type ProviderDescriptor = {
  type?: unknown;
};

const BFL_REFERENCE_IMAGE_MODEL_REGEX =
  /^(?:bfl\/)?(?:flux-2-(?:max|flex|pro(?:-preview)?|klein-(?:4b|9b)(?:-preview)?)|flux-kontext-(?:pro|max))$/i;

function readStringField(source: unknown, key: string) {
  if (!source || typeof source !== "object" || !(key in source)) return "";
  return String((source as Record<string, unknown>)[key] || "").trim();
}

function normalizeBflModelId(value: unknown) {
  const trimmed = String(value || "").trim();
  if (!trimmed) return "";

  return trimmed
    .replace(/^https?:\/\/[^/]+/i, "")
    .replace(/^\/?(?:bfl\/)?v1\//i, "")
    .replace(/^\/+/, "")
    .replace(/[?#].*$/, "");
}

export function isKnownReferenceImageModel(
  model: ModelDescriptor | null | undefined,
  provider?: ProviderDescriptor | null
) {
  const providerType = String(provider?.type || "").trim();
  if (providerType !== "bfl-image") return false;

  const candidates = [
    model?.modelId,
    readStringField(model?.defaultParams, "endpoint"),
    readStringField(model?.defaultParams, "submitEndpoint"),
    readStringField(model?.capabilities, "endpoint"),
  ];

  const capabilitiesEndpoints =
    model?.capabilities &&
    typeof model.capabilities === "object" &&
    "endpoints" in model.capabilities
      ? (model.capabilities as { endpoints?: unknown }).endpoints
      : undefined;

  candidates.push(readStringField(capabilitiesEndpoints, "submit"));

  return candidates
    .map(normalizeBflModelId)
    .some((modelId) => BFL_REFERENCE_IMAGE_MODEL_REGEX.test(modelId));
}

export function modelLacksReferenceImageInputSupport(
  model: ModelDescriptor | null | undefined,
  provider?: ProviderDescriptor | null
) {
  if (isKnownReferenceImageModel(model, provider)) return false;
  return model?.supportsVision === false;
}

export function modelSupportsReferenceImageInput(
  model: ModelDescriptor | null | undefined,
  provider?: ProviderDescriptor | null
) {
  return !modelLacksReferenceImageInputSupport(model, provider);
}