@pasko70/pibo 1.9.7 → 1.9.9

package/dist/tools/browser-use-leases.js

@@ -1,516 +0,0 @@
-import { spawn } from 'node:child_process';
-import { existsSync, readFileSync } from 'node:fs';
-import { cp, mkdir, rm, writeFile } from 'node:fs/promises';
-import { join, resolve } from 'node:path';
-import { ErrorCode, formatCliError } from '../cli-errors.js';
-import { BROWSER_USE_DEFAULT_PROFILE } from './browser-use-wrapper.js';
-import { browserPoolPaths, loadBrowserPoolState, releaseBrowserPoolLease, saveBrowserPoolState, withBrowserPoolLock, } from './browser-pool.js';
-const REGISTRY_VERSION = 1;
-const DEFAULT_APP = 'pibo-chat';
-const DEFAULT_TTL_MINUTES = 8 * 60;
-const DEFAULT_TEMPLATE_PROFILE = 'pibo-auth-template';
-const AUTH_POOL_DIR = 'auth-pool';
-function shellQuote(value) {
-    return `'${value.replaceAll("'", "'\\''")}'`;
-}
-function nowIso() {
-    return new Date().toISOString();
-}
-function addMinutes(date, minutes) {
-    return new Date(date.getTime() + minutes * 60_000);
-}
-function sanitizeIdPart(value) {
-    return value
-        .trim()
-        .toLowerCase()
-        .replace(/[^a-z0-9_.-]+/g, '-')
-        .replace(/^-+|-+$/g, '') || 'default';
-}
-function registryPath(status) {
-    return join(status.homeDir, AUTH_POOL_DIR, 'leases.json');
-}
-function lockDir(status) {
-    return join(status.homeDir, AUTH_POOL_DIR, '.leases.lock');
-}
-function appPoolDir(status, app) {
-    return join(status.homeDir, AUTH_POOL_DIR, sanitizeIdPart(app));
-}
-function browserUseHome(status) {
-    return status.homeDir;
-}
-function browserPoolWorkerId() {
-    return process.env.PIBO_BROWSER_POOL_WORKER_ID || process.env.PIBO_COMPUTE_WORKER_ID || process.env.HOSTNAME || 'local';
-}
-function browserPoolId() {
-    return process.env.PIBO_BROWSER_POOL_ID || 'default';
-}
-function browserPoolRootDir(status) {
-    return process.env.PIBO_BROWSER_POOL_ROOT || join(browserUseHome(status), 'pibo-browser-pool');
-}
-function browserPoolLeaseIdForSession(sessionName) {
-    return `browser-use:${safeSessionName(sessionName)}`;
-}
-function browserPoolIdentityForLease(status, lease) {
-    return {
-        workerId: lease.browserPoolWorkerId || browserPoolWorkerId(),
-        poolId: lease.browserPoolId || browserPoolId(),
-    };
-}
-function browserPoolRootForLease(status, lease) {
-    return lease.browserPoolRootDir || browserPoolRootDir(status);
-}
-function authTemplateDir(status) {
-    if (process.env.PIBO_BROWSER_USE_AUTH_TEMPLATE_DIR) {
-        return process.env.PIBO_BROWSER_USE_AUTH_TEMPLATE_DIR;
-    }
-    return join(status.homeDir, 'chrome-profiles', DEFAULT_TEMPLATE_PROFILE);
-}
-function defaultTemplateDir(status) {
-    const preferred = authTemplateDir(status);
-    if (existsSync(preferred))
-        return preferred;
-    // Never fall back to the active default profile (it may be running).
-    // Create an empty template directory so leases get clean profiles.
-    return preferred;
-}
-export function readRegistry(status) {
-    const path = registryPath(status);
-    if (!existsSync(path))
-        return { version: REGISTRY_VERSION, leases: [] };
-    const parsed = JSON.parse(readFileSync(path, 'utf-8'));
-    if (!parsed || typeof parsed !== 'object' || !Array.isArray(parsed.leases)) {
-        throw new Error(`Invalid browser-use lease registry: ${path}`);
-    }
-    return {
-        version: REGISTRY_VERSION,
-        leases: parsed.leases,
-    };
-}
-async function writeRegistry(status, registry) {
-    const path = registryPath(status);
-    await mkdir(join(status.homeDir, AUTH_POOL_DIR), { recursive: true });
-    await writeFile(path, `${JSON.stringify({ ...registry, version: REGISTRY_VERSION }, null, 2)}\n`);
-}
-async function withRegistryLock(status, action) {
-    const lock = lockDir(status);
-    await mkdir(join(status.homeDir, AUTH_POOL_DIR), { recursive: true });
-    let acquired = false;
-    for (let attempt = 0; attempt < 50; attempt += 1) {
-        try {
-            await mkdir(lock);
-            acquired = true;
-            break;
-        }
-        catch (error) {
-            if (error.code !== 'EEXIST')
-                throw error;
-            await new Promise((resolve) => setTimeout(resolve, 50));
-        }
-    }
-    if (!acquired) {
-        throw new Error('Timed out waiting for browser-use lease registry lock');
-    }
-    try {
-        return await action();
-    }
-    finally {
-        await rm(lock, { recursive: true, force: true });
-    }
-}
-export function isExpired(lease, now = new Date()) {
-    return new Date(lease.expiresAt).getTime() <= now.getTime();
-}
-function safeSessionName(sessionName) {
-    return sessionName.replace(/[^A-Za-z0-9_.-]/g, '_');
-}
-function cdpPidFile(status, sessionName) {
-    return join(status.homeDir, 'pibo-cdp', `${safeSessionName(sessionName)}.pid`);
-}
-function processIsAlive(pid) {
-    try {
-        process.kill(pid, 0);
-        return true;
-    }
-    catch {
-        return false;
-    }
-}
-function readPid(path) {
-    try {
-        const value = Number.parseInt(readFileSync(path, 'utf-8'), 10);
-        return Number.isFinite(value) && value > 0 ? value : undefined;
-    }
-    catch {
-        return undefined;
-    }
-}
-function leaseProcessIsAlive(status, lease) {
-    const pid = readPid(cdpPidFile(status, lease.sessionName));
-    return pid ? processIsAlive(pid) : false;
-}
-function killLeaseProcess(status, lease) {
-    const pid = readPid(cdpPidFile(status, lease.sessionName));
-    if (!pid)
-        return;
-    try {
-        process.kill(pid, 'SIGTERM');
-    }
-    catch {
-        // Process already exited.
-    }
-}
-async function releaseAssociatedBrowserPoolLease(status, lease) {
-    if (!lease.browserPoolLeaseId)
-        return;
-    const identity = browserPoolIdentityForLease(status, lease);
-    const paths = browserPoolPaths(browserPoolRootForLease(status, lease), identity);
-    await releaseBrowserPoolLease(paths, identity, { leaseId: lease.browserPoolLeaseId });
-}
-async function reapAssociatedBrowserPoolLeaseProcess(status, lease) {
-    if (!lease.browserPoolLeaseId)
-        return false;
-    const identity = browserPoolIdentityForLease(status, lease);
-    const paths = browserPoolPaths(browserPoolRootForLease(status, lease), identity);
-    return withBrowserPoolLock(paths.lockPath, { holder: `auth-lease-reap:${lease.id}` }, async () => {
-        const state = await loadBrowserPoolState(paths.statePath, { ...identity, onMissing: 'empty', onMalformed: 'empty' });
-        if (!browserPoolStateMatchesAuthLease(state, lease))
-            return false;
-        const terminated = await terminateManagedBrowserProcess(state);
-        const reapedAt = nowIso();
-        await saveBrowserPoolState(paths.statePath, {
-            ...state,
-            state: terminated ? 'stale' : state.state,
-            activeLeaseId: undefined,
-            activeLeaseCount: 0,
-            holder: undefined,
-            lastUsedAt: reapedAt,
-            idleExpiresAt: undefined,
-            cleanupStatus: terminated ? 'success' : 'skipped',
-            lastError: terminated ? 'Expired auth lease reaped managed browser process' : 'No live managed browser process to reap',
-        });
-        return terminated;
-    });
-}
-function browserPoolStateMatchesAuthLease(state, lease) {
-    return state.state === 'leased'
-        && state.activeLeaseId === lease.browserPoolLeaseId
-        && Boolean(state.userDataDir)
-        && resolve(state.userDataDir) === resolve(lease.userDataDir);
-}
-async function terminateManagedBrowserProcess(state) {
-    if (!state.pid || !processIsAlive(state.pid))
-        return false;
-    const target = state.processGroupId && state.processGroupId === state.pid ? -state.processGroupId : state.pid;
-    try {
-        process.kill(target, 'SIGTERM');
-    }
-    catch {
-        return false;
-    }
-    if (!(await waitForProcessExit(state.pid, 1000))) {
-        try {
-            process.kill(target, 'SIGKILL');
-        }
-        catch {
-            // Process may have exited after TERM.
-        }
-        await waitForProcessExit(state.pid, 1000);
-    }
-    return !processIsAlive(state.pid);
-}
-async function waitForProcessExit(pid, timeoutMs) {
-    const startedAt = Date.now();
-    while (Date.now() - startedAt < timeoutMs) {
-        if (!processIsAlive(pid))
-            return true;
-        await new Promise((resolve) => setTimeout(resolve, 50));
-    }
-    return !processIsAlive(pid);
-}
-async function copyTemplateProfile(templateDir, userDataDir) {
-    if (existsSync(templateDir)) {
-        assertTemplateIsNotRunning(templateDir);
-        await rm(userDataDir, { recursive: true, force: true });
-        await mkdir(userDataDir, { recursive: true });
-        await cp(templateDir, userDataDir, {
-            recursive: true,
-            filter: (source) => {
-                const base = source.split(/[\\/]/).pop() ?? '';
-                return !base.startsWith('Singleton') && base !== 'DevToolsActivePort';
-            },
-        });
-        return;
-    }
-    // Template does not exist yet: create a clean empty profile directory.
-    // Chrome will initialise a fresh profile on first start.
-    await rm(userDataDir, { recursive: true, force: true });
-    await mkdir(userDataDir, { recursive: true });
-}
-function assertTemplateIsNotRunning(templateDir) {
-    for (const fileName of ['SingletonLock', 'SingletonCookie', 'SingletonSocket']) {
-        if (!existsSync(join(templateDir, fileName)))
-            continue;
-        throw new Error(formatCliError({
-            code: ErrorCode.CLIENT_ERROR,
-            type: 'BROWSER_USE_AUTH_TEMPLATE_RUNNING',
-            message: `Browser-use auth template appears to be running: ${templateDir}`,
-            details: `Found ${fileName}`,
-            suggestion: 'Close the template browser before acquiring leases, or pass --template-dir for a closed template profile.',
-        }));
-    }
-}
-function nextSlotNumber(leases, app) {
-    const prefix = `${sanitizeIdPart(app)}-slot-`;
-    const used = new Set(leases
-        .filter((lease) => lease.app === app)
-        .map((lease) => Number.parseInt(lease.id.slice(prefix.length), 10))
-        .filter((value) => Number.isFinite(value)));
-    for (let value = 1; value < 10_000; value += 1) {
-        if (!used.has(value))
-            return value;
-    }
-    throw new Error(`No browser-use slots available for ${app}`);
-}
-function createLease(status, app, holder, ttlMinutes, profileName, leases) {
-    const slot = nextSlotNumber(leases, app);
-    const slotName = `slot-${String(slot).padStart(3, '0')}`;
-    const appId = sanitizeIdPart(app);
-    const id = `${appId}-${slotName}`;
-    const createdAt = nowIso();
-    const sessionName = `pibo-auth-${id}`;
-    return {
-        id,
-        app,
-        holder,
-        sessionName,
-        userDataDir: join(appPoolDir(status, app), slotName),
-        profileName,
-        status: 'active',
-        browserPoolLeaseId: browserPoolLeaseIdForSession(sessionName),
-        browserPoolWorkerId: browserPoolWorkerId(),
-        browserPoolId: browserPoolId(),
-        browserPoolRootDir: browserPoolRootDir(status),
-        createdAt,
-        updatedAt: createdAt,
-        expiresAt: addMinutes(new Date(createdAt), ttlMinutes).toISOString(),
-    };
-}
-function selectReusableLease(status, leases, app) {
-    return leases.find((lease) => lease.app === app &&
-        (lease.status === 'released' || isExpired(lease)) &&
-        !leaseProcessIsAlive(status, lease));
-}
-function printLeaseEnv(status, lease) {
-    console.log(`export BROWSER_USE_HOME=${shellQuote(browserUseHome(status))}`);
-    console.log(`export PIBO_BROWSER_USE_LEASE_ID=${shellQuote(lease.id)}`);
-    console.log(`export PIBO_BROWSER_USE_SESSION=${shellQuote(lease.sessionName)}`);
-    console.log(`export PIBO_BROWSER_USE_CHROME_USER_DATA_DIR=${shellQuote(lease.userDataDir)}`);
-    console.log(`export PIBO_BROWSER_USE_DEFAULT_PROFILE=${shellQuote(lease.profileName)}`);
-    if (lease.browserPoolLeaseId)
-        console.log(`export PIBO_BROWSER_POOL_LEASE_ID=${shellQuote(lease.browserPoolLeaseId)}`);
-    console.log(`# browser-use --session "$PIBO_BROWSER_USE_SESSION" state`);
-}
-function printLeaseJson(status, lease) {
-    console.log(JSON.stringify({
-        ...lease,
-        browserUseHome: browserUseHome(status),
-        exports: {
-            BROWSER_USE_HOME: browserUseHome(status),
-            PIBO_BROWSER_USE_LEASE_ID: lease.id,
-            PIBO_BROWSER_USE_SESSION: lease.sessionName,
-            PIBO_BROWSER_USE_CHROME_USER_DATA_DIR: lease.userDataDir,
-            PIBO_BROWSER_USE_DEFAULT_PROFILE: lease.profileName,
-            ...(lease.browserPoolLeaseId ? { PIBO_BROWSER_POOL_LEASE_ID: lease.browserPoolLeaseId } : {}),
-        },
-    }, null, 2));
-}
-export async function acquireBrowserUseLease(context, options = {}) {
-    const app = options.app?.trim() || DEFAULT_APP;
-    const holder = options.holder?.trim() || process.env.PIBO_BROWSER_USE_LEASE_HOLDER || process.env.USER || 'unknown';
-    const ttlMinutes = options.ttlMinutes ?? DEFAULT_TTL_MINUTES;
-    const profileName = options.profileName?.trim() || BROWSER_USE_DEFAULT_PROFILE;
-    const templateDir = options.templateDir?.trim() || defaultTemplateDir(context.status);
-    await withRegistryLock(context.status, async () => {
-        const registry = readRegistry(context.status);
-        const reapedCount = await reapStaleLeasesInRegistry(context.status, registry);
-        let lease = selectReusableLease(context.status, registry.leases, app);
-        if (lease) {
-            await copyTemplateProfile(templateDir, lease.userDataDir);
-            lease.holder = holder;
-            lease.status = 'active';
-            lease.profileName = profileName;
-            lease.browserPoolLeaseId = browserPoolLeaseIdForSession(lease.sessionName);
-            lease.browserPoolWorkerId = browserPoolWorkerId();
-            lease.browserPoolId = browserPoolId();
-            lease.browserPoolRootDir = browserPoolRootDir(context.status);
-            lease.updatedAt = nowIso();
-            lease.expiresAt = addMinutes(new Date(), ttlMinutes).toISOString();
-        }
-        else {
-            const activeCount = registry.leases.filter((item) => item.app === app &&
-                item.status === 'active' &&
-                !isExpired(item)).length;
-            if (options.maxSlots !== undefined && activeCount >= options.maxSlots) {
-                throw new Error(formatCliError({
-                    code: ErrorCode.CLIENT_ERROR,
-                    type: 'BROWSER_USE_AUTH_POOL_EXHAUSTED',
-                    message: `No browser-use auth slots available for ${app}`,
-                    details: `active=${activeCount} max=${options.maxSlots}`,
-                    suggestion: 'Release a lease or rerun with a higher --max-slots value.',
-                }));
-            }
-            lease = createLease(context.status, app, holder, ttlMinutes, profileName, registry.leases);
-            await copyTemplateProfile(templateDir, lease.userDataDir);
-            registry.leases.push(lease);
-        }
-        await writeRegistry(context.status, registry);
-        if (reapedCount > 0 && !options.json) {
-            console.log(`Reaped ${reapedCount} stale lease${reapedCount === 1 ? '' : 's'}`);
-        }
-        if (options.json)
-            printLeaseJson(context.status, lease);
-        else
-            printLeaseEnv(context.status, lease);
-        if (!options.noWarmup) {
-            const warmup = await warmupBrowserUseLease(context, lease);
-            if (!warmup.success && !options.json) {
-                console.log(`Warning: Browser warm-up failed: ${warmup.error}`);
-            }
-        }
-    });
-}
-async function warmupBrowserUseLease(context, lease, timeoutMs = 15000) {
-    const wrapperPath = join(context.status.homeDir, 'bin', 'browser-use');
-    if (!existsSync(wrapperPath)) {
-        return { success: false, error: 'browser-use wrapper not found' };
-    }
-    const env = {
-        ...process.env,
-        BROWSER_USE_HOME: browserUseHome(context.status),
-        PIBO_BROWSER_USE_SESSION: lease.sessionName,
-        PIBO_BROWSER_USE_CHROME_USER_DATA_DIR: lease.userDataDir,
-        PIBO_BROWSER_USE_DEFAULT_PROFILE: lease.profileName,
-        ...(lease.browserPoolLeaseId ? { PIBO_BROWSER_POOL_LEASE_ID: lease.browserPoolLeaseId } : {}),
-    };
-    return new Promise((resolve) => {
-        const child = spawn(wrapperPath, ['--session', lease.sessionName, '--pibo-ensure-chrome'], {
-            env,
-            stdio: ['ignore', 'pipe', 'pipe'],
-        });
-        let stdout = '';
-        let stderr = '';
-        child.stdout.on('data', (chunk) => { stdout += chunk; });
-        child.stderr.on('data', (chunk) => { stderr += chunk; });
-        const timer = setTimeout(() => {
-            child.kill('SIGTERM');
-            resolve({ success: false, error: `Warm-up timed out after ${timeoutMs}ms` });
-        }, timeoutMs);
-        child.on('error', (error) => {
-            clearTimeout(timer);
-            resolve({ success: false, error: error.message });
-        });
-        child.on('close', (code) => {
-            clearTimeout(timer);
-            if (code === 0) {
-                const cdpUrl = stdout.trim();
-                resolve({ success: true, cdpUrl: cdpUrl || undefined });
-            }
-            else {
-                resolve({ success: false, error: stderr || `Chrome start exited with code ${code}` });
-            }
-        });
-    });
-}
-async function reapStaleLeasesInRegistry(status, registry) {
-    let count = 0;
-    for (const lease of registry.leases) {
-        if (lease.status !== 'active' || !isExpired(lease))
-            continue;
-        if (lease.browserPoolLeaseId) {
-            await reapAssociatedBrowserPoolLeaseProcess(status, lease);
-        }
-        else if (leaseProcessIsAlive(status, lease)) {
-            killLeaseProcess(status, lease);
-        }
-        lease.status = 'released';
-        lease.updatedAt = nowIso();
-        count += 1;
-    }
-    return count;
-}
-export async function listBrowserUseLeases(context, json = false) {
-    let reapedCount = 0;
-    // Auto-reap stale leases before listing
-    await withRegistryLock(context.status, async () => {
-        const registry = readRegistry(context.status);
-        reapedCount = await reapStaleLeasesInRegistry(context.status, registry);
-        if (reapedCount > 0) {
-            await writeRegistry(context.status, registry);
-        }
-    });
-    const registry = readRegistry(context.status);
-    const rows = registry.leases.map((lease) => ({
-        ...lease,
-        expired: isExpired(lease),
-        processAlive: leaseProcessIsAlive(context.status, lease),
-    }));
-    if (json) {
-        console.log(JSON.stringify(rows, null, 2));
-        return;
-    }
-    if (rows.length === 0) {
-        console.log('No browser-use auth leases.');
-        return;
-    }
-    for (const row of rows) {
-        const state = row.status === 'active' && !row.expired ? 'active' : row.expired ? 'expired' : row.status;
-        console.log(`${row.id}\t${state}\t${row.holder}\t${row.sessionName}\t${row.userDataDir}`);
-    }
-}
-export async function releaseBrowserUseLease(context, id, options = {}) {
-    await withRegistryLock(context.status, async () => {
-        const registry = readRegistry(context.status);
-        const lease = registry.leases.find((candidate) => candidate.id === id);
-        if (!lease) {
-            throw new Error(formatCliError({
-                code: ErrorCode.CLIENT_ERROR,
-                type: 'BROWSER_USE_AUTH_LEASE_NOT_FOUND',
-                message: `Browser-use auth lease "${id}" not found`,
-                suggestion: 'Run `pibo tools browser-use lease list`.',
-            }));
-        }
-        if (lease.browserPoolLeaseId) {
-            await releaseAssociatedBrowserPoolLease(context.status, lease);
-        }
-        else {
-            killLeaseProcess(context.status, lease);
-        }
-        lease.status = 'released';
-        lease.updatedAt = nowIso();
-        if (options.deleteProfile) {
-            await rm(lease.userDataDir, { recursive: true, force: true });
-        }
-        await writeRegistry(context.status, registry);
-        console.log(`Released ${lease.id}`);
-    });
-}
-export async function reapStaleBrowserUseLeases(context) {
-    await withRegistryLock(context.status, async () => {
-        const registry = readRegistry(context.status);
-        const count = await reapStaleLeasesInRegistry(context.status, registry);
-        await writeRegistry(context.status, registry);
-        console.log(`Reaped ${count} stale browser-use auth lease${count === 1 ? '' : 's'}`);
-    });
-}
-export function printBrowserUseAuthTemplatePath(status) {
-    console.log(authTemplateDir(status));
-}
-export async function printBrowserUseAuthTemplateEnv(status) {
-    const templateDir = authTemplateDir(status);
-    await mkdir(templateDir, { recursive: true });
-    console.log(`export BROWSER_USE_HOME=${shellQuote(browserUseHome(status))}`);
-    console.log(`export PIBO_BROWSER_USE_SESSION=${shellQuote('pibo-auth-template')}`);
-    console.log(`export PIBO_BROWSER_USE_CHROME_USER_DATA_DIR=${shellQuote(templateDir)}`);
-    console.log(`export PIBO_BROWSER_USE_DEFAULT_PROFILE=${shellQuote(BROWSER_USE_DEFAULT_PROFILE)}`);
-    console.log(`# browser-use --headed --session "$PIBO_BROWSER_USE_SESSION" open <chat-url>`);
-}