Merge pull request #16 from lnbits/add_test_button

Test Endpoint

__init__.py

@@ -1,5 +1,6 @@
 import asyncio
 from typing import List
+
 from fastapi import APIRouter
 from starlette.staticfiles import StaticFiles
 

helpers.py

@@ -0,0 +1,19 @@
+from bech32 import bech32_decode, convertbits
+
+
+def normalize_public_key(pubkey: str) -> str:
+    if pubkey.startswith("npub1"):
+        _, decoded_data = bech32_decode(pubkey)
+        if not decoded_data:
+            raise ValueError("Public Key is not valid npub")
+
+        decoded_data_bits = convertbits(decoded_data, 5, 8, False)
+        if not decoded_data_bits:
+            raise ValueError("Public Key is not valid npub")
+        return bytes(decoded_data_bits).hex()
+
+    # check if valid hex
+    if len(pubkey) != 64:
+        raise ValueError("Public Key is not valid hex")
+    int(pubkey, 16)
+    return pubkey

models.py

@@ -50,6 +50,16 @@ class Filters(BaseModel):
     __root__: List[Filter]
 
 
+class TestMessage(BaseModel):
+    sender_private_key: Optional[str]
+    reciever_public_key: str
+    message: str
+
+class TestMessageResponse(BaseModel):
+    private_key: str
+    public_key: str
+    event_json: str
+
 # class nostrKeys(BaseModel):
 #     pubkey: str
 #     privkey: str

services.py

@@ -4,6 +4,7 @@
 
 from fastapi import WebSocket, WebSocketDisconnect
 from loguru import logger
+
 from lnbits.helpers import urlsafe_short_hash
 
 from .models import Event, Filter, Filters, Relay, RelayList

tasks.py

@@ -1,6 +1,6 @@
 import asyncio
-import ssl
 import json
+import ssl
 import threading
 
 from .crud import get_relays
@@ -11,8 +11,8 @@
 from .services import (
     nostr,
     received_subscription_eosenotices,
-    received_subscription_notices,
     received_subscription_events,
+    received_subscription_notices,
 )