using System;
using System.Collections.Concurrent;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;
using BurnOutSharp.ExecutableType.Microsoft;
using BurnOutSharp.Matching;

namespace BurnOutSharp.PackerType
{
    public class InnoSetup : IContentCheck, IScannable
    {
        /// <inheritdoc/>
        public bool ShouldScan(byte[] magic) => true;

        /// <inheritdoc/>
        public string CheckContents(string file, byte[] fileContent, bool includeDebug, PortableExecutable pex, NewExecutable nex)
        {
            // Try to read the contents as a PE executable
            if (pex != null)
            {
                var sections = pex?.SectionTable;
                if (sections == null)
                    return null;
                
                // Get the DATA/.data section, if it exists
                if (pex.DataSectionRaw != null)
                {
                    var matchers = new List<ContentMatchSet>
                    {
                        // Inno Setup Setup Data (
                        new ContentMatchSet(new byte?[]
                        {
                            0x49, 0x6E, 0x6E, 0x6F, 0x20, 0x53, 0x65, 0x74,
                            0x75, 0x70, 0x20, 0x53, 0x65, 0x74, 0x75, 0x70,
                            0x20, 0x44, 0x61, 0x74, 0x61, 0x20, 0x28
                        }, GetVersion, "Inno Setup"),
                    };

                    string match = MatchUtil.GetFirstMatch(file, pex.DataSectionRaw, matchers, includeDebug);
                    if (!string.IsNullOrWhiteSpace(match))
                        return match;
                }

                // Get the DOS stub from the executable, if possible
                var stub = pex?.DOSStubHeader;
                if (stub == null)
                    return null;
                
                // Check for "Inno" in the reserved words
                if (stub.Reserved2[4] == 0x6E49 && stub.Reserved2[5] == 0x6F6E)
                {
                    string version = GetOldVersion(file, fileContent);
                    if (!string.IsNullOrWhiteSpace(version))
                        return $"Inno Setup {version}";
                    
                    return "Inno Setup (Unknown Version)";
                }
            }

            // Try to read the contents as an NE executable
            if (nex != null)
            {
                // Get the DOS stub from the executable, if possible
                var stub = nex?.DOSStubHeader;
                if (stub == null)
                    return null;
                
                // Check for "Inno" in the reserved words
                if (stub.Reserved2[4] == 0x6E49 && stub.Reserved2[5] == 0x6F6E)
                {
                    string version = GetOldVersion(file, fileContent);
                    if (!string.IsNullOrWhiteSpace(version))
                        return $"Inno Setup {version}";
                    
                    return "Inno Setup (Unknown Version)";
                }
            }

            return null;
        }

        /// <inheritdoc/>
        public ConcurrentDictionary<string, ConcurrentQueue<string>> Scan(Scanner scanner, string file)
        {
            if (!File.Exists(file))
                return null;

            using (var fs = File.OpenRead(file))
            {
                return Scan(scanner, fs, file);
            }
        }

        // TOOO: Add Inno Setup extraction
        // https://github.com/dscharrer/InnoExtract
        /// <inheritdoc/>
        public ConcurrentDictionary<string, ConcurrentQueue<string>> Scan(Scanner scanner, Stream stream, string file)
        {
            return null;
        }

        public static string GetVersion(string file, byte[] fileContent, List<int> positions)
        {
            try
            {
                int index = positions[0];
                index += 23;

                var versionBytes = new ReadOnlySpan<byte>(fileContent, index, 16).ToArray();
                var onlyVersion = versionBytes.TakeWhile(b => b != ')').ToArray();

                index += onlyVersion.Length + 2;
                var unicodeBytes = new ReadOnlySpan<byte>(fileContent, index, 3).ToArray();
                string version = Encoding.ASCII.GetString(onlyVersion);

                if (unicodeBytes.SequenceEqual(new byte[] { 0x28, 0x75, 0x29 }))
                    return (version + " (Unicode)");

                return version;
            }
            catch
            {
                return "(Unknown Version)";
            }
        }

        private static string GetOldVersion(string file, byte[] fileContent)
        {
            // TODO: Only sample is an NE executable - verify if this is in PE as well or where in the NE this lives
            var matchers = new List<ContentMatchSet>
            {
                // "rDlPtS02" + (char)0x87 + "eVx"
                new ContentMatchSet(new byte?[] { 0x72, 0x44, 0x6C, 0x50, 0x74, 0x53, 0x30, 0x32, 0x87, 0x65, 0x56, 0x78 }, "1.2.16 or earlier"),
            };

            string match = MatchUtil.GetFirstMatch(file, fileContent, matchers, false);
            return match ?? "Unknown 1.X"; 
        }
    }
}
