feat(main): integrate /proxy and /ca_bundle commands into REPL

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: lzwjava

iclaw/completer.py

@@ -7,6 +7,8 @@
     "/model_provider",
     "/model",
     "/search_provider",
+    "/proxy",
+    "/ca_bundle",
     "/copy",
     "/status",
     "/help",

iclaw/config.py

@@ -37,7 +37,7 @@ def load_session_settings() -> dict:
 
 
 def save_session_settings(
-    model_provider, current_model, search_provider, proxy=None, ca_bundle=None
+    *, model_provider, current_model, search_provider, proxy=None, ca_bundle=None
 ) -> None:
     config = _load_config()
     config["model_provider"] = model_provider

iclaw/main.py

@@ -6,8 +6,10 @@
 
 from prompt_toolkit import PromptSession
 
+from iclaw import http
 from iclaw.at_mention import resolve_at_mentions
 from iclaw.commands.model import handle_model_command, handle_model_provider_command
+from iclaw.commands.proxy import handle_ca_bundle_command, handle_proxy_command
 from iclaw.commands.search_provider import handle_search_provider_command
 from iclaw.commands.utils import handle_copy_command
 from iclaw.completer import IclawCompleter
@@ -28,6 +30,8 @@
     ("/model_provider", "Select and authenticate with the model provider"),
     ("/model", "Select specific model from your provider"),
     ("/search_provider", "Select the web search provider"),
+    ("/proxy", "Set HTTP/HTTPS proxy (usage: /proxy [url|off])"),
+    ("/ca_bundle", "Set CA bundle for HTTPS (usage: /ca_bundle [path|off])"),
     ("/copy", "Copy last Copilot response to clipboard"),
     ("/status", "Show current settings"),
     ("/help", "Show available commands"),
@@ -44,6 +48,9 @@ def main():
     model_provider = settings["model_provider"]
     current_model = settings["current_model"]
     search_provider = settings["search_provider"]
+    proxy = settings["proxy"]
+    ca_bundle = settings["ca_bundle"]
+    http.reconfigure(proxy=proxy, ca_bundle=ca_bundle)
 
     if github_token:
         print("Connecting to GitHub Copilot...")
@@ -91,20 +98,66 @@ def main():
                 copilot_token = t
                 github_token = load_github_token()
                 token_expiry = time.monotonic() + TOKEN_REFRESH_INTERVAL
-                save_session_settings(model_provider, current_model, search_provider)
+                save_session_settings(
+                    model_provider=model_provider,
+                    current_model=current_model,
+                    search_provider=search_provider,
+                    proxy=proxy,
+                    ca_bundle=ca_bundle,
+                )
             continue
         if user_input == "/model":
             current_model = handle_model_command(copilot_token, current_model)
-            save_session_settings(model_provider, current_model, search_provider)
+            save_session_settings(
+                model_provider=model_provider,
+                current_model=current_model,
+                search_provider=search_provider,
+                proxy=proxy,
+                ca_bundle=ca_bundle,
+            )
             continue
         if user_input == "/search_provider":
             search_provider = handle_search_provider_command(search_provider)
-            save_session_settings(model_provider, current_model, search_provider)
+            save_session_settings(
+                model_provider=model_provider,
+                current_model=current_model,
+                search_provider=search_provider,
+                proxy=proxy,
+                ca_bundle=ca_bundle,
+            )
+            continue
+        if user_input == "/proxy" or user_input.startswith("/proxy "):
+            parts = user_input.split(maxsplit=1)
+            arg = parts[1] if len(parts) > 1 else None
+            proxy = handle_proxy_command(proxy, arg)
+            http.reconfigure(proxy=proxy, ca_bundle=ca_bundle)
+            save_session_settings(
+                model_provider=model_provider,
+                current_model=current_model,
+                search_provider=search_provider,
+                proxy=proxy,
+                ca_bundle=ca_bundle,
+            )
+            continue
+        if user_input == "/ca_bundle" or user_input.startswith("/ca_bundle "):
+            parts = user_input.split(maxsplit=1)
+            arg = parts[1] if len(parts) > 1 else None
+            ca_bundle = handle_ca_bundle_command(ca_bundle, arg)
+            http.reconfigure(proxy=proxy, ca_bundle=ca_bundle)
+            save_session_settings(
+                model_provider=model_provider,
+                current_model=current_model,
+                search_provider=search_provider,
+                proxy=proxy,
+                ca_bundle=ca_bundle,
+            )
             continue
         if user_input == "/status":
             print(f"  model_provider:  {model_provider}")
             print(f"  model:           {current_model}")
             print(f"  search_provider: {search_provider}")
