Ant-Browser/backend/automation_script_run_playwright.go

package backend

import (
	"context"
	"fmt"
	"strings"

	"ant-chrome/backend/internal/automation"
)

func automationSelectorProfileID(selector map[string]any) string {
	if selector == nil {
		return ""
	}

	profileID, _ := selector["profileId"].(string)
	return strings.TrimSpace(profileID)
}

func (a *App) ensurePlaywrightTargetReady(selector map[string]any) error {
	profileID := automationSelectorProfileID(selector)
	if profileID == "" {
		return nil
	}

	if _, err := a.BrowserInstanceStart(profileID); err != nil {
		return fmt.Errorf("预启动脚本目标实例失败: %w", err)
	}
	return nil
}

func (a *App) runPlaywrightScript(ctx context.Context, script automation.ScriptRecord, input automation.ScriptRunRequest) (string, string, string) {
	if ctx == nil {
		ctx = context.Background()
	}
	if a.automationMgr == nil {
		return "", "脚本执行失败", "automation runtime manager is not initialized"
	}
	if a.config == nil || !a.config.Automation.Enabled {
		return "", "脚本执行失败", "自动化支持尚未启用"
	}
	if err := ctx.Err(); err != nil {
		return "", "脚本执行失败", automationRunContextErrorMessage(err)
	}
	if err := a.automationMgr.EnsureInstalled(ctx); err != nil {
		return "", "脚本执行失败", err.Error()
	}
	if err := ctx.Err(); err != nil {
		return "", "脚本执行失败", automationRunContextErrorMessage(err)
	}

	state := a.automationMgr.CurrentState()
	if !state.Ready {
		return "", "脚本执行失败", "自动化运行时尚未就绪"
	}

	paramsText := resolveAutomationRunJSONText(input.ParamsText, script.ParamsText, input.UseScriptParams)

	selector, targetSummary, err := a.resolveAutomationEffectiveSelector(script, input, false)
	if err != nil {
		return "", "脚本执行失败", err.Error()
	}
	if err := a.ensurePlaywrightTargetReady(selector); err != nil {
		return "", "脚本执行失败", err.Error()
	}
	if err := ctx.Err(); err != nil {
		return "", "脚本执行失败", automationRunContextErrorMessage(err)
	}
	params, err := parseAutomationJSONObject(paramsText, false)
	if err != nil {
		return "", "脚本执行失败", err.Error()
	}

	baseURL, authHeader, authValue, err := a.automationDemoEndpoint()
	if err != nil {
		return "", "脚本执行失败", err.Error()
	}

	scriptPath, artifactDir, cleanup, err := a.preparePlaywrightScriptWorkspace(state.RuntimeDir, script)
	if err != nil {
		return "", "脚本执行失败", err.Error()
	}
	defer cleanup()
	if err := ctx.Err(); err != nil {
		return "", "脚本执行失败", automationRunContextErrorMessage(err)
	}

	taskResult, err := a.automationMgr.RunScriptTask(ctx, automation.ScriptTaskRequest{
		TaskKey:          "script:" + script.ID,
		ScriptPath:       scriptPath,
		Selector:         selector,
		Params:           params,
		LaunchBaseURL:    baseURL,
		LaunchAuthHeader: authHeader,
		LaunchAuthValue:  authValue,
		ArtifactDir:      artifactDir,
		Timeout:          automationScriptRunTimeout(input),
	})
	if err != nil {
		return "", "脚本执行失败", err.Error()
	}
	if !taskResult.OK {
		errorText := strings.TrimSpace(taskResult.Error)
		if errorText == "" {
			errorText = "playwright script returned ok=false"
		}
		return taskResult.ResultText, appendAutomationRunSummary(taskResult.Summary, targetSummary), errorText
	}
	return taskResult.ResultText, appendAutomationRunSummary(taskResult.Summary, targetSummary), ""
}