from flask import Flask, render_template, request, redirect
from db import get_db_connection
from utils import generate_client_code

app = Flask(
    __name__,
    template_folder="../templates",
    static_folder="../static",
)

@app.route("/")
def index():
    return """
    <h1>OTB Billing</h1>
    <p>Version 0.0.5</p>
    <p><a href="/clients">Clients</a></p>
    """

@app.route("/clients")
def clients():

    conn = get_db_connection()
    cursor = conn.cursor(dictionary=True)

    cursor.execute("SELECT * FROM clients ORDER BY id DESC")
    clients = cursor.fetchall()

    conn.close()

    return render_template("clients/list.html", clients=clients)


@app.route("/clients/new", methods=["GET","POST"])
def new_client():

    if request.method == "POST":

        company_name = request.form["company_name"]
        contact_name = request.form["contact_name"]
        email = request.form["email"]
        phone = request.form["phone"]

        conn = get_db_connection()
        cursor = conn.cursor(dictionary=True)

        cursor.execute("SELECT MAX(id) as last_id FROM clients")
        result = cursor.fetchone()

        last_number = result["last_id"] if result["last_id"] else 0

        client_code = generate_client_code(company_name, last_number)

        cursor = conn.cursor()

        cursor.execute(
            """
            INSERT INTO clients
            (client_code, company_name, contact_name, email, phone)
            VALUES (%s,%s,%s,%s,%s)
            """,
            (client_code, company_name, contact_name, email, phone)
        )

        conn.commit()
        conn.close()

        return redirect("/clients")

    return render_template("clients/new.html")


if __name__ == "__main__":
    app.run(host="0.0.0.0", port=5050)