serein-flow/SereinWAT/SereinWAT.cs

using OpenQA.Selenium;
using OpenQA.Selenium.Chrome;
using OpenQA.Selenium.Edge;
using OpenQA.Selenium.Firefox;
using OpenQA.Selenium.IE;
using Serein.NodeFlow;
using static System.Net.Mime.MediaTypeNames;

namespace Serein.Module
{
    public enum DriverType
    {
        Chrome,
        Edge,
        IE,
        Firefox,
    }
    public enum ByType
    {
        XPath,
        Id,
        Class,
        Name,
        CssSelector,
        PartialLinkText,
    }
    public enum ScriptOp
    {
        Add,
        Modify,
        Delete,
    }
    public enum ActionType
    {
        Click,
        DoubleClick,
        RightClick,
        SendKeys
    }


    /// <summary>
    /// 网页自动化测试
    /// Web test automation
    /// </summary>
    [DynamicFlow]
    public class WebSelenium
    {

        public WebDriver WebDriver { get; set; }

        
        public DriverType DriverType { get; set; }

        //public ChromeDriver Driver { get; set; }
        #region Init and Exit
        [MethodDetail(DynamicNodeType.Init)]
        public void Init(DynamicContext context)
        {
        }

        [MethodDetail(DynamicNodeType.Exit)]
        public void Exit(DynamicContext context)
        {
            WebDriver?.Quit();
        }
        #endregion

        [MethodDetail(DynamicNodeType.Action,"等待")]
        public void Wait(int time = 1000)
        {
            Thread.Sleep(time);
        }

        [MethodDetail(DynamicNodeType.Action,"启动浏览器")]
        public WebDriver OpenDriver( bool isVisible = true, DriverType driverType = DriverType.Chrome)
        {
            if(driverType == DriverType.Chrome)
            {
                ChromeOptions options = new ChromeOptions();
                if (!isVisible)
                {
                    options.AddArgument("headless"); // 添加无头模式参数
                    options.AddArgument("disable-gpu"); // 需要禁用 GPU
                    options.LeaveBrowserRunning = true;  // 设置浏览器不自动关闭
                }
                WebDriver = new ChromeDriver(options);
                return WebDriver;
            }
            else if (driverType == DriverType.Edge)
            {
                EdgeOptions options = new EdgeOptions();
                if (!isVisible)
                {
                    options.AddArgument("headless"); // 添加无头模式参数
                    options.AddArgument("disable-gpu"); // 需要禁用 GPU
                    options.LeaveBrowserRunning = true;  // 设置浏览器不自动关闭
                }
                WebDriver = new EdgeDriver(options);
                return WebDriver;
            }
            else if (driverType == DriverType.IE)
            {
                InternetExplorerOptions options = new InternetExplorerOptions();
                // IE浏览器不支持无头模式，因此这里不添加无头模式参数
                WebDriver = new InternetExplorerDriver(options);
                return WebDriver;
            }
            else if (driverType == DriverType.Firefox)
            {
                FirefoxOptions options = new FirefoxOptions();
                if (!isVisible)
                {
                    options.AddArgument("-headless"); // 添加无头模式参数
                                                      // Firefox没有直接的LeaveBrowserRunning选项，但可以使用调试器暂停关闭
                }

                // FirefoxDriver 没有直接的 LeaveBrowserRunning 选项
                WebDriver = new FirefoxDriver(options);
                return WebDriver;
            }
            else
            {
                throw new InvalidOperationException("");
            }
            
        }

        [MethodDetail(DynamicNodeType.Action,"进入网页")]
        public void ToPage( string url)
        {
            if (url.StartsWith("https://") || url.StartsWith("http://"))
            {
                WebDriver.Navigate().GoToUrl($"{url}");
            }
            else
            {
                throw new Exception("请输入完整的Url。Please enter the full Url.");
            }
        }


