//
// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License.  You may obtain a copy of the License at
//
//   http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.
//

using System;
using System.Collections.Generic;
using System.IO;
using System.Xml.Serialization;
using Microsoft.Win32;


namespace NPanday.Plugin.Settings
{
    [Serializable]
    [ClassAttribute(Phase = "validate", Goal = "generate-settings")]
    public sealed class SettingsGeneratorMojo : AbstractMojo
    {
        public SettingsGeneratorMojo() { }

        [FieldAttribute("npandaySettingsPath", Expression = "${npanday.settings}", Type = "java.lang.String")]
        public string npandaySettingsPath;

        public override Type GetMojoImplementationType()
        {
            return this.GetType();
        }

        public override void Execute()
        {
            string outputFile;
            if (String.IsNullOrEmpty(npandaySettingsPath))
            {
                npandaySettingsPath = PathUtil.GetHomeM2Folder();
            }
            outputFile = PathUtil.BuildSettingsFilePath(npandaySettingsPath);

            XmlSerializer serializer = new XmlSerializer(typeof(npandaySettings));

            npandaySettings settings = new npandaySettings();
            settings.operatingSystem = Environment.OSVersion.ToString();

            RegistryKey monoRegistryKey = Registry.LocalMachine.OpenSubKey(@"SOFTWARE\Novell\Mono");
            RegistryKey microsoftRegistryKey = Registry.LocalMachine.OpenSubKey(@"SOFTWARE\Microsoft\.NETFramework");

            string defaultMonoCLR = (monoRegistryKey != null) ? (string)monoRegistryKey.GetValue("DefaultCLR") : null;

            settings.defaultSetup = GetDefaultSetup(defaultMonoCLR,
                (string)microsoftRegistryKey.GetValue("InstallRoot"));

            npandaySettingsVendorsVendor[] microsoftVendors = null;
            try
            {
                microsoftVendors = GetVendorsForMicrosoft(microsoftRegistryKey);
            }
            catch (ExecutionException e)
            {
                Console.WriteLine(e.ToString());
            }

            npandaySettingsVendorsVendor[] monoVendors = null;
            npandaySettingsVendorsVendor dotGnuVendor = null;

            if (File.Exists(outputFile))
            {
                try
                {
                    monoVendors = GetVendorsForMono(monoRegistryKey, defaultMonoCLR);
                }
                catch (ExecutionException e)
                {
                    if (microsoftVendors == null)
                        Console.WriteLine(e.ToString());
                }

                try
                {
                    dotGnuVendor = GetVendorForGnu(Environment.GetEnvironmentVariable("CSCC_LIB_PATH"));
                }
                catch (ExecutionException e)
                {
                    if (microsoftVendors == null)
                        Console.WriteLine(e.ToString());
                }
            }
            int monoVendorsLength = (monoVendors == null) ? 0 : monoVendors.Length;
            int dotGnuVendorLength = (dotGnuVendor == null) ? 0 : 1;
            int microsoftVendorsLength = (microsoftVendors == null) ? 0 : microsoftVendors.Length;

            npandaySettingsVendorsVendor[] vendors =
                new npandaySettingsVendorsVendor[microsoftVendorsLength + monoVendorsLength + dotGnuVendorLength];

            int copyLocation = 0;
            if (microsoftVendors != null)
            {
                microsoftVendors.CopyTo(vendors, copyLocation);
                copyLocation += microsoftVendors.Length;
            }
            if (monoVendors != null)
            {
                monoVendors.CopyTo(vendors, copyLocation);
                copyLocation += monoVendors.Length;
            }
            if (dotGnuVendor != null)
                vendors[copyLocation] = dotGnuVendor;

            settings.vendors = vendors;
            TextWriter writer = new StreamWriter(@outputFile);
            serializer.Serialize(writer, settings);
            writer.Close();
        }

        protected npandaySettingsDefaultSetup GetDefaultSetup(string defaultMonoCLR,
                                                             string installRoot)
        {
            npandaySettingsDefaultSetup defaultSetup = new npandaySettingsDefaultSetup();
            if (installRoot == null)
            {
                defaultSetup.vendorName = "MONO";
                defaultSetup.vendorVersion = defaultMonoCLR;
                defaultSetup.frameworkVersion = "2.0.50727";
                return (defaultMonoCLR != null) ? defaultSetup : null;
            }
            bool dirInfo11 = new DirectoryInfo(Path.Combine(installRoot, "v1.1.4322")).Exists;
            bool dirInfo20 = new DirectoryInfo(Path.Combine(installRoot, "v2.0.50727")).Exists;
            bool dirInfo35 = new DirectoryInfo(Path.Combine(installRoot, "v3.5")).Exists;
            bool dirInfo40 = new DirectoryInfo(Path.Combine(installRoot, "v4.0.30319")).Exists;


            if (!dirInfo11 && !dirInfo20 && !dirInfo35 && !dirInfo40)
                return null;

            defaultSetup.vendorName = "MICROSOFT";
            defaultSetup.vendorVersion = (dirInfo20) ? "2.0.50727" : ((dirInfo35) ? "3.5" : ((dirInfo40) ? "4.0" : "1.1.4322"));
            defaultSetup.frameworkVersion = defaultSetup.vendorVersion;
            return defaultSetup;
        }

