diff options
| -rw-r--r-- | dailypuzzle.py | 59 | ||||
| -rw-r--r-- | puzzle.sh | 335 | ||||
| -rw-r--r-- | requirements.txt | 2 |
3 files changed, 396 insertions, 0 deletions
diff --git a/dailypuzzle.py b/dailypuzzle.py new file mode 100644 index 0000000..d4f884e --- /dev/null +++ b/dailypuzzle.py @@ -0,0 +1,59 @@ +#!/usr/bin/env python3 +import chess +import chess.pgn +import io +import json +import os +import time +import urllib.request + +OUT = "/var/www/aryanmustafa/api/dailypuzzle" +INTERVAL = 3600 # check every hour; Lichess updates once per day + +os.makedirs(os.path.dirname(OUT), exist_ok=True) + +while True: + try: + req = urllib.request.Request( + "https://lichess.org/api/puzzle/daily", + headers={"Accept": "application/json"}, + ) + with urllib.request.urlopen(req, timeout=10) as r: + raw = json.loads(r.read()) + + puzzle = raw["puzzle"] + game = raw["game"] + + # Lichess returns the full game PGN; the puzzle starts at initialPly. + # Replay up to that ply with python-chess to get the exact FEN. + pgn_text = game["pgn"] + initial_ply = puzzle["initialPly"] + + parsed = chess.pgn.read_game(io.StringIO(pgn_text)) + board = parsed.board() + for i, move in enumerate(parsed.mainline_moves()): + board.push(move) + if i >= initial_ply: + break + + result = { + "id": puzzle["id"], + "fen": board.fen(), + "side": "white" if board.turn == chess.WHITE else "black", + "solution": puzzle["solution"], # UCI list e.g. ["e2e4","d7d5"] + "rating": puzzle.get("rating"), + "themes": puzzle.get("themes", []), + "url": f"https://lichess.org/training/{puzzle['id']}", + } + + tmp = OUT + ".tmp" + with open(tmp, "w") as f: + json.dump(result, f) + os.replace(tmp, OUT) # atomic write + + print(f"updated: puzzle {result['id']}", flush=True) + + except Exception as e: + print(f"error: {e}", flush=True) + + time.sleep(INTERVAL) diff --git a/puzzle.sh b/puzzle.sh new file mode 100644 index 0000000..004ae87 --- /dev/null +++ b/puzzle.sh @@ -0,0 +1,335 @@ +#!/usr/bin/env bash +# chess.aryanmustafa.com/play — Daily Chess Puzzle +# Usage: curl -sL chess.aryanmustafa.com/play | bash + +set -euo pipefail + +API="https://aryanmustafa.com/api/dailypuzzle" + +# ── Colours ──────────────────────────────────────────────────────────────────── +ESC=$'\033' +RESET="${ESC}[0m" +BOLD="${ESC}[1m" + +LIGHT_BG="${ESC}[48;5;229m" # warm cream +DARK_BG="${ESC}[48;5;94m" # brown + +CHRISTMAS=false +for arg in "$@"; do + case "$arg" in + --christmas) CHRISTMAS=true ;; + esac +done + +if $CHRISTMAS; then + WHITE_PIECE="${ESC}[1;92m" # bold bright green + BLACK_PIECE="${ESC}[1;91m" # bold bright red +else + WHITE_PIECE="${ESC}[1;38;5;208m" # bold orange + BLACK_PIECE="${ESC}[38;5;93m${ESC}[1m" # bold purple +fi + +CYAN="${ESC}[36m" +YELLOW="${ESC}[33m" +GREEN="${ESC}[32m" +RED="${ESC}[31m" +DIM="${ESC}[2m" + +# ── Helpers ──────────────────────────────────────────────────────────────────── +die() { echo "${RED}Error: $*${RESET}" >&2; exit 1; } +require_cmd() { command -v "$1" &>/dev/null || die "'$1' is required but not installed."; } + +require_cmd curl +require_cmd python3 + +# ── Fetch puzzle ─────────────────────────────────────────────────────────────── +echo "" +echo "${BOLD}${CYAN} ♟ Daily Chess Puzzle${RESET}" +echo "${DIM} Fetching from lichess via aryanmustafa.com…${RESET}" +echo "" + +PUZZLE_JSON=$(curl -sf "$API") || die "Could not reach $API" + +PUZZLE_ID=$(echo "$PUZZLE_JSON" | python3 -c "import sys,json; d=json.load(sys.stdin); print(d['id'])") +FEN=$(echo "$PUZZLE_JSON" | python3 -c "import sys,json; d=json.load(sys.stdin); print(d['fen'])") +SIDE=$(echo "$PUZZLE_JSON" | python3 -c "import sys,json; d=json.load(sys.stdin); print(d['side'])") +RATING=$(echo "$PUZZLE_JSON" | python3 -c "import sys,json; d=json.load(sys.stdin); print(d.get('rating','?'))") +THEMES=$(echo "$PUZZLE_JSON" | python3 -c "import sys,json; d=json.load(sys.stdin); print(', '.join(d.get('themes',[])))") +SOLUTION=$(echo "$PUZZLE_JSON" | python3 -c "import sys,json; d=json.load(sys.stdin); print(' '.join(d['solution']))") +URL=$(echo "$PUZZLE_JSON" | python3 -c "import sys,json; d=json.load(sys.stdin); print(d['url'])") + +read -ra SOL_ALL <<< "$SOLUTION" +# solution[0] is the opponent's move that opens the puzzle (applied silently, shown on board) +# solution[1:] are the moves the player must find +SETUP_MOVE="" +SOL_ARRAY=("${SOL_ALL[@]}") +SOL_LEN=${#SOL_ARRAY[@]} + +# ── Python board engine ──────────────────────────────────────────────────────── +py_engine() { + python3 - "$@" <<'PYEOF' +import sys + +def parse_fen(fen): + parts = fen.split() + rows = parts[0].split('/') + board = {} + for r, row in enumerate(rows): + c = 0 + for ch in row: + if ch.isdigit(): + c += int(ch) + else: + board[(r, c)] = ch + c += 1 + side = parts[1] + return board, side + +def board_to_str(board): + lines = [] + for r in range(8): + row = [] + for c in range(8): + row.append(board.get((r, c), '.')) + lines.append(''.join(row)) + return '\n'.join(lines) + +def uci_to_coords(uci): + fc, fr, tc, tr = uci[0], uci[1], uci[2], uci[3] + return (8 - int(fr), ord(fc) - ord('a')), (8 - int(tr), ord(tc) - ord('a')) + +def apply_move(board, uci): + src, dst = uci_to_coords(uci) + piece = board.get(src) + if piece is None: + return None, "No piece at source square" + new_board = dict(board) + if piece in ('K', 'k') and abs(src[1] - dst[1]) == 2: + new_board[dst] = piece + del new_board[src] + if dst[1] == 6: + rook_src, rook_dst = (src[0], 7), (src[0], 5) + else: + rook_src, rook_dst = (src[0], 0), (src[0], 3) + rook = new_board.get(rook_src) + if rook: + new_board[rook_dst] = rook + del new_board[rook_src] + else: + if piece in ('P', 'p') and src[1] != dst[1] and dst not in board: + new_board.pop((src[0], dst[1]), None) + new_board[dst] = piece + del new_board[src] + if len(uci) == 5: + promo = uci[4] + new_board[dst] = promo.upper() if piece.isupper() else promo.lower() + elif piece == 'P' and dst[0] == 0: + new_board[dst] = 'Q' + elif piece == 'p' and dst[0] == 7: + new_board[dst] = 'q' + return new_board, None + +def board_to_fen(board, side): + rows = [] + for r in range(8): + empty = 0 + row_str = '' + for c in range(8): + p = board.get((r, c)) + if p: + if empty: row_str += str(empty); empty = 0 + row_str += p + else: + empty += 1 + if empty: row_str += str(empty) + rows.append(row_str) + new_side = 'b' if side == 'w' else 'w' + return '/'.join(rows) + f' {new_side} - - 0 1' + +def validate_uci(uci): + uci = uci.strip().lower() + if len(uci) not in (4, 5): + return None, "Move must be 4-5 chars (e.g. e2e4)" + if uci[0] not in 'abcdefgh' or uci[2] not in 'abcdefgh': + return None, "Invalid file (use a-h)" + if uci[1] not in '12345678' or uci[3] not in '12345678': + return None, "Invalid rank (use 1-8)" + if len(uci) == 5 and uci[4] not in 'qrbn': + return None, "Invalid promotion piece (q/r/b/n)" + return uci, None + +cmd = sys.argv[1] + +if cmd == "show": + board, _ = parse_fen(sys.argv[2]) + print(board_to_str(board)) + +elif cmd == "apply": + board, side = parse_fen(sys.argv[2]) + uci, err = validate_uci(sys.argv[3]) + if err: + print(f"ERR:{err}"); sys.exit(0) + new_board, err = apply_move(board, uci) + if err: + print(f"ERR:{err}"); sys.exit(0) + print(board_to_fen(new_board, side)) + +elif cmd == "hint_piece": + board, _ = parse_fen(sys.argv[2]) + src, _ = uci_to_coords(sys.argv[3]) + piece = board.get(src, '?') + file_ = chr(ord('a') + src[1]) + rank_ = str(8 - src[0]) + print(f"{piece}@{file_}{rank_}") +PYEOF +} + +# ── Draw board ───────────────────────────────────────────────────────────────── +draw_board() { + local fen="$1" + local board_str + board_str=$(py_engine show "$fen") + + piece_char() { + case "$1" in + P|p) echo "P" ;; N|n) echo "N" ;; B|b) echo "B" ;; + R|r) echo "R" ;; Q|q) echo "Q" ;; K|k) echo "K" ;; + *) echo " " ;; + esac + } + + echo "" + echo " a b c d e f g h" + echo " ┌───┬───┬───┬───┬───┬───┬───┬───┐" + + local row_idx=0 + while IFS= read -r row_line; do + local rank=$((8 - row_idx)) + printf "%d │" "$rank" + + local col_idx=0 + while [ $col_idx -lt 8 ]; do + local ch="${row_line:$col_idx:1}" + local pc; pc=$(piece_char "$ch") + + local is_light=$(( (row_idx + col_idx) % 2 )) + local sq_bg + if [ "$is_light" -eq 0 ]; then + sq_bg="$LIGHT_BG" + else + sq_bg="$DARK_BG" + fi + + # White pieces = uppercase letters; black pieces = lowercase + local fg + if [[ "$ch" =~ [A-Z] ]]; then + fg="$WHITE_PIECE" + else + fg="$BLACK_PIECE" + fi + + printf "%s%s %s %s│" "$sq_bg" "$fg" "$pc" "$RESET" + col_idx=$((col_idx + 1)) + done + + printf " %d\n" "$rank" + if [ "$row_idx" -lt 7 ]; then + echo " ├───┼───┼───┼───┼───┼───┼───┼───┤" + fi + row_idx=$((row_idx + 1)) + done <<< "$board_str" + + echo " └───┴───┴───┴───┴───┴───┴───┴───┘" + echo " a b c d e f g h" + echo "" +} + +# ── Setup ────────────────────────────────────────────────────────────────────── +SIDE_DISPLAY=$(echo "$SIDE" | awk '{print toupper(substr($0,1,1))substr($0,2)}') + +echo " ${BOLD}Puzzle #${PUZZLE_ID}${RESET} ${DIM}Rating: ${RATING} Themes: ${THEMES}${RESET}" +echo " ${DIM}${URL}${RESET}" +echo "" +echo " ${YELLOW}${BOLD}${SIDE_DISPLAY} to move.${RESET} Find the best sequence (${SOL_LEN} moves)." +echo " ${DIM}Enter moves in UCI format (e.g. e2e4). Type 'hint' or 'quit'.${RESET}" + +# Apply opponent's opening move so the board shows the puzzle start position +current_fen="$FEN" +move_num=0 + +draw_board "$current_fen" + +# ── Puzzle loop ──────────────────────────────────────────────────────────────── +declare -A PIECE_NAMES=( + [P]="Pawn" [N]="Knight" [B]="Bishop" [R]="Rook" [Q]="Queen" [K]="King" + [p]="Pawn" [n]="Knight" [b]="Bishop" [r]="Rook" [q]="Queen" [k]="King" +) + +while [ "$move_num" -lt "$SOL_LEN" ]; do + expected="${SOL_ARRAY[$move_num]}" + + printf " ${BOLD}Move %d/%d${RESET} › " "$((move_num + 1))" "$SOL_LEN" + read -r user_input </dev/tty + + user_input=$(echo "$user_input" | tr '[:upper:]' '[:lower:]' | tr -d '[:space:]') + + case "$user_input" in + quit|q|exit) + echo "" + echo " ${DIM}Puzzle abandoned. Solution: ${SOLUTION}${RESET}" + echo " ${DIM}${URL}${RESET}" + echo "" + exit 0 + ;; + hint|h) + hint_raw=$(py_engine hint_piece "$current_fen" "$expected") + piece_sym="${hint_raw%%@*}" + sq="${hint_raw##*@}" + full_name="${PIECE_NAMES[$piece_sym]:-Piece}" + echo " ${YELLOW}Hint: Move the ${full_name} on ${sq}.${RESET}" + continue + ;; + "") + continue + ;; + esac + + if ! [[ "$user_input" =~ ^[a-h][1-8][a-h][1-8][qrbn]?$ ]]; then + echo " ${RED}Invalid format. Use UCI notation, e.g. e2e4 or a7a8q.${RESET}" + continue + fi + + if [ "$user_input" = "$expected" ]; then + move_num=$((move_num + 1)) + + new_fen=$(py_engine apply "$current_fen" "$user_input") + if [[ "$new_fen" == ERR:* ]]; then + echo " ${RED}${new_fen#ERR:}${RESET}" + move_num=$((move_num - 1)) + continue + fi + current_fen="$new_fen" + + if [ "$move_num" -lt "$SOL_LEN" ]; then + opp_move="${SOL_ARRAY[$move_num]}" + echo " ${GREEN}✓ Correct!${RESET} ${DIM}Opponent plays ${opp_move}…${RESET}" + new_fen=$(py_engine apply "$current_fen" "$opp_move") + if [[ "$new_fen" != ERR:* ]]; then + current_fen="$new_fen" + fi + move_num=$((move_num + 1)) + draw_board "$current_fen" + else + echo " ${GREEN}✓ Correct!${RESET}" + draw_board "$current_fen" + fi + else + echo " ${RED}✗ Not quite — try again.${RESET}" + fi +done + +echo "" +echo " ${BOLD}${GREEN}Puzzle solved! ♟${RESET}" +echo " ${DIM}Full solution: ${SOLUTION}${RESET}" +echo " ${DIM}Review on Lichess: ${URL}${RESET}" +echo "" diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..43125ab --- /dev/null +++ b/requirements.txt @@ -0,0 +1,2 @@ +chess + |