main.py

from flask import Flask, flash, g, redirect, render_template, request,session, url_for
from admins import *
from github import *
from cloudflare import *
import sqlite3
from routes.authentication import * 

connection = sqlite3.connect("database.db")

connection.execute("CREATE TABLE IF NOT EXISTS users (token TEXT, username TEXT, subdomains TEXT)")
connection.commit()
connection.close()
def use_database(query: str, values:tuple=None):
    
    connection = sqlite3.connect("database.db")
    res = connection.execute(query, values)
    returned_value = None
    if "select" in query.lower():
        returned_value = res.fetchone()
    connection.commit()
    connection.close()
    return returned_value

app = Flask(__name__)
app.secret_key = 'somesecretkeythatonlyishouldknow'
app.config['GITHUB_CLIENT_ID'] = CLIENT_ID
app.config['GITHUB_CLIENT_SECRET'] = CLIENT_SECRET
github = GitHub(app)




cloudflare = {}
for domain in CLOUDFLARE_DOMAINS:
    cloudflare[domain['url']] = Cloudflare(api_token=CLOUDFLARE_API_TOKEN,account_id=CLOUDFLARE_ACCOUNT_ID,zone_id=domain['cloudflare_zone_id'])


#load_github_sites(app=app) #loads sites from github api





@app.route('/')
def indexnormal():
    return render_template('home.html')

@app.route('/signin')
def signin():
    return github.authorize()

@app.route('/claim')
def claim():
    return render_template("claim.html")

# ADMIN WEBSITE CODE
@app.before_request
def before_request():
    g.user = None
    

    if 'user_id' in session:
        user = [x for x in ADMIN_ACCTS if x.id == session['user_id']][0]
        g.user = user


@app.route('/admin', methods=['GET', 'POST']) #admin site soon
def admin():
    if not g.user:
        return redirect(url_for('login'))
    subdomains = []

    for domain in CLOUDFLARE_DOMAINS:
        yes = cloudflare[domain['url']].getDNSrecords()
        for ye in yes:
            subdomains.append({"name": ye['name'], "type": ye['type'], "content": ye['content'], "id": ye['id'], "proxied": ye['proxied']})
    return render_template('admin.html', subdomains=subdomains, account_id=CLOUDFLARE_ACCOUNT_ID)

@app.route('/dashboard', methods=['GET', 'POST'])
def dashboard():
    if request.method == "POST":
        data = {}
        data["dns_record"] = request.form["dns_record"]
        data["type"] = request.form.get("type")
        data["url"] = request.form.get("url")
        data["dns_content"] = request.form.get("dns_content")
        subdomains = use_database("SELECT subdomains FROM users where token = ?", (session['id'],))
        if data['type'].lower() == "a":
            
            use_database("UPDATE users SET subdomains = ? WHERE token = ? '", (f"{subdomains}<>{data['dns_record']}.{data['url']}", session['id'],))
            cloudflare[data['url']].insert_A_record(data["dns_record"], data["dns_content"], PROXIED=False)
        elif data['type'].lower() == "cname":
            use_database("UPDATE users SET subdomains = ? WHERE token = ? '", (f"{subdomains}<>{data['dns_record']}.{data['url']}", session['id'],))
            
            cloudflare[data['url']].insert_CNAME_record(data["dns_record"], data["dns_content"], PROXIED=False)    
        else:
            return "wrong type"
    all_sub_domains=[]
    for all_domain in CLOUDFLARE_DOMAINS:
        records = cloudflare[all_domain['url']].getDNSrecords()
        for record in records:
            all_sub_domains.append({"name": record['name'], "type": record['type'], "content": record['content'], "id": record['id'], "proxied": record['proxied']})
    
    
    domains = use_database("SELECT subdomains from users where token = ?", (session['id'],))
    user_subdomains = []
    print(domains)
    if domains[0] is None:
        return render_template('admin.html', subdomains=None, account_id=CLOUDFLARE_ACCOUNT_ID)
        
    
    for domain in domains[0].split("<>"):
        for possible_domain in all_sub_domains:
            if possible_domain['name'] == domain:
                user_subdomains.append(possible_domain)
    return render_template('admin.html', subdomains=user_subdomains, account_id=CLOUDFLARE_ACCOUNT_ID)

@app.route('/control', methods=['GET', 'POST']) #admin site soon
def control(output:str = "N/A"):
    if not g.user:
        return redirect(url_for('login'))
    
    if request.method == "POST":
        data = {}
        data["dns_record"] = request.form["dns_record"]
        data["type"] = request.form.get("type")
        data["url"] = request.form.get("url")
        data["dns_content"] = request.form.get("dns_content")
        if data['type'].lower() == "a":
            cloudflare[data['url']].insert_A_record(data["dns_record"], data["dns_content"], PROXIED=False)
        elif data['type'].lower() == "cname":
            cloudflare[data['url']].insert_CNAME_record(data["dns_record"], data["dns_content"], PROXIED=False)    
        else:
            return "wrong type"
        
        
        
    
    return render_template("control.html", output=output, urls=CLOUDFLARE_DOMAINS)
    
@app.route('/loginadmin', methods=['GET', 'POST'])
def loginadmin():
    if request.method == "POST":
        session.pop('user_id', None)
        
        email = request.form["email"]
        pw = request.form["password"]

        user = [x for x in ADMIN_ACCTS if x.username == email][0]
        if user and user.password == pw:
            session['user_id'] = user.id
            return redirect(url_for('admin'))
        
        return redirect(url_for('loginadmin'))
        
    return render_template("login.html")

if __name__ == '__main__':
    #from waitress import serve
    #serve(app, host="0.0.0.0", port=8080)
    app.register_blueprint(authentication)
    app.run(host='0.0.0.0')