izbi-dns/app/auth.py

import sqlite3

def init_db(path: str) -> None:
    with sqlite3.connect(path) as connection:
        cursor = connection.cursor()
        cursor.execute('''
            CREATE TABLE IF NOT EXISTS Tokens (
                TOKEN TEXT PRIMARY KEY NOT NULL,
                USERID INTEGER NOT NULL,
                NAME TEXT NOT NULL
            )
        ''')

        cursor.execute('''
            CREATE TABLE IF NOT EXISTS Domains (
                DOMAIN TEXT PRIMARY KEY NOT NULL,
                USERID INTEGER NOT NULL
            )
        ''')

        connection.commit()
        cursor.close()

def check_token_domain(db_path: str, token: str, domain: str) -> bool:
    with sqlite3.connect(db_path) as connection:
        cursor = connection.cursor()
        cursor.execute('''
            SELECT COUNT(*) FROM Tokens
            JOIN Domains ON Tokens.userid = Domains.userid
            WHERE token = ?
                AND domain = ?
        ''', (token,domain))

    return cursor.fetchone()[0] == 1