first commit

Lua/ScriptApi.cs

@@ -0,0 +1,280 @@
+using MoonSharp.Interpreter;
+
+namespace anjianjingling.Lua
+{
+    /// <summary>
+    /// 把自动化能力注册进 Lua 全局环境。
+    /// 两套写法并存：
+    ///   1) 全局简写（按键精灵风格）  moveTo(100,200)  leftClick()  delay(500)
+    ///   2) 命名空间写法              mouse.moveTo(100,200)  mouse.click()  key.press("ENTER")
+    /// </summary>
+    internal static class ScriptApi
+    {
+        private const double DefaultSim = 0.85;
+        private const int DefaultTolerance = 10;
+
+        public static void Bind(Script script, LuaEngine engine, ILuaHost host)
+        {
+            var g = script.Globals;
+
+            // ===================== 基础 =====================
+            g["log"] = (Action<string>)(m => host.Log(m));
+            g["delay"] = (Action<double>)(ms => engine.Sleep(ms));
+            g["sleep"] = (Action<double>)(ms => engine.Sleep(ms));
+            g["delayRandom"] = (Action<double, double>)((a, b) =>
+            {
+                int lo = (int)Math.Min(a, b), hi = (int)Math.Max(a, b);
+                engine.Sleep(lo >= hi ? lo : Random.Shared.Next(lo, hi + 1));
+            });
+            g["random"] = (Func<double, double, double>)((a, b) =>
+            {
+                int lo = (int)Math.Min(a, b), hi = (int)Math.Max(a, b);
+                return lo >= hi ? lo : Random.Shared.Next(lo, hi + 1);
+            });
+            g["now"] = (Func<double>)(() => DateTimeOffset.Now.ToUnixTimeMilliseconds());
+            g["stop"] = (Action)(() => throw new AbortSignal(false));
+
+            // ===================== 鼠标 =====================
+            var mouse = new Table(script);
+            mouse["moveTo"] = (Action<int, int>)((x, y) => { engine.Tick(); host.MouseMove(x, y); });
+            mouse["moveR"] = (Action<int, int>)((dx, dy) => { engine.Tick(); host.MouseMoveR(dx, dy); });
+            mouse["down"] = (Action<string>)(b => { engine.Tick(); host.MouseDown(NormBtn(b)); });
+            mouse["up"] = (Action<string>)(b => { engine.Tick(); host.MouseUp(NormBtn(b)); });
+            mouse["wheel"] = (Action<int>)(d => { engine.Tick(); host.MouseWheel(d); });
+            mouse["drag"] = (Action<int, int, int, int, int>)((x1, y1, x2, y2, ms) =>
+            {
+                engine.Tick();
+                host.MouseDrag(x1, y1, x2, y2, ms <= 0 ? 400 : ms);
+            });
+            mouse["pos"] = (Func<DynValue>)(() =>
+            {
+                engine.Tick();
+                var p = host.MousePosition();
+                return DynValue.NewTuple(DynValue.NewNumber(p.X), DynValue.NewNumber(p.Y));
+            });
+
+            // click 支持多种参数形态
+            mouse["click"] = DynValue.NewCallback((ctx, a) =>
+            {
+                engine.Tick();
+                string btn = "left";
+                int? x = null, y = null;
+                if (a.Count >= 3 && a[0].Type == DataType.Number)
+                {
+                    x = (int)Math.Round(a[0].Number);
+                    y = (int)Math.Round(a[1].Number);
+                    btn = a[2].Type == DataType.String ? a[2].String : "left";
+                }
+                else if (a.Count >= 2 && a[0].Type == DataType.Number)
+                {
+                    x = (int)Math.Round(a[0].Number);
+                    y = (int)Math.Round(a[1].Number);
+                }
+                else if (a.Count >= 1 && a[0].Type == DataType.String)
+                {
+                    btn = a[0].String;
+                }
+                host.MouseClick(NormBtn(btn), x, y);
+                return DynValue.Nil;
+            });
+
+            mouse["doubleClick"] = DynValue.NewCallback((ctx, a) =>
+            {
+                engine.Tick();
+                int? x = a.Count >= 2 && a[0].Type == DataType.Number ? (int)Math.Round(a[0].Number) : null;
+                int? y = a.Count >= 2 && a[0].Type == DataType.Number ? (int)Math.Round(a[1].Number) : null;
+                host.MouseDoubleClick(x, y);
+                return DynValue.Nil;
+            });
+            g["mouse"] = mouse;
+
+            // ===================== 键盘 =====================
+            var key = new Table(script);
+            key["press"] = (Action<string>)(k => { engine.Tick(); host.KeyPress(k); });
+            key["down"] = (Action<string>)(k => { engine.Tick(); host.KeyDown(k); });
+            key["up"] = (Action<string>)(k => { engine.Tick(); host.KeyUp(k); });
+            key["hotkey"] = (Action<string>)(k => { engine.Tick(); host.HotKey(k); });
+            key["type"] = (Action<string>)(t => { engine.Tick(); host.TypeText(t); });
+            g["key"] = key;
+
+            // ===================== 屏幕 / 视觉 =====================
+            var screen = new Table(script);
+            screen["size"] = (Func<DynValue>)(() =>
+            {
+                engine.Tick();
+                var (w, h) = host.ScreenSize();
+                return DynValue.NewTuple(DynValue.NewNumber(w), DynValue.NewNumber(h));
+            });
+            screen["color"] = (Func<int, int, string>)((x, y) => { engine.Tick(); return host.ColorAt(x, y); });
+            screen["shot"] = (Action<string>)(p => { engine.Tick(); host.ScreenShot(p); });
+            screen["pick"] = (Func<string, string>)(p =>
+            {
+                engine.Tick();
+                return host.PickRegion(string.IsNullOrWhiteSpace(p) ? null : p) ?? "";
+            });
+
+            screen["findColor"] = DynValue.NewCallback((ctx, a) =>
+            {
+                engine.Tick();
+                string hex = a.Count > 0 && a[0].Type == DataType.String ? a[0].String : "FF0000";
+                int tol = a.Count > 1 && a[1].Type == DataType.Number ? (int)Math.Round(a[1].Number) : DefaultTolerance;
+                var (w, h) = host.ScreenSize();
+                int x1 = a.Count > 2 ? (int)Math.Round(a[2].Number) : 0;
+                int y1 = a.Count > 3 ? (int)Math.Round(a[3].Number) : 0;
+                int x2 = a.Count > 4 ? (int)Math.Round(a[4].Number) : w - 1;
+                int y2 = a.Count > 5 ? (int)Math.Round(a[5].Number) : h - 1;
+                var p = host.FindColor(x1, y1, x2, y2, hex, tol);
+                return DynValue.NewTuple(DynValue.NewNumber(p.X), DynValue.NewNumber(p.Y));
+            });
+
+            screen["findImage"] = DynValue.NewCallback((ctx, a) =>
+            {
+                engine.Tick();
+                string path = a.Count > 0 && a[0].Type == DataType.String ? a[0].String : "";
+                double sim = a.Count > 1 && a[1].Type == DataType.Number ? a[1].Number : DefaultSim;
+                var (w, h) = host.ScreenSize();
+                int x1 = a.Count > 2 ? (int)Math.Round(a[2].Number) : 0;
+                int y1 = a.Count > 3 ? (int)Math.Round(a[3].Number) : 0;
+                int x2 = a.Count > 4 ? (int)Math.Round(a[4].Number) : w - 1;
+                int y2 = a.Count > 5 ? (int)Math.Round(a[5].Number) : h - 1;
+                var r = host.FindImage(x1, y1, x2, y2, path, sim);
+                return DynValue.NewTuple(
+                    DynValue.NewNumber(r.X), DynValue.NewNumber(r.Y), DynValue.NewNumber(r.Score));
+            });
+
+            screen["waitImage"] = DynValue.NewCallback((ctx, a) =>
+            {
+                string path = a.Count > 0 && a[0].Type == DataType.String ? a[0].String : "";
+                int timeout = a.Count > 1 && a[1].Type == DataType.Number ? (int)Math.Round(a[1].Number) : 10000;
+                double sim = a.Count > 2 && a[2].Type == DataType.Number ? a[2].Number : DefaultSim;
+
+                var (w, h) = host.ScreenSize();
+                var sw = System.Diagnostics.Stopwatch.StartNew();
+                int rx = -1, ry = -1;
+                double best = 0;
+                while (sw.ElapsedMilliseconds < timeout)
+                {
+                    engine.Tick();
+                    var r = host.FindImage(0, 0, w - 1, h - 1, path, sim);
+                    best = r.Score;
+                    if (r.X >= 0) { rx = r.X; ry = r.Y; break; }
+                    engine.Sleep(300);
+                }
+                host.Log(rx >= 0
+                    ? $"等待图片出现：找到 ({rx},{ry}) 相似度 {best:0.00}"
+                    : $"等待图片超时：{path}（最高 {best:0.00}）");
+                return DynValue.NewTuple(
+                    DynValue.NewNumber(rx), DynValue.NewNumber(ry), DynValue.NewNumber(best));
+            });
+
+            screen["exists"] = DynValue.NewCallback((ctx, a) =>
+            {
+                engine.Tick();
+                string path = a.Count > 0 && a[0].Type == DataType.String ? a[0].String : "";
+                double sim = a.Count > 1 && a[1].Type == DataType.Number ? a[1].Number : DefaultSim;
+                var (w, h) = host.ScreenSize();
+                var r = host.FindImage(0, 0, w - 1, h - 1, path, sim);
+                return DynValue.NewBoolean(r.X >= 0);
+            });
+            g["screen"] = screen;
+
+            // ===================== 窗口 / 系统 =====================
+            var win = new Table(script);
+            win["launch"] = (Action<string, string>)((p, args) => { engine.Tick(); host.Launch(p, args); });
+            win["activate"] = (Func<string, bool>)(t => { engine.Tick(); return host.ActivateWindow(t); });
+            win["list"] = (Func<DynValue>)(() =>
+            {
+                engine.Tick();
+                var list = new Table(script);
+                var arr = host.ListWindows();
+                for (int i = 0; i < arr.Length; i++) list[i + 1] = arr[i];
+                return DynValue.NewTable(list);
+            });
+            g["win"] = win;
+
+            var sys = new Table(script);
+            sys["time"] = (Func<double>)(() => DateTimeOffset.Now.ToUnixTimeMilliseconds());
+            sys["sleep"] = (Action<double>)(ms => engine.Sleep(ms));
+            sys["speed"] = (Func<double, double>)(s => { engine.Speed = s <= 0 ? 1 : s; return engine.Speed; });
+            g["sys"] = sys;
+
+            // ===================== 剪贴板 =====================
+            var clip = new Table(script);
+            clip["get"] = (Func<string>)(() => host.ClipGet());
+            clip["set"] = (Action<string>)(t => host.ClipSet(t));
+            g["clip"] = clip;
+
+            // ===================== 浏览器 =====================
+            var browser = new Table(script);
+            browser["open"] = (Action<string>)(u => { engine.Tick(); host.BrowserOpen(u); });
+            browser["click"] = (Func<string, bool>)(s => { engine.Tick(); return host.BrowserClick(s); });
+            browser["type"] = (Func<string, string, bool>)((s, t) => { engine.Tick(); return host.BrowserType(s, t); });
+            browser["wait"] = (Func<string, int, bool>)((s, ms) => { engine.Tick(); return host.BrowserWaitFor(s, ms <= 0 ? 10000 : ms); });
+            browser["text"] = (Func<string, string>)(s => { engine.Tick(); return host.BrowserGetText(s); });
+            browser["js"] = (Func<string, string>)(js => { engine.Tick(); return host.BrowserExecJs(js); });
+            browser["shot"] = (Action<string>)(p => { engine.Tick(); host.BrowserShot(p); });
+            browser["back"] = (Action)(() => { engine.Tick(); host.BrowserBack(); });
+            browser["forward"] = (Action)(() => { engine.Tick(); host.BrowserForward(); });
+            browser["reload"] = (Action)(() => { engine.Tick(); host.BrowserReload(); });
+            g["browser"] = browser;
+
+            // ===================== 全局简写 =====================
+            g["moveTo"] = mouse["moveTo"];
+            g["moveR"] = mouse["moveR"];
+            g["click"] = mouse["click"];
+            g["leftClick"] = DynValue.NewCallback((ctx, a) =>
+            {
+                engine.Tick();
+                int? x = a.Count >= 2 ? (int)Math.Round(a[0].Number) : null;
+                int? y = a.Count >= 2 ? (int)Math.Round(a[1].Number) : null;
+                host.MouseClick("left", x, y);
+                return DynValue.Nil;
+            });
+            g["rightClick"] = DynValue.NewCallback((ctx, a) =>
+            {
+                engine.Tick();
+                int? x = a.Count >= 2 ? (int)Math.Round(a[0].Number) : null;
+                int? y = a.Count >= 2 ? (int)Math.Round(a[1].Number) : null;
+                host.MouseClick("right", x, y);
+                return DynValue.Nil;
+            });
+            g["middleClick"] = DynValue.NewCallback((ctx, a) =>
+            {
+                engine.Tick();
+                int? x = a.Count >= 2 ? (int)Math.Round(a[0].Number) : null;
+                int? y = a.Count >= 2 ? (int)Math.Round(a[1].Number) : null;
+                host.MouseClick("middle", x, y);
+                return DynValue.Nil;
+            });
+            g["doubleClick"] = mouse["doubleClick"];
+            g["wheel"] = mouse["wheel"];
+            g["drag"] = mouse["drag"];
+
+            g["keyPress"] = key["press"];
+            g["keyDown"] = key["down"];
+            g["keyUp"] = key["up"];
+            g["hotkey"] = key["hotkey"];
+            g["typeText"] = key["type"];
+            g["sayText"] = key["type"];
+
+            g["findImage"] = screen["findImage"];
+            g["findColor"] = screen["findColor"];
+            g["waitImage"] = screen["waitImage"];
+            g["exists"] = screen["exists"];
+            g["colorAt"] = screen["color"];
+            g["screenshot"] = screen["shot"];
+
+            g["run"] = win["launch"];
+            g["activate"] = win["activate"];
+            g["clipGet"] = clip["get"];
+            g["clipSet"] = clip["set"];
+        }
+
+        private static string NormBtn(string b) => b.Trim().ToLowerInvariant() switch
+        {
+            "右" or "right" or "r" or "rightclick" => "right",
+            "中" or "middle" or "m" => "middle",
+            _ => "left"
+        };
+    }
+}