remote-claude 0.1.5 → 0.1.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "remote-claude",
-  "version": "0.1.5",
+  "version": "0.1.6",
   "description": "双端共享 Claude CLI 工具",
   "bin": {
     "remote-claude": "bin/remote-claude",

package/remote_claude.py

@@ -30,7 +30,8 @@ from utils.session import (
     list_active_sessions, is_session_active, cleanup_session,
     is_lark_running, get_lark_pid, get_lark_status, get_lark_pid_file,
     save_lark_status, cleanup_lark,
-    USER_DATA_DIR, ensure_user_data_dir, get_lark_log_file
+    USER_DATA_DIR, ensure_user_data_dir, get_lark_log_file,
+    get_env_snapshot_path
 )
 
 
@@ -56,6 +57,15 @@ def cmd_start(args):
 
     ensure_socket_dir()
 
+    # 将当前 shell 的完整环境变量保存到快照文件（权限 0600 防止密钥泄露）
+    # tmux new-session 继承的是 tmux 服务器的全局环境，而非调用方 shell 的环境，
+    # 通过快照文件将完整环境传递给 server.py 的 _start_pty()
+    import json
+    env_snapshot_path = get_env_snapshot_path(session_name)
+    env_fd = os.open(str(env_snapshot_path), os.O_WRONLY | os.O_CREAT | os.O_TRUNC, 0o600)
+    with os.fdopen(env_fd, 'w') as f:
+        json.dump(dict(os.environ), f)
+
     # 构建 server 命令
     server_script = SCRIPT_DIR / "server" / "server.py"
     claude_args = args.claude_args if args.claude_args else []

package/server/server.py

@@ -606,9 +606,32 @@ class ProxyServer:
 
     def _start_pty(self):
         """启动 PTY 并运行 Claude"""
-        pid, fd = pty.fork()
+        # 加载环境变量快照（从 cmd_start 保存的快照文件恢复调用方 shell 的完整环境）
+        from utils.session import get_env_snapshot_path
+        import json as _json
+        env_snapshot_path = get_env_snapshot_path(self.session_name)
+        _extra_env = {}
+        try:
+            with open(env_snapshot_path) as _f:
+                _extra_env = _json.load(_f)
+        except Exception:
+            pass
+
+        try:
+            pid, fd = pty.fork()
+        except OSError:
+            env_snapshot_path.unlink(missing_ok=True)
+            raise
 
         if pid == 0:
+            # 子进程：恢复调用方 shell 的完整环境变量
+            for k, v in _extra_env.items():
+                os.environ[k] = v
+            # 环境已加载到内存，立即删除快照文件（execvp 前销毁）
+            try:
+                env_snapshot_path.unlink()
+            except Exception:
+                pass
             # 恢复 TERM 以支持 kitty keyboard protocol（Shift+Enter 等扩展键）
             # tmux 会将 TERM 改为 tmux-256color，导致 Claude CLI 不启用 kitty protocol
             os.environ['TERM'] = 'xterm-256color'
@@ -616,6 +639,7 @@ class ProxyServer:
             for key in ('TMUX', 'TMUX_PANE'):
                 os.environ.pop(key, None)
             os.execvp("claude", ["claude"] + self.claude_args)
+            os._exit(1)  # execvp 失败时兜底退出
         else:
             # 父进程
             self.master_fd = fd

package/utils/session.py

@@ -60,6 +60,11 @@ def get_mq_path(session_name: str) -> Path:
     return SOCKET_DIR / f"{_safe_filename(session_name)}.mq"
 
 
+def get_env_snapshot_path(session_name: str) -> Path:
+    """获取环境变量快照文件路径（cmd_start 写入，_start_pty 读取后删除）"""
+    return SOCKET_DIR / f"{_safe_filename(session_name)}_env.json"
+
+
 def ensure_socket_dir():
     """确保 socket 目录存在"""
     SOCKET_DIR.mkdir(parents=True, exist_ok=True)
@@ -248,6 +253,7 @@ def cleanup_session(session_name: str):
     sock_path.unlink(missing_ok=True)
     pid_file.unlink(missing_ok=True)
     get_mq_path(session_name).unlink(missing_ok=True)
+    get_env_snapshot_path(session_name).unlink(missing_ok=True)
 
 
 def is_session_active(session_name: str) -> bool: