import sqlite3

conn = sqlite3.connect("user.db")
cursor = conn.cursor()

cursor.execute("""
    CREATE TABLE IF NOT EXISTS users(
        discord_id TEXT PRIMARY KEY,
        puuid TEXT NOT NULL,
        game_name TEXT,
        tag_line TEXT
    )
""")
conn.commit()


def save_user(discord_id: str, puuid: str, game_name: str, tag_line: str):
    cursor.execute(
        """
        INSERT INTO users(discord_id,puuid,game_name,tag_line)
        VALUEs (?,?,?,?)
        ON CONFLICT(discord_id) DO UPDATE SET
         puuid = excluded.puuid,
         game_name=excluded.game_name,
         tag_line=excluded.tag_line
    """,
        (discord_id, puuid, game_name, tag_line),
    )
    conn.commit()


def get_user(discord_id: str):
    cursor.execute("SELECT * FROM users WHERE discord_id = ?", (discord_id,))
    return cursor.fetchone()