        protected npandaySettingsVendorsVendor GetVendorForGnu(String libPath)
        {
            if (libPath == null)
                throw new ExecutionException("NPANDAY-9011-000: Could not detect GNU vendor: No CSCC_LIB_PATH Found");

            if (libPath.EndsWith("lib" + Path.DirectorySeparatorChar + "cscc" + Path.DirectorySeparatorChar + "lib"))
            {
                string installR = new DirectoryInfo(libPath).Parent.Parent.Parent.FullName;
                string[] tokenizedInstallRoot = installR.Split(Path.DirectorySeparatorChar);
                string vendorVersion = tokenizedInstallRoot[tokenizedInstallRoot.Length - 1];
                if (!isValidVersion(vendorVersion))
                {
                    throw new ExecutionException("NPANDAY-9011-001: Invalid version format for dotGNU: Version = " +
                        vendorVersion + ", Root = " + installR);
                }

                npandaySettingsVendorsVendor vendor = new npandaySettingsVendorsVendor();
                vendor.vendorName = "DotGNU";
                vendor.vendorVersion = vendorVersion;
                npandaySettingsVendorsVendorFrameworksFramework[] vendorFrameworks
                    = new npandaySettingsVendorsVendorFrameworksFramework[1];
                npandaySettingsVendorsVendorFrameworksFramework vf = new npandaySettingsVendorsVendorFrameworksFramework();
                vf.installRoot = Path.Combine(installR, "bin");
                vf.frameworkVersion = "2.0.50727";//doesn't matter
                vendorFrameworks[0] = vf; ;
                vendor.frameworks = vendorFrameworks;
                return vendor;
            }
            throw new ExecutionException("NPANDAY-9011-002: CSCC_LIB_PATH found but could not determine vendor information");
        }

