from flask import Flask, render_template, request, redirect, url_for, jsonify
import sqlite3
import string
import random
import os
import socket

ADMIN_HOST = "web-oops-admin"
ADMIN_PORT = 3001

app = Flask(__name__)



# Database setup
def init_db():
    conn = sqlite3.connect('urls.db')
    c = conn.cursor()
    c.execute('''CREATE TABLE IF NOT EXISTS urls
                 (id INTEGER PRIMARY KEY AUTOINCREMENT,
                  original_url TEXT NOT NULL,
                  short_code TEXT UNIQUE NOT NULL,
                  clicks INTEGER DEFAULT 0,
                  created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP)''')
    conn.commit()
    conn.close()

def generate_short_code(length=6):
    characters = string.ascii_letters + string.digits
    return ''.join(random.choice(characters) for _ in range(length))

def get_db_connection():
    conn = sqlite3.connect('urls.db')
    conn.row_factory = sqlite3.Row
    return conn

@app.route('/', methods=['GET', 'POST'])
def index():
    message = None
    shortened_url = None
    
    if request.method == 'POST':
        original_url = request.form['original_url']


        url = original_url.lower()
        while "script" in url:
            url = url.replace("script", "")
        
        # Generate unique short code
        while True:
            short_code = generate_short_code()
            conn = get_db_connection()
            existing = conn.execute('SELECT id FROM urls WHERE short_code = ?', 
                                  (short_code,)).fetchone()
            if not existing:
                break
            conn.close()
        
        # Save to database
        conn = get_db_connection()
        conn.execute('INSERT INTO urls (original_url, short_code) VALUES (?, ?)',
                    (original_url, short_code))
        conn.commit()
        conn.close()
        
        shortened_url = request.host_url + short_code
        message = "URL shortened successfully!"
    
    return render_template("index.html", 
                                message=message, 
                                shortened_url=shortened_url)

@app.post('/report')
def report():
    submit_id = request.form["submit_id"]
    submit_id = submit_id.split("/")[-1]
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    s.connect((ADMIN_HOST, ADMIN_PORT))
    s.sendall(submit_id.encode())
    s.close()
    return render_template("index.html", 
                                report_message="Reported successfully.")


@app.route('/<short_code>')
def redirect_url(short_code):
    conn = get_db_connection()
    url_data = conn.execute('SELECT original_url FROM urls WHERE short_code = ?', 
                           (short_code,)).fetchone()
    
    if url_data:
        # Increment click counter
        conn.execute('UPDATE urls SET clicks = clicks + 1 WHERE short_code = ?', 
                    (short_code,))
        conn.commit()
        conn.close()
        return render_template("redir.html", url=url_data["original_url"]), 200
    else:
        conn.close()
        return render_template("not_found.html"), 404

init_db()