using SharpOdinClient.structs;
using SharpOdinClient.util;

private readonly Odin _odin = new Odin();

public MainWindow()
{
    InitializeComponent();
    _odin.Log += OnOdinLog;
    _odin.ProgressChanged += OnOdinProgress;
}

private void OnOdinProgress(string filename, long max, long value, long writtenSize)
{
    // filename: partition/file being flashed
    // max: total size
    // value / writtenSize: bytes written
}

private void OnOdinLog(string text, utils.MsgType color)
{
    // text: log line
    // color: message type for UI coloring
}

var device = await _odin.FindDownloadModePort();

Console.WriteLine(device.Name);  // device name
Console.WriteLine(device.COM);    // COM port number
Console.WriteLine(device.VID);    // USB VID
Console.WriteLine(device.PID);    // USB PID

if (await _odin.FindAndSetDownloadMode())
{
  // ready for LOKE_Initialize / flash operations
}

var info = await _odin.DVIF();
await _odin.PrintInfo();

if (await _odin.FindAndSetDownloadMode())
{
    await _odin.PrintInfo();

    if (await _odin.IsOdin())
    {
        if (await _odin.LOKE_Initialize(0))
        {
            var pit = await _odin.Read_Pit();
            if (pit.Result)
            {
                byte[] pitBytes = pit.data;           // raw PIT file bytes
                var entries = pit.Pit;                // partition table entries
            }
        }
    }
}

if (await _odin.FindAndSetDownloadMode())
{
    await _odin.PrintInfo();

    if (await _odin.IsOdin())
    {
        if (await _odin.LOKE_Initialize(0))
        {
            bool ok = (await _odin.Write_Pit("path\\to\\device.pit")).status;
        }
    }
}

public async Task<bool> FlashFirmware(List<string> tarPaths)
{
    var flashFiles = new List<FileFlash>();

    foreach (var tarPath in tarPaths)
    {
        var members = _odin.tar.TarInformation(tarPath);
        foreach (var member in members)
        {
            if (AlreadyAdded(member, flashFiles)) continue;

            var ext = Path.GetExtension(member.Filename);
            var file = new FileFlash
            {
                Enable   = true,
                FileName = member.Filename,
                FilePath = tarPath
            };

            if (ext == ".pit")
            {
                // PIT inside package — handle repartition separately if needed
            }
            else if (ext == ".lz4")
            {
                file.RawSize = _odin.CalculateLz4SizeFromTar(tarPath, member.Filename);
            }
            else
            {
                file.RawSize = member.Filesize;
            }

            flashFiles.Add(file);
        }
    }

    if (flashFiles.Count == 0) return false;

    long totalSize = flashFiles.Sum(f => f.RawSize);

    if (!await _odin.FindAndSetDownloadMode()) return false;
    await _odin.PrintInfo();

    if (!await _odin.IsOdin()) return false;
    if (!await _odin.LOKE_Initialize(totalSize)) return false;

    // Optional: repartition if PIT is inside the package
    var pitInTar = flashFiles.Find(x => x.FileName.EndsWith(".pit", StringComparison.OrdinalIgnoreCase));
    if (pitInTar != null)
    {
        // await _odin.Write_Pit(pitInTar.FilePath);
    }

    var readPit = await _odin.Read_Pit();
    if (!readPit.Result) return false;

    const int efsClear = 0;
    const int bootUpdate = 1;

    if (!await _odin.FlashFirmware(flashFiles, readPit.Pit, efsClear, bootUpdate, autoReboot: true))
        return false;

    return await _odin.PDAToNormal();
}

public async Task<bool> FlashSingleFile(string filePath, string partitionFileName)
{
    var flashFile = new FileFlash
    {
        Enable   = true,
        FileName = partitionFileName,   // e.g. "boot.img"
        FilePath = filePath,
        RawSize  = new FileInfo(filePath).Length
    };

    if (!await _odin.FindAndSetDownloadMode()) return false;
    await _odin.PrintInfo();

    if (!await _odin.IsOdin()) return false;
    if (!await _odin.LOKE_Initialize(flashFile.RawSize)) return false;

    var readPit = await _odin.Read_Pit();
    if (!readPit.Result) return false;

    const int efsClear = 0;
    const int bootUpdate = 0;

    if (!await _odin.FlashSingleFile(flashFile, readPit.Pit, efsClear, bootUpdate, autoReboot: true))
        return false;

    return await _odin.PDAToNormal();
}