highperfocused/lnbits
1
0
Fork 0
mirror of https://github.com/lnbits/lnbits.git synced 2025-09-20 13:04:23 +02:00
Code Issues Packages Projects Releases Wiki Activity

FakeWallet: provide payment preimage

Browse Source
This commit is contained in:
callebtc
2023-11-13 09:06:59 -03:00
committed by Pavol Rusnak
parent f989eba2c9
commit 3e91200d3f
1 changed files with 23 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
lnbits/wallets/fake.py
View File

@@ -1,9 +1,8 @@
import asyncio import asyncio
import hashlib import hashlib
import random
from datetime import datetime from datetime import datetime
from os import urandom from os import urandom
from typing import AsyncGenerator, Optional from typing import AsyncGenerator, Dict, Optional, Set
from bolt11 import ( from bolt11 import (
Bolt11, Bolt11,
@@ -29,6 +28,8 @@ from .base import (
class FakeWallet(Wallet): class FakeWallet(Wallet):
queue: asyncio.Queue = asyncio.Queue(0) queue: asyncio.Queue = asyncio.Queue(0)
payment_secrets: Dict[str, str] = dict()
paid_invoices: Set[str] = set()
secret: str = settings.fake_wallet_secret secret: str = settings.fake_wallet_secret
privkey: str = hashlib.pbkdf2_hmac( privkey: str = hashlib.pbkdf2_hmac(
"sha256", "sha256",
@@ -70,20 +71,18 @@ class FakeWallet(Wallet):
if expiry: if expiry:
tags.add(TagChar.expire_time, expiry) tags.add(TagChar.expire_time, expiry)
# random hash
checking_id = (
self.privkey[:6]
+ hashlib.sha256(str(random.getrandbits(256)).encode()).hexdigest()[6:]
)
tags.add(TagChar.payment_hash, checking_id)
if payment_secret: if payment_secret:
secret = payment_secret.hex() secret = payment_secret.hex()
else: else:
secret = urandom(32).hex() secret = urandom(32).hex()
tags.add(TagChar.payment_secret, secret) tags.add(TagChar.payment_secret, secret)
payment_hash = hashlib.sha256(secret.encode()).hexdigest()
tags.add(TagChar.payment_hash, payment_hash)
self.payment_secrets[payment_hash] = secret
bolt11 = Bolt11( bolt11 = Bolt11(
currency="bc", currency="bc",
amount_msat=MilliSatoshi(amount * 1000), amount_msat=MilliSatoshi(amount * 1000),
@@ -93,7 +92,9 @@ class FakeWallet(Wallet):
payment_request = encode(bolt11, self.privkey) payment_request = encode(bolt11, self.privkey)
return InvoiceResponse(True, checking_id, payment_request) return InvoiceResponse(
ok=True, checking_id=payment_hash, payment_request=payment_request
)
async def pay_invoice(self, bolt11: str, _: int) -> PaymentResponse: async def pay_invoice(self, bolt11: str, _: int) -> PaymentResponse:
try: try:
@@ -101,16 +102,23 @@ class FakeWallet(Wallet):
except Bolt11Exception as exc: except Bolt11Exception as exc:
return PaymentResponse(ok=False, error_message=str(exc)) return PaymentResponse(ok=False, error_message=str(exc))
if invoice.payment_hash[:6] == self.privkey[:6]: if invoice.payment_hash in self.payment_secrets:
await self.queue.put(invoice) await self.queue.put(invoice)
return PaymentResponse(True, invoice.payment_hash, 0) self.paid_invoices.add(invoice.payment_hash)
return PaymentResponse(
ok=True,
checking_id=invoice.payment_hash,
fee_msat=0,
preimage=self.payment_secrets.get(invoice.payment_hash) or "0" * 64,
)
else: else:
return PaymentResponse( return PaymentResponse(
ok=False, error_message="Only internal invoices can be used!" ok=False, error_message="Only internal invoices can be used!"
) )
async def get_invoice_status(self, _: str) -> PaymentStatus: async def get_invoice_status(self, checking_id: str) -> PaymentStatus:
return PaymentStatus(None) paid = checking_id in self.paid_invoices
return PaymentStatus(paid)
async def get_payment_status(self, _: str) -> PaymentStatus: async def get_payment_status(self, _: str) -> PaymentStatus:
return PaymentStatus(None) return PaymentStatus(None)