migrations.py

from coincurve import PrivateKey


async def m001_initial(db):
    """
    Initial tables
    """
    await db.execute(
        """
        CREATE TABLE nwcprovider.keys (
            pubkey TEXT PRIMARY KEY,
            wallet TEXT NOT NULL,
            description TEXT NOT NULL,
            expires_at INTEGER NOT NULL,
            permissions TEXT NOT NULL,
            created_at INTEGER NOT NULL
        );
        """
    )

    await db.execute(
        f"""
        CREATE TABLE nwcprovider.spent (
            id {db.serial_primary_key},
            pubkey TEXT NOT NULL,
            amount_msats INTEGER NOT NULL,
            created_at INTEGER NOT NULL,
            FOREIGN KEY(pubkey)
            REFERENCES {db.references_schema}keys(pubkey)
            ON DELETE CASCADE
        );
        """
    )

    await db.execute(
        f"""
        CREATE TABLE nwcprovider.budgets (
            id {db.serial_primary_key},
            pubkey TEXT NOT NULL,
            budget_msats INTEGER NOT NULL,
            refresh_window INTEGER NOT NULL,
            created_at INTEGER NOT NULL,
            FOREIGN KEY(pubkey)
            REFERENCES {db.references_schema}keys(pubkey)
            ON DELETE CASCADE
        );
        """
    )


async def m002_config(db):
    """
    Config table
    """
    await db.execute(
        """
        CREATE TABLE nwcprovider.config (
            key TEXT PRIMARY KEY,
            value TEXT NOT NULL
        );
        """
    )


async def m003_default_config(db):
    """
    Default config
    """
    await db.execute(
        """
        INSERT INTO nwcprovider.config (key, value) VALUES ('relay', 'nostrclient')
        ON CONFLICT (key) DO UPDATE SET value = EXCLUDED.value;
        """
    )
    private_key = PrivateKey()
    await db.execute(
        """
        INSERT INTO nwcprovider.config (key, value)
        VALUES ('provider_key', :provider_key)
        ON CONFLICT (key) DO UPDATE SET value = EXCLUDED.value;
        """,
        {"provider_key": private_key.to_hex()},
    )


async def m004_default_config2(db):
    """
    Default config
    """
    await db.execute(
        """
        INSERT INTO nwcprovider.config (key, value) VALUES ('relay_alias', :value)
        ON CONFLICT (key) DO UPDATE SET value = EXCLUDED.value;
        """,
        {"value": ""},
    )


async def m005_key_last_used(db):
    """
    Add last_used to keys
    """
    await db.execute(
        """
        ALTER TABLE nwcprovider.keys ADD COLUMN last_used INTEGER;
        """
    )


async def m006_default_config3(db):
    """
    Default config
    """
    await db.execute(
        """
        INSERT INTO nwcprovider.config (key, value)
        VALUES ('handle_missed_events', :value)
        ON CONFLICT (key) DO UPDATE SET value = EXCLUDED.value;
        """,
        {"value": "0"},
    )