import sqlite3
import os
from flask import Flask, render_template, request, jsonify

app = Flask(__name__)

DB_PATH = '/opt/flaskapp/scores.db'

def get_db():
    conn = sqlite3.connect(DB_PATH)
    conn.row_factory = sqlite3.Row
    return conn

def init_db():
    with get_db() as conn:
        conn.execute('''
            CREATE TABLE IF NOT EXISTS scores (
                id        INTEGER PRIMARY KEY AUTOINCREMENT,
                name      TEXT    NOT NULL,
                score     INTEGER NOT NULL,
                level     INTEGER NOT NULL,
                lines     INTEGER NOT NULL,
                domain    TEXT,
                ts        DATETIME DEFAULT CURRENT_TIMESTAMP
            )
        ''')
        conn.commit()

init_db()

@app.route("/", defaults={"path": ""})
@app.route("/<path:path>")
def blocked(path):
    if path == "leaderboard":
        return leaderboard()
    domain = request.host or request.args.get("domain", "that site")
    domain = domain.split(":")[0]
    return render_template("blocked.html", domain=domain, reason="blocked by network policy")

@app.route("/score", methods=["POST"])
def submit_score():
    data = request.get_json()
    name  = (data.get("name") or "ANONYMOUS")[:16].upper().strip()
    score = int(data.get("score", 0))
    level = int(data.get("level", 1))
    lines = int(data.get("lines", 0))
    domain = (data.get("domain") or "unknown")[:64]
    with get_db() as conn:
        conn.execute(
            "INSERT INTO scores (name, score, level, lines, domain) VALUES (?,?,?,?,?)",
            (name, score, level, lines, domain)
        )
        conn.commit()
    return jsonify({"ok": True})

@app.route("/leaderboard")
def leaderboard():
    with get_db() as conn:
        rows = conn.execute(
            "SELECT name, score, level, lines, domain, ts FROM scores ORDER BY score DESC LIMIT 20"
        ).fetchall()
    return render_template("leaderboard.html", scores=rows)

if __name__ == "__main__":
    app.run(host="0.0.0.0", port=80, debug=True)