#!/usr/bin/env bash

set -euo pipefail

ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
cd "$ROOT_DIR"

check_contains() {
    local file="$1"
    local needle="$2"
    grep -Fq "$needle" "$file" || {
        echo "missing in ${file}: ${needle}"
        exit 1
    }
}

check_contains "scripts/run_intel_pipeline.sh" 'PIPELINE_SOURCE_SET="openrouter,moonshot,deepseek,openai,zhipu,baidu,bytedance,tencent_subscription,aliyun_subscription'
check_contains "scripts/run_real_pipeline.sh" 'PIPELINE_SOURCE_SET="openrouter,moonshot,deepseek,openai,zhipu,baidu,bytedance,tencent_subscription,aliyun_subscription'
check_contains "scripts/run_daily.sh" 'PIPELINE_SOURCE_SET="openrouter,moonshot,deepseek,openai,zhipu,baidu,bytedance,tencent_subscription,aliyun_subscription'

check_contains "scripts/run_intel_pipeline.sh" 'run_or_fail "tencent_subscription" "腾讯云套餐导入失败"'
check_contains "scripts/run_real_pipeline.sh" 'merge_failed_source_keys "tencent_subscription"'
check_contains "scripts/run_real_pipeline.sh" 'record_failure "腾讯云套餐导入失败"'
check_contains "scripts/run_daily.sh" 'merge_failed_source_keys "tencent_subscription"'
check_contains "scripts/run_daily.sh" 'error_exit "腾讯云套餐导入失败"'
check_contains "scripts/run_intel_pipeline.sh" 'run_or_fail "qwen_pricing" "通义千问价格导入失败"'
check_contains "scripts/run_intel_pipeline.sh" 'run_or_fail "hunyuan_pricing" "腾讯混元价格导入失败"'
check_contains "scripts/run_intel_pipeline.sh" 'run_or_fail "mobile_cloud_pricing" "移动云 MoMA 价格导入失败"'
check_contains "scripts/run_intel_pipeline.sh" 'run_or_fail "youdao_pricing" "网易有道价格导入失败"'
check_contains "scripts/run_intel_pipeline.sh" 'run_or_fail "platform360_pricing" "360 智脑价格导入失败"'
check_contains "scripts/run_intel_pipeline.sh" 'run_or_fail "cucloud_pricing" "联通云 Token Plan 价格导入失败"'
check_contains "scripts/run_intel_pipeline.sh" 'run_or_fail "huawei_maas_pricing" "华为云 MaaS 价格导入失败"'
check_contains "scripts/run_intel_pipeline.sh" 'run_or_fail "baichuan_pricing" "百川价格导入失败"'
check_contains "scripts/run_intel_pipeline.sh" 'run_or_fail "lingyiwanwu_pricing" "零一万物价格导入失败"'
check_contains "scripts/run_intel_pipeline.sh" 'run_or_fail "sensenova_pricing" "商汤 SenseNova 价格导入失败"'
check_contains "scripts/run_intel_pipeline.sh" 'run_or_fail "xfyun_pricing" "讯飞价格导入失败"'
check_contains "scripts/run_intel_pipeline.sh" 'run_or_fail "bytedance_pricing" "火山方舟价格导入失败"'
check_contains "scripts/run_real_pipeline.sh" 'merge_failed_source_keys "qwen_pricing"'
check_contains "scripts/run_real_pipeline.sh" 'merge_failed_source_keys "hunyuan_pricing"'
check_contains "scripts/run_real_pipeline.sh" 'merge_failed_source_keys "mobile_cloud_pricing"'
check_contains "scripts/run_real_pipeline.sh" 'merge_failed_source_keys "youdao_pricing"'
check_contains "scripts/run_real_pipeline.sh" 'merge_failed_source_keys "platform360_pricing"'
check_contains "scripts/run_real_pipeline.sh" 'merge_failed_source_keys "cucloud_pricing"'
check_contains "scripts/run_real_pipeline.sh" 'merge_failed_source_keys "huawei_maas_pricing"'
check_contains "scripts/run_real_pipeline.sh" 'merge_failed_source_keys "baichuan_pricing"'
check_contains "scripts/run_real_pipeline.sh" 'merge_failed_source_keys "lingyiwanwu_pricing"'
check_contains "scripts/run_real_pipeline.sh" 'merge_failed_source_keys "sensenova_pricing"'
check_contains "scripts/run_real_pipeline.sh" 'merge_failed_source_keys "xfyun_pricing"'
check_contains "scripts/run_real_pipeline.sh" 'merge_failed_source_keys "bytedance_pricing"'
check_contains "scripts/run_daily.sh" 'merge_failed_source_keys "qwen_pricing"'
check_contains "scripts/run_daily.sh" 'merge_failed_source_keys "hunyuan_pricing"'
check_contains "scripts/run_daily.sh" 'merge_failed_source_keys "mobile_cloud_pricing"'
check_contains "scripts/run_daily.sh" 'merge_failed_source_keys "youdao_pricing"'
check_contains "scripts/run_daily.sh" 'merge_failed_source_keys "platform360_pricing"'
check_contains "scripts/run_daily.sh" 'merge_failed_source_keys "cucloud_pricing"'
check_contains "scripts/run_daily.sh" 'merge_failed_source_keys "huawei_maas_pricing"'
check_contains "scripts/run_daily.sh" 'merge_failed_source_keys "baichuan_pricing"'
check_contains "scripts/run_daily.sh" 'merge_failed_source_keys "lingyiwanwu_pricing"'
check_contains "scripts/run_daily.sh" 'merge_failed_source_keys "sensenova_pricing"'
check_contains "scripts/run_daily.sh" 'merge_failed_source_keys "xfyun_pricing"'
check_contains "scripts/run_daily.sh" 'merge_failed_source_keys "bytedance_pricing"'


check_contains "scripts/verify_importer_smoke.sh" 'run_smoke "tencent-live"'
check_contains "scripts/verify_importer_smoke.sh" 'run_smoke "qwen-fixture"'
check_contains "scripts/verify_importer_smoke.sh" 'run_smoke "hunyuan-fixture"'
check_contains "scripts/verify_importer_smoke.sh" 'run_smoke "mobile-cloud-fixture"'
check_contains "scripts/verify_importer_smoke.sh" 'run_smoke "youdao-fixture"'
check_contains "scripts/verify_importer_smoke.sh" 'run_smoke "platform360-fixture"'
check_contains "scripts/verify_importer_smoke.sh" 'run_smoke "cucloud-pricing-fixture"'
check_contains "scripts/verify_importer_smoke.sh" 'run_smoke "huawei-maas-fixture"'
check_contains "scripts/verify_importer_smoke.sh" 'run_smoke "baichuan-fixture"'
check_contains "scripts/verify_importer_smoke.sh" 'run_smoke "lingyiwanwu-fixture"'
check_contains "scripts/verify_importer_smoke.sh" 'run_smoke "sensenova-fixture"'
check_contains "scripts/verify_importer_smoke.sh" 'run_smoke "xfyun-fixture"'
check_contains "scripts/verify_importer_smoke.sh" 'run_smoke "bytedance-fixture"'

echo "pipeline_runtime_alignment_test: PASS"