halo-agent 1.1.0

package/package.json

@@ -0,0 +1,49 @@
+{
+  "name": "halo-agent",
+  "version": "1.1.0",
+  "description": "HALO local apply agent — auto-fills job applications using your real Chrome session",
+  "main": "index.js",
+  "bin": {
+    "halo-agent": "./index.js"
+  },
+  "scripts": {
+    "start": "node index.js start",
+    "init": "node index.js init",
+    "pair": "node index.js pair",
+    "install-autostart": "node index.js install-autostart",
+    "uninstall-autostart": "node index.js uninstall-autostart"
+  },
+  "files": [
+    "index.js",
+    "config.js",
+    "browser.js",
+    "poller.js",
+    "orchestrator.js",
+    "localServer.js",
+    "filler.js",
+    "scanPage.js",
+    "captcha.js",
+    "vision.js",
+    "manusAutomate.js",
+    "README.md"
+  ],
+  "keywords": [
+    "halo",
+    "job-application",
+    "auto-apply",
+    "playwright",
+    "cli"
+  ],
+  "license": "MIT",
+  "dependencies": {
+    "playwright": "^1.44.0",
+    "ws": "^8.17.0",
+    "node-fetch": "^3.3.2",
+    "ora": "^8.0.1",
+    "inquirer": "^9.2.22",
+    "chalk": "^5.3.0"
+  },
+  "engines": {
+    "node": ">=18"
+  }
+}

package/poller.js

@@ -0,0 +1,172 @@
+'use strict';
+
+/**
+ * Polls HALO backend every 5 seconds for queued jobs.
+ * When a job is found, hands it to the orchestrator.
+ * Runs sequentially — one job at a time to avoid Chrome chaos.
+ */
+
+const { runJob } = require('./orchestrator');
+const { cfAccessHeaders } = require('./config');
+
+let active = false;
+let sessionId = null;
+
+async function startPolling(chromeConn, config) {
+  console.log('[poller] Started. Polling for queued jobs every 5s...');
+
+  // Register agent session with backend
+  sessionId = await registerSession(config);
+
+  const heartbeatInterval = setInterval(() => heartbeat(config, sessionId), 15000);
+
+  while (true) {
+    await new Promise(r => setTimeout(r, 5000));
+    if (active) continue; // already processing a job
+
+    const item = await fetchNextJob(config);
+    if (!item) continue;
+
+    active = true;
+    console.log(`[poller] Picked up job: ${item.company} - ${item.title} (${item.id})`);
+
+    // Update session status
+    await heartbeat(config, sessionId, 'filling', item.job_id);
+
+    const reportStatus = makeReporter(config, item.id, sessionId);
+    try {
+      await runJob(item, chromeConn, config, reportStatus);
+    } catch (err) {
+      console.error('[poller] runJob threw:', err.message);
+      // Last-resort safety net — orchestrator usually classifies, but if a
+      // throw escaped without classification, fall back to 'generic'.
+      await reportStatus('NEEDS_ATTENTION', {
+        needs_attention_reason: err.message,
+        intervention_type: 'generic',
+        step: 'NEEDS_ATTENTION',
+        step_detail: err.message?.slice(0, 120),
+      }).catch(() => {});
+    } finally {
+      active = false;
+      await heartbeat(config, sessionId, 'idle', null);
+    }
+  }
+}
+
+// Maps the coarse status the orchestrator already emits to the fine-grained
+// step value the SSE feed renders. The orchestrator can override either by
+// passing { step, step_detail } explicitly in extra.
+const STATUS_TO_STEP = {
+  PENDING: 'ANALYZING',
+  IN_PROGRESS: 'FILLING',
+  REVIEWING: 'REVIEWING',
+  DONE: 'DONE',
+  NEEDS_ATTENTION: 'NEEDS_ATTENTION',
+  CANCELLED: 'DONE',
+};
+
+function makeReporter(config, queueId, sessionId) {
+  return async function reportStatus(status, extra = {}) {
+    // Existing coarse-status PATCH — preserves all current backend behaviour
+    // (auto-create submission on DONE, etc). Untouched by the SSE work.
+    try {
+      await fetch(`${config.apiUrl}/apply-queue/${queueId}`, {
+        method: 'PATCH',
+        headers: {
+          Authorization: `Bearer ${config.token}`,
+          'Content-Type': 'application/json',
+          ...cfAccessHeaders(config),
+        },
+        body: JSON.stringify({ status, ...extra }),
+      });
+    } catch (e) {
+      console.warn('[poller] Could not report status:', e.message);
+    }
+
+    // Granular step PATCH — drives the live SSE feed. Fire-and-forget; if the
+    // backend is on an older version that doesn't have /agent yet, this 404s
+    // silently and the user just sees the coarse status as before.
+    const step = extra.step || STATUS_TO_STEP[status] || status;
+    const detail = extra.step_detail || extra.detail || extra.needs_attention_reason || null;
+    const fields_filled = typeof extra.fields_filled === 'number' ? extra.fields_filled : undefined;
+
+    try {
+      await fetch(`${config.apiUrl}/agent/queue/${queueId}/step`, {
+        method: 'PATCH',
+        headers: {
+          Authorization: `Bearer ${config.token}`,
+          'Content-Type': 'application/json',
+          ...cfAccessHeaders(config),
+        },
+        body: JSON.stringify({
+          step,
+          detail,
+          ...(fields_filled !== undefined ? { fields_filled } : {}),
+          ...(sessionId ? { session_id: sessionId } : {}),
+        }),
+      });
+    } catch { /* fire-and-forget; non-critical */ }
+  };
+}
+
+async function fetchNextJob(config) {
+  try {
+    const res = await fetch(`${config.apiUrl}/apply-queue/next`, {
+      headers: {
+        Authorization: `Bearer ${config.token}`,
+        ...cfAccessHeaders(config),
+      },
+    });
+    if (!res.ok) return null;
+    const data = await res.json();
+    return data.item || null;
+  } catch {
+    return null;
+  }
+}
+
+async function registerSession(config) {
+  try {
+    const res = await fetch(`${config.apiUrl}/apply-queue/agent/sessions`, {
+      method: 'POST',
+      headers: {
+        Authorization: `Bearer ${config.token}`,
+        'Content-Type': 'application/json',
+        ...cfAccessHeaders(config),
+      },
+      body: JSON.stringify({
+        platform: process.platform,
+        agent_version: require('./package.json').version,
+        status: 'idle',
+      }),
+    });
+    if (!res.ok) {
+      const text = await res.text().catch(() => res.status);
+      console.warn(`[poller] Session registration failed (${res.status}):`, text);
+      return null;
+    }
+    const data = await res.json();
+    console.log('[poller] Session registered:', data.session_id);
+    return data.session_id;
+  } catch (e) {
+    console.warn('[poller] Session registration error:', e.message);
+    return null;
+  }
+}
+
+async function heartbeat(config, sessionId, status = 'idle', currentJobId = null) {
+  if (!sessionId) return;
+  try {
+    await fetch(`${config.apiUrl}/apply-queue/agent/sessions`, {
+      method: 'POST',
+      headers: {
+        Authorization: `Bearer ${config.token}`,
+        'Content-Type': 'application/json',
+        ...cfAccessHeaders(config),
+      },
+      body: JSON.stringify({ id: sessionId, status, current_job_id: currentJobId }),
+    });
+  } catch {}
+}
+
+module.exports = { startPolling };