#!/bin/sh
set -eu

ROOT="$(CDPATH= cd -- "$(dirname -- "$0")/../.." && pwd)"
cd "$ROOT"

python3 scripts/gen_skill_shortcut_sections.py --check

chat_skill="skills/multi/dingtalk-chat/SKILL.md"
doc_skill="skills/multi/dingtalk-doc/SKILL.md"
event_skill="skills/multi/dingtalk-event/SKILL.md"
mono_skill="skills/mono/SKILL.md"
runtime_contract="skills/multi/dingtalk-shared/references/runtime-contract.md"
chat_target_bytes=10000
chat_max_overage_percent=10
chat_max_bytes=$((chat_target_bytes * (100 + chat_max_overage_percent) / 100))
doc_max_bytes=10000
event_max_bytes=10000
runtime_contract_max_bytes=3000

chat_bytes="$(wc -c < "$chat_skill" | tr -d ' ')"
if [ "$chat_bytes" -gt "$chat_max_bytes" ]; then
	printf '%s\n' \
		"skill context budget exceeded: $chat_skill is ${chat_bytes} bytes (target ${chat_target_bytes}, max ${chat_max_bytes} with ${chat_max_overage_percent}% allowance)" >&2
	exit 1
fi

doc_bytes="$(wc -c < "$doc_skill" | tr -d ' ')"
if [ "$doc_bytes" -gt "$doc_max_bytes" ]; then
	printf '%s\n' \
		"skill context budget exceeded: $doc_skill is ${doc_bytes} bytes (max ${doc_max_bytes})" >&2
	exit 1
fi

event_bytes="$(wc -c < "$event_skill" | tr -d ' ')"
if [ "$event_bytes" -gt "$event_max_bytes" ]; then
	printf '%s\n' \
		"skill context budget exceeded: $event_skill is ${event_bytes} bytes (max ${event_max_bytes})" >&2
	exit 1
fi

runtime_contract_bytes="$(wc -c < "$runtime_contract" | tr -d ' ')"
if [ "$runtime_contract_bytes" -gt "$runtime_contract_max_bytes" ]; then
	printf '%s\n' \
		"skill context budget exceeded: $runtime_contract is ${runtime_contract_bytes} bytes (max ${runtime_contract_max_bytes})" >&2
	exit 1
fi

shortcut_rows="$(
	awk '
		/<!-- VISIBLE_SHORTCUTS_START -->/ { in_block = 1; next }
		/<!-- VISIBLE_SHORTCUTS_END -->/ { in_block = 0 }
		in_block && /^\| `dws chat \+/ { count++ }
		END { print count + 0 }
	' "$chat_skill"
)"
if [ "$shortcut_rows" -ne 0 ]; then
	printf '%s\n' \
		"skill context budget exceeded: $chat_skill re-expanded $shortcut_rows shortcut rows" >&2
	exit 1
fi

doc_shortcut_rows="$(
	awk '
		/<!-- VISIBLE_SHORTCUTS_START -->/ { in_block = 1; next }
		/<!-- VISIBLE_SHORTCUTS_END -->/ { in_block = 0 }
		in_block && /^\| `dws doc \+/ { count++ }
		END { print count + 0 }
	' "$doc_skill"
)"
if [ "$doc_shortcut_rows" -ne 0 ]; then
	printf '%s\n' \
		"skill context budget exceeded: $doc_skill re-expanded $doc_shortcut_rows shortcut rows" >&2
	exit 1
fi

for required_heading in \
	"## 最小 DWS 执行契约" \
	"## Golden Route" \
	"## 关键结果语义" \
	"## 按需加载" \
	"## 错误最短路径"
do
	if ! grep -Fq "$required_heading" "$chat_skill"; then
		printf '%s\n' \
			"skill value regression: $chat_skill is missing $required_heading" >&2
		exit 1
	fi
done

for required_heading in \
	"## 最小 DWS 执行契约" \
	"## Golden Route" \
	"## 关键结果语义" \
	"## 按需加载" \
	"## 错误最短路径"
do
	if ! grep -Fq "$required_heading" "$doc_skill"; then
		printf '%s\n' \
			"skill value regression: $doc_skill is missing $required_heading" >&2
		exit 1
	fi
done

for required_route in \
	"+search" \
	"+fetch" \
	"+create" \
	"+update" \
	"+checkpoint-update" \
	"+export"
do
	if ! grep -Fq -- "$required_route" "$doc_skill"; then
		printf '%s\n' \
			"skill route regression: $doc_skill is missing $required_route" >&2
		exit 1
	fi
done

for forbidden_route in \
	"## 标准 SOP" \
	"doc_create_and_write.py" \
	"只在任务命中时读取一个精确 reference" \
	"MUST 先用 Read" \
	"先用 Read 工具完整读取"
do
	if grep -Fq "$forbidden_route" "$doc_skill"; then
		printf '%s\n' \
			"skill route regression: $doc_skill restored legacy route: $forbidden_route" >&2
		exit 1
	fi
done

for required_reference_rule in \
	"Golden Route 已给出命令且参数足够时，禁止读取 reference" \
	"才最多读取一个 reference" \
	"\`+create\`、\`+fetch\`、\`+update\` append/overwrite、\`+export\`、\`+import\` 禁止读取 reference"
do
	if ! grep -Fq "$required_reference_rule" "$doc_skill"; then
		printf '%s\n' \
			"skill reference budget regression: $doc_skill is missing $required_reference_rule" >&2
		exit 1
	fi
done

if grep -Fq "../dingtalk-shared/SKILL.md" "$doc_skill"; then
	printf '%s\n' \
		"skill context regression: $doc_skill requires full dingtalk-shared cold-start loading" >&2
	exit 1
fi

for required_route in \
	"+dm" \
	"+send-to-group" \
	"+messages-send" \
	"+chat-messages" \
	"+search-msg" \
	"--download-resources" \
	"+conversation-list-top"
do
	if ! grep -Fq -- "$required_route" "$chat_skill"; then
		printf '%s\n' \
			"skill route regression: $chat_skill is missing $required_route" >&2
		exit 1
	fi
done

for forbidden_route in \
	"## 标准 SOP" \
	"dws aisearch person --keyword" \
	"dws aisearch person --query" \
	"dws chat message send-by-webhook" \
	"dt_media_upload" \
	"MUST 先用 Read"
do
	if grep -Fq "$forbidden_route" "$chat_skill"; then
		printf '%s\n' \
			"skill route regression: $chat_skill restored legacy route: $forbidden_route" >&2
		exit 1
	fi
done

if grep -Fq "../dingtalk-shared/SKILL.md" "$chat_skill"; then
	printf '%s\n' \
		"skill context regression: $chat_skill requires full dingtalk-shared cold-start loading" >&2
	exit 1
fi

if grep -Fq "充分阅读产品参考文件" "$mono_skill"; then
	printf '%s\n' \
		"skill context budget regression: $mono_skill requires full product-reference loading" >&2
	exit 1
fi

printf '%s\n' \
	"skill context budget: ok (chat_bytes=$chat_bytes target=$chat_target_bytes max=$chat_max_bytes allowance=${chat_max_overage_percent}% doc_bytes=$doc_bytes doc_max=$doc_max_bytes event_bytes=$event_bytes event_max=$event_max_bytes runtime_contract_bytes=$runtime_contract_bytes runtime_contract_max=$runtime_contract_max_bytes shortcut_rows=$shortcut_rows doc_shortcut_rows=$doc_shortcut_rows)"