mirror of
https://github.com/lnbits/lnbits-legend.git
synced 2025-02-26 15:42:30 +01:00
156 lines
3.9 KiB
Python
156 lines
3.9 KiB
Python
from typing import List, Optional, Union
|
|
|
|
from lnbits.helpers import urlsafe_short_hash
|
|
import shortuuid
|
|
from . import db
|
|
from .models import (
|
|
nostrKeys,
|
|
nostrNotes,
|
|
nostrCreateRelays,
|
|
nostrRelays,
|
|
nostrConnections,
|
|
nostrCreateConnections,
|
|
nostrRelayList,
|
|
)
|
|
|
|
###############KEYS##################
|
|
|
|
|
|
async def create_nostrkeys(data: nostrKeys) -> nostrKeys:
|
|
await db.execute(
|
|
"""
|
|
INSERT INTO nostradmin.keys (
|
|
pubkey,
|
|
privkey
|
|
)
|
|
VALUES (?, ?)
|
|
""",
|
|
(data.pubkey, data.privkey),
|
|
)
|
|
return await get_nostrkeys(nostrkey_id)
|
|
|
|
|
|
async def get_nostrkeys(pubkey: str) -> nostrKeys:
|
|
row = await db.fetchone("SELECT * FROM nostradmin.keys WHERE pubkey = ?", (pubkey,))
|
|
return nostrKeys(**row) if row else None
|
|
|
|
|
|
###############NOTES##################
|
|
|
|
|
|
async def create_nostrnotes(data: nostrNotes) -> nostrNotes:
|
|
await db.execute(
|
|
"""
|
|
INSERT INTO nostradmin.notes (
|
|
id,
|
|
pubkey,
|
|
created_at,
|
|
kind,
|
|
tags,
|
|
content,
|
|
sig
|
|
)
|
|
VALUES (?, ?, ?, ?, ?, ?, ?)
|
|
""",
|
|
(
|
|
data.id,
|
|
data.pubkey,
|
|
data.created_at,
|
|
data.kind,
|
|
data.tags,
|
|
data.content,
|
|
data.sig,
|
|
),
|
|
)
|
|
return await get_nostrnotes(data.id)
|
|
|
|
|
|
async def get_nostrnotes(nostrnote_id: str) -> nostrNotes:
|
|
row = await db.fetchone("SELECT * FROM nostradmin.notes WHERE id = ?", (nostrnote_id,))
|
|
return nostrNotes(**row) if row else None
|
|
|
|
|
|
###############RELAYS##################
|
|
|
|
|
|
async def create_nostrrelays(data: nostrCreateRelays) -> nostrCreateRelays:
|
|
nostrrelay_id = shortuuid.uuid(name=data.relay)
|
|
|
|
if await get_nostrrelay(nostrrelay_id):
|
|
return "error"
|
|
await db.execute(
|
|
"""
|
|
INSERT INTO nostradmin.relays (
|
|
id,
|
|
relay
|
|
)
|
|
VALUES (?, ?)
|
|
""",
|
|
(nostrrelay_id, data.relay),
|
|
)
|
|
return await get_nostrrelay(nostrrelay_id)
|
|
|
|
|
|
async def get_nostrrelays() -> nostrRelays:
|
|
rows = await db.fetchall("SELECT * FROM nostradmin.relays")
|
|
return [nostrRelays(**row) for row in rows]
|
|
|
|
|
|
async def get_nostrrelay(nostrrelay_id: str) -> nostrRelays:
|
|
row = await db.fetchone("SELECT * FROM nostradmin.relays WHERE id = ?", (nostrrelay_id,))
|
|
return nostrRelays(**row) if row else None
|
|
|
|
|
|
async def update_nostrrelayallowlist(allowlist: str) -> nostrRelayList:
|
|
await db.execute(
|
|
"""
|
|
UPDATE nostradmin.relaylist SET
|
|
allowlist = ?
|
|
WHERE id = ?
|
|
""",
|
|
(allowlist, 1),
|
|
)
|
|
return await get_nostrrelaylist()
|
|
|
|
async def update_nostrrelaydenylist(denylist: str) -> nostrRelayList:
|
|
await db.execute(
|
|
"""
|
|
UPDATE nostradmin.relaylist SET
|
|
denylist = ?
|
|
WHERE id = ?
|
|
""",
|
|
(denylist, 1),
|
|
)
|
|
return await get_nostrrelaylist()
|
|
|
|
async def get_nostrrelaylist() -> nostrRelayList:
|
|
row = await db.fetchone("SELECT * FROM nostradmin.relaylist WHERE id = ?", (1,))
|
|
return nostrRelayList(**row) if row else None
|
|
|
|
|
|
###############CONNECTIONS##################
|
|
|
|
|
|
async def create_nostrconnections(
|
|
data: nostrCreateConnections
|
|
) -> nostrCreateConnections:
|
|
nostrrelay_id = shortuuid.uuid(name=data.relayid + data.pubkey)
|
|
await db.execute(
|
|
"""
|
|
INSERT INTO nostradmin.connections (
|
|
id,
|
|
pubkey,
|
|
relayid
|
|
)
|
|
VALUES (?, ?, ?)
|
|
""",
|
|
(data.id, data.pubkey, data.relayid),
|
|
)
|
|
return await get_nostrconnections(data.id)
|
|
|
|
|
|
async def get_nostrconnections(nostrconnections_id: str) -> nostrConnections:
|
|
row = await db.fetchone(
|
|
"SELECT * FROM nostradmin.connections WHERE id = ?", (nostrconnections_id,)
|
|
)
|
|
return nostrConnections(**row) if row else None
|