Compare commits
4 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 71b20e6e4a | |||
| 7709292f1c | |||
| b93b2ea36e | |||
| 44eafff757 |
16
CHANGELOG.md
16
CHANGELOG.md
@@ -3,6 +3,22 @@
|
|||||||
本文件对应技能市场 **「更新日志」** Tab。
|
本文件对应技能市场 **「更新日志」** Tab。
|
||||||
发布 CI 会按当前 tag 版本提取匹配的 `##` 小节。
|
发布 CI 会按当前 tag 版本提取匹配的 `##` 小节。
|
||||||
|
|
||||||
|
## 2.0.29
|
||||||
|
|
||||||
|
- 「打开浏览器」有可恢复会话时改用 `--restore-last-session` 强制恢复标签与 Session Cookie;不再裸写受 Secure Preferences HMAC 保护的 `restore_on_startup`(该写法会被 Chrome 否掉并落到新标签页)
|
||||||
|
|
||||||
|
## 2.0.28
|
||||||
|
|
||||||
|
- 「打开浏览器」识别现代 Chrome `Default/Sessions/Session_*`(及 `Tabs_*`),有可恢复会话时不再强制带平台首页 URL,避免冲掉 Session Cookie 登录态
|
||||||
|
|
||||||
|
## 2.0.27
|
||||||
|
|
||||||
|
- 「打开浏览器」:Profile 已占用时复用窗口;加固 Preferences/Local State「继续上次会话」;有历史会话时冷启动不带 URL,避免冲掉登录态
|
||||||
|
|
||||||
|
## 2.0.26
|
||||||
|
|
||||||
|
- 「打开浏览器」启动前写入 Profile「继续上次会话」,减轻独立 Profile 下 Session Cookie 关窗后丢失(科力普等站点更明显)
|
||||||
|
|
||||||
## 2.0.25
|
## 2.0.25
|
||||||
|
|
||||||
- 「账号租约」表新增行右键「强制释放租约」:仅状态为 active 时显示,确认后立刻释放,无需租约令牌
|
- 「账号租约」表新增行右键「强制释放租约」:仅状态为 active 时显示,确认后立刻释放,无需租约令牌
|
||||||
|
|||||||
2
SKILL.md
2
SKILL.md
@@ -1,7 +1,7 @@
|
|||||||
---
|
---
|
||||||
name: 账号与凭据管理
|
name: 账号与凭据管理
|
||||||
description: "通用账号 & 凭据 & 登录流程管理器。管理网页/RPA 账号、API Key/Token;支持 9 种登录策略(账密自动、扫码、2FA、SSO 等);提供加密本地存储(Fernet);提供 rpa_helpers 库供业务 skill 同进程调用 ensure_logged_in 完成多平台自动登录。涵盖物流、LLM、内容、开发、API 五类平台。"
|
description: "通用账号 & 凭据 & 登录流程管理器。管理网页/RPA 账号、API Key/Token;支持 9 种登录策略(账密自动、扫码、2FA、SSO 等);提供加密本地存储(Fernet);提供 rpa_helpers 库供业务 skill 同进程调用 ensure_logged_in 完成多平台自动登录。涵盖物流、LLM、内容、开发、API 五类平台。"
|
||||||
version: 2.0.25
|
version: 2.0.29
|
||||||
author: 深圳匠厂科技有限公司
|
author: 深圳匠厂科技有限公司
|
||||||
metadata:
|
metadata:
|
||||||
openclaw:
|
openclaw:
|
||||||
|
|||||||
@@ -59,4 +59,4 @@ description: "通用账号 & 凭据 & 登录流程管理器。管理网页/RPA
|
|||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
**作者**:深圳匠厂科技有限公司 · **版本**:2.0.25
|
**作者**:深圳匠厂科技有限公司 · **版本**:2.0.27
|
||||||
|
|||||||
@@ -14,6 +14,10 @@ import sys
|
|||||||
|
|
||||||
from db.connection import get_conn, init_db
|
from db.connection import get_conn, init_db
|
||||||
from db.accounts_repo import get_account_by_id, get_platform_from_db
|
from db.accounts_repo import get_account_by_id, get_platform_from_db
|
||||||
|
from service.chrome_profile_prefs import (
|
||||||
|
ensure_session_restore,
|
||||||
|
profile_has_restorable_session,
|
||||||
|
)
|
||||||
from util.logging_config import get_skill_logger
|
from util.logging_config import get_skill_logger
|
||||||
from util.platforms import PLATFORMS
|
from util.platforms import PLATFORMS
|
||||||
|
|
||||||
@@ -83,21 +87,29 @@ HUMAN_CHROME_FLAGS = (
|
|||||||
"--disable-search-engine-choice-screen",
|
"--disable-search-engine-choice-screen",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# 强制恢复上次会话(含 Session Cookie)。不依赖 Preferences.restore_on_startup
|
||||||
|
#(该项受 Secure Preferences HMAC 保护,裸写会被 Chrome 否掉)。
|
||||||
|
RESTORE_LAST_SESSION_FLAG = "--restore-last-session"
|
||||||
|
|
||||||
def human_chrome_launch_args(profile_dir, url=None):
|
|
||||||
|
def human_chrome_launch_args(profile_dir, url=None, restore_session=False):
|
||||||
"""chrome.exe 之后的参数(Popen 列表形式,路径不额外加引号)。"""
|
"""chrome.exe 之后的参数(Popen 列表形式,路径不额外加引号)。"""
|
||||||
profile_abs = os.path.abspath(profile_dir)
|
profile_abs = os.path.abspath(profile_dir)
|
||||||
args = [f"--user-data-dir={profile_abs}", *HUMAN_CHROME_FLAGS]
|
args = [f"--user-data-dir={profile_abs}", *HUMAN_CHROME_FLAGS]
|
||||||
|
if restore_session:
|
||||||
|
args.append(RESTORE_LAST_SESSION_FLAG)
|
||||||
cleaned = (url or "").strip()
|
cleaned = (url or "").strip()
|
||||||
if cleaned:
|
if cleaned:
|
||||||
args.append(cleaned)
|
args.append(cleaned)
|
||||||
return args
|
return args
|
||||||
|
|
||||||
|
|
||||||
def human_chrome_shortcut_arguments(profile_dir, url=None):
|
def human_chrome_shortcut_arguments(profile_dir, url=None, restore_session=False):
|
||||||
"""Windows .lnk Arguments 字符串(user-data-dir 带引号)。"""
|
"""Windows .lnk Arguments 字符串(user-data-dir 带引号)。"""
|
||||||
profile_abs = os.path.abspath(profile_dir)
|
profile_abs = os.path.abspath(profile_dir)
|
||||||
parts = [f'--user-data-dir="{profile_abs}"', *HUMAN_CHROME_FLAGS]
|
parts = [f'--user-data-dir="{profile_abs}"', *HUMAN_CHROME_FLAGS]
|
||||||
|
if restore_session:
|
||||||
|
parts.append(RESTORE_LAST_SESSION_FLAG)
|
||||||
cleaned = (url or "").strip()
|
cleaned = (url or "").strip()
|
||||||
if cleaned:
|
if cleaned:
|
||||||
parts.append(cleaned)
|
parts.append(cleaned)
|
||||||
@@ -191,6 +203,51 @@ def cmd_reveal_profile(account_id):
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def _list_profile_browser_pids(profile_dir: str):
|
||||||
|
"""找出已占用本 Profile 的 chrome/msedge PID(与采集技能同逻辑)。"""
|
||||||
|
import json as _json
|
||||||
|
|
||||||
|
profile_abs = os.path.abspath(profile_dir)
|
||||||
|
needle = os.path.normcase(profile_abs)
|
||||||
|
if sys.platform != "win32":
|
||||||
|
return []
|
||||||
|
ps = (
|
||||||
|
"Get-CimInstance Win32_Process -Filter "
|
||||||
|
"\"name='chrome.exe' or name='msedge.exe'\" |"
|
||||||
|
" Select-Object ProcessId,CommandLine | ConvertTo-Json -Compress"
|
||||||
|
)
|
||||||
|
try:
|
||||||
|
completed = subprocess.run(
|
||||||
|
["powershell", "-NoProfile", "-Command", ps],
|
||||||
|
capture_output=True,
|
||||||
|
text=True,
|
||||||
|
timeout=20,
|
||||||
|
check=False,
|
||||||
|
)
|
||||||
|
except Exception:
|
||||||
|
return []
|
||||||
|
raw = (completed.stdout or "").strip()
|
||||||
|
if not raw:
|
||||||
|
return []
|
||||||
|
try:
|
||||||
|
payload = _json.loads(raw)
|
||||||
|
except Exception:
|
||||||
|
return []
|
||||||
|
rows = payload if isinstance(payload, list) else [payload]
|
||||||
|
pids = []
|
||||||
|
for row in rows:
|
||||||
|
if not isinstance(row, dict):
|
||||||
|
continue
|
||||||
|
cmdline = str(row.get("CommandLine") or "")
|
||||||
|
if needle not in os.path.normcase(cmdline):
|
||||||
|
continue
|
||||||
|
try:
|
||||||
|
pids.append(int(row.get("ProcessId")))
|
||||||
|
except (TypeError, ValueError):
|
||||||
|
continue
|
||||||
|
return sorted(set(pids))
|
||||||
|
|
||||||
|
|
||||||
def cmd_open(account_id):
|
def cmd_open(account_id):
|
||||||
"""
|
"""
|
||||||
打开该账号的持久化浏览器,仅用于肉眼核对 / 手工登录 / 抓 DOM。
|
打开该账号的持久化浏览器,仅用于肉眼核对 / 手工登录 / 抓 DOM。
|
||||||
@@ -225,20 +282,68 @@ def cmd_open(account_id):
|
|||||||
profile_abs = os.path.abspath(profile_dir)
|
profile_abs = os.path.abspath(profile_dir)
|
||||||
os.makedirs(profile_abs, exist_ok=True)
|
os.makedirs(profile_abs, exist_ok=True)
|
||||||
|
|
||||||
|
existing = _list_profile_browser_pids(profile_abs)
|
||||||
|
if existing:
|
||||||
|
log.info(
|
||||||
|
"browser_open_reuse account_id=%s profile_dir=%s pids=%s",
|
||||||
|
account_id,
|
||||||
|
profile_abs,
|
||||||
|
existing,
|
||||||
|
)
|
||||||
|
print(
|
||||||
|
json.dumps(
|
||||||
|
{
|
||||||
|
"type": "result",
|
||||||
|
"schema_version": 1,
|
||||||
|
"status": "success",
|
||||||
|
"message": "已复用打开中的浏览器窗口",
|
||||||
|
"success": True,
|
||||||
|
"data": {
|
||||||
|
"account_id": int(target["id"]) if target.get("id") is not None else account_id,
|
||||||
|
"account_label": target.get("account_label"),
|
||||||
|
"browser": "Google Chrome",
|
||||||
|
"browser_exe": browser_exe,
|
||||||
|
"profile_dir": profile_abs,
|
||||||
|
"url": resolve_account_browser_url(target),
|
||||||
|
"pid": existing[0],
|
||||||
|
"reused": True,
|
||||||
|
"pids": existing,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
ensure_ascii=False,
|
||||||
|
)
|
||||||
|
)
|
||||||
|
return
|
||||||
|
|
||||||
|
# 清理 exit_type 等(不写 MAC 保护的 restore_on_startup);有会话时靠 --restore-last-session
|
||||||
|
prefs_result = ensure_session_restore(profile_abs)
|
||||||
|
log.info("browser_open_prefs %s", prefs_result)
|
||||||
|
|
||||||
conn = get_conn()
|
conn = get_conn()
|
||||||
try:
|
try:
|
||||||
url = resolve_account_browser_url(target, conn=conn)
|
url = resolve_account_browser_url(target, conn=conn)
|
||||||
finally:
|
finally:
|
||||||
conn.close()
|
conn.close()
|
||||||
|
|
||||||
# 与 profile_shortcut 同一套人用参数:独立 user-data-dir 保存登录态
|
# 有可恢复会话:不带 URL + --restore-last-session(恢复标签与 Session Cookie)。
|
||||||
cmd = [browser_exe, *human_chrome_launch_args(profile_abs, url)]
|
# 首次/空 Profile:仍打开平台首页便于登录。
|
||||||
|
has_session = profile_has_restorable_session(profile_abs)
|
||||||
|
launch_url = None if has_session else url
|
||||||
|
cmd = [
|
||||||
|
browser_exe,
|
||||||
|
*human_chrome_launch_args(
|
||||||
|
profile_abs, launch_url, restore_session=has_session
|
||||||
|
),
|
||||||
|
]
|
||||||
|
|
||||||
log.info(
|
log.info(
|
||||||
"browser_open_launch exe=%s profile_dir=%s url=%s platform=%s",
|
"browser_open_launch exe=%s profile_dir=%s launch_url=%s has_session=%s "
|
||||||
|
"restore_last_session=%s platform=%s",
|
||||||
browser_exe,
|
browser_exe,
|
||||||
profile_abs,
|
profile_abs,
|
||||||
url,
|
launch_url or "(restore)",
|
||||||
|
has_session,
|
||||||
|
has_session,
|
||||||
target.get("platform_key"),
|
target.get("platform_key"),
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -289,6 +394,9 @@ def cmd_open(account_id):
|
|||||||
"profile_dir": profile_abs,
|
"profile_dir": profile_abs,
|
||||||
"url": url,
|
"url": url,
|
||||||
"pid": child.pid,
|
"pid": child.pid,
|
||||||
|
"has_session": has_session,
|
||||||
|
"restore_last_session": has_session,
|
||||||
|
"session_restore_on_startup": prefs_result.get("restore_after"),
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
ensure_ascii=False,
|
ensure_ascii=False,
|
||||||
|
|||||||
163
scripts/service/chrome_profile_prefs.py
Normal file
163
scripts/service/chrome_profile_prefs.py
Normal file
@@ -0,0 +1,163 @@
|
|||||||
|
"""Chrome/Edge 独立 Profile:启动前清理退出标记,便于会话文件可被恢复。
|
||||||
|
|
||||||
|
注意:``session.restore_on_startup`` 受 Secure Preferences HMAC 保护,
|
||||||
|
裸写 Preferences 会被 Chrome 校验否掉并回落到新标签页。真正强制恢复会话
|
||||||
|
应在启动参数使用 ``--restore-last-session``(见 browser_service)。
|
||||||
|
"""
|
||||||
|
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import json
|
||||||
|
import os
|
||||||
|
from typing import Any, Dict
|
||||||
|
|
||||||
|
from util.logging_config import get_skill_logger
|
||||||
|
|
||||||
|
# Chromium ≤ roughly mid-era: session files lived under Default/
|
||||||
|
_LEGACY_SESSION_FILES = (
|
||||||
|
"Current Session",
|
||||||
|
"Last Session",
|
||||||
|
"Current Tabs",
|
||||||
|
"Last Tabs",
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def profile_has_restorable_session(profile_dir: str) -> bool:
|
||||||
|
"""True when the profile has on-disk session data Chrome can restore.
|
||||||
|
|
||||||
|
Modern Chrome/Edge write ``Default/Sessions/Session_*`` (and ``Tabs_*``).
|
||||||
|
Older builds used ``Default/Current Session`` etc. Either form counts.
|
||||||
|
"""
|
||||||
|
default_dir = os.path.join(os.path.abspath(profile_dir), "Default")
|
||||||
|
if any(
|
||||||
|
os.path.isfile(os.path.join(default_dir, name))
|
||||||
|
for name in _LEGACY_SESSION_FILES
|
||||||
|
):
|
||||||
|
return True
|
||||||
|
|
||||||
|
sessions_dir = os.path.join(default_dir, "Sessions")
|
||||||
|
if not os.path.isdir(sessions_dir):
|
||||||
|
return False
|
||||||
|
try:
|
||||||
|
names = os.listdir(sessions_dir)
|
||||||
|
except OSError:
|
||||||
|
return False
|
||||||
|
return any(
|
||||||
|
name.startswith("Session_") or name.startswith("Tabs_")
|
||||||
|
for name in names
|
||||||
|
if os.path.isfile(os.path.join(sessions_dir, name))
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def preferences_path(profile_dir: str) -> str:
|
||||||
|
return os.path.join(os.path.abspath(profile_dir), "Default", "Preferences")
|
||||||
|
|
||||||
|
|
||||||
|
def local_state_path(profile_dir: str) -> str:
|
||||||
|
return os.path.join(os.path.abspath(profile_dir), "Local State")
|
||||||
|
|
||||||
|
|
||||||
|
def _patch_preferences_dict(data: Dict[str, Any]) -> Dict[str, Any]:
|
||||||
|
"""只改不受 Secure Preferences MAC 保护的退出标记;不动 restore_on_startup。"""
|
||||||
|
restore_before = int((data.get("session") or {}).get("restore_on_startup") or 0)
|
||||||
|
exit_before = str((data.get("profile") or {}).get("exit_type") or "")
|
||||||
|
|
||||||
|
profile = data.setdefault("profile", {})
|
||||||
|
if str(profile.get("exit_type") or "").lower() == "crashed":
|
||||||
|
profile["exit_type"] = "Normal"
|
||||||
|
profile["exited_cleanly"] = True
|
||||||
|
|
||||||
|
browser = data.setdefault("browser", {})
|
||||||
|
browser["has_seen_welcome_page"] = True
|
||||||
|
browser["check_default_browser"] = False
|
||||||
|
|
||||||
|
return {"restore_before": restore_before, "exit_type_before": exit_before}
|
||||||
|
|
||||||
|
|
||||||
|
def _patch_local_state(profile_dir: str, log) -> Dict[str, Any]:
|
||||||
|
path = local_state_path(profile_dir)
|
||||||
|
if not os.path.isfile(path):
|
||||||
|
return {"ok": True, "skipped": True, "path": path}
|
||||||
|
try:
|
||||||
|
with open(path, "r", encoding="utf-8") as fh:
|
||||||
|
data = json.load(fh)
|
||||||
|
except Exception as exc:
|
||||||
|
log.warning("chrome_local_state_read_failed path=%s err=%s", path, exc)
|
||||||
|
return {"ok": False, "path": path, "error": str(exc)}
|
||||||
|
|
||||||
|
profile = data.setdefault("profile", {})
|
||||||
|
info = profile.setdefault("info_cache", {}).setdefault("Default", {})
|
||||||
|
info["active_time"] = float(info.get("active_time") or 0.0) or 1.0
|
||||||
|
exited = profile.get("exited_cleanly", True)
|
||||||
|
profile["exited_cleanly"] = True
|
||||||
|
try:
|
||||||
|
with open(path, "w", encoding="utf-8") as fh:
|
||||||
|
json.dump(data, fh, ensure_ascii=False, separators=(",", ":"))
|
||||||
|
except Exception as exc:
|
||||||
|
log.warning("chrome_local_state_write_failed path=%s err=%s", path, exc)
|
||||||
|
return {"ok": False, "path": path, "error": str(exc), "exited_cleanly_before": exited}
|
||||||
|
|
||||||
|
log.info("chrome_local_state_patched path=%s exited_cleanly %s->True", path, exited)
|
||||||
|
return {"ok": True, "path": path, "exited_cleanly_before": exited}
|
||||||
|
|
||||||
|
|
||||||
|
def ensure_session_restore(profile_dir: str) -> Dict[str, Any]:
|
||||||
|
"""启动前清理 exit_type / exited_cleanly(不受 MAC 保护);不写 restore_on_startup。"""
|
||||||
|
log = get_skill_logger()
|
||||||
|
profile_abs = os.path.abspath(profile_dir)
|
||||||
|
prefs = preferences_path(profile_abs)
|
||||||
|
os.makedirs(os.path.dirname(prefs), exist_ok=True)
|
||||||
|
data: Dict[str, Any] = {}
|
||||||
|
existed = os.path.isfile(prefs)
|
||||||
|
if existed:
|
||||||
|
try:
|
||||||
|
with open(prefs, "r", encoding="utf-8") as fh:
|
||||||
|
data = json.load(fh)
|
||||||
|
except Exception:
|
||||||
|
log.exception("chrome_prefs_read_failed path=%s", prefs)
|
||||||
|
data = {}
|
||||||
|
|
||||||
|
meta = _patch_preferences_dict(data)
|
||||||
|
before = int(meta["restore_before"])
|
||||||
|
exit_before = str(meta["exit_type_before"] or "")
|
||||||
|
|
||||||
|
try:
|
||||||
|
with open(prefs, "w", encoding="utf-8") as fh:
|
||||||
|
json.dump(data, fh, ensure_ascii=False, separators=(",", ":"))
|
||||||
|
except Exception as exc:
|
||||||
|
log.warning("chrome_prefs_write_failed path=%s err=%s", prefs, exc)
|
||||||
|
return {
|
||||||
|
"ok": False,
|
||||||
|
"path": prefs,
|
||||||
|
"existed": existed,
|
||||||
|
"restore_before": before,
|
||||||
|
"restore_after": before,
|
||||||
|
"exit_type_before": exit_before,
|
||||||
|
"error": str(exc),
|
||||||
|
}
|
||||||
|
|
||||||
|
after = int((data.get("session") or {}).get("restore_on_startup") or 0)
|
||||||
|
exit_after = str((data.get("profile") or {}).get("exit_type") or "")
|
||||||
|
local = _patch_local_state(profile_abs, log)
|
||||||
|
log.info(
|
||||||
|
"chrome_session_restore_prepared profile_dir=%s prefs=%s existed=%s "
|
||||||
|
"restore_on_startup_unchanged=%s exit_type %s->%s local_state=%s",
|
||||||
|
profile_abs,
|
||||||
|
prefs,
|
||||||
|
existed,
|
||||||
|
after,
|
||||||
|
exit_before or "-",
|
||||||
|
exit_after or "-",
|
||||||
|
local,
|
||||||
|
)
|
||||||
|
return {
|
||||||
|
"ok": True,
|
||||||
|
"path": prefs,
|
||||||
|
"existed": existed,
|
||||||
|
"restore_before": before,
|
||||||
|
"restore_after": after,
|
||||||
|
"exit_type_before": exit_before,
|
||||||
|
"exit_type_after": exit_after,
|
||||||
|
"local_state": local,
|
||||||
|
"prefs_restore_on_startup_touched": False,
|
||||||
|
}
|
||||||
@@ -18,16 +18,76 @@ class TestPlaywrightLaunchHelpers:
|
|||||||
assert "--start-maximized" in args
|
assert "--start-maximized" in args
|
||||||
|
|
||||||
|
|
||||||
|
class TestEnsureSessionRestoreDoesNotTouchProtectedPref:
|
||||||
|
def test_leaves_restore_on_startup_unchanged(self, tmp_path, fake_env):
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
from service.chrome_profile_prefs import ensure_session_restore, preferences_path
|
||||||
|
from util.logging_config import setup_skill_logging
|
||||||
|
|
||||||
|
setup_skill_logging("account-manager", "openclaw.skill.account_manager_test")
|
||||||
|
|
||||||
|
profile = tmp_path / "prof"
|
||||||
|
prefs = Path(preferences_path(str(profile)))
|
||||||
|
prefs.parent.mkdir(parents=True, exist_ok=True)
|
||||||
|
prefs.write_text(
|
||||||
|
json.dumps(
|
||||||
|
{
|
||||||
|
"session": {"restore_on_startup": 5},
|
||||||
|
"profile": {"exit_type": "Crashed", "exited_cleanly": False},
|
||||||
|
}
|
||||||
|
),
|
||||||
|
encoding="utf-8",
|
||||||
|
)
|
||||||
|
|
||||||
|
result = ensure_session_restore(str(profile))
|
||||||
|
assert result["ok"] is True
|
||||||
|
assert result["prefs_restore_on_startup_touched"] is False
|
||||||
|
data = json.loads(prefs.read_text(encoding="utf-8"))
|
||||||
|
assert data["session"]["restore_on_startup"] == 5
|
||||||
|
assert data["profile"]["exit_type"] == "Normal"
|
||||||
|
assert data["profile"]["exited_cleanly"] is True
|
||||||
|
|
||||||
|
|
||||||
|
class TestProfileHasRestorableSession:
|
||||||
|
def test_empty_profile_false(self, tmp_path):
|
||||||
|
from service.chrome_profile_prefs import profile_has_restorable_session
|
||||||
|
|
||||||
|
assert profile_has_restorable_session(str(tmp_path / "prof")) is False
|
||||||
|
|
||||||
|
def test_legacy_current_session(self, tmp_path):
|
||||||
|
from service.chrome_profile_prefs import profile_has_restorable_session
|
||||||
|
|
||||||
|
default = tmp_path / "prof" / "Default"
|
||||||
|
default.mkdir(parents=True)
|
||||||
|
(default / "Current Session").write_bytes(b"x")
|
||||||
|
assert profile_has_restorable_session(str(tmp_path / "prof")) is True
|
||||||
|
|
||||||
|
def test_modern_sessions_dir(self, tmp_path):
|
||||||
|
from service.chrome_profile_prefs import profile_has_restorable_session
|
||||||
|
|
||||||
|
sessions = tmp_path / "prof" / "Default" / "Sessions"
|
||||||
|
sessions.mkdir(parents=True)
|
||||||
|
(sessions / "Session_13434010951759956").write_bytes(b"sess")
|
||||||
|
assert profile_has_restorable_session(str(tmp_path / "prof")) is True
|
||||||
|
|
||||||
|
def test_modern_tabs_only(self, tmp_path):
|
||||||
|
from service.chrome_profile_prefs import profile_has_restorable_session
|
||||||
|
|
||||||
|
sessions = tmp_path / "prof" / "Default" / "Sessions"
|
||||||
|
sessions.mkdir(parents=True)
|
||||||
|
(sessions / "Tabs_13434010952144072").write_bytes(b"tabs")
|
||||||
|
assert profile_has_restorable_session(str(tmp_path / "prof")) is True
|
||||||
|
|
||||||
|
|
||||||
class TestCmdOpenDirectBrowser:
|
class TestCmdOpenDirectBrowser:
|
||||||
"""account open = same-process Chrome/Edge launch (like reveal-profile)."""
|
"""account open = same-process Chrome/Edge launch (like reveal-profile)."""
|
||||||
|
|
||||||
def test_cmd_open_launches_browser_exe(self, clean_db, tmp_path, capsys, monkeypatch):
|
def _seed_account(self, clean_db, tmp_path, profile):
|
||||||
from db.accounts_repo import insert_account
|
from db.accounts_repo import insert_account
|
||||||
from service import browser_service
|
|
||||||
from util.platforms import seed_platforms
|
from util.platforms import seed_platforms
|
||||||
|
|
||||||
seed_platforms(clean_db)
|
seed_platforms(clean_db)
|
||||||
profile = tmp_path / "prof"
|
|
||||||
aid = insert_account(
|
aid = insert_account(
|
||||||
clean_db,
|
clean_db,
|
||||||
platform_key="sohu",
|
platform_key="sohu",
|
||||||
@@ -45,7 +105,9 @@ class TestCmdOpenDirectBrowser:
|
|||||||
auth_strategy="password_auto",
|
auth_strategy="password_auto",
|
||||||
)
|
)
|
||||||
clean_db.commit()
|
clean_db.commit()
|
||||||
|
return aid
|
||||||
|
|
||||||
|
def _patch_launch(self, monkeypatch, browser_service, tmp_path):
|
||||||
fake_exe = str(tmp_path / "chrome.exe")
|
fake_exe = str(tmp_path / "chrome.exe")
|
||||||
monkeypatch.setattr(
|
monkeypatch.setattr(
|
||||||
browser_service,
|
browser_service,
|
||||||
@@ -64,6 +126,14 @@ class TestCmdOpenDirectBrowser:
|
|||||||
return FakeProc()
|
return FakeProc()
|
||||||
|
|
||||||
monkeypatch.setattr(browser_service.subprocess, "Popen", fake_popen)
|
monkeypatch.setattr(browser_service.subprocess, "Popen", fake_popen)
|
||||||
|
return fake_exe, spawned
|
||||||
|
|
||||||
|
def test_cmd_open_launches_browser_exe(self, clean_db, tmp_path, capsys, monkeypatch):
|
||||||
|
from service import browser_service
|
||||||
|
|
||||||
|
profile = tmp_path / "prof"
|
||||||
|
aid = self._seed_account(clean_db, tmp_path, profile)
|
||||||
|
fake_exe, spawned = self._patch_launch(monkeypatch, browser_service, tmp_path)
|
||||||
|
|
||||||
browser_service.cmd_open(str(aid))
|
browser_service.cmd_open(str(aid))
|
||||||
out = capsys.readouterr().out.strip()
|
out = capsys.readouterr().out.strip()
|
||||||
@@ -88,6 +158,30 @@ class TestCmdOpenDirectBrowser:
|
|||||||
assert "open_child_runner" not in joined
|
assert "open_child_runner" not in joined
|
||||||
assert "python" not in joined.lower()
|
assert "python" not in joined.lower()
|
||||||
|
|
||||||
|
def test_cmd_open_skips_url_when_modern_sessions_exist(
|
||||||
|
self, clean_db, tmp_path, capsys, monkeypatch
|
||||||
|
):
|
||||||
|
"""Modern Chrome Sessions/ must restore via --restore-last-session, no homepage URL."""
|
||||||
|
from service import browser_service
|
||||||
|
|
||||||
|
profile = tmp_path / "prof"
|
||||||
|
sessions = profile / "Default" / "Sessions"
|
||||||
|
sessions.mkdir(parents=True)
|
||||||
|
(sessions / "Session_13434010951759956").write_bytes(b"sess")
|
||||||
|
aid = self._seed_account(clean_db, tmp_path, profile)
|
||||||
|
_fake_exe, spawned = self._patch_launch(monkeypatch, browser_service, tmp_path)
|
||||||
|
|
||||||
|
browser_service.cmd_open(str(aid))
|
||||||
|
out = capsys.readouterr().out.strip()
|
||||||
|
payload = json.loads(out.splitlines()[-1])
|
||||||
|
|
||||||
|
assert payload["success"] is True
|
||||||
|
assert payload["data"]["has_session"] is True
|
||||||
|
assert payload["data"]["restore_last_session"] is True
|
||||||
|
assert "https://example.com" not in spawned["cmd"]
|
||||||
|
assert "--restore-last-session" in spawned["cmd"]
|
||||||
|
assert spawned["cmd"][-1] == "--restore-last-session" or spawned["cmd"][-1].startswith("--")
|
||||||
|
|
||||||
|
|
||||||
class TestHumanChromeLaunchArgs:
|
class TestHumanChromeLaunchArgs:
|
||||||
def test_flags_then_url(self, tmp_path):
|
def test_flags_then_url(self, tmp_path):
|
||||||
@@ -98,9 +192,17 @@ class TestHumanChromeLaunchArgs:
|
|||||||
assert args[0].startswith("--user-data-dir=")
|
assert args[0].startswith("--user-data-dir=")
|
||||||
for flag in HUMAN_CHROME_FLAGS:
|
for flag in HUMAN_CHROME_FLAGS:
|
||||||
assert flag in args
|
assert flag in args
|
||||||
|
assert "--restore-last-session" not in args
|
||||||
assert args[-1] == "https://chat.deepseek.com"
|
assert args[-1] == "https://chat.deepseek.com"
|
||||||
assert "--enable-automation" not in args
|
assert "--enable-automation" not in args
|
||||||
|
|
||||||
|
def test_restore_session_flag_without_url(self, tmp_path):
|
||||||
|
from service.browser_service import human_chrome_launch_args
|
||||||
|
|
||||||
|
args = human_chrome_launch_args(str(tmp_path / "prof"), None, restore_session=True)
|
||||||
|
assert "--restore-last-session" in args
|
||||||
|
assert not any(a.startswith("http") for a in args)
|
||||||
|
|
||||||
def test_shortcut_arguments_quote_profile_and_include_url(self, tmp_path):
|
def test_shortcut_arguments_quote_profile_and_include_url(self, tmp_path):
|
||||||
from service.browser_service import human_chrome_shortcut_arguments
|
from service.browser_service import human_chrome_shortcut_arguments
|
||||||
|
|
||||||
@@ -110,8 +212,16 @@ class TestHumanChromeLaunchArgs:
|
|||||||
assert '--user-data-dir="' in text
|
assert '--user-data-dir="' in text
|
||||||
assert "--start-maximized" in text
|
assert "--start-maximized" in text
|
||||||
assert "--no-first-run" in text
|
assert "--no-first-run" in text
|
||||||
|
assert "--restore-last-session" not in text
|
||||||
assert text.endswith("https://example.com")
|
assert text.endswith("https://example.com")
|
||||||
|
|
||||||
|
def test_shortcut_restore_session(self, tmp_path):
|
||||||
|
from service.browser_service import human_chrome_shortcut_arguments
|
||||||
|
|
||||||
|
text = human_chrome_shortcut_arguments(str(tmp_path / "p"), None, restore_session=True)
|
||||||
|
assert "--restore-last-session" in text
|
||||||
|
assert "http" not in text
|
||||||
|
|
||||||
|
|
||||||
class TestResolveChromiumChannel:
|
class TestResolveChromiumChannel:
|
||||||
"""resolve_chromium_channel returns a channel string when browsers exist."""
|
"""resolve_chromium_channel returns a channel string when browsers exist."""
|
||||||
|
|||||||
Reference in New Issue
Block a user