mirror of
https://gitee.com/mateos/mateclaw.git
synced 2026-09-13 11:13:43 +08:00
735 lines
29 KiB
Java
735 lines
29 KiB
Java
package vip.mate.tool.builtin;
|
||
|
||
import cn.hutool.http.HttpUtil;
|
||
import cn.hutool.json.JSONArray;
|
||
import cn.hutool.json.JSONObject;
|
||
import cn.hutool.json.JSONUtil;
|
||
import com.microsoft.playwright.*;
|
||
import com.microsoft.playwright.options.LoadState;
|
||
import jakarta.annotation.PreDestroy;
|
||
import lombok.extern.slf4j.Slf4j;
|
||
import org.springframework.ai.tool.annotation.Tool;
|
||
import org.springframework.ai.tool.annotation.ToolParam;
|
||
import org.springframework.stereotype.Component;
|
||
|
||
import java.net.Socket;
|
||
import java.nio.file.Paths;
|
||
import java.util.ArrayList;
|
||
import java.util.Base64;
|
||
import java.util.List;
|
||
import java.util.Locale;
|
||
import java.util.concurrent.*;
|
||
|
||
/**
|
||
* 浏览器自动化工具
|
||
* 基于 Playwright Java，实现 action-based 浏览器自动化 API。
|
||
* 支持 start / stop / open / snapshot / screenshot / click / type / eval / connect_cdp / list_cdp_targets。
|
||
*/
|
||
@Slf4j
|
||
@Component
|
||
public class BrowserUseTool {
|
||
|
||
private static final long IDLE_TIMEOUT_MINUTES = 30;
|
||
private static final int MAX_SNAPSHOT_LENGTH = 20_000;
|
||
private static final int CDP_SCAN_PORT_MIN = 9000;
|
||
private static final int CDP_SCAN_PORT_MAX = 10000;
|
||
|
||
private static final boolean IS_WINDOWS = System.getProperty("os.name", "")
|
||
.toLowerCase(Locale.ROOT).contains("win");
|
||
|
||
/**
|
||
* 共享 Playwright 实例（Node.js 进程）。
|
||
* Playwright.create() 启动一个 Node.js 子进程，耗时 1-2 秒。
|
||
* 复用同一实例可将后续 start/connect_cdp 的延迟从 ~98s 降至 ~1s。
|
||
*/
|
||
private volatile Playwright sharedPlaywright;
|
||
private final Object playwrightLock = new Object();
|
||
|
||
private final ConcurrentHashMap<String, BrowserSession> sessions = new ConcurrentHashMap<>();
|
||
private final ScheduledExecutorService scheduler = Executors.newSingleThreadScheduledExecutor(r -> {
|
||
Thread t = new Thread(r, "browser-idle-watchdog");
|
||
t.setDaemon(true);
|
||
return t;
|
||
});
|
||
|
||
@Tool(description = """
|
||
Control a browser (Playwright). Default is headless. Use headed=true with action=start for a visible window.
|
||
Typical flow: start → open(url) → snapshot → click/type → stop.
|
||
For CDP: connect_cdp(url="http://localhost:9222") to attach to an existing Chrome, or list_cdp_targets to scan.
|
||
|
||
Supported actions:
|
||
- start: Launch a new browser. Optional headed=true for visible window.
|
||
- stop: Close browser. If connected via CDP, only disconnects (Chrome keeps running).
|
||
- open: Navigate to a URL. Requires url parameter. Auto-starts browser if not running.
|
||
- snapshot: Get page text content, interactive elements, and title.
|
||
- screenshot: Take a screenshot. Optional path to save file; returns base64 if no path.
|
||
- click: Click an element. Requires selector (CSS selector).
|
||
- type: Type text into an element. Requires selector and text.
|
||
- eval: Execute JavaScript on the page. Requires code parameter.
|
||
- connect_cdp: Connect to an existing Chrome via CDP. Requires url (e.g. "http://localhost:9222").
|
||
- list_cdp_targets: Scan local ports (9000-10000) for CDP endpoints. Optional cdpPort for single port.
|
||
- navigate_back: Go back in browser history.
|
||
""")
|
||
public String browser_use(
|
||
@ToolParam(description = "Action: start|stop|open|snapshot|screenshot|click|type|eval|connect_cdp|list_cdp_targets|navigate_back") String action,
|
||
@ToolParam(description = "URL to navigate to (for open), or CDP base URL (for connect_cdp, e.g. http://localhost:9222)", required = false) String url,
|
||
@ToolParam(description = "CSS selector for target element (for click/type)", required = false) String selector,
|
||
@ToolParam(description = "Text to type (for action=type)", required = false) String text,
|
||
@ToolParam(description = "JavaScript code to execute (for action=eval)", required = false) String code,
|
||
@ToolParam(description = "File path to save screenshot (for action=screenshot)", required = false) String path,
|
||
@ToolParam(description = "Launch visible browser window (for action=start, default false)", required = false) Boolean headed,
|
||
@ToolParam(description = "Single CDP port to scan (for action=list_cdp_targets)", required = false) Integer cdpPort
|
||
) {
|
||
if (action == null || action.isBlank()) {
|
||
return error("action is required");
|
||
}
|
||
|
||
String sessionKey = "default";
|
||
log.info("[BrowserUse] action={}, url={}, selector={}, headed={}, cdpPort={}", action, url, selector, headed, cdpPort);
|
||
|
||
try {
|
||
return switch (action.toLowerCase().trim()) {
|
||
case "start" -> doStart(sessionKey, Boolean.TRUE.equals(headed));
|
||
case "stop" -> doStop(sessionKey);
|
||
case "open" -> doOpen(sessionKey, url);
|
||
case "snapshot" -> doSnapshot(sessionKey);
|
||
case "screenshot" -> doScreenshot(sessionKey, path);
|
||
case "click" -> doClick(sessionKey, selector);
|
||
case "type" -> doType(sessionKey, selector, text);
|
||
case "eval" -> doEval(sessionKey, code);
|
||
case "connect_cdp" -> doConnectCdp(sessionKey, url);
|
||
case "list_cdp_targets" -> doListCdpTargets(cdpPort);
|
||
case "navigate_back" -> doNavigateBack(sessionKey);
|
||
default -> error("Unknown action: " + action + ". Supported: start, stop, open, snapshot, screenshot, click, type, eval, connect_cdp, list_cdp_targets, navigate_back");
|
||
};
|
||
} catch (PlaywrightException e) {
|
||
log.error("[BrowserUse] Playwright error: {}", e.getMessage());
|
||
return error("Browser error: " + e.getMessage());
|
||
} catch (Exception e) {
|
||
log.error("[BrowserUse] Unexpected error: {}", e.getMessage(), e);
|
||
return error("Unexpected error: " + e.getMessage());
|
||
}
|
||
}
|
||
|
||
// ==================== Playwright Lifecycle ====================
|
||
|
||
/**
|
||
* 获取或创建共享 Playwright 实例（双重检查锁定）。
|
||
* 首次调用约 1-2s（启动 Node.js），后续调用 ~0ms。
|
||
*/
|
||
private Playwright getOrCreatePlaywright() {
|
||
Playwright pw = sharedPlaywright;
|
||
if (pw != null) {
|
||
return pw;
|
||
}
|
||
synchronized (playwrightLock) {
|
||
pw = sharedPlaywright;
|
||
if (pw != null) {
|
||
return pw;
|
||
}
|
||
log.info("[BrowserUse] Creating shared Playwright instance...");
|
||
long start = System.currentTimeMillis();
|
||
pw = Playwright.create();
|
||
sharedPlaywright = pw;
|
||
log.info("[BrowserUse] Playwright instance created in {}ms", System.currentTimeMillis() - start);
|
||
return pw;
|
||
}
|
||
}
|
||
|
||
// ==================== Action Handlers ====================
|
||
|
||
private String doStart(String sessionKey, boolean headed) {
|
||
BrowserSession existing = sessions.get(sessionKey);
|
||
if (existing != null && existing.isAlive()) {
|
||
if (existing.headed == headed) {
|
||
existing.touch();
|
||
return ok("Browser already running (headed=" + headed + ")");
|
||
}
|
||
doStop(sessionKey);
|
||
}
|
||
|
||
log.info("[BrowserUse] Starting browser (headed={})", headed);
|
||
long startTime = System.currentTimeMillis();
|
||
|
||
Playwright pw = getOrCreatePlaywright();
|
||
BrowserType.LaunchOptions launchOptions = new BrowserType.LaunchOptions()
|
||
.setHeadless(!headed);
|
||
|
||
// 平台特定启动参数
|
||
List<String> extraArgs = chromiumLaunchArgs();
|
||
if (!extraArgs.isEmpty()) {
|
||
launchOptions.setArgs(extraArgs);
|
||
log.debug("[BrowserUse] Chromium extra args: {}", extraArgs);
|
||
}
|
||
|
||
Browser browser = pw.chromium().launch(launchOptions);
|
||
BrowserContext context = browser.newContext(new Browser.NewContextOptions()
|
||
.setViewportSize(1280, 800)
|
||
.setLocale("zh-CN"));
|
||
Page page = context.newPage();
|
||
|
||
BrowserSession session = new BrowserSession(browser, context, page, headed, false, null);
|
||
sessions.put(sessionKey, session);
|
||
scheduleIdleCheck(sessionKey);
|
||
|
||
long elapsed = System.currentTimeMillis() - startTime;
|
||
log.info("[BrowserUse] Browser started successfully (headed={}) in {}ms", headed, elapsed);
|
||
return ok("Browser started (headed=" + headed + ") in " + elapsed + "ms. Use action=open with url to navigate.");
|
||
}
|
||
|
||
private String doConnectCdp(String sessionKey, String cdpUrl) {
|
||
if (cdpUrl == null || cdpUrl.isBlank()) {
|
||
return error("url is required for action=connect_cdp (e.g. http://127.0.0.1:9222)");
|
||
}
|
||
|
||
// Stop existing session if any
|
||
BrowserSession existing = sessions.get(sessionKey);
|
||
if (existing != null) {
|
||
doStop(sessionKey);
|
||
}
|
||
|
||
// Normalize CDP URL and force IPv4 to avoid ECONNREFUSED ::1 on macOS
|
||
String normalizedCdpUrl = cdpUrl.trim();
|
||
if (!normalizedCdpUrl.startsWith("http")) {
|
||
normalizedCdpUrl = "http://" + normalizedCdpUrl;
|
||
}
|
||
normalizedCdpUrl = normalizedCdpUrl.replace("://localhost:", "://127.0.0.1:");
|
||
normalizedCdpUrl = normalizedCdpUrl.replace("://localhost/", "://127.0.0.1/");
|
||
if (normalizedCdpUrl.endsWith("://localhost")) {
|
||
normalizedCdpUrl = normalizedCdpUrl.replace("://localhost", "://127.0.0.1");
|
||
}
|
||
|
||
log.info("[BrowserUse] Connecting to CDP at: {}", normalizedCdpUrl);
|
||
long startTime = System.currentTimeMillis();
|
||
|
||
Playwright pw = getOrCreatePlaywright();
|
||
Browser browser = pw.chromium().connectOverCDP(normalizedCdpUrl);
|
||
|
||
// Get existing contexts and pages
|
||
List<BrowserContext> contexts = browser.contexts();
|
||
BrowserContext context;
|
||
Page page;
|
||
|
||
if (!contexts.isEmpty()) {
|
||
context = contexts.get(0);
|
||
List<Page> pages = context.pages();
|
||
page = pages.isEmpty() ? context.newPage() : pages.get(0);
|
||
} else {
|
||
context = browser.newContext();
|
||
page = context.newPage();
|
||
}
|
||
|
||
BrowserSession session = new BrowserSession(browser, context, page, true, true, normalizedCdpUrl);
|
||
sessions.put(sessionKey, session);
|
||
scheduleIdleCheck(sessionKey);
|
||
|
||
String title = page.title();
|
||
String currentUrl = page.url();
|
||
long elapsed = System.currentTimeMillis() - startTime;
|
||
|
||
log.info("[BrowserUse] Connected to CDP at {} in {}ms (page: {} - {})", normalizedCdpUrl, elapsed, currentUrl, title);
|
||
|
||
JSONObject result = new JSONObject();
|
||
result.set("ok", true);
|
||
result.set("cdpUrl", normalizedCdpUrl);
|
||
result.set("currentUrl", currentUrl);
|
||
result.set("currentTitle", title);
|
||
result.set("pagesCount", context.pages().size());
|
||
result.set("message", "Connected to Chrome via CDP at " + normalizedCdpUrl + ". Current page: " + title);
|
||
return JSONUtil.toJsonPrettyStr(result);
|
||
}
|
||
|
||
private String doListCdpTargets(Integer cdpPort) {
|
||
log.info("[BrowserUse] Scanning for CDP targets (port={})", cdpPort);
|
||
|
||
JSONArray targets = new JSONArray();
|
||
|
||
if (cdpPort != null && cdpPort > 0) {
|
||
// Scan single port
|
||
JSONObject target = probeCdpPort(cdpPort);
|
||
if (target != null) {
|
||
targets.add(target);
|
||
}
|
||
} else {
|
||
// Scan port range
|
||
for (int port = CDP_SCAN_PORT_MIN; port <= CDP_SCAN_PORT_MAX; port++) {
|
||
if (isPortOpen(port)) {
|
||
JSONObject target = probeCdpPort(port);
|
||
if (target != null) {
|
||
targets.add(target);
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
log.info("[BrowserUse] Found {} CDP target(s)", targets.size());
|
||
|
||
JSONObject result = new JSONObject();
|
||
result.set("ok", true);
|
||
result.set("targets", targets);
|
||
result.set("count", targets.size());
|
||
if (targets.isEmpty()) {
|
||
result.set("message", "No CDP targets found. Start Chrome with --remote-debugging-port=9222 first.");
|
||
} else {
|
||
result.set("message", "Found " + targets.size() + " CDP target(s). Use connect_cdp with the url to connect.");
|
||
}
|
||
return JSONUtil.toJsonPrettyStr(result);
|
||
}
|
||
|
||
private String doStop(String sessionKey) {
|
||
BrowserSession session = sessions.remove(sessionKey);
|
||
if (session == null) {
|
||
return ok("No browser running");
|
||
}
|
||
|
||
String cdpUrl = session.cdpUrl;
|
||
boolean wasCdp = session.connectedViaCdp;
|
||
session.close(); // Only closes Browser/Context, not the shared Playwright instance
|
||
|
||
if (wasCdp) {
|
||
log.info("[BrowserUse] Disconnected from CDP (Chrome keeps running at {})", cdpUrl);
|
||
return ok("Disconnected from CDP. Chrome process at " + cdpUrl + " keeps running.");
|
||
} else {
|
||
log.info("[BrowserUse] Browser stopped");
|
||
return ok("Browser stopped and resources released");
|
||
}
|
||
}
|
||
|
||
private String doOpen(String sessionKey, String url) {
|
||
if (url == null || url.isBlank()) {
|
||
return error("url is required for action=open");
|
||
}
|
||
|
||
BrowserSession session = getSession(sessionKey);
|
||
if (session == null) {
|
||
doStart(sessionKey, false);
|
||
session = getSession(sessionKey);
|
||
}
|
||
|
||
session.touch();
|
||
Page page = session.page;
|
||
|
||
String normalizedUrl = url.trim();
|
||
if (!normalizedUrl.matches("^https?://.*")) {
|
||
normalizedUrl = "https://" + normalizedUrl;
|
||
}
|
||
|
||
page.navigate(normalizedUrl);
|
||
page.waitForLoadState(LoadState.DOMCONTENTLOADED);
|
||
|
||
String title = page.title();
|
||
String currentUrl = page.url();
|
||
|
||
log.info("[BrowserUse] Opened: {} (title={})", currentUrl, title);
|
||
|
||
JSONObject result = new JSONObject();
|
||
result.set("ok", true);
|
||
result.set("title", title);
|
||
result.set("url", currentUrl);
|
||
result.set("message", "Page loaded: " + title);
|
||
return JSONUtil.toJsonPrettyStr(result);
|
||
}
|
||
|
||
private String doNavigateBack(String sessionKey) {
|
||
BrowserSession session = requireSession(sessionKey);
|
||
if (session == null) {
|
||
return error("No browser running. Use action=start first.");
|
||
}
|
||
|
||
session.touch();
|
||
session.page.goBack();
|
||
|
||
String title = session.page.title();
|
||
String url = session.page.url();
|
||
|
||
log.info("[BrowserUse] Navigated back to: {} ({})", url, title);
|
||
|
||
JSONObject result = new JSONObject();
|
||
result.set("ok", true);
|
||
result.set("title", title);
|
||
result.set("url", url);
|
||
result.set("message", "Navigated back to: " + title);
|
||
return JSONUtil.toJsonPrettyStr(result);
|
||
}
|
||
|
||
private String doSnapshot(String sessionKey) {
|
||
BrowserSession session = requireSession(sessionKey);
|
||
if (session == null) {
|
||
return error("No browser running. Use action=start first.");
|
||
}
|
||
|
||
session.touch();
|
||
Page page = session.page;
|
||
|
||
String title = page.title();
|
||
String url = page.url();
|
||
|
||
String textContent = page.evaluate("""
|
||
(() => {
|
||
function getVisibleText(node, depth) {
|
||
if (depth > 10) return '';
|
||
const results = [];
|
||
if (node.nodeType === Node.TEXT_NODE) {
|
||
const text = node.textContent.trim();
|
||
if (text) results.push(text);
|
||
} else if (node.nodeType === Node.ELEMENT_NODE) {
|
||
const el = node;
|
||
const style = window.getComputedStyle(el);
|
||
if (style.display === 'none' || style.visibility === 'hidden') return '';
|
||
const tag = el.tagName.toLowerCase();
|
||
if (['a', 'button', 'input', 'select', 'textarea'].includes(tag)) {
|
||
const id = el.id ? '#' + el.id : '';
|
||
const cls = el.className && typeof el.className === 'string'
|
||
? '.' + el.className.trim().split(/\\s+/).slice(0, 2).join('.')
|
||
: '';
|
||
const text = el.textContent ? el.textContent.trim().substring(0, 80) : '';
|
||
const href = el.getAttribute('href') || '';
|
||
const placeholder = el.getAttribute('placeholder') || '';
|
||
const selector = tag + id + cls;
|
||
let desc = '[' + selector + ']';
|
||
if (text) desc += ' "' + text + '"';
|
||
if (href) desc += ' href=' + href;
|
||
if (placeholder) desc += ' placeholder=' + placeholder;
|
||
results.push(desc);
|
||
}
|
||
for (const child of el.childNodes) {
|
||
const childText = getVisibleText(child, depth + 1);
|
||
if (childText) results.push(childText);
|
||
}
|
||
}
|
||
return results.join('\\n');
|
||
}
|
||
const text = getVisibleText(document.body, 0);
|
||
return text.substring(0, %d);
|
||
})()
|
||
""".formatted(MAX_SNAPSHOT_LENGTH)).toString();
|
||
|
||
JSONObject result = new JSONObject();
|
||
result.set("ok", true);
|
||
result.set("title", title);
|
||
result.set("url", url);
|
||
result.set("content", textContent);
|
||
return JSONUtil.toJsonPrettyStr(result);
|
||
}
|
||
|
||
private String doScreenshot(String sessionKey, String path) {
|
||
BrowserSession session = requireSession(sessionKey);
|
||
if (session == null) {
|
||
return error("No browser running. Use action=start first.");
|
||
}
|
||
|
||
session.touch();
|
||
Page page = session.page;
|
||
|
||
Page.ScreenshotOptions opts = new Page.ScreenshotOptions().setFullPage(false);
|
||
|
||
if (path != null && !path.isBlank()) {
|
||
opts.setPath(Paths.get(path));
|
||
page.screenshot(opts);
|
||
log.info("[BrowserUse] Screenshot saved to: {}", path);
|
||
|
||
JSONObject result = new JSONObject();
|
||
result.set("ok", true);
|
||
result.set("path", path);
|
||
result.set("message", "Screenshot saved to " + path);
|
||
return JSONUtil.toJsonPrettyStr(result);
|
||
} else {
|
||
byte[] bytes = page.screenshot(opts);
|
||
String base64 = Base64.getEncoder().encodeToString(bytes);
|
||
log.info("[BrowserUse] Screenshot captured ({} bytes)", bytes.length);
|
||
|
||
JSONObject result = new JSONObject();
|
||
result.set("ok", true);
|
||
result.set("format", "png");
|
||
result.set("base64", base64);
|
||
result.set("size", bytes.length);
|
||
result.set("message", "Screenshot captured (" + bytes.length + " bytes)");
|
||
return JSONUtil.toJsonPrettyStr(result);
|
||
}
|
||
}
|
||
|
||
private String doClick(String sessionKey, String selector) {
|
||
if (selector == null || selector.isBlank()) {
|
||
return error("selector is required for action=click");
|
||
}
|
||
|
||
BrowserSession session = requireSession(sessionKey);
|
||
if (session == null) {
|
||
return error("No browser running. Use action=start first.");
|
||
}
|
||
|
||
session.touch();
|
||
Page page = session.page;
|
||
|
||
page.click(selector);
|
||
page.waitForLoadState(LoadState.DOMCONTENTLOADED);
|
||
|
||
String title = page.title();
|
||
String url = page.url();
|
||
|
||
log.info("[BrowserUse] Clicked: {} (page now: {})", selector, url);
|
||
|
||
JSONObject result = new JSONObject();
|
||
result.set("ok", true);
|
||
result.set("selector", selector);
|
||
result.set("currentUrl", url);
|
||
result.set("currentTitle", title);
|
||
result.set("message", "Clicked element: " + selector);
|
||
return JSONUtil.toJsonPrettyStr(result);
|
||
}
|
||
|
||
private String doType(String sessionKey, String selector, String text) {
|
||
if (selector == null || selector.isBlank()) {
|
||
return error("selector is required for action=type");
|
||
}
|
||
if (text == null) {
|
||
return error("text is required for action=type");
|
||
}
|
||
|
||
BrowserSession session = requireSession(sessionKey);
|
||
if (session == null) {
|
||
return error("No browser running. Use action=start first.");
|
||
}
|
||
|
||
session.touch();
|
||
Page page = session.page;
|
||
|
||
page.fill(selector, text);
|
||
|
||
log.info("[BrowserUse] Typed into: {} ({} chars)", selector, text.length());
|
||
|
||
JSONObject result = new JSONObject();
|
||
result.set("ok", true);
|
||
result.set("selector", selector);
|
||
result.set("textLength", text.length());
|
||
result.set("message", "Typed " + text.length() + " characters into " + selector);
|
||
return JSONUtil.toJsonPrettyStr(result);
|
||
}
|
||
|
||
private String doEval(String sessionKey, String code) {
|
||
if (code == null || code.isBlank()) {
|
||
return error("code is required for action=eval");
|
||
}
|
||
|
||
BrowserSession session = requireSession(sessionKey);
|
||
if (session == null) {
|
||
return error("No browser running. Use action=start first.");
|
||
}
|
||
|
||
session.touch();
|
||
Page page = session.page;
|
||
|
||
Object evalResult = page.evaluate(code);
|
||
String resultStr = evalResult != null ? evalResult.toString() : "null";
|
||
|
||
if (resultStr.length() > 10_000) {
|
||
resultStr = resultStr.substring(0, 10_000) + "\n... [truncated]";
|
||
}
|
||
|
||
log.info("[BrowserUse] Eval executed ({} chars result)", resultStr.length());
|
||
|
||
JSONObject result = new JSONObject();
|
||
result.set("ok", true);
|
||
result.set("result", resultStr);
|
||
return JSONUtil.toJsonPrettyStr(result);
|
||
}
|
||
|
||
// ==================== Platform Helpers ====================
|
||
|
||
/**
|
||
* 返回 Chromium 在当前平台下需要的额外启动参数。
|
||
* <p>
|
||
* Windows: --no-sandbox（沙箱兼容性）+ --disable-gpu（GPU 硬件加速问题）
|
||
* 容器环境: --no-sandbox + --disable-dev-shm-usage（共享内存不足）
|
||
*/
|
||
private static List<String> chromiumLaunchArgs() {
|
||
List<String> args = new ArrayList<>();
|
||
boolean inContainer = isRunningInContainer();
|
||
|
||
if (IS_WINDOWS || inContainer) {
|
||
args.add("--no-sandbox");
|
||
}
|
||
if (inContainer) {
|
||
args.add("--disable-dev-shm-usage");
|
||
}
|
||
if (IS_WINDOWS) {
|
||
args.add("--disable-gpu");
|
||
}
|
||
return args;
|
||
}
|
||
|
||
/**
|
||
* 检测是否运行在 Docker/容器环境中。
|
||
*/
|
||
private static boolean isRunningInContainer() {
|
||
try {
|
||
// Docker 容器中通常存在 /.dockerenv 文件
|
||
if (java.nio.file.Files.exists(java.nio.file.Path.of("/.dockerenv"))) {
|
||
return true;
|
||
}
|
||
// 或者 /proc/1/cgroup 包含 docker/kubepods
|
||
java.nio.file.Path cgroup = java.nio.file.Path.of("/proc/1/cgroup");
|
||
if (java.nio.file.Files.exists(cgroup)) {
|
||
String content = java.nio.file.Files.readString(cgroup);
|
||
return content.contains("docker") || content.contains("kubepods") || content.contains("containerd");
|
||
}
|
||
} catch (Exception ignored) {}
|
||
return false;
|
||
}
|
||
|
||
// ==================== CDP Helpers ====================
|
||
|
||
private boolean isPortOpen(int port) {
|
||
try (Socket socket = new Socket()) {
|
||
socket.connect(new java.net.InetSocketAddress("127.0.0.1", port), 100);
|
||
return true;
|
||
} catch (Exception e) {
|
||
return false;
|
||
}
|
||
}
|
||
|
||
private JSONObject probeCdpPort(int port) {
|
||
try {
|
||
String jsonUrl = "http://127.0.0.1:" + port + "/json/version";
|
||
String response = HttpUtil.get(jsonUrl, 2000);
|
||
if (response != null && response.contains("webSocketDebuggerUrl")) {
|
||
JSONObject version = JSONUtil.parseObj(response);
|
||
JSONObject target = new JSONObject();
|
||
target.set("port", port);
|
||
target.set("url", "http://127.0.0.1:" + port);
|
||
target.set("browser", version.getStr("Browser", "unknown"));
|
||
target.set("webSocketDebuggerUrl", version.getStr("webSocketDebuggerUrl", ""));
|
||
return target;
|
||
}
|
||
} catch (Exception e) {
|
||
log.debug("[BrowserUse] Port {} is not a CDP endpoint: {}", port, e.getMessage());
|
||
}
|
||
return null;
|
||
}
|
||
|
||
// ==================== Session Management ====================
|
||
|
||
private BrowserSession getSession(String sessionKey) {
|
||
BrowserSession session = sessions.get(sessionKey);
|
||
if (session != null && !session.isAlive()) {
|
||
sessions.remove(sessionKey);
|
||
session.close();
|
||
return null;
|
||
}
|
||
return session;
|
||
}
|
||
|
||
private BrowserSession requireSession(String sessionKey) {
|
||
return getSession(sessionKey);
|
||
}
|
||
|
||
private void scheduleIdleCheck(String sessionKey) {
|
||
scheduler.scheduleAtFixedRate(() -> {
|
||
BrowserSession session = sessions.get(sessionKey);
|
||
if (session == null) return;
|
||
long idleMinutes = (System.currentTimeMillis() - session.lastActivity) / 60_000;
|
||
if (idleMinutes >= IDLE_TIMEOUT_MINUTES) {
|
||
log.info("[BrowserUse] Idle timeout ({}min), stopping session: {}", idleMinutes, sessionKey);
|
||
doStop(sessionKey);
|
||
}
|
||
}, IDLE_TIMEOUT_MINUTES, 5, TimeUnit.MINUTES);
|
||
}
|
||
|
||
@PreDestroy
|
||
public void cleanup() {
|
||
log.info("[BrowserUse] Cleaning up all browser sessions");
|
||
scheduler.shutdownNow();
|
||
sessions.forEach((key, session) -> {
|
||
try {
|
||
session.close();
|
||
} catch (Exception e) {
|
||
log.warn("[BrowserUse] Error closing session {}: {}", key, e.getMessage());
|
||
}
|
||
});
|
||
sessions.clear();
|
||
|
||
// Shutdown the shared Playwright Node.js process
|
||
synchronized (playwrightLock) {
|
||
if (sharedPlaywright != null) {
|
||
try {
|
||
sharedPlaywright.close();
|
||
log.info("[BrowserUse] Shared Playwright instance closed");
|
||
} catch (Exception e) {
|
||
log.warn("[BrowserUse] Error closing Playwright: {}", e.getMessage());
|
||
}
|
||
sharedPlaywright = null;
|
||
}
|
||
}
|
||
}
|
||
|
||
// ==================== Helper Methods ====================
|
||
|
||
private String ok(String message) {
|
||
JSONObject result = new JSONObject();
|
||
result.set("ok", true);
|
||
result.set("message", message);
|
||
return JSONUtil.toJsonPrettyStr(result);
|
||
}
|
||
|
||
private String error(String message) {
|
||
JSONObject result = new JSONObject();
|
||
result.set("ok", false);
|
||
result.set("error", message);
|
||
return JSONUtil.toJsonPrettyStr(result);
|
||
}
|
||
|
||
// ==================== Inner Class ====================
|
||
|
||
/**
|
||
* 浏览器会话（不持有 Playwright 实例，Playwright 由外层共享管理）
|
||
*/
|
||
private static class BrowserSession {
|
||
final Browser browser;
|
||
final BrowserContext context;
|
||
volatile Page page;
|
||
final boolean headed;
|
||
final boolean connectedViaCdp;
|
||
final String cdpUrl;
|
||
volatile long lastActivity;
|
||
|
||
BrowserSession(Browser browser, BrowserContext context, Page page,
|
||
boolean headed, boolean connectedViaCdp, String cdpUrl) {
|
||
this.browser = browser;
|
||
this.context = context;
|
||
this.page = page;
|
||
this.headed = headed;
|
||
this.connectedViaCdp = connectedViaCdp;
|
||
this.cdpUrl = cdpUrl;
|
||
this.lastActivity = System.currentTimeMillis();
|
||
}
|
||
|
||
void touch() {
|
||
this.lastActivity = System.currentTimeMillis();
|
||
}
|
||
|
||
boolean isAlive() {
|
||
return browser != null && browser.isConnected();
|
||
}
|
||
|
||
/**
|
||
* 关闭浏览器会话（不关闭共享 Playwright）。
|
||
* CDP 模式：仅断开连接，Chrome 进程继续运行。
|
||
* Launch 模式：关闭 context + browser（终止 Chromium 进程）。
|
||
*/
|
||
void close() {
|
||
if (connectedViaCdp) {
|
||
try {
|
||
if (browser != null) browser.close();
|
||
} catch (Exception ignored) {}
|
||
} else {
|
||
try {
|
||
if (context != null) context.close();
|
||
} catch (Exception ignored) {}
|
||
try {
|
||
if (browser != null) browser.close();
|
||
} catch (Exception ignored) {}
|
||
}
|
||
}
|
||
}
|
||
}
|