mirror of
https://github.com/believethehype/nostrdvm.git
synced 2025-10-09 18:22:50 +02:00
132 lines
5.2 KiB
Python
132 lines
5.2 KiB
Python
# ADMINISTRARIVE DB MANAGEMENT
|
|
|
|
from nostr_sdk import Keys, PublicKey, Client, EventId
|
|
|
|
from nostr_dvm.utils.database_utils import get_from_sql_table, list_db, delete_from_sql_table, update_sql_table, \
|
|
get_or_add_user, clean_db
|
|
from nostr_dvm.utils.dvmconfig import DVMConfig
|
|
from nostr_dvm.utils.nip65_utils import nip65_announce_relays
|
|
from nostr_dvm.utils.nip88_utils import nip88_announce_tier, fetch_nip88_parameters_for_deletion, fetch_nip88_event, \
|
|
check_and_set_tiereventid_nip88
|
|
from nostr_dvm.utils.nip89_utils import nip89_announce_tasks, fetch_nip89_parameters_for_deletion
|
|
from nostr_dvm.utils.nostr_utils import update_profile
|
|
from nostr_dvm.utils.nut_wallet_utils import NutZapWallet
|
|
|
|
|
|
class AdminConfig:
|
|
REBROADCAST_NIP89: bool = False
|
|
REBROADCAST_NIP88: bool = False
|
|
UPDATE_PROFILE: bool = False
|
|
REBROADCAST_NIP65_RELAY_LIST: bool = False
|
|
DELETE_NIP89: bool = False
|
|
DELETE_NIP88: bool = False
|
|
FETCH_NIP88: bool = False
|
|
WHITELISTUSER: bool = False
|
|
UNWHITELISTUSER: bool = False
|
|
BLACKLISTUSER: bool = False
|
|
DELETEUSER: bool = False
|
|
LISTDATABASE: bool = False
|
|
ClEANDB: bool = False
|
|
MELT_ON_STARTUP: bool = False
|
|
POW: bool = False
|
|
INDEX: str = "1"
|
|
LUD16: str = ""
|
|
|
|
USERNPUBS: list = []
|
|
|
|
EVENTID: str = ""
|
|
PRIVKEY: str = ""
|
|
|
|
|
|
async def admin_make_database_updates(adminconfig: AdminConfig = None, dvmconfig: DVMConfig = None,
|
|
client: Client = None):
|
|
# This is called on start of Server, Admin function to manually whitelist/blacklist/add balance/delete users
|
|
if adminconfig is None or dvmconfig is None:
|
|
return
|
|
|
|
if not isinstance(adminconfig, AdminConfig):
|
|
return
|
|
|
|
if ((adminconfig.WHITELISTUSER is True or adminconfig.UNWHITELISTUSER is True or adminconfig.BLACKLISTUSER is True or adminconfig.DELETEUSER is True)
|
|
and adminconfig.USERNPUBS == []):
|
|
return
|
|
|
|
if adminconfig.UPDATE_PROFILE and (dvmconfig.NIP89 is None):
|
|
return
|
|
|
|
if adminconfig.DELETE_NIP89 and (adminconfig.EVENTID == "" or adminconfig.EVENTID == ""):
|
|
return
|
|
|
|
db = dvmconfig.DB
|
|
|
|
for npub in adminconfig.USERNPUBS:
|
|
publickey = PublicKey.parse(npub).to_hex()
|
|
|
|
if adminconfig.WHITELISTUSER:
|
|
user = await get_or_add_user(db, publickey, client=client, config=dvmconfig)
|
|
update_sql_table(db, user.npub, user.balance, True, False, user.nip05, user.lud16, user.name,
|
|
user.lastactive, user.subscribed)
|
|
user = get_from_sql_table(db, publickey)
|
|
print(str(user.name) + " is whitelisted: " + str(user.iswhitelisted))
|
|
|
|
if adminconfig.UNWHITELISTUSER:
|
|
user = get_from_sql_table(db, publickey)
|
|
update_sql_table(db, user.npub, user.balance, False, False, user.nip05, user.lud16, user.name,
|
|
user.lastactive, user.subscribed)
|
|
|
|
if adminconfig.BLACKLISTUSER:
|
|
user = get_from_sql_table(db, publickey)
|
|
update_sql_table(db, user.npub, user.balance, False, True, user.nip05, user.lud16, user.name,
|
|
user.lastactive, user.subscribed)
|
|
|
|
if adminconfig.DELETEUSER:
|
|
delete_from_sql_table(db, publickey)
|
|
|
|
|
|
|
|
if adminconfig.ClEANDB:
|
|
clean_db(db)
|
|
|
|
if adminconfig.LISTDATABASE:
|
|
list_db(db)
|
|
|
|
if adminconfig.MELT_ON_STARTUP:
|
|
nutzap_wallet = NutZapWallet()
|
|
keys = Keys.parse(adminconfig.PRIVKEY)
|
|
nut_wallet = await nutzap_wallet.get_nut_wallet(client, keys)
|
|
lud16 = adminconfig.LUD16
|
|
npub = keys.public_key().to_hex()
|
|
await nutzap_wallet.melt_cashu(nut_wallet, DVMConfig.NUZAP_MINTS[0], nut_wallet.balance, client, keys, lud16, npub)
|
|
await nutzap_wallet.get_nut_wallet(client, keys)
|
|
|
|
if adminconfig.REBROADCAST_NIP89:
|
|
await nip89_announce_tasks(dvmconfig, client=client)
|
|
|
|
if adminconfig.REBROADCAST_NIP65_RELAY_LIST:
|
|
await nip65_announce_relays(dvmconfig, client=client)
|
|
|
|
if adminconfig.REBROADCAST_NIP88:
|
|
annotier_id = await nip88_announce_tier(dvmconfig, client=client)
|
|
check_and_set_tiereventid_nip88(dvmconfig.IDENTIFIER, adminconfig.INDEX, annotier_id.to_hex())
|
|
|
|
if adminconfig.DELETE_NIP89:
|
|
event_id = EventId.parse(adminconfig.EVENTID).to_hex()
|
|
keys = Keys.parse(
|
|
adminconfig.PRIVKEY) # Private key from sender of Event (e.g. the key of an nip89 announcement you want to delete)
|
|
await fetch_nip89_parameters_for_deletion(keys, event_id, client, dvmconfig, adminconfig.POW)
|
|
|
|
if adminconfig.DELETE_NIP88:
|
|
event_id = EventId.parse(adminconfig.EVENTID).to_hex()
|
|
keys = Keys.parse(
|
|
adminconfig.PRIVKEY) # Private key from sender of Event (e.g. the key of an nip89 announcement you want to delete)
|
|
await fetch_nip88_parameters_for_deletion(keys, event_id, client, dvmconfig)
|
|
|
|
if adminconfig.FETCH_NIP88:
|
|
event_id = EventId.parse(adminconfig.EVENTID).to_hex()
|
|
keys = Keys.parse(
|
|
adminconfig.PRIVKEY)
|
|
await fetch_nip88_event(keys, event_id, client, dvmconfig)
|
|
|
|
if adminconfig.UPDATE_PROFILE:
|
|
await update_profile(dvmconfig, client, lud16=dvmconfig.LN_ADDRESS)
|