#!/usr/bin/env bash
# Regression test for scripts/skill-size-check.sh: enforces agentskills.io's
# 500-line/5,000-word(proxy-for-token) SKILL.md size ceiling.
set -euo pipefail

REPO_ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
SCRIPT="$REPO_ROOT/scripts/skill-size-check.sh"
PASS=0
FAIL=0

pass() { echo "  PASS: $1"; PASS=$((PASS + 1)); }
fail() { echo "  FAIL: $1"; FAIL=$((FAIL + 1)); }

TMPDIR="$(mktemp -d)"
trap 'rm -rf "$TMPDIR"' EXIT

make_fixture() {
  local name="$1" lines="$2" words_per_line="$3" file
  file="$TMPDIR/$name.md"
  {
    echo "---"
    echo "name: $name"
    echo "description: Test fixture."
    echo "---"
    for ((i = 1; i <= lines; i++)); do
      w=""
      for ((j = 1; j <= words_per_line; j++)); do
        w="$w word"
      done
      echo "$w"
    done
  } > "$file"
  echo "$file"
}

echo ""
echo "--- passes a file under both limits ---"
SMALL="$(make_fixture small 10 5)"
if "$SCRIPT" "$SMALL"; then
  pass "file under both limits exits 0"
else
  fail "file under both limits should have exited 0"
fi

echo ""
echo "--- fails a file over the line limit ---"
MANY_LINES="$(make_fixture many-lines 600 1)"
if "$SCRIPT" "$MANY_LINES" 2>/dev/null; then
  fail "file over the 500-line ceiling should have exited non-zero"
else
  pass "file over the 500-line ceiling exits non-zero"
fi

echo ""
echo "--- fails a file over the word-count limit ---"
MANY_WORDS="$(make_fixture many-words 10 600)"
if "$SCRIPT" "$MANY_WORDS" 2>/dev/null; then
  fail "file over the 5,000-word ceiling should have exited non-zero"
else
  pass "file over the 5,000-word ceiling exits non-zero"
fi

# Boundary-pair tests below read the script's current MAX_WORDS rather than
# hardcoding it, so they don't silently drift if the threshold changes again.
MAX_WORDS="$(grep -oE '^MAX_WORDS=[0-9]+' "$SCRIPT" | cut -d= -f2)"
MAX_LINES="$(grep -oE '^MAX_LINES=[0-9]+' "$SCRIPT" | cut -d= -f2)"

# make_line_fixture builds a file with an exact total line count (frontmatter
# included), independent of word count, for the line-boundary tests.
make_line_fixture() {
  local name="$1" total_lines="$2" file body_lines
  file="$TMPDIR/$name.md"
  {
    echo "---"
    echo "name: $name"
    echo "description: Test fixture."
    echo "---"
  } > "$file"
  body_lines=$((total_lines - 4))
  for ((i = 1; i <= body_lines; i++)); do
    echo "word"
  done >> "$file"
  echo "$file"
}

# The line ceiling is exclusive of the limit itself ("stay under $MAX_LINES
# lines", per skill-authoring.md), enforced via `>=` — so $((MAX_LINES - 1))
# must pass and $MAX_LINES itself must already fail.
echo ""
echo "--- passes a file at $((MAX_LINES - 1)) lines, just under the $MAX_LINES-line boundary ---"
AT_LINES="$(make_line_fixture at-line-limit "$((MAX_LINES - 1))")"
ACTUAL_LINES=$(awk 'END{print NR}' "$AT_LINES")
if [[ "$ACTUAL_LINES" -ne "$((MAX_LINES - 1))" ]]; then
  fail "fixture has $ACTUAL_LINES lines, expected exactly $((MAX_LINES - 1))"
elif "$SCRIPT" "$AT_LINES"; then
  pass "file at $((MAX_LINES - 1)) lines exits 0"
else
  fail "file at $((MAX_LINES - 1)) lines should have exited 0"
fi

echo ""
echo "--- fails a file at exactly the $MAX_LINES-line boundary ---"
OVER_LINES="$(make_line_fixture over-line-limit "$MAX_LINES")"
ACTUAL_OVER_LINES=$(awk 'END{print NR}' "$OVER_LINES")
if [[ "$ACTUAL_OVER_LINES" -ne "$MAX_LINES" ]]; then
  fail "fixture has $ACTUAL_OVER_LINES lines, expected exactly $MAX_LINES"
elif "$SCRIPT" "$OVER_LINES" 2>/dev/null; then
  fail "file at exactly $MAX_LINES lines should have exited non-zero (>= ceiling, the boundary bug this test guards against)"
else
  pass "file at exactly $MAX_LINES lines exits non-zero"
fi

# make_word_fixture builds a file with an exact total word count (frontmatter
# words included, since the script's `wc -w` counts the whole file) by padding
# a body line with just enough "word" tokens to close the gap to the target.
make_word_fixture() {
  local name="$1" target="$2" file cur remaining body
  file="$TMPDIR/$name.md"
  {
    echo "---"
    echo "name: $name"
    echo "description: Test fixture."
    echo "---"
  } > "$file"
  cur=$(wc -w < "$file")
  remaining=$((target - cur))
  body=""
  for ((i = 1; i <= remaining; i++)); do
    body="$body word"
  done
  echo "$body" >> "$file"
  echo "$file"
}

echo ""
echo "--- passes a file at exactly the $MAX_WORDS-word boundary ---"
AT_WORDS="$(make_word_fixture at-word-limit "$MAX_WORDS")"
ACTUAL_WORDS=$(wc -w < "$AT_WORDS")
if [[ "$ACTUAL_WORDS" -ne "$MAX_WORDS" ]]; then
  fail "fixture has $ACTUAL_WORDS words, expected exactly $MAX_WORDS"
elif "$SCRIPT" "$AT_WORDS"; then
  pass "file at exactly $MAX_WORDS words exits 0"
else
  fail "file at exactly $MAX_WORDS words should have exited 0"
fi

echo ""
echo "--- fails a file one word over the $MAX_WORDS-word boundary ---"
OVER_WORDS="$(make_word_fixture over-word-limit "$((MAX_WORDS + 1))")"
ACTUAL_OVER_WORDS=$(wc -w < "$OVER_WORDS")
if [[ "$ACTUAL_OVER_WORDS" -ne "$((MAX_WORDS + 1))" ]]; then
  fail "fixture has $ACTUAL_OVER_WORDS words, expected exactly $((MAX_WORDS + 1))"
elif "$SCRIPT" "$OVER_WORDS" 2>/dev/null; then
  fail "file at $((MAX_WORDS + 1)) words should have exited non-zero"
else
  pass "file at $((MAX_WORDS + 1)) words exits non-zero"
fi

echo ""
echo "Results: $PASS passed, $FAIL failed"
[[ $FAIL -eq 0 ]]