barrett/cp.nvim
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

initial commit

Browse source
This commit is contained in:
Barrett Ruth 2025-09-11 23:52:32 -05:00
commit dcb7debff6
29 changed files with 1276 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

85
templates/scripts/scrape.sh Executable file
View file

@ -0,0 +1,85 @@
#!/bin/sh
CONTEST="$1"
PROBLEM="$2"
PROBLEM_LETTER="$3"
if [ -z "$CONTEST" ] || [ -z "$PROBLEM" ]; then
echo "Usage: make scrape <contest> <problem_id> [problem_letter]"
echo "Available contests: cses, atcoder, codeforces"
echo "Examples:"
echo " make scrape cses 1068"
echo " make scrape atcoder abc042 a"
echo " make scrape codeforces 1234 A"
exit
fi
test -d io && true || mkdir -p io
TMPFILE=$(mktemp)
ORIGDIR=$(pwd)
case "$CONTEST" in
cses)
cd "$(dirname "$0")/../.." && uv run scrapers/cses.py "$PROBLEM" > "$TMPFILE"
if [ $? -eq 0 ]; then
cd "$ORIGDIR"
awk '/^---INPUT---$/ {getline; while ($0 != "---OUTPUT---") {print; getline}} END {}' "$TMPFILE" > "io/$PROBLEM.in"
awk '/^---OUTPUT---$/ {getline; while ($0 != "---END---") {print; getline}} END {}' "$TMPFILE" > "io/$PROBLEM.expected"
echo "Scraped problem $PROBLEM to io/$PROBLEM.in and io/$PROBLEM.expected"
else
echo "Failed to scrape problem $PROBLEM"
cat "$TMPFILE"
rm "$TMPFILE"
exit
fi
;;
atcoder)
if [ -z "$PROBLEM_LETTER" ]; then
echo "AtCoder requires problem letter (e.g., make scrape atcoder abc042 a)"
rm "$TMPFILE"
exit
fi
FULL_PROBLEM_ID="${PROBLEM}${PROBLEM_LETTER}"
cd "$(dirname "$0")/../.." && uv run scrapers/atcoder.py "$PROBLEM" "$PROBLEM_LETTER" > "$TMPFILE"
if [ $? -eq 0 ]; then
cd "$ORIGDIR"
awk '/^---INPUT---$/ {getline; while ($0 != "---OUTPUT---") {print; getline}} END {}' "$TMPFILE" > "io/$FULL_PROBLEM_ID.in"
awk '/^---OUTPUT---$/ {getline; while ($0 != "---END---") {print; getline}} END {}' "$TMPFILE" > "io/$FULL_PROBLEM_ID.expected"
echo "Scraped problem $FULL_PROBLEM_ID to io/$FULL_PROBLEM_ID.in and io/$FULL_PROBLEM_ID.expected"
else
echo "Failed to scrape problem $FULL_PROBLEM_ID"
cat "$TMPFILE"
rm "$TMPFILE"
exit
fi
;;
codeforces)
if [ -z "$PROBLEM_LETTER" ]; then
echo "Codeforces requires problem letter (e.g., make scrape codeforces 1234 A)"
rm "$TMPFILE"
exit
fi
FULL_PROBLEM_ID="${PROBLEM}${PROBLEM_LETTER}"
cd "$(dirname "$0")/../.." && uv run scrapers/codeforces.py "$PROBLEM" "$PROBLEM_LETTER" > "$TMPFILE"
if [ $? -eq 0 ]; then
cd "$ORIGDIR"
awk '/^---INPUT---$/ {getline; while ($0 != "---OUTPUT---") {print; getline}} END {}' "$TMPFILE" > "io/$FULL_PROBLEM_ID.in"
awk '/^---OUTPUT---$/ {getline; while ($0 != "---END---") {print; getline}} END {}' "$TMPFILE" > "io/$FULL_PROBLEM_ID.expected"
echo "Scraped problem $FULL_PROBLEM_ID to io/$FULL_PROBLEM_ID.in and io/$FULL_PROBLEM_ID.expected"
else
echo "Failed to scrape problem $FULL_PROBLEM_ID"
echo "You can manually add test cases to io/$FULL_PROBLEM_ID.in and io/$FULL_PROBLEM_ID.expected"
cat "$TMPFILE"
rm "$TMPFILE"
exit
fi
;;
*)
echo "Unknown contest type: $CONTEST"
echo "Available contests: cses, atcoder, codeforces"
rm "$TMPFILE"
exit
;;
esac
rm "$TMPFILE"