        private npandaySettingsVendorsVendor[] GetVendorsForMicrosoft(RegistryKey microsoftRegistryKey)
        {
            if (microsoftRegistryKey == null)
                throw new ExecutionException("NPANDAY-9011-006: Microsoft installation could not be found.");

            string installRoot = (string)microsoftRegistryKey.GetValue("InstallRoot");

            DotnetSdkLocator sdkLocator = new DotnetSdkLocator();

            string sdkInstallRoot11 = sdkLocator.Find1_1();
            string sdkInstallRoot20 = sdkLocator.Find2_0();
            string sdkInstallRoot35 = sdkLocator.Find3_5();
            string sdkInstallRoot40 = sdkLocator.Find4_0();

            if (installRoot == null) throw new ExecutionException("NPANDAY-9011-005");

            List<npandaySettingsVendorsVendor> vendors = new List<npandaySettingsVendorsVendor>();
            DirectoryInfo dirInfo11 = new DirectoryInfo(Path.Combine(installRoot, "v1.1.4322"));
            DirectoryInfo dirInfo20 = new DirectoryInfo(Path.Combine(installRoot, "v2.0.50727"));
            DirectoryInfo dirInfo30 = new DirectoryInfo(Path.Combine(installRoot, "v3.0"));
            DirectoryInfo dirInfo35 = new DirectoryInfo(Path.Combine(installRoot, "v3.5"));
            DirectoryInfo dirInfo40 = new DirectoryInfo(Path.Combine(installRoot, "v4.0.30319"));

            if (dirInfo11.Exists)
            {
                npandaySettingsVendorsVendor vendor = new npandaySettingsVendorsVendor();
                vendor.vendorName = "MICROSOFT";
                vendor.vendorVersion = "1.1.4322";
                npandaySettingsVendorsVendorFrameworksFramework[] vendorFrameworks
                    = new npandaySettingsVendorsVendorFrameworksFramework[1];
                npandaySettingsVendorsVendorFrameworksFramework vf
                    = new npandaySettingsVendorsVendorFrameworksFramework();
                vf.installRoot = dirInfo11.FullName;
                vf.frameworkVersion = "1.1.4322";

                vendorFrameworks[0] = vf;
                vf.sdkInstallRoot = sdkInstallRoot11;
                FindAndAssignExecutablePaths(vf);
                vendor.frameworks = vendorFrameworks;

                vendors.Add(vendor);
            }
            if (dirInfo20.Exists)
            {
                npandaySettingsVendorsVendor vendor = new npandaySettingsVendorsVendor();
                vendor.vendorName = "MICROSOFT";
                vendor.vendorVersion = "2.0.50727";
                npandaySettingsVendorsVendorFrameworksFramework[] vendorFrameworks
                    = new npandaySettingsVendorsVendorFrameworksFramework[1];
                npandaySettingsVendorsVendorFrameworksFramework vf = new npandaySettingsVendorsVendorFrameworksFramework();
                vf.installRoot = dirInfo20.FullName;
                vf.frameworkVersion = "2.0.50727";
                vendorFrameworks[0] = vf;
                vf.sdkInstallRoot = sdkInstallRoot20;
                FindAndAssignExecutablePaths(vf);

                vendor.frameworks = vendorFrameworks;
                vendors.Add(vendor);
            }
            if (dirInfo30.Exists)
            {
                npandaySettingsVendorsVendor vendor = new npandaySettingsVendorsVendor();
                vendor.vendorName = "MICROSOFT";
                vendor.vendorVersion = "3.0";
                npandaySettingsVendorsVendorFrameworksFramework[] vendorFrameworks = new npandaySettingsVendorsVendorFrameworksFramework[1];
                npandaySettingsVendorsVendorFrameworksFramework vf = new npandaySettingsVendorsVendorFrameworksFramework();
                vf.installRoot = dirInfo30.FullName;
                vf.frameworkVersion = "3.0";
                vendorFrameworks[0] = vf;
                vf.sdkInstallRoot = sdkInstallRoot20;

                // 3.0 does not have it's own compilers, so the bins from 2.0 will do the job here
                vf.executablePaths = new string[] { dirInfo20.FullName };

                FindAndAssignExecutablePaths(vf);
                vendor.frameworks = vendorFrameworks;
                vendors.Add(vendor);
            }
            if (dirInfo35.Exists)
            {
                npandaySettingsVendorsVendor vendor = new npandaySettingsVendorsVendor();
                vendor.vendorName = "MICROSOFT";
                vendor.vendorVersion = "3.5";
                npandaySettingsVendorsVendorFrameworksFramework[] vendorFrameworks = new npandaySettingsVendorsVendorFrameworksFramework[1];
                npandaySettingsVendorsVendorFrameworksFramework vf = new npandaySettingsVendorsVendorFrameworksFramework();
                vf.installRoot = dirInfo35.FullName;
                vf.frameworkVersion = "3.5";
                vendorFrameworks[0] = vf;
                vf.sdkInstallRoot = sdkInstallRoot35;
                FindAndAssignExecutablePaths(vf);
                vendor.frameworks = vendorFrameworks;
                vendors.Add(vendor);
            }
            if (dirInfo40.Exists)
            {
                npandaySettingsVendorsVendor vendor = new npandaySettingsVendorsVendor();
                vendor.vendorName = "MICROSOFT";
                vendor.vendorVersion = "4.0";
                npandaySettingsVendorsVendorFrameworksFramework[] vendorFrameworks = new npandaySettingsVendorsVendorFrameworksFramework[1];
                npandaySettingsVendorsVendorFrameworksFramework vf = new npandaySettingsVendorsVendorFrameworksFramework();
                vf.installRoot = dirInfo40.FullName;
                vf.frameworkVersion = "4.0";
                vendorFrameworks[0] = vf;
                vf.sdkInstallRoot = sdkInstallRoot40;
                FindAndAssignExecutablePaths(vf);
                vendor.frameworks = vendorFrameworks;
                vendors.Add(vendor);
            }

            foreach (string sdk in microsoftRegistryKey.GetSubKeyNames())
            {
                if (sdk == "v4.5.50709")
                {
                    npandaySettingsVendorsVendor vendor = new npandaySettingsVendorsVendor();
                    vendor.vendorName = "MICROSOFT";
                    vendor.vendorVersion = "4.5";
                    npandaySettingsVendorsVendorFrameworksFramework[] vendorFrameworks = new npandaySettingsVendorsVendorFrameworksFramework[1];
                    npandaySettingsVendorsVendorFrameworksFramework vf = new npandaySettingsVendorsVendorFrameworksFramework();
                    vf.installRoot = dirInfo40.FullName;
                    vf.frameworkVersion = "4.5";
                    vendorFrameworks[0] = vf;
                    vf.sdkInstallRoot = sdkInstallRoot40;
                    FindAndAssignExecutablePaths(vf);
                    vendor.frameworks = vendorFrameworks;
                    vendors.Add(vendor);
                }
                else if (sdk == "v4.5.1")
                {
                    npandaySettingsVendorsVendor vendor = new npandaySettingsVendorsVendor();
                    vendor.vendorName = "MICROSOFT";
                    vendor.vendorVersion = "4.5.1";
                    npandaySettingsVendorsVendorFrameworksFramework[] vendorFrameworks = new npandaySettingsVendorsVendorFrameworksFramework[1];
                    npandaySettingsVendorsVendorFrameworksFramework vf = new npandaySettingsVendorsVendorFrameworksFramework();
                    vf.installRoot = dirInfo40.FullName;
                    vf.frameworkVersion = "4.5.1";
                    vendorFrameworks[0] = vf;
                    vf.sdkInstallRoot = sdkInstallRoot40;
                    FindAndAssignExecutablePaths(vf);
                    vendor.frameworks = vendorFrameworks;
                    vendors.Add(vendor);
                }
            }
            return vendors.ToArray();
        }

