Fix CapSolver key enablement migration

Author: esokullu

src/chrome/src/agent/agent.js

@@ -30,6 +30,7 @@ import {
 import * as trace from '../trace/recorder.js';
 import { tracesToMarkdown } from './trace-export.js';
 import { solveCaptcha, detectCaptcha, injectToken, captchaParamError, captchaTypesMatch, captchaWebsiteUrl } from './captcha-solver.js';
+import { isCapsolverEnabled, normalizeCapsolverApiKey } from './capsolver-config.js';
 import { captchaChallengeKey, detectChallengeDialog, detectChallengeDialogInPage } from './captcha-gate.js';
 import { applyCaptchaFrameVisibility } from './captcha-frame-runtime.js';
 import { getRecordingStateFresh as recorderStateFresh } from '../recorder/host.js';
@@ -245,8 +246,8 @@ export class Agent extends LoopDetector {
     this.userMemoryRecords = [];
     this.userMemoryMaxPromptChars = USER_MEMORY_DEFAULT_MAX_PROMPT_CHARS;
 
-    // CapSolver integration. Off by default. When enabled AND an API key
-    // is set, the system prompt grows a "[CAPTCHA SOLVER]" note that
+    // CapSolver integration. A valid saved API key enables the
+    // "[CAPTCHA SOLVER]" system-prompt note, which
     // tells the model to try `solve_captcha` once before falling back to
     // asking the user. The agent reads the key from chrome.storage.local
     // at call time so rotating the key doesn't require a restart.
@@ -14658,9 +14659,8 @@ Rules: no prose intro, no conclusion, no "this screenshot shows...", no layout d
     }
 
     // ─── CAPTCHA solver ──────────────────────────────────────────────
-    // Only meaningfully wired when the user has enabled CapSolver in
-    // Settings. We re-check on every call so flipping the toggle or
-    // rotating the key takes effect without a restart.
+    // A valid saved key is the enable control. Re-check it on every call so
+    // rotating or clearing the key takes effect without a restart.
     if (name === 'solve_captcha') {
       let dispatched = false;
       const noDispatchFailure = (error) => ({
@@ -14670,13 +14670,10 @@ Rules: no prose intro, no conclusion, no "this screenshot shows...", no layout d
         error,
       });
       try {
-        const stored = await chrome.storage.local.get(['captchaSolverEnabled', 'capsolverApiKey']);
-        if (!stored.captchaSolverEnabled) {
-          return noDispatchFailure('CapSolver is not enabled. Ask the user to enable it in Settings → General → Advanced, or fall back to asking them to solve the captcha manually.');
-        }
-        const apiKey = (stored.capsolverApiKey || '').trim();
-        if (!apiKey) {
-          return noDispatchFailure('CapSolver is enabled but no API key is configured. Ask the user to set one in Settings → General → Advanced, or fall back to asking them to solve the captcha manually.');
+        const stored = await chrome.storage.local.get(['capsolverApiKey', 'captchaSolverEnabled']);
+        const apiKey = normalizeCapsolverApiKey(stored.capsolverApiKey);
+        if (!isCapsolverEnabled(apiKey, stored.captchaSolverEnabled)) {
+          return noDispatchFailure('CapSolver is not enabled with a valid API key. Ask the user to save their key in Settings → General → Advanced, or fall back to asking them to solve the captcha manually.');
         }
 
         // Use the top-level URL as a fallback. Frame-aware detection below

src/chrome/src/agent/capsolver-config.js

@@ -0,0 +1,17 @@
+// CapSolver account keys currently use the CAP- prefix shown in Settings.
+// Keep this check intentionally structural: account validity and balance are
+// verified by the CapSolver API, while this prevents empty or obviously
+// malformed values from enabling paid CAPTCHA solving.
+export const CAPSOLVER_API_KEY_PATTERN = /^CAP-[A-Za-z0-9_-]{20,}$/;
+
+export function normalizeCapsolverApiKey(value) {
+  return typeof value === 'string' ? value.trim() : '';
+}
+
+export function isValidCapsolverApiKey(value) {
+  return CAPSOLVER_API_KEY_PATTERN.test(normalizeCapsolverApiKey(value));
+}
+
+export function isCapsolverEnabled(apiKey, enabled) {
+  return enabled === true && isValidCapsolverApiKey(apiKey);
+}

src/chrome/src/agent/tools.js

@@ -1054,7 +1054,7 @@ export const AGENT_TOOLS = [
     type: 'function',
     function: {
       name: 'solve_captcha',
-      description: 'Solve a CAPTCHA on the current page using the CapSolver API (only available when the user has enabled CapSolver and provided an API key in Settings → General → Advanced). The tool scans every frame, ranks the active visible widget above hidden/background integrations, and fills missing type/site-key parameters from the selected frame. Returns the token, selected frame/type, and targeted injection/callback status; verify page progress afterward because token injection alone does not prove the challenge cleared. On failure (no CapSolver key configured, ambiguous candidates, unknown type, API error, timeout) the tool returns `{ success: false, error: "..." }` — use only the exact frameUrl, websiteKey, frameId, or framePath discriminator offered by the ambiguity result; otherwise ask the user to solve it manually.',
+      description: 'Solve a CAPTCHA on the current page using the CapSolver API (available when the user has saved a valid API key in Settings → General → Advanced). The tool scans every frame, ranks the active visible widget above hidden/background integrations, and fills missing type/site-key parameters from the selected frame. Returns the token, selected frame/type, and targeted injection/callback status; verify page progress afterward because token injection alone does not prove the challenge cleared. On failure (no CapSolver key configured, ambiguous candidates, unknown type, API error, timeout) the tool returns `{ success: false, error: "..." }` — use only the exact frameUrl, websiteKey, frameId, or framePath discriminator offered by the ambiguity result; otherwise ask the user to solve it manually.',
       parameters: {
         type: 'object',
         properties: {

src/chrome/src/background.js

@@ -27,6 +27,7 @@ import {
   getClaudeOAuthStatus,
 } from './providers/oauth-claude.js';
 import { getBalance as capsolverGetBalance } from './agent/captcha-solver.js';
+import { isCapsolverEnabled } from './agent/capsolver-config.js';
 import { createCloudRunController } from './cloud-runs.js';
 import { ensureOffscreen } from './offscreen/ensure.js';
 import {
@@ -735,14 +736,15 @@ async function loadCustomSkills() {
 }
 const customSkillsReady = loadCustomSkills();
 
-// CapSolver opt-in. We only need the toggle here — the API key is read at
-// call time inside the agent's solve_captcha handler so rotating it via
-// the settings page is picked up without a restart.
+// A valid key plus explicit consent enables CapSolver. Requiring the existing
+// boolean preserves legacy profiles that saved a key while the old switch was
+// off; pressing Save Key in the new UI sets consent to true.
 async function loadCaptchaSolver() {
-  const stored = await chrome.storage.local.get('captchaSolverEnabled');
-  if (stored.captchaSolverEnabled != null) {
-    agent.captchaSolverEnabled = !!stored.captchaSolverEnabled;
-  }
+  const stored = await chrome.storage.local.get(['capsolverApiKey', 'captchaSolverEnabled']);
+  agent.captchaSolverEnabled = isCapsolverEnabled(
+    stored.capsolverApiKey,
+    stored.captchaSolverEnabled,
+  );
 }
 loadCaptchaSolver();
 
@@ -931,9 +933,10 @@ chrome.storage.onChanged.addListener((changes) => {
     }
     refreshPrompts = true;
   }
-  if (changes.captchaSolverEnabled) {
-    agent.captchaSolverEnabled = !!changes.captchaSolverEnabled.newValue;
-    refreshPrompts = true;
+  if (changes.capsolverApiKey || changes.captchaSolverEnabled) {
+    loadCaptchaSolver()
+      .then(() => agent._refreshSystemPrompts())
+      .catch((error) => console.warn('[WebBrain] CapSolver setting could not be refreshed', error));
   }
   if (changes.planBeforeActMode || changes.planBeforeAct) {
     applyPlanBeforeActMode(normalizePlanBeforeActMode({

src/chrome/src/ui/locales/ar.js

@@ -458,17 +458,17 @@ export default {
   "st.transcription.connected": "تم الاتصال! النموذج: {model}",
   "st.transcription.failed": "فشل: {error}",
   "st.transcription.fill_required": "املأ عنوان API الأساسي والنموذج أولًا.",
-  "st.captcha.desc_html": "دع الوكيل يحلّ اختبارات CAPTCHA تلقائيًا عبر واجهة <a href=\"https://capsolver.com\" target=\"_blank\" style=\"color:var(--accent);\">CapSolver</a>. يدعم reCAPTCHA v2/v3 و hCaptcha و Cloudflare Turnstile. مُعطّل افتراضيًا — عند التعطيل، يتوقّف الوكيل ويطلب منك حلّ الاختبار بنفسك. تتقاضى CapSolver رسومًا لكل عملية حلّ (~$0.001–$0.003)؛ تستخدم حسابك ومفتاح API الخاص بك.",
+  "st.captcha.desc_html": "دع الوكيل يحلّ اختبارات CAPTCHA تلقائيًا عبر واجهة <a href=\"https://capsolver.com\" target=\"_blank\" style=\"color:var(--accent);\">CapSolver</a>. يدعم reCAPTCHA v2/v3 وhCaptcha وCloudflare Turnstile. يؤدي حفظ مفتاح API صالح إلى تمكين CapSolver تلقائيًا؛ وبدونه، يتوقّف الوكيل ويطلب منك حلّ الاختبار بنفسك. تتقاضى CapSolver رسومًا لكل عملية حلّ (~$0.001–$0.003)؛ تستخدم حسابك ومفتاح API الخاص بك.",
   "st.captcha.enabled.label": "تفعيل CapSolver",
   "st.captcha.enabled.desc": "عندما يصادف الوكيل اختبار CAPTCHA فإنه يستدعي CapSolver مرّة واحدة قبل اللجوء إلى سؤالك. يتطلّب مفتاح API أدناه.",
   "st.captcha.api_key.label": "مفتاح CapSolver API",
   "st.captcha.save": "حفظ المفتاح",
   "st.captcha.check_balance": "التحقّق من الرصيد",
   "st.captcha.clear": "مسح",
-  "st.captcha.saved": "تم الحفظ.",
+  "st.captcha.saved": "تم الحفظ. تم تمكين CapSolver.",
   "st.captcha.cleared": "تم المسح.",
   "st.captcha.checking": "جارٍ التحقّق من الرصيد…",
-  "st.captcha.need_key": "أدخل مفتاح API أولًا.",
+  "st.captcha.need_key": "أدخل مفتاح CapSolver API صالحًا يبدأ بـ CAP-.",
   "st.captcha.balance_ok": "موافق — الرصيد: {balance}",
   "st.captcha.balance_fail": "فشل: {error}",
   "st.captcha.security_html": "<strong>تنبيه:</strong> يُخزَّن مفتاح API <strong>بصيغة نصية واضحة</strong> في التخزين المحلي للمتصفح. تتقاضى CapSolver رسومًا من حسابك عن كل عملية حلّ؛ ولن يستدعيها الوكيل إلا عندما يحجب اختبار CAPTCHA خطوةً فعلًا (مرّة واحدة كحدّ أقصى لكل مواجهة — لن يعيد المحاولة عند الفشل). تحظر شروط خدمة بعض المواقع حلّ اختبارات CAPTCHA آليًا؛ استخدم حسن تقديرك.",

src/chrome/src/ui/locales/bn.js

@@ -796,17 +796,17 @@ export default {
   'st.memory.reason.not_found': "কোনো সংরক্ষিত মেমরি সেই আইডি নেই।",
   'st.memory.security_html': "<strong>গোপনীয়তা:</strong> ব্যবহারকারীর মেমরি এই ব্রাউজার প্রোফাইলে প্লেইনটেক্সটে সংরক্ষণ করা হয়। সক্রিয় করা হলে, সিস্টেম প্রম্পটের অংশ হিসাবে আপনি যে LLM প্রদানকারীকে কনফিগার করেন সক্রিয় মেমরি রেকর্ডগুলি পাঠানো হয়৷ এখানে পাসওয়ার্ড, API কী, টোকেন, রিকভারি কোড বা সংবেদনশীল গোপনীয়তা সংরক্ষণ করবেন না।",
 
-  'st.captcha.desc_html': "এজেন্টকে স্বয়ংক্রিয়ভাবে এর মাধ্যমে ক্যাপচা সমাধান করতে দিন <a href=\"https://capsolver.com\" target=\"_blank\" style=\"color:var(--accent);\">CapSolver</a> API reCAPTCHA v2/v3, hCaptcha এবং Cloudflare টার্নস্টাইল সমর্থন করে। ডিফল্টরূপে বন্ধ — বন্ধ হয়ে গেলে, এজেন্ট থামে এবং আপনাকে নিজেই ক্যাপচা সমাধান করতে বলে। সমাধান প্রতি CapSolver চার্জ (~$0.001–$0.003); আপনি আপনার নিজের অ্যাকাউন্ট এবং API কী আনুন।",
+  'st.captcha.desc_html': "এজেন্টকে <a href=\"https://capsolver.com\" target=\"_blank\" style=\"color:var(--accent);\">CapSolver</a> API-এর মাধ্যমে স্বয়ংক্রিয়ভাবে CAPTCHA সমাধান করতে দিন। এটি reCAPTCHA v2/v3, hCaptcha এবং Cloudflare Turnstile সমর্থন করে। একটি বৈধ API কী সংরক্ষণ করলে CapSolver স্বয়ংক্রিয়ভাবে সক্রিয় হয়; কী না থাকলে এজেন্ট থামে এবং আপনাকে নিজেই CAPTCHA সমাধান করতে বলে। প্রতিটি সমাধানের জন্য CapSolver চার্জ করে (~$0.001–$0.003); আপনি নিজের অ্যাকাউন্ট ও API কী ব্যবহার করেন।",
   'st.captcha.enabled.label': "CapSolver সক্ষম করুন",
   'st.captcha.enabled.desc': "যখন এজেন্ট একটি ক্যাপচা হিট করে তখন এটি আপনাকে জিজ্ঞাসা করার আগে একবার ক্যাপসোলভারকে কল করবে। নীচে একটি API কী প্রয়োজন৷",
   'st.captcha.api_key.label': "CapSolver API কী",
   'st.captcha.save': "কী সংরক্ষণ করুন",
   'st.captcha.check_balance': "ব্যালেন্স চেক করুন",
   'st.captcha.clear': "পরিষ্কার",
-  'st.captcha.saved': "সংরক্ষিত",
+  'st.captcha.saved': "সংরক্ষিত। CapSolver সক্রিয় হয়েছে।",
   'st.captcha.cleared': "সাফ করা হয়েছে।",
   'st.captcha.checking': "ব্যালেন্স চেক করা হচ্ছে...",
-  'st.captcha.need_key': "প্রথমে একটি API কী লিখুন।",
+  'st.captcha.need_key': "CAP- দিয়ে শুরু হওয়া একটি বৈধ CapSolver API কী লিখুন।",
   'st.captcha.balance_ok': "ঠিক আছে — ব্যালেন্স: {balance}",
   'st.captcha.balance_fail': "ব্যর্থ হয়েছে: {error}",
   'st.captcha.security_html': "<strong>হেড-আপ:</strong> API কী সংরক্ষণ করা হয় <strong>প্লেইনটেক্সট</strong> ব্রাউজার স্থানীয় স্টোরেজে। CapSolver প্রতিটি সমাধানের জন্য আপনার অ্যাকাউন্ট চার্জ করে; এজেন্ট কেবল তখনই এটিকে কল করবে যখন একটি ক্যাপচা প্রকৃতপক্ষে একটি পদক্ষেপকে ব্লক করে (প্রতি এনকাউন্টারে সর্বোচ্চ একবার - এটি ব্যর্থ হলে পুনরায় চেষ্টা করবে না)। কিছু সাইটের পরিষেবার শর্তাবলী স্বয়ংক্রিয় ক্যাপচা সমাধান নিষিদ্ধ করে; আপনার রায় ব্যবহার করুন।",

src/chrome/src/ui/locales/de.js

@@ -738,17 +738,17 @@ export default {
   'st.memory.security_html': '<strong>Datenschutz:</strong> Benutzermemorie wird als Klartext in diesem Browser-Profil gespeichert...',
 
   // --- CAPTCHA settings ---
-  'st.captcha.desc_html': 'Lassen Sie den Agenten CAPTCHAs automatisch über die <a href="https://capsolver.com" target="_blank" style="color:var(--accent);">CapSolver</a>-API lösen...',
+  'st.captcha.desc_html': 'Lassen Sie den Agenten CAPTCHAs automatisch über die <a href="https://capsolver.com" target="_blank" style="color:var(--accent);">CapSolver</a>-API lösen. Unterstützt reCAPTCHA v2/v3, hCaptcha und Cloudflare Turnstile. Das Speichern eines gültigen API-Schlüssels aktiviert CapSolver automatisch; ohne Schlüssel hält der Agent an und bittet Sie, das CAPTCHA selbst zu lösen. CapSolver berechnet jede Lösung (~$0.001–$0.003); Sie verwenden Ihr eigenes Konto und Ihren eigenen API-Schlüssel.',
   'st.captcha.enabled.label': 'CapSolver aktivieren',
   'st.captcha.enabled.desc': 'Wenn der Agent auf ein CAPTCHA trifft, wird CapSolver einmal aufgerufen, bevor er Sie zur Lösung auffordert.',
   'st.captcha.api_key.label': 'CapSolver API-Schlüssel',
   'st.captcha.save': 'Schlüssel speichern',
   'st.captcha.check_balance': 'Guthaben prüfen',
   'st.captcha.clear': 'Löschen',
-  'st.captcha.saved': 'Gespeichert.',
+  'st.captcha.saved': 'Gespeichert. CapSolver ist aktiviert.',
   'st.captcha.cleared': 'Gelöscht.',
   'st.captcha.checking': 'Guthaben wird überprüft…',
-  'st.captcha.need_key': 'Geben Sie zuerst einen API-Schlüssel ein.',
+  'st.captcha.need_key': 'Geben Sie einen gültigen CapSolver-API-Schlüssel ein, der mit CAP- beginnt.',
   'st.captcha.balance_ok': 'OK — Guthaben: {balance}',
   'st.captcha.balance_fail': 'Fehlgeschlagen: {error}',
   'st.captcha.security_html': '<strong>Hinweis:</strong> Der API-Schlüssel wird als Klartext im Browser-Speicher gespeichert...',