import sqlite3


def initialize_db():
    conn = sqlite3.connect('selena.db')
    cursor = conn.cursor()

    # Birthdays table
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS birthdays (
            user_id TEXT PRIMARY KEY,
            birthday TEXT
        )
    ''')

    # Currency table
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS currency (
            user_id TEXT PRIMARY KEY,
            balance INTEGER,
            last_earned TIMESTAMP
        )
    ''')

    # Followed channels table
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS followed_channels (
            twitch_name TEXT PRIMARY KEY,
            discord_channel_id INTEGER
        )
    ''')
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS followed_youtube_channels (
            youtube_channel_id TEXT PRIMARY KEY,
            discord_channel_id INTEGER
        )
    ''')
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS youtube_status (
            youtube_channel_id TEXT PRIMARY KEY,
            last_video_id TEXT
        )
    ''')
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS user_xp (
            user_id INTEGER PRIMARY KEY,
            xp INTEGER,
            level INTEGER
        )
    ''')

    conn.commit()
    conn.close()


def get_connection():
    return sqlite3.connect('selena.db')