From 577a3932f5bac8314c7da9136e81e16fcb1c7908 Mon Sep 17 00:00:00 2001
From: bitkarrot <73979971+bitkarrot@users.noreply.github.com>
Date: Thu, 2 Mar 2023 15:51:42 -0800
Subject: [PATCH]  initial crud, index form

---
 README.md                   |  4 ++-
 crud.py                     | 62 +++++++++++++++++++++++++++++++++++--
 static/js/index.js          |  6 +++-
 templates/lnurlp/index.html | 16 ++++++++++
 4 files changed, 83 insertions(+), 5 deletions(-)

diff --git a/README.md b/README.md
index 515cf90..ec603ec 100644
--- a/README.md
+++ b/README.md
@@ -26,4 +26,6 @@ LNURL is a range of lightning-network standards that allow us to use lightning-n
    - you can now open your LNURLp and copy the LNURL, get the shareable link or print it\
      ![view lnurlp](https://i.imgur.com/4n41S7T.jpg)
 
-3. Optional - add Lightning Address
\ No newline at end of file
+3. Optional - add Lightning Address
+   - attach a username to your lnurlp to create a lightning address
+   - the LN address format will be username@lnbits-domain-name
diff --git a/crud.py b/crud.py
index 4acb4a4..2362601 100644
--- a/crud.py
+++ b/crud.py
@@ -1,10 +1,12 @@
+import re
 from typing import List, Optional, Union
 
 from lnbits.helpers import urlsafe_short_hash
 
-from . import db
+from . import db, maindb
 from .models import CreatePayLinkData, PayLink
 
+from loguru import logger
 
 async def create_pay_link(data: CreatePayLinkData, wallet_id: str) -> PayLink:
     link_id = urlsafe_short_hash()[:6]
@@ -26,9 +28,11 @@ async def create_pay_link(data: CreatePayLinkData, wallet_id: str) -> PayLink:
             success_url,
             comment_chars,
             currency,
-            fiat_base_multiplier
+            fiat_base_multiplier,
+            username
+
         )
-        VALUES (?, ?, ?, ?, ?, 0, 0, ?, ?, ?, ?, ?, ?, ?, ?)
+        VALUES (?, ?, ?, ?, ?, 0, 0, ?, ?, ?, ?, ?, ?, ?, ?, ?)
         """,
         (
             link_id,
@@ -44,6 +48,7 @@ async def create_pay_link(data: CreatePayLinkData, wallet_id: str) -> PayLink:
             data.comment_chars,
             data.currency,
             data.fiat_base_multiplier,
+            data.username,
         ),
     )
     assert result
@@ -53,6 +58,57 @@ async def create_pay_link(data: CreatePayLinkData, wallet_id: str) -> PayLink:
     return link
 
 
+
+async def check_lnaddress_update(username: str, id: str) -> bool:
+    # check no duplicates for lnaddress when updating an username
+    row = await db.fetchall(
+        "SELECT username FROM lnurlp.pay_links WHERE username = ? AND id = ?",
+        (username, id),
+    )
+    logger.info("number of rows from username search")
+    logger.info(len(row))
+    if len(row) > 1:
+        assert False, "Lightning Address Already exists. Try a different One?"
+        return
+    else:
+        return True
+
+
+async def check_lnaddress_exists(username: str) -> bool:
+    # check if lnaddress username exists in the database when creating a new entry
+    row = await db.fetchall(
+        "SELECT username FROM lnurlp.pay_links WHERE username = ?", (username,)
+    )
+    logger.info("number of rows from lnaddress search")
+    if row:
+        assert False, "Lighting Address Already exists. Try a different One?"
+        return
+    else:
+        return True
+
+
+async def check_lnaddress_format(username: str) -> bool:
+    # check username complies with lnaddress specification
+    if not re.match("^[a-z0-9-_.]{3,15}$", username):
+        assert False, "Only letters a-z0-9-_. allowed, min 3 and max 15 characters!"
+        return
+    return True
+
+async def get_wallet_key(wallet_id: str) -> str:
+    row = await maindb.fetchone("SELECT inkey FROM wallets WHERE id = ?", (wallet_id,))
+    if row is not None:
+        return row[0]
+    else:
+        assert False, "Cannot locate wallet invoice key"
+        return
+    
+async def get_address_data(username: str) -> Optional[PayLink]:
+    row = await db.fetchone(
+        "SELECT * FROM lnurl.pay_links WHERE username = ?", (username,)
+    )
+    return PayLink.from_row(row) if row else None
+
+
 async def get_pay_link(link_id: str) -> Optional[PayLink]:
     row = await db.fetchone("SELECT * FROM lnurlp.pay_links WHERE id = ?", (link_id,))
     return PayLink.from_row(row) if row else None
diff --git a/static/js/index.js b/static/js/index.js
index c1372be..0075f28 100644
--- a/static/js/index.js
+++ b/static/js/index.js
@@ -9,6 +9,8 @@ var locationPath = [
   window.location.pathname
 ].join('')
 
+var domain = window.location.hostname
+
 var mapPayLink = obj => {
   obj._data = _.clone(obj)
   obj.date = Quasar.utils.date.formatDate(
@@ -26,6 +28,7 @@ new Vue({
   mixins: [windowMixin],
   data() {
     return {
+      domain: domain,
       currencies: [],
       fiatRates: {},
       checker: null,
@@ -137,7 +140,8 @@ new Vue({
           'success_text',
           'success_url',
           'comment_chars',
-          'currency'
+          'currency',
+          'username'
         ),
         (value, key) =>
           (key === 'webhook_url' ||
diff --git a/templates/lnurlp/index.html b/templates/lnurlp/index.html
index 3fbd344..f1ffacc 100644
--- a/templates/lnurlp/index.html
+++ b/templates/lnurlp/index.html
@@ -29,6 +29,7 @@
             <q-tr :props="props">
               <q-th auto-width></q-th>
               <q-th auto-width>Description</q-th>
+              <q-th auto-width>Lightning Address</q-th>
               <q-th auto-width>Amount</q-th>
               <q-th auto-width>Currency</q-th>
               <q-th auto-width></q-th>
@@ -148,6 +149,21 @@
           label="Wallet *"
         >
         </q-select>
+        <div class="row">
+          <div class="col">
+           <q-input
+            filled
+            dense
+            v-model.trim="formDialog.data.lnaddress"
+            type="text"
+            label="Lightning Username"
+          >
+          </div>
+          <div class="col"  style="margin-top: 10px">
+            <span class="label"> @{% raw %} {{domain}} {% endraw %} </span>
+          </div>
+          </q-input>
+        </div>
         <q-input
           filled
           dense