+            print(f"  proxy:           {proxy or '(not set)'}")
+            print(f"  ca_bundle:       {ca_bundle or '(system default)'}")
             print(f"  cwd:             {os.getcwd()}")
             print()
             continue

tests/test_main.py

@@ -20,10 +20,11 @@ def _ps(*inputs):
 
 
 class TestMain(unittest.TestCase):
+    @patch("iclaw.main.http")
     @patch("iclaw.main.chat")
     @patch("iclaw.main.load_github_token")
     @patch("iclaw.main.PromptSession")
-    def test_main_cli(self, mock_ps, mock_load, mock_chat):
+    def test_main_cli(self, mock_ps, mock_load, mock_chat, mock_http):
         mock_load.return_value = "gt"
         mock_ps.return_value = _mock_session(".exit")
         with (
@@ -50,95 +51,108 @@ def test_load_github_token_invalid_json(self):
             mp.read_text.return_value = "not json"
             self.assertIsNone(main.load_github_token())
 
+    @patch("iclaw.main.http")
     @patch("iclaw.main.load_github_token", return_value=None)
     @patch("iclaw.main.PromptSession")
-    def test_main_no_token(self, mock_ps, mock_load):
+    def test_main_no_token(self, mock_ps, mock_load, mock_http):
         mock_ps.return_value = _mock_session(".exit")
         with patch("sys.stdout"), patch("sys.stderr"):
             main.main()
 
+    @patch("iclaw.main.http")
     @patch("iclaw.main.load_github_token", return_value="gt")
     @patch("iclaw.main.get_copilot_token", side_effect=Exception("fail"))
     @patch("iclaw.main.PromptSession")
-    def test_main_copilot_token_error(self, mock_ps, mock_cp, mock_load):
+    def test_main_copilot_token_error(self, mock_ps, mock_cp, mock_load, mock_http):
         mock_ps.return_value = _mock_session(".exit")
         with patch("sys.stdout"), patch("sys.stderr"):
             main.main()
 
+    @patch("iclaw.main.http")
     @patch("iclaw.main.load_github_token", return_value="gt")
     @patch("iclaw.main.get_copilot_token", return_value="ct")
     @patch("iclaw.main.PromptSession")
-    def test_main_empty_and_help(self, mock_ps, mock_cp, mock_load):
+    def test_main_empty_and_help(self, mock_ps, mock_cp, mock_load, mock_http):
         mock_ps.return_value = _mock_session("", "/help", "/", ".exit")
         with patch("sys.stdout"), patch("iclaw.main.time.monotonic", return_value=0):
             main.main()
 
+    @patch("iclaw.main.http")
     @patch("iclaw.main.load_github_token", return_value="gt")
     @patch("iclaw.main.get_copilot_token", return_value="ct")
     @patch("iclaw.main.PromptSession")
-    def test_main_eof(self, mock_ps, mock_cp, mock_load):
+    def test_main_eof(self, mock_ps, mock_cp, mock_load, mock_http):
         mock_ps.return_value = _mock_session(EOFError())
         with patch("sys.stdout"), patch("iclaw.main.time.monotonic", return_value=0):
             main.main()
 
+    @patch("iclaw.main.http")
     @patch("iclaw.main.load_github_token", return_value=None)
     @patch("iclaw.main.PromptSession")
-    def test_main_not_authenticated(self, mock_ps, mock_load):
+    def test_main_not_authenticated(self, mock_ps, mock_load, mock_http):
         mock_ps.return_value = _mock_session("hello", ".exit")
         with patch("sys.stdout"), patch("sys.stderr"):
             main.main()
 
+    @patch("iclaw.main.http")
     @patch("iclaw.main.load_github_token", return_value="gt")
     @patch("iclaw.main.get_copilot_token", return_value="ct")
     @patch("iclaw.main.handle_copy_command")
     @patch("iclaw.main.PromptSession")
-    def test_main_copy(self, mock_ps, mock_copy, mock_cp, mock_load):
+    def test_main_copy(self, mock_ps, mock_copy, mock_cp, mock_load, mock_http):
         mock_ps.return_value = _mock_session("/copy", ".exit")
         with patch("sys.stdout"), patch("iclaw.main.time.monotonic", return_value=0):
             main.main()
         mock_copy.assert_called_once()
 
+    @patch("iclaw.main.http")
     @patch("iclaw.main.load_github_token", return_value="gt")
     @patch("iclaw.main.get_copilot_token", return_value="ct")
     @patch("iclaw.main.handle_model_provider_command", return_value=("copilot", None))
     @patch("iclaw.main.PromptSession")
-    def test_main_model_provider(self, mock_ps, mock_mp, mock_cp, mock_load):
+    def test_main_model_provider(self, mock_ps, mock_mp, mock_cp, mock_load, mock_http):
         mock_ps.return_value = _mock_session("/model_provider", ".exit")
         with patch("sys.stdout"), patch("iclaw.main.time.monotonic", return_value=0):
             main.main()
 
+    @patch("iclaw.main.http")
     @patch("iclaw.main.load_github_token", return_value="gt")
     @patch("iclaw.main.get_copilot_token", return_value="ct")
     @patch("iclaw.main.handle_model_command", return_value="gpt-4o")
     @patch("iclaw.main.PromptSession")
-    def test_main_model(self, mock_ps, mock_mc, mock_cp, mock_load):
+    def test_main_model(self, mock_ps, mock_mc, mock_cp, mock_load, mock_http):
         mock_ps.return_value = _mock_session("/model", ".exit")
         with patch("sys.stdout"), patch("iclaw.main.time.monotonic", return_value=0):
             main.main()
 
+    @patch("iclaw.main.http")
     @patch("iclaw.main.load_github_token", return_value="gt")
     @patch("iclaw.main.get_copilot_token", return_value="ct")
     @patch("iclaw.main.handle_search_provider_command", return_value="bing")
     @patch("iclaw.main.PromptSession")
-    def test_main_search_provider(self, mock_ps, mock_sp, mock_cp, mock_load):
+    def test_main_search_provider(
+        self, mock_ps, mock_sp, mock_cp, mock_load, mock_http
+    ):
         mock_ps.return_value = _mock_session("/search_provider", ".exit")
         with patch("sys.stdout"), patch("iclaw.main.time.monotonic", return_value=0):
             main.main()
 
+    @patch("iclaw.main.http")
     @patch("iclaw.main.load_github_token", return_value="gt")
     @patch("iclaw.main.get_copilot_token", return_value="ct")
     @patch("iclaw.main.chat", return_value={"content": "hi"})
     @patch("iclaw.main.PromptSession")
-    def test_main_chat(self, mock_ps, mock_chat, mock_cp, mock_load):
+    def test_main_chat(self, mock_ps, mock_chat, mock_cp, mock_load, mock_http):
         mock_ps.return_value = _mock_session("hello", ".exit")
         with patch("sys.stdout"), patch("iclaw.main.time.monotonic", return_value=0):
             main.main()
 
+    @patch("iclaw.main.http")
     @patch("iclaw.main.load_github_token", return_value="gt")
     @patch("iclaw.main.get_copilot_token", return_value="ct")
     @patch("iclaw.main.chat", side_effect=Exception("API error"))
     @patch("iclaw.main.PromptSession")
-    def test_main_chat_error(self, mock_ps, mock_chat, mock_cp, mock_load):
+    def test_main_chat_error(self, mock_ps, mock_chat, mock_cp, mock_load, mock_http):
         mock_ps.return_value = _mock_session("hello", ".exit")
         with (
             patch("sys.stdout"),
@@ -147,13 +161,14 @@ def test_main_chat_error(self, mock_ps, mock_chat, mock_cp, mock_load):
         ):
             main.main()
 
+    @patch("iclaw.main.http")
     @patch("iclaw.main.load_github_token", return_value="gt")
     @patch("iclaw.main.get_copilot_token", return_value="ct")
     @patch("iclaw.main.chat")
     @patch("iclaw.main.web_search", return_value="search results")
     @patch("iclaw.main.PromptSession")
     def test_main_tool_call_web_search(
-        self, mock_ps, mock_ws, mock_chat, mock_cp, mock_load
+        self, mock_ps, mock_ws, mock_chat, mock_cp, mock_load, mock_http
     ):
         mock_ps.return_value = _mock_session("hello", ".exit")
         mock_chat.side_effect = [
@@ -175,13 +190,14 @@ def test_main_tool_call_web_search(
             main.main()
         mock_ws.assert_called_once()
 
+    @patch("iclaw.main.http")
     @patch("iclaw.main.load_github_token", return_value="gt")
     @patch("iclaw.main.get_copilot_token", return_value="ct")
     @patch("iclaw.main.chat")
     @patch("iclaw.main.exec", return_value="output")
     @patch("iclaw.main.PromptSession")
     def test_main_tool_call_exec(
-        self, mock_ps, mock_exec, mock_chat, mock_cp, mock_load
+        self, mock_ps, mock_exec, mock_chat, mock_cp, mock_load, mock_http
     ):
         mock_ps.return_value = _mock_session("hello", ".exit")
         mock_chat.side_effect = [
@@ -202,11 +218,14 @@ def test_main_tool_call_exec(
         with patch("sys.stdout"), patch("iclaw.main.time.monotonic", return_value=0):
             main.main()
 
+    @patch("iclaw.main.http")
     @patch("iclaw.main.load_github_token", return_value="gt")
     @patch("iclaw.main.get_copilot_token", return_value="ct")
     @patch("iclaw.main.chat")
     @patch("iclaw.main.PromptSession")
-    def test_main_tool_call_edit(self, mock_ps, mock_chat, mock_cp, mock_load):
+    def test_main_tool_call_edit(
+        self, mock_ps, mock_chat, mock_cp, mock_load, mock_http
+    ):
         mock_ps.return_value = _mock_session("hello", ".exit")
         mock_chat.side_effect = [
             {
@@ -236,11 +255,14 @@ def test_main_tool_call_edit(self, mock_ps, mock_chat, mock_cp, mock_load):
             if os.path.exists(tmp):
                 os.remove(tmp)
 
+    @patch("iclaw.main.http")
     @patch("iclaw.main.load_github_token", return_value="gt")
     @patch("iclaw.main.get_copilot_token", return_value="ct")
     @patch("iclaw.main.chat", return_value={"content": "hi"})
     @patch("iclaw.main.PromptSession")
-    def test_main_token_refresh(self, mock_ps, mock_chat, mock_cp, mock_load):
+    def test_main_token_refresh(
+        self, mock_ps, mock_chat, mock_cp, mock_load, mock_http
+    ):
         mock_ps.return_value = _mock_session("hello", ".exit")
         with (
             patch("sys.stdout"),
@@ -250,18 +272,30 @@ def test_main_token_refresh(self, mock_ps, mock_chat, mock_cp, mock_load):
         # get_copilot_token called twice: once at startup, once on refresh
         self.assertEqual(mock_cp.call_count, 2)
 
+    @patch("iclaw.main.http")
     @patch("iclaw.main.load_github_token", return_value="gt")
     @patch("iclaw.main.get_copilot_token", return_value="ct")
     @patch(
         "iclaw.main.handle_model_provider_command",
         return_value=("copilot", "new_token"),
     )
     @patch("iclaw.main.PromptSession")
-    def test_main_model_provider_with_token(self, mock_ps, mock_mp, mock_cp, mock_load):
+    def test_main_model_provider_with_token(
+        self, mock_ps, mock_mp, mock_cp, mock_load, mock_http
+    ):
         mock_ps.return_value = _mock_session("/model_provider", ".exit")
         with patch("sys.stdout"), patch("iclaw.main.time.monotonic", return_value=0):
             main.main()
 
+    @patch("iclaw.main.http")
+    @patch("iclaw.main.load_github_token", return_value="gt")
+    @patch("iclaw.main.get_copilot_token", return_value="ct")
+    @patch("iclaw.main.PromptSession")
+    def test_main_status(self, mock_ps, mock_cp, mock_load, mock_http):
+        mock_ps.return_value = _mock_session("/status", ".exit")
+        with patch("sys.stdout"), patch("iclaw.main.time.monotonic", return_value=0):
+            main.main()
+
 
 if __name__ == "__main__":
     unittest.main()