import sqlite3
import os

DB_PATH = 'noirflow.db'

def get_db_connection():
    conn = sqlite3.connect(DB_PATH)
    conn.row_factory = sqlite3.Row
    return conn

def init_db():
    conn = get_db_connection()
    # Table for node instances on the board
    conn.execute('''
        CREATE TABLE IF NOT EXISTS node_instances (
            id TEXT PRIMARY KEY,
            node_type_id TEXT,
            x INTEGER,
            y INTEGER,
            data TEXT
        )
    ''')
    # Table for connections between nodes
    conn.execute('''
        CREATE TABLE IF NOT EXISTS connections (
            id TEXT PRIMARY KEY,
            from_node TEXT,
            from_slot TEXT,
            to_node TEXT,
            to_slot TEXT
        )
    ''')
    conn.commit()
    conn.close()