Genarrative/apps/admin-web/src/pages/adminEditorGenerationPricing.ts

import type {
  AdminUpsertEditorGenerationPricingRequest,
  EditorGenerationModel3dEndpointPricingPayload,
  EditorGenerationModel3dPricingPayload,
  EditorGenerationPricingConfigPayload,
} from '../api/adminApiTypes';

/** 3D 端点在两个并列段里的字段名与后台展示名。 */
export const EDITOR_MODEL3D_ENDPOINT_SECTIONS = [
  { key: 'textToModelPricing', label: '文生 3D' },
  { key: 'imageToModelPricing', label: '图生 3D' },
] as const satisfies ReadonlyArray<{
  key: keyof EditorGenerationModel3dPricingPayload;
  label: string;
}>;

/**
 * 3D 加价项键到中文标签。键来自接口返回，这里只负责展示；遇到未登记的键回退显示原键名，
 * 不因为标签缺失就把该项漏掉。
 */
const EDITOR_MODEL3D_ADD_ON_LABELS: Record<string, string> = {
  hdTexture: '高清贴图',
  ultraTexture: '超高清贴图',
  hdGeometry: '高清几何',
  quadMesh: '四边形网格',
  smartLowPoly: '智能低面',
  generateParts: '分件生成',
};

export function editorModel3dAddOnLabel(addOn: string) {
  return EDITOR_MODEL3D_ADD_ON_LABELS[addOn] ?? addOn;
}

export function isPositiveMudPoints(value: number | null | undefined) {
  return typeof value === 'number' && Number.isInteger(value) && value >= 1;
}

/**
 * 提交前校验：所有泥点数值都必须是不小于 1 的整数，返回不合法项的展示名。
 *
 * 后端仍会独立校验一遍；这里只是不把明显非法的 payload 发出去，避免保存后才报错。
 */
export function collectInvalidEditorGenerationPricingLabels(
  pricing: EditorGenerationPricingConfigPayload,
) {
  const labels: string[] = [];
  for (const [model, modelPricing] of Object.entries(pricing.models)) {
    const tieredPrices = modelPricing.prices;
    // `prices: {}` 没有档位可校验，不能算「这一项走分层计价」——否则底价会被顺手跳过校验。
    if (tieredPrices && Object.keys(tieredPrices).length > 0) {
      for (const [tier, price] of Object.entries(tieredPrices)) {
        if (!isPositiveMudPoints(price)) {
          labels.push(`${model} ${tier}`);
        }
      }
      continue;
    }
    if (!isPositiveMudPoints(modelPricing.price)) {
      labels.push(`${model} 泥点`);
    }
  }

  const model3d = pricing.model3d;
  if (model3d) {
    for (const { key, label } of EDITOR_MODEL3D_ENDPOINT_SECTIONS) {
      // 段与子表都来自接口返回：缺段或缺子表时只跳过这一段，不能让整页校验崩掉。
      const endpointPricing = model3d[key];
      if (!endpointPricing) {
        continue;
      }
      for (const [version, price] of Object.entries(
        endpointPricing.versionPrices ?? {},
      )) {
        if (!isPositiveMudPoints(price.noTexture)) {
          labels.push(`${label} ${version} 无贴图`);
        }
        if (!isPositiveMudPoints(price.texture)) {
          labels.push(`${label} ${version} 带贴图`);
        }
      }
      for (const [addOn, price] of Object.entries(
        endpointPricing.addOnPrices ?? {},
      )) {
        if (!isPositiveMudPoints(price)) {
          labels.push(`${label} ${editorModel3dAddOnLabel(addOn)}`);
        }
      }
    }
  }
  return labels;
}

/**
 * 组装保存请求：`models` 与 3D 两段整段回传，不带定价版本（保存即覆盖，服务端不比对版本）。
 * 3D 段缺失时回传 `null`——服务端会以 400 拒绝，后台不允许在这里静默新建或丢弃整段。
 */
export function buildAdminEditorGenerationPricingRequest(
  pricing: EditorGenerationPricingConfigPayload,
): AdminUpsertEditorGenerationPricingRequest {
  return {
    models: pricing.models,
    model3d: pricing.model3d ?? null,
  };
}

/** 单个端点段的不可变更新：底价与加价项各改一处，其余原样保留。 */
export function updateEditorModel3dEndpointPricing(
  model3d: EditorGenerationModel3dPricingPayload,
  section: keyof EditorGenerationModel3dPricingPayload,
  updater: (
    current: EditorGenerationModel3dEndpointPricingPayload,
  ) => EditorGenerationModel3dEndpointPricingPayload,
): EditorGenerationModel3dPricingPayload {
  return { ...model3d, [section]: updater(model3d[section]) };
}