using NTwain;
using System;
using System.Collections;
using System.Linq;
using System.Reflection;
using System.Threading;
using TWAINWorkingGroup;

namespace Net5Console
{
    class Program
    {
        [STAThread]
        static void Main(string[] args)
        {
            Console.WriteLine("Starting twain test in console...");
            Console.WriteLine();

            using (var session = new TwainSession(Assembly.GetExecutingAssembly(), null, IntPtr.Zero))
            using (var hold = new ManualResetEventSlim())
            {
                session.DeviceEvent += (sender, e) =>
                {
                    Console.WriteLine($"Got device event " + (TWDE)e.Event);
                    Console.WriteLine();
                };
                session.ScanEvent += (sender, closing) =>
                {
                    Console.WriteLine($"Got scan event " + closing);
                    Console.WriteLine();

                    // don't care, just end it
                    TW_PENDINGXFERS pending = default;
                    var sts = session.TWAIN.DatPendingxfers(DG.CONTROL, MSG.RESET, ref pending);

                    TW_USERINTERFACE twuserinterface = default;
                    if (session.TWAIN.DatUserinterface(DG.CONTROL, MSG.DISABLEDS, ref twuserinterface) == STS.SUCCESS)
                    {
                        Console.WriteLine("Disabled device.");
                        Console.WriteLine();
                    }
                    else
                    {
                        Console.Error.WriteLine("Failed to disabled device.");
                        Console.WriteLine();
                    }
                    hold.Set();
                };

                if (session.Open() == TWAINWorkingGroup.STS.SUCCESS)
                {
                    Console.WriteLine("Opened DSM");
                    Console.WriteLine();

                    Console.WriteLine("Default device:");
                    Console.WriteLine($"\t{session.DefaultDevice}");
                    Console.WriteLine();

                    Console.WriteLine("All devices:");
                    TW_IDENTITY dsToUse = default;
                    foreach (var dev in session.GetDevices())
                    {
                        Console.WriteLine($"\t{dev}");
                        if (dev.ProductName == "TWAIN2 FreeImage Software Scanner")
                        {
                            dsToUse = dev;
                        }
                    }
                    Console.WriteLine();

                    session.CurrentDevice = dsToUse;
                    if (session.CurrentDevice.HasValue)
                    {
                        Console.WriteLine("Current device after opening attempt:");
                        Console.WriteLine($"\t{session.CurrentDevice}");
                        Console.WriteLine();

                        var caps = session.Capabilities;
                        Console.WriteLine("All device caps:");
                        foreach (var cap in caps.CAP_SUPPORTEDCAPS.GetValues())
                        {
                            WriteCapInfo(caps, cap);
                        }
                        Console.WriteLine();

                        var sts = session.StartCapture(false);
                        if (sts == STS.SUCCESS)
                        {
                            Console.Error.WriteLine("Waiting for capture to complete.");
                            Console.WriteLine();

                            while (!hold.IsSet) Thread.Sleep(100);
                        }
                        else
                        {
                            Console.Error.WriteLine("Failed to start capture: " + sts);
                            Console.WriteLine();
                        }
                    }
                    else
                    {
                        Console.WriteLine("No devices opened.");
                        Console.WriteLine();
                    }
                }
                else
                {
                    Console.Error.WriteLine("Failed to open DSM");
                }
                Console.WriteLine("Test Ended");
            }

        }

        private static void WriteCapInfo(Capabilities caps, CAP cap)
        {
            // use reflection due to unknown generics
            var propInfo = typeof(Capabilities).GetProperty(cap.ToString());
            if (propInfo == null) return;

            var capWrapper = propInfo.GetValue(caps);
            var wrapType = capWrapper.GetType();
            var label = (string)wrapType.GetMethod(nameof(CapWrapper<int>.GetLabel)).Invoke(capWrapper, null);
            var supports = (TWQC)wrapType.GetMethod(nameof(CapWrapper<int>.QuerySupport)).Invoke(capWrapper, null);

            Console.WriteLine($"\t{label ?? cap.ToString()}: {supports}");
            Console.WriteLine($"\t\tDefault: {wrapType.GetMethod(nameof(CapWrapper<int>.GetDefault)).Invoke(capWrapper, null)}");
            Console.WriteLine($"\t\tCurrent: {wrapType.GetMethod(nameof(CapWrapper<int>.GetCurrent)).Invoke(capWrapper, null)}");
            bool first = true;
            foreach (var val in (IEnumerable)wrapType.GetMethod(nameof(CapWrapper<int>.GetValues)).Invoke(capWrapper, null))
            {
                if (first)
                {
                    Console.WriteLine($"\t\tValues:\t{val}");
                    first = false;
                }
                else
                {
                    Console.WriteLine($"\t\t\t{val}");
                }
            }
        }
    }
}