diff --git a/AUTO_CREDIT_CHANGES.md b/AUTO_CREDIT_CHANGES.md new file mode 100644 index 00000000..408f6987 --- /dev/null +++ b/AUTO_CREDIT_CHANGES.md @@ -0,0 +1,88 @@ +# LNBits Auto-Credit Changes + +## Overview +Modified LNBits server to automatically credit new accounts with 1 million satoshis (1,000,000 sats) when they are created. + +## Changes Made + +### 1. Modified `lnbits/core/services/users.py` + +**Added imports:** +- `get_wallet` from `..crud` +- `update_wallet_balance` from `.payments` + +**Modified `create_user_account_no_ckeck` function:** +- Changed `create_wallet` call to capture the returned wallet object +- Added automatic credit of 1,000,000 sats after wallet creation +- Added error handling and logging for the credit operation + +**Code changes:** +```python +# Before: +await create_wallet( + user_id=account.id, + wallet_name=wallet_name or settings.lnbits_default_wallet_name, +) + +# After: +wallet = await create_wallet( + user_id=account.id, + wallet_name=wallet_name or settings.lnbits_default_wallet_name, +) + +# Credit new account with 1 million satoshis +try: + await update_wallet_balance(wallet, 1_000_000) + logger.info(f"Credited new account {account.id} with 1,000,000 sats") +except Exception as e: + logger.error(f"Failed to credit new account {account.id} with 1,000,000 sats: {e}") +``` + +### 2. Updated Tests in `tests/api/test_auth.py` + +**Modified test functions:** +- `test_register_ok`: Added balance verification for regular user registration +- `test_register_nostr_ok`: Added balance verification for Nostr authentication + +**Added assertions:** +```python +# Check that the wallet has 1 million satoshis +wallet = user.wallets[0] +assert wallet.balance == 1_000_000, f"Expected 1,000,000 sats balance, got {wallet.balance} sats" +``` + +## Affected Account Creation Paths + +The automatic credit will be applied to all new accounts created through: + +1. **Regular user registration** (`/api/v1/auth/register`) +2. **Nostr authentication** (`/api/v1/auth/nostr`) +3. **SSO login** (when new account is created) +4. **API account creation** (`/api/v1/account`) +5. **Admin user creation** (via admin interface) + +## Excluded Paths + +- **Superuser/Admin account creation** (`init_admin_settings`): This function creates the admin account directly and bypasses the user creation flow, so it won't receive the automatic credit. + +## Testing + +To test the changes: + +1. Install dependencies: `poetry install` +2. Run the modified tests: `poetry run pytest tests/api/test_auth.py::test_register_ok -v` +3. Run Nostr test: `poetry run pytest tests/api/test_auth.py::test_register_nostr_ok -v` + +## Logging + +The system will log: +- Success: `"Credited new account {account.id} with 1,000,000 sats"` +- Failure: `"Failed to credit new account {account.id} with 1,000,000 sats: {error}"` + +## Notes + +- The credit uses the existing `update_wallet_balance` function which creates an internal payment record +- The credit is applied after wallet creation but before user extensions are set up +- Error handling ensures that account creation continues even if the credit fails +- The credit amount is hardcoded to 1,000,000 sats (1MM sats) + diff --git a/extensions.json b/extensions.json index 48918bc8..5ef33282 100644 --- a/extensions.json +++ b/extensions.json @@ -14,6 +14,17 @@ "archive": "https://git.atitlan.io/aiolabs/satmachineadmin/archive/v0.0.1.zip", "hash": "e9b61de40e940f05aca460a0560cd3cc8b98f2f0d861e37cef13b046eaec0b7c" }, + { + "id": "satmachineadmin", + "repo": "https://git.atitlan.io/aiolabs/satmachineadmin", + "name": "Satoshi Machine Admin", + "min_lnbits_version": "1.4.0", + "version": "0.0.3", + "short_description": "Admin Dashboard for Satoshi Machine", + "icon": "https://git.atitlan.io/aiolabs/satmachineadmin/raw/branch/main/static/image/aio.png", + "archive": "https://git.atitlan.io/aiolabs/satmachineadmin/archive/v0.0.3.zip", + "hash": "0cf520ee62037298320d8c0caa6f15d89858447d826af53472cb369023eda46d" + }, { "id": "satmachineclient", "repo": "https://git.atitlan.io/aiolabs/satmachineclient", @@ -34,6 +45,16 @@ "archive": "https://git.atitlan.io/aiolabs/events/archive/v0.0.1.zip", "hash": "410ac9c340551aabe88e0dc7393ed30dc7f4703f1345af5e6b02b0ebc5337d52" }, + { + "id": "events", + "repo": "https://git.atitlan.io/aiolabs/events", + "name": "AIO Events", + "version": "0.0.2", + "short_description": "AIO fork of aiolabs/events", + "icon": "https://git.atitlan.io/aiolabs/events/raw/branch/main/static/image/events.png", + "archive": "https://git.atitlan.io/aiolabs/events/archive/v0.0.2.zip", + "hash": "4ac140b97b164f81c482cc19da07a79e961685ae146dc89cba6d12e7d5e80716" + }, { "id": "nostrclient", "repo": "https://git.atitlan.io/aiolabs/nostrclient", @@ -58,11 +79,11 @@ "id": "nostrrelay", "repo": "https://git.atitlan.io/aiolabs/nostrrelay", "name": "AIO Nostrrelay", - "version": "0.0.1", + "version": "0.0.2", "short_description": "AIO fork of aiolabs/nostrrelay", "icon": "https://git.atitlan.io/aiolabs/nostrrelay/raw/branch/main/static/image/nostrrelay.png", - "archive": "https://git.atitlan.io/aiolabs/nostrrelay/archive/v0.0.1.zip", - "hash": "2b1ba6a3b7a79f6595f075c332b82a0ba64c608637ad5bcf23182e40fdf7f68f" + "archive": "https://git.atitlan.io/aiolabs/nostrrelay/archive/v0.0.2.zip", + "hash": "6e353358314b740679e55e2ff98c7024a97c6df8bf27d1784e8cee448407bebd" }, { "id": "castle", diff --git a/lnbits/core/services/users.py b/lnbits/core/services/users.py index d21a8aff..16d6884c 100644 --- a/lnbits/core/services/users.py +++ b/lnbits/core/services/users.py @@ -26,6 +26,7 @@ from ..crud import ( get_super_settings, get_user_extensions, get_user_from_account, + get_wallet, update_account, update_super_user, update_user_extension, @@ -37,6 +38,7 @@ from ..models import ( UserExtra, ) from .settings import update_cached_settings +from .payments import update_wallet_balance async def create_user_account( @@ -80,6 +82,13 @@ async def create_user_account_no_ckeck( conn=conn, ) + # Credit new account with 1 million satoshis + try: + await update_wallet_balance(wallet, 1_000_000, conn=conn) + logger.info(f"Credited new account {account.id} with 1,000,000 sats") + except Exception as e: + logger.error(f"Failed to credit new account {account.id} with 1,000,000 sats: {e}") + user_extensions = (default_exts or []) + settings.lnbits_user_default_extensions for ext_id in user_extensions: try: diff --git a/lnbits/wallets/lndrest.py b/lnbits/wallets/lndrest.py index 861448d6..da72a61f 100644 --- a/lnbits/wallets/lndrest.py +++ b/lnbits/wallets/lndrest.py @@ -178,7 +178,7 @@ class LndRestWallet(Wallet): "no_inflight_updates": True, } if settings.lnd_rest_allow_self_payment: - req["allow_self_payment"] = 1 + req["allow_self_payment"] = True try: r = await self.client.post( diff --git a/tests/api/test_auth.py b/tests/api/test_auth.py index 57cacd03..8b1e7ba2 100644 --- a/tests/api/test_auth.py +++ b/tests/api/test_auth.py @@ -295,6 +295,10 @@ async def test_register_ok(http_client: AsyncClient): assert ( len(user.wallets) == 1 ), f"Expected 1 default wallet, not {len(user.wallets)}." + + # Check that the wallet has 1 million satoshis + wallet = user.wallets[0] + assert wallet.balance == 1_000_000, f"Expected 1,000,000 sats balance, got {wallet.balance} sats" @pytest.mark.anyio @@ -588,6 +592,10 @@ async def test_register_nostr_ok(http_client: AsyncClient, settings: Settings): assert ( len(user.wallets) == 1 ), f"Expected 1 default wallet, not {len(user.wallets)}." + + # Check that the wallet has 1 million satoshis + wallet = user.wallets[0] + assert wallet.balance == 1_000_000, f"Expected 1,000,000 sats balance, got {wallet.balance} sats" @pytest.mark.anyio