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

#r "nuget:EyeAuras.ImGuiSdk, 0.0.6"
#r "nuget:ImGui.NET, 1.91.6.1"

using EyeAuras.ImGuiSdk;
using ImGuiNET;
using System.Numerics;
using System;
using System.Drawing;
using System.Linq;
using EyeAuras.OpenCVAuras.Scripting;
using EyeAuras.Roxy.Api;
using EyeAuras.Scripting.Api;
using EyeAuras.Shared.Services;
using PoeShared.Scaffolding;

{
    AddNewExtension<ImGuiContainerExtensions>();

    var targetWindow = GetService<IWindowSelector>().GetWindow("Aim Trainer");

    using var cv = GetService<IComputerVisionExperimentalScriptingApi>()
        .ForWindow(targetWindow);

    var sendInput = GetService<ISendInputScriptingApi>();
    sendInput.TargetWindow = null; // ниже передаём экранные координаты

    using var osd = GetService<IImGuiExperimentalApi>();

    var mlModelPath = @"https://s3.eyeauras.net/media/2025/03/AimLab_20240213193604OOBfW2f00U5K.onnx";

    var renderGate = new object();
    var renderStopped = false;
    var renderRegistration = osd.AddRenderer(() =>
    {
        lock (renderGate)
        {
            if (renderStopped || cancellationToken.IsCancellationRequested || !targetWindow.IsForeground) return;

            var windowBounds = targetWindow.DwmFrameBounds;
            var drawList = ImGui.GetBackgroundDrawList();
            drawList.AddRect(
                new Vector2(windowBounds.Left, windowBounds.Top),
                new Vector2(windowBounds.Right, windowBounds.Bottom),
                Color.Purple.ToImGuiColor());

            var result = cv.MLSearchRaw(mlModelPath);
            if (!result.Detected.Predictions.Any()) return;

            var currentPosition = result
              .Detected
              .Predictions
              .First()
              .Rectangle
              .Transform(result.ViewportTransforms.WorldToScreen);

            Log.Info($"Found via ML @ {currentPosition}");

            if (!currentPosition.IsEmpty)
            {
                var targetCenter = currentPosition.ToWinRectangle().Center();

                drawList.AddRect(
                    new Vector2(currentPosition.Left, currentPosition.Top),
                    new Vector2(currentPosition.Right, currentPosition.Bottom),
                    Color.AliceBlue.ToImGuiColor());

                drawList.AddCircle(targetCenter.ToVector2(), 10, Color.Red.ToImGuiColor());

                sendInput.MouseMoveTo(targetCenter);
                sendInput.MouseClick();
            }
        }
    });

    try
    {
        cancellationToken.WaitHandle.WaitOne();
    }
    finally
    {
        lock (renderGate)
        {
            // Wait for any current callback, then prevent all later work.
            renderStopped = true;
        }
        renderRegistration.Dispose();
    }
}