#!/usr/bin/env bash
# tkcheck -- find TK/TODO/XXX/FIXME/TBD markers across a project
set -euo pipefail

target="${1:-.}"
pattern='\bTK\b|TODO|XXX|FIXME|\bTBD\b|\bNOTE\b|\bHACK\b'

echo "Scanning for markers in: $target"
echo ""

if [ -f "$target" ]; then
    grep -n -E --color=always "$pattern" "$target" || echo "No markers found."
else
    results=$(grep -rn -E --color=always "$pattern" "$target" \
        --include='*.md' --include='*.mdx' --include='*.txt' \
        --include='*.tex' --include='*.rst' 2>/dev/null) || true
    if [ -n "$results" ]; then
        echo "$results"
        echo ""
        count=$(echo "$results" | wc -l)
        echo "$count marker(s) found."
    else
        echo "No markers found."
    fi
fi