feat: añadir soporte para /t web con tokens de login y util crypto
Co-authored-by: aider (openrouter/openai/gpt-5) <aider@aider.chat>webui
parent
1e188c2e96
commit
c8c4fdd927
@ -0,0 +1,20 @@
|
||||
/**
|
||||
* Utilidades criptográficas sin dependencias externas.
|
||||
* - randomTokenBase64Url: genera un token aleatorio (base64url, sin relleno).
|
||||
* - sha256Hex: calcula SHA-256 y devuelve en hex.
|
||||
*/
|
||||
|
||||
export function randomTokenBase64Url(bytes: number = 32): string {
|
||||
const arr = new Uint8Array(bytes);
|
||||
crypto.getRandomValues(arr);
|
||||
const b64 = Buffer.from(arr).toString('base64');
|
||||
// base64url (RFC 4648) sin padding
|
||||
return b64.replace(/\+/g, '-').replace(/\//g, '_').replace(/=+$/g, '');
|
||||
}
|
||||
|
||||
export async function sha256Hex(input: string): Promise<string> {
|
||||
const data = new TextEncoder().encode(input);
|
||||
const hashBuf = await crypto.subtle.digest('SHA-256', data);
|
||||
const bytes = new Uint8Array(hashBuf);
|
||||
return Array.from(bytes).map(b => b.toString(16).padStart(2, '0')).join('');
|
||||
}
|
||||
@ -0,0 +1,131 @@
|
||||
import { describe, test, expect, beforeEach, afterEach } from 'bun:test';
|
||||
import { Database } from 'bun:sqlite';
|
||||
import { initializeDatabase } from '../../../src/db';
|
||||
import { CommandService } from '../../../src/services/command';
|
||||
import { sha256Hex } from '../../../src/utils/crypto';
|
||||
import { Metrics } from '../../../src/services/metrics';
|
||||
|
||||
const envBackup = { ...process.env };
|
||||
let memdb: Database;
|
||||
|
||||
describe('CommandService - /t web (emisión de token de login)', () => {
|
||||
beforeEach(() => {
|
||||
process.env = {
|
||||
...envBackup,
|
||||
NODE_ENV: 'test',
|
||||
TZ: 'Europe/Madrid',
|
||||
WEB_BASE_URL: 'https://app.example.test'
|
||||
};
|
||||
Metrics.reset?.();
|
||||
memdb = new Database(':memory:');
|
||||
initializeDatabase(memdb);
|
||||
(CommandService as any).dbInstance = memdb;
|
||||
});
|
||||
|
||||
afterEach(() => {
|
||||
process.env = envBackup;
|
||||
try { memdb.close(); } catch {}
|
||||
});
|
||||
|
||||
test('DM feliz: devuelve URL con token y persiste hash en web_tokens', async () => {
|
||||
const res = await CommandService.handle({
|
||||
sender: '34600123456',
|
||||
groupId: '34600123456@s.whatsapp.net', // DM (no @g.us)
|
||||
message: '/t web',
|
||||
mentions: []
|
||||
});
|
||||
expect(Array.isArray(res)).toBe(true);
|
||||
expect(res.length).toBe(1);
|
||||
expect(res[0].recipient).toBe('34600123456');
|
||||
expect(res[0].message).toContain('https://app.example.test');
|
||||
|
||||
const m = res[0].message.match(/https?:\/\/\S+/);
|
||||
expect(m).toBeTruthy();
|
||||
const url = new URL(m![0]);
|
||||
expect(url.pathname).toBe('/login');
|
||||
const token = url.searchParams.get('token') || '';
|
||||
expect(token.length).toBeGreaterThan(0);
|
||||
|
||||
const hash = await sha256Hex(token);
|
||||
const row = memdb.prepare(`
|
||||
SELECT user_id, token_hash, used_at, expires_at
|
||||
FROM web_tokens
|
||||
WHERE user_id = ? AND token_hash = ?
|
||||
`).get('34600123456', hash) as any;
|
||||
|
||||
expect(row).toBeTruthy();
|
||||
expect(row.user_id).toBe('34600123456');
|
||||
expect(row.token_hash).toBe(hash);
|
||||
expect(row.used_at).toBeNull();
|
||||
// expires_at debe ser en el futuro
|
||||
const now = new Date();
|
||||
const exp = new Date(String(row.expires_at).replace(' ', 'T') + 'Z');
|
||||
expect(exp.getTime()).toBeGreaterThan(now.getTime() + 9 * 60 * 1000 - 10 * 1000); // ~>= 9min50s
|
||||
expect(Metrics.get('web_tokens_issued_total')).toBe(1);
|
||||
});
|
||||
|
||||
test('En grupo: responde que debe usarse por privado y no inserta token', async () => {
|
||||
const res = await CommandService.handle({
|
||||
sender: '34600123456',
|
||||
groupId: '123@g.us',
|
||||
message: '/t web',
|
||||
mentions: []
|
||||
});
|
||||
expect(res.length).toBe(1);
|
||||
expect(res[0].recipient).toBe('34600123456');
|
||||
expect(res[0].message.toLowerCase()).toContain('privado');
|
||||
|
||||
const cnt = memdb.prepare(`SELECT COUNT(*) AS c FROM web_tokens WHERE user_id = ?`).get('34600123456') as any;
|
||||
expect(Number(cnt.c)).toBe(0);
|
||||
});
|
||||
|
||||
test('Sin WEB_BASE_URL: error claro y no inserta token', async () => {
|
||||
delete process.env.WEB_BASE_URL;
|
||||
|
||||
const res = await CommandService.handle({
|
||||
sender: '34600123456',
|
||||
groupId: '34600123456@s.whatsapp.net',
|
||||
message: '/t web',
|
||||
mentions: []
|
||||
});
|
||||
expect(res.length).toBe(1);
|
||||
expect(res[0].message).toContain('no está configurada');
|
||||
const cnt = memdb.prepare(`SELECT COUNT(*) AS c FROM web_tokens WHERE user_id = ?`).get('34600123456') as any;
|
||||
expect(Number(cnt.c)).toBe(0);
|
||||
});
|
||||
|
||||
test('Token vigente: se invalida y se emite uno nuevo (queda solo 1 activo)', async () => {
|
||||
// Primera emisión
|
||||
{
|
||||
const r1 = await CommandService.handle({
|
||||
sender: '34600123456',
|
||||
groupId: '34600123456@s.whatsapp.net',
|
||||
message: '/t web',
|
||||
mentions: []
|
||||
});
|
||||
expect(r1.length).toBe(1);
|
||||
}
|
||||
|
||||
// Segunda emisión (debe invalidar el anterior)
|
||||
{
|
||||
const r2 = await CommandService.handle({
|
||||
sender: '34600123456',
|
||||
groupId: '34600123456@s.whatsapp.net',
|
||||
message: '/t web',
|
||||
mentions: []
|
||||
});
|
||||
expect(r2.length).toBe(1);
|
||||
}
|
||||
|
||||
const counts = memdb.prepare(`
|
||||
SELECT
|
||||
SUM(CASE WHEN used_at IS NULL THEN 1 ELSE 0 END) AS active,
|
||||
COUNT(*) AS total
|
||||
FROM web_tokens
|
||||
WHERE user_id = ?
|
||||
`).get('34600123456') as any;
|
||||
|
||||
expect(Number(counts.total)).toBeGreaterThanOrEqual(2);
|
||||
expect(Number(counts.active)).toBe(1);
|
||||
});
|
||||
});
|
||||
Loading…
Reference in New Issue