        [MethodDetail(DynamicNodeType.Action,"定位元素")]
        public IWebElement FindElement( string key = "",  ByType byType = ByType.XPath,  int index = 0)
        {
            By by = byType switch
            {
                ByType.Id => By.Id(key),
                ByType.XPath => By.XPath(key),
                ByType.Class => By.ClassName(key),
                ByType.Name => By.Name(key),
                ByType.CssSelector => By.CssSelector(key),
                ByType.PartialLinkText => By.PartialLinkText(key),
            };
            if(index == -1)
            {
                return WebDriver.FindElements(by).First();
            }
            else
            {
                return WebDriver.FindElements(by)[index];
            }
        }

        [MethodDetail(DynamicNodeType.Action, "定位并操作元素")]
        public IWebElement FindAndUseElement( ByType byType = ByType.XPath,
                                              string key = "",
                                              ActionType actionType = ActionType.Click,
                                              string text = "",
                                              int index = 0,
                                              int waitTime = 0)
        {
            Thread.Sleep(waitTime);
            By by = byType switch
            {
                ByType.Id => By.Id(key),
                ByType.XPath => By.XPath(key),
                ByType.Class => By.ClassName(key),
                ByType.Name => By.Name(key),
                ByType.CssSelector => By.CssSelector(key),
                ByType.PartialLinkText => By.PartialLinkText(key),
            };
            var element = WebDriver.FindElements(by)[index];
            Thread.Sleep(waitTime);
            var actions = new OpenQA.Selenium.Interactions.Actions(WebDriver);
            switch (actionType)
            {
                case ActionType.Click:
                    actions.Click(element).Perform();
                    break;
                case ActionType.DoubleClick:
                    actions.DoubleClick(element).Perform();
                    break;
                case ActionType.RightClick:
                    actions.ContextClick(element).Perform();
                    break;
                case ActionType.SendKeys:
                    element.Click();
                    element.Clear();
                    element.SendKeys(text);
                    break;
            }

            return element;
        }

        [MethodDetail(DynamicNodeType.Action, "操作元素")]
        public void PerformAction(IWebElement element,  ActionType actionType = ActionType.Click,  string text = "")
        {
            var actions = new OpenQA.Selenium.Interactions.Actions(WebDriver);

            switch (actionType)
            {
                case ActionType.Click:
                    actions.Click(element).Perform();
                    break;
                case ActionType.DoubleClick:
                    actions.DoubleClick(element).Perform();
                    break;
                case ActionType.RightClick:
                    actions.ContextClick(element).Perform();
                    break;
                case ActionType.SendKeys:
                    element.Click();
                    element.Clear();
                    element.SendKeys(text);
                    break;
            }
        }


        [MethodDetail(DynamicNodeType.Action,"Js操作元素属性")]
        public void AddAttribute(IWebElement element,  ScriptOp scriptOp = ScriptOp.Modify, string attributeName = "",  string value = "")
        {
            if(scriptOp == ScriptOp.Add)
            {
                WebDriver.ExecuteScript($"arguments[0].{attributeName} = arguments[1];", element, value);
            }
            else if (scriptOp == ScriptOp.Modify)
            {
                element.SetAttribute(WebDriver, attributeName, value);
                string newHref = element.GetAttribute("href");
                Console.WriteLine("New href value: " + newHref);

                WebDriver.ExecuteScript("arguments[0].setAttribute(arguments[1], arguments[2]);", element, attributeName, value);
            }
            else if (scriptOp == ScriptOp.Delete)
            {
                WebDriver.ExecuteScript("arguments[0].removeAttribute(arguments[1]);", element, attributeName);
            }
        }


        [MethodDetail(DynamicNodeType.Action, "Js获取元素属性")]
        public string GetAttribute(IWebElement element,  string attributeName = "")
        {
            return element.GetAttribute(attributeName);
        }
    }

    public static class MyExtension
    {
        public static void SetAttribute(this IWebElement element, IWebDriver driver, string attributeName, string value)
        {
            IJavaScriptExecutor js = (IJavaScriptExecutor)driver;
            js.ExecuteScript($"arguments[0].setAttribute('{attributeName}', '{value}');", element);
        }
    }
}