1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
// iOS drops a web push subscription after three pushes that showed no
// notification, and nothing used to make a new one: the wallet's background
// switch stayed on, getSubscription() came back empty, and the notifier kept
// a dead endpoint, so NWC requests stopped waking the closed app until the
// user toggled it off and on (reported by PullCap). Opening the app, or
// coming back to it, must subscribe again and tell the notifier.
//
// Run: bun tools/push-resubscribe-test.js
import puppeteer from './puppeteer.js';
import { generateMnemonic } from '@scure/bip39';
import { wordlist } from '@scure/bip39/wordlists/english';
import { cacheKeyFor } from '../src/wallet.js';
import { buildHtml } from '../build.js';
const sleep = (ms) => new Promise((r) => setTimeout(r, ms));
let ok = true;
const check = (n, c, d = '') => { console.log(` ${c ? '✓' : '✗'} ${n}${d ? ' — ' + d : ''}`); if (!c) ok = false; };
const html = await buildHtml({ minify: true, pwa: false });
const server = Bun.serve({ port: 5263, fetch: () => new Response(html, { headers: { 'content-type': 'text/html' } }) });
const browser = await puppeteer.launch({ executablePath: '/usr/bin/google-chrome', headless: 'new', args: ['--no-sandbox'] });
const page = await browser.newPage();
const KEY = 'BP' + 'A'.repeat(85); // any 65-byte key, base64url
await page.evaluateOnNewDocument((KEY) => {
const bytes = () => Uint8Array.from(atob(KEY.replace(/-/g, '+').replace(/_/g, '/')), (c) => c.charCodeAt(0));
window.__subs = 0; window.__regs = [];
let sub = null; // iOS has already dropped it
const make = () => ({
endpoint: 'https://web.push.apple.com/sub' + window.__subs,
options: { applicationServerKey: bytes().buffer },
toJSON() { return { endpoint: this.endpoint, keys: { p256dh: 'x', auth: 'y' } }; },
unsubscribe: () => { sub = null; return Promise.resolve(true); },
});
window.__drop = () => { sub = null; };
const reg = {
pushManager: {
getSubscription: () => Promise.resolve(sub),
subscribe: () => { window.__subs++; sub = make(); return Promise.resolve(sub); },
},
showNotification: () => Promise.resolve(),
};
Object.defineProperty(navigator.serviceWorker, 'ready', { get: () => Promise.resolve(reg) });
Notification.requestPermission = () => Promise.resolve('granted');
Object.defineProperty(Notification, 'permission', { get: () => 'granted' });
const f = window.fetch;
window.fetch = async (u, o) => {
const url = String(u);
if (url.startsWith('https://nwcpush.coinos.io/vapid')) return new Response(JSON.stringify({ publicKey: KEY }));
if (url.startsWith('https://nwcpush.coinos.io/register')) {
const b = JSON.parse(o.body);
if (b.servicePubkeys) window.__regs.push(b.subscription.endpoint);
return new Response('{"ok":true}');
}
return f(u, o);
};
}, KEY);
const click = (t) => page.evaluate((x) => { const e = [...document.querySelectorAll('button')].find((n) => n.textContent.trim().toLowerCase().includes(x)); if (e) { e.click(); return true; } return false; }, t);
const waitText = async (x, ms = 25000) => { for (let i = 0; i < ms/250; i++) { if ((await page.evaluate(() => document.body.innerText)).toLowerCase().includes(x)) return true; await sleep(250); } return false; };
try {
await page.setViewport({ width: 390, height: 844 });
await page.goto('http://localhost:5263/', { waitUntil: 'domcontentloaded' });
await page.evaluate(() => localStorage.setItem('btc-wallet-network', 'regtest'));
await page.reload({ waitUntil: 'domcontentloaded' });
await sleep(400);
await click('create a new wallet'); await sleep(500);
await click('to import'); await sleep(400);
await page.waitForSelector('textarea');
const mn = generateMnemonic(wordlist);
await page.type('textarea', mn);
await click('open wallet');
await waitText('receive', 20000);
const base = cacheKeyFor(mn + '\n');
// a wallet answering in the background, whose subscription iOS took away
await page.evaluate((k) => localStorage.setItem(k + ':nwc', JSON.stringify({
conns: [{ id: 'x', name: 'test', servicePk: 'a'.repeat(64), sk: 'b'.repeat(64), maxSat: 1000, budget: {} }],
background: true,
})), base);
await page.reload({ waitUntil: 'domcontentloaded' });
await waitText('receive', 20000);
await sleep(3000);
let s = await page.evaluate(() => ({ subs: window.__subs, regs: window.__regs }));
check('opening the app subscribes again', s.subs >= 1, `${s.subs} subscribe()`);
check('...and registers the new endpoint for the connection', s.regs.includes('https://web.push.apple.com/sub' + s.subs), s.regs.join(' '));
// dropped again while the app sat in the background; it resumes, no reload
await page.evaluate(() => { window.__drop(); window.__regs.length = 0; });
await page.evaluate(() => {
Object.defineProperty(document, 'visibilityState', { configurable: true, get: () => 'visible' });
document.dispatchEvent(new Event('visibilitychange'));
});
await sleep(3000);
const before = s.subs;
s = await page.evaluate(() => ({ subs: window.__subs, regs: window.__regs }));
check('coming back to the app subscribes again', s.subs > before, `${s.subs} subscribe()`);
check('...and registers it', s.regs.includes('https://web.push.apple.com/sub' + s.subs), s.regs.join(' '));
// a live subscription is left alone
await page.evaluate(() => { window.__regs.length = 0; document.dispatchEvent(new Event('visibilitychange')); });
await sleep(3000);
const again = await page.evaluate(() => ({ subs: window.__subs, regs: window.__regs }));
check('a subscription still there is reused', again.subs === s.subs && again.regs.length >= 1);
} finally { await browser.close(); server.stop(true); }
console.log(ok ? '\n✅ a dropped push subscription comes back' : '\n❌ failed');
process.exit(ok ? 0 : 1);
|