        private void FindAndAssignExecutablePaths(npandaySettingsVendorsVendorFrameworksFramework x)
        {
            if (x.sdkInstallRoot != null)
            {
                if (File.Exists(Path.Combine(x.sdkInstallRoot, "xsd.exe")))
                {
                    return;
                }
                string binDir = Path.Combine(x.sdkInstallRoot, "bin");
                if (File.Exists(Path.Combine(binDir, "xsd.exe")))
                {
                    if (x.executablePaths == null)
                        x.executablePaths = new string[] { binDir };
                    else
                    {
                        List<string> list = new List<string>(x.executablePaths);
                        list.Add(binDir);
                        x.executablePaths = list.ToArray();
                    }
                }
            }
        }

        private npandaySettingsVendorsVendor[] GetVendorsForMono(RegistryKey monoRegistryKey, string defaultMonoCLR)
        {
            if (monoRegistryKey == null)
                throw new ExecutionException("NPANDAY-9011-007: Mono installation could not be found.");

            npandaySettingsVendorsVendor[] vendors = new npandaySettingsVendorsVendor[monoRegistryKey.SubKeyCount];
            int i = 0;
            foreach (string keyName in monoRegistryKey.GetSubKeyNames())
            {
                string sdkInstallRoot = (string)monoRegistryKey.OpenSubKey(keyName).GetValue("SdkInstallRoot");
                if (sdkInstallRoot == null)
                    throw new ExecutionException("NPANDAY-9011-004: Could not find install root key for mono");
                string installRoot = Path.Combine(sdkInstallRoot, "bin");
                npandaySettingsVendorsVendorFrameworksFramework[] vendorFrameworks = new npandaySettingsVendorsVendorFrameworksFramework[2];
                npandaySettingsVendorsVendorFrameworksFramework vf11 = new npandaySettingsVendorsVendorFrameworksFramework();
                vf11.installRoot = installRoot;
                vf11.frameworkVersion = "1.1.4322";
                vendorFrameworks[0] = vf11;

                npandaySettingsVendorsVendorFrameworksFramework vf20 = new npandaySettingsVendorsVendorFrameworksFramework();
                vf20.installRoot = installRoot;
                vf20.frameworkVersion = "2.0.50727";
                vendorFrameworks[1] = vf20;

                npandaySettingsVendorsVendorFrameworksFramework vf35 = new npandaySettingsVendorsVendorFrameworksFramework();
                vf35.installRoot = installRoot;
                vf35.frameworkVersion = "3.5";
                vendorFrameworks[2] = vf35;

                npandaySettingsVendorsVendorFrameworksFramework vf40 = new npandaySettingsVendorsVendorFrameworksFramework();
                vf40.installRoot = installRoot;
                vf40.frameworkVersion = "4.0";
                vendorFrameworks[3] = vf40;

                npandaySettingsVendorsVendor vendor = new npandaySettingsVendorsVendor();
                vendor.vendorName = "MONO";
                vendor.vendorVersion = keyName;
                vendor.frameworks = vendorFrameworks;
                if (defaultMonoCLR.Equals(keyName)) vendor.isDefault = "true";
                vendors[i++] = vendor;
            }
            return vendors;
        }

        private bool isValidVersion(String version)
        {
            string[] vendorVersionToken = version.Split('.');
            foreach (string token in vendorVersionToken)
            {
                try
                {
                    Single.Parse(token);
                }
                catch (Exception)
                {
                    return false;
                }
            }
            return true;
        }
    }
}