from dataclasses import dataclass
from .connect import get_connection


def init_db():
    conn = get_connection()
    cursor = conn.cursor()
    cursor.execute("""
        CREATE TABLE IF NOT EXISTS songs (
            mbid TEXT PRIMARY KEY,
            title TEXT NOT NULL,
            artist TEXT NOT NULL,
            tags TEXT NOT NULL,
            lastfm_image_id TEXT NOT NULL,
            youtube_id TEXT NOT NULL
        );
    """)
    conn.commit()
    conn.close()


@dataclass
class Song:
    mbid: str
    title: str
    artist: str
    tags: list[str]
    image_id: str
    youtube_id: str