//this is a simple 2D aimbot + ESP for https://aimtrainer.io/target-tracking

using System.Drawing;
using System.Linq;
using EyeAuras.Blazor.Controls.ScreenOverlay;
using EyeAuras.OpenCVAuras.Scripting;
using EyeAuras.Roxy.Api;
using EyeAuras.Shared.Scripting;
using EyeAuras.Shared.Services;
using PoeShared.Scaffolding;

//we're going to work with a specific window, lets find it
var targetWindow = GetService<IWindowSelector>().GetWindow("Tracking - Aim Trainer");

{
    //this API is used to do Computer Vision stuff (image/text/color/ml search)
    using var cv = GetService<IComputerVisionExperimentalScriptingApi>()
        .ForWindow(targetWindow); // selected window

    //this API is used to generate mouse movements
    var sendInput = GetService<ISendInputScriptingApi>();
    sendInput.TargetWindow = null; // ниже передаём экранные координаты

    //pre-trained model we're going to use to find the target
    var mlModelPath = @"https://s3.eyeauras.net/media/2025/03/AimLab_20240213193604OOBfW2f00U5K.onnx";

    using var osd = GetService<IOnScreenCanvasScriptingApi>()
        .Create();

    try
    {
        using var windowRect = osd.AddRectangle()
            .WithBorderColor(Color.Purple)
            .WithBackground(Color.Transparent);
        using var targetRect = osd.AddRectangle()
            .WithBorderColor(Color.Red)
            .WithBackground(Color.Transparent);

        while (!cancellationToken.WaitHandle.WaitOne(10))
        {
            if (!targetWindow.IsForeground)
            {
                //do nothing if the window is not in the foreground
                continue;
            }
            //this method will be called many-many times per second, until OSD is closed
            windowRect.WithRect(targetWindow.DwmFrameBounds);

            //get predictions from the model
            var result = cv.MLSearchRaw(mlModelPath);
            if (!result.Detected.Predictions.Any())
            {
                //not found
                continue;
            }

            var currentPosition = result //WindowImageProcessedEventArgs
              .Detected //get detection results
              .Predictions //get predictions
              .First() //more specifically, first one
              .Rectangle //get bounding box of that prediction IN LOCAL(aka World) coordinates
              .Transform(result.ViewportTransforms.WorldToScreen); //transform World coordinates to Screen coordinates
            Log.Info($"Found via ML @ {currentPosition}");

            if (!currentPosition.IsEmpty)
            {
                //found the image
                var targetCenter = currentPosition.ToWinRectangle().Center();

                targetRect.WithRect(currentPosition.ToWinRectangle());

                sendInput.MouseMoveTo(targetCenter);
                sendInput.MouseClick();
            }
        }
    }
    finally
    {
        osd.Clear();
    }
}