mirror of
https://github.com/believethehype/nostrdvm.git
synced 2025-11-19 12:27:07 +01:00
85 lines
2.8 KiB
Python
85 lines
2.8 KiB
Python
import os
|
|
from datetime import timedelta
|
|
from hashlib import sha256
|
|
from pathlib import Path
|
|
|
|
import dotenv
|
|
from nostr_sdk import Tag, Keys, EventBuilder, Filter, Alphabet, PublicKey, Event
|
|
|
|
from utils.definitions import EventDefinitions
|
|
from utils.nostr_utils import send_event
|
|
|
|
|
|
class NIP89Config:
|
|
DTAG: str = ""
|
|
NAME: str = ""
|
|
KIND: int = None
|
|
PK: str = ""
|
|
CONTENT: str = ""
|
|
|
|
|
|
def nip89_create_d_tag(name, pubkey, image):
|
|
key_str = str(name + image + pubkey)
|
|
d_tag = sha256(key_str.encode('utf-8')).hexdigest()[:16]
|
|
return d_tag
|
|
|
|
|
|
|
|
def nip89_announce_tasks(dvm_config, client):
|
|
k_tag = Tag.parse(["k", str(dvm_config.NIP89.KIND)])
|
|
d_tag = Tag.parse(["d", dvm_config.NIP89.DTAG])
|
|
keys = Keys.from_sk_str(dvm_config.NIP89.PK)
|
|
content = dvm_config.NIP89.CONTENT
|
|
event = EventBuilder(EventDefinitions.KIND_ANNOUNCEMENT, content, [k_tag, d_tag]).to_event(keys)
|
|
send_event(event, client=client, dvm_config=dvm_config)
|
|
print("Announced NIP 89 for " + dvm_config.NIP89.NAME)
|
|
|
|
|
|
def nip89_fetch_all_dvms(client):
|
|
ktags = []
|
|
for i in range(5000, 5999):
|
|
ktags.append(str(i))
|
|
|
|
filter = Filter().kind(EventDefinitions.KIND_ANNOUNCEMENT).custom_tag(Alphabet.K, ktags)
|
|
events = client.get_events_of([filter], timedelta(seconds=5))
|
|
for event in events:
|
|
print(event.as_json())
|
|
|
|
|
|
def nip89_fetch_events_pubkey(client, pubkey, kind):
|
|
ktags = [str(kind)]
|
|
# for i in range(5000, 5999):
|
|
# ktags.append(str(i))
|
|
nip89filter = (Filter().kind(EventDefinitions.KIND_ANNOUNCEMENT).author(PublicKey.from_hex(pubkey)).
|
|
custom_tag(Alphabet.K, ktags))
|
|
events = client.get_events_of([nip89filter], timedelta(seconds=2))
|
|
|
|
dvms = {}
|
|
for event in events:
|
|
if dvms.get(event.pubkey().to_hex()):
|
|
if dvms.get(event.pubkey().to_hex()).created_at().as_secs() < event.created_at().as_secs():
|
|
dvms[event.pubkey().to_hex()] = event
|
|
else:
|
|
dvms[event.pubkey().to_hex()] = event
|
|
|
|
# should be one element of the kind now
|
|
for dvm in dvms:
|
|
return dvms[dvm].content()
|
|
|
|
|
|
def check_and_set_d_tag(identifier, name, pk, imageurl):
|
|
if not os.getenv("NIP89_DTAG_" + identifier.upper()):
|
|
new_dtag = nip89_create_d_tag(name, Keys.from_sk_str(pk).public_key().to_hex(),
|
|
imageurl)
|
|
nip89_add_dtag_to_env_file("NIP89_DTAG_" + identifier.upper(), new_dtag)
|
|
print("Some new dtag:" + new_dtag)
|
|
return new_dtag
|
|
else:
|
|
return os.getenv("NIP89_DTAG_" + identifier.upper())
|
|
|
|
def nip89_add_dtag_to_env_file(dtag, oskey):
|
|
env_path = Path('.env')
|
|
if env_path.is_file():
|
|
print(f'loading environment from {env_path.resolve()}')
|
|
dotenv.load_dotenv(env_path, verbose=True, override=True)
|
|
dotenv.set_key(env_path, dtag, oskey) |