from config import Config

_DEFAULT_GROUP = "30тс"


def save_user(user_id: int, group: str = _DEFAULT_GROUP):
    from .DB import get_db

    db = get_db()
    cur = db.cursor()
    cur.execute("INSERT INTO users (user_id, user_group) VALUES (?, ?)", (user_id, group))
    db.commit()
    cur.close()
    db.close()


def set_group(user_id: int, group: str = _DEFAULT_GROUP):
    from .DB import get_db

    db = get_db()
    cur = db.cursor()
    cur.execute("SELECT 1 FROM users WHERE user_id = ?", (user_id,))
    exists = cur.fetchone()

    if exists:
        cur.execute("UPDATE users SET user_group = ? WHERE user_id = ?", (group, user_id))
    else:
        cur.execute("INSERT INTO users (user_id, user_group) VALUES (?, ?)", (user_id, group))

    db.commit()
    cur.close()
    db.close()


def get_group(user_id: int, default: str = _DEFAULT_GROUP) -> str:
    from .DB import get_db

    db = get_db()
    cur = db.cursor()
    cur.execute("SELECT user_group FROM users WHERE user_id = ?", (user_id,))
    row = cur.fetchone()
    if row:
        group = row[0]
    else:
        cur.execute("INSERT INTO users (user_id, user_group) VALUES (?, ?)", (user_id, default))
        db.commit()
        group = default
    cur.close()
    db.close()
    return group