lnurlp: support success_text and success_url.

lnbits/extensions/lnurlp/models.py

@@ -1,9 +1,11 @@
 import json
+from urllib.parse import urlparse, urlunparse, parse_qs, urlencode, ParseResult
 from quart import url_for
+from typing import NamedTuple, Optional, Dict
+from sqlite3 import Row
 from lnurl import Lnurl, encode as lnurl_encode
 from lnurl.types import LnurlPayMetadata
-from sqlite3 import Row
-from typing import NamedTuple
+from lnurl.models import LnurlPaySuccessAction, MessageAction, UrlAction
 
 
 class PayLink(NamedTuple):
@@ -31,6 +33,18 @@ class PayLink(NamedTuple):
     def lnurlpay_metadata(self) -> LnurlPayMetadata:
         return LnurlPayMetadata(json.dumps([["text/plain", self.description]]))
 
+    def success_action(self, payment_hash: str) -> Optional[LnurlPaySuccessAction]:
+        if self.success_url:
+            url: ParseResult = urlparse(self.success_url)
+            qs: Dict = parse_qs(url.query)
+            qs["payment_hash"] = payment_hash
+            url = url._replace(query=urlencode(qs))
+            return UrlAction(url=urlunparse(url), description=self.success_text)
+        elif self.success_text:
+            return MessageAction(message=self.success_text)
+        else:
+            return None
+
 
 class Invoice(NamedTuple):
     payment_hash: str