//
// 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.IO;
using System.Collections;
using System.Xml.Serialization;
using Microsoft.Win32;

using NPanday.Plugin;

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

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

        public override void Execute()
        {
        	string outputFile = Environment.GetEnvironmentVariable("USERPROFILE") + "/.m2/npanday-settings.xml";
            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;
            try
            {
                monoVendors = GetVendorsForMono(monoRegistryKey, defaultMonoCLR);
            }
            catch(ExecutionException e)
            {
                Console.WriteLine(e.ToString());
            }

            npandaySettingsVendorsVendor dotGnuVendor = null;
            try
            {
                dotGnuVendor = GetVendorForGnu(Environment.GetEnvironmentVariable("CSCC_LIB_PATH"));
            }
            catch(ExecutionException e)
            {
                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;
                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;

            if(installRoot != null)
            {
                if(!dirInfo11 && !dirInfo20)
                    return null;
                defaultSetup.vendorName = "MICROSOFT";
                defaultSetup.vendorVersion = (dirInfo20) ? "2.0.50727" :  "1.1.4322";
                defaultSetup.frameworkVersion = defaultSetup.vendorVersion;
                return defaultSetup;
            }
            else if(defaultMonoCLR != null)
            {
                defaultSetup.vendorName = "MONO";
                defaultSetup.vendorVersion = defaultMonoCLR;
                defaultSetup.frameworkVersion = "2.0.50727";
                return defaultSetup;
            }
            return null;
        }

        protected npandaySettingsVendorsVendor GetVendorForGnu(String libPath)
        {
            if(libPath == null)
                throw new ExecutionException("NPANDAY-9011-000: 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");
            string sdkInstallRoot11 = (string) microsoftRegistryKey.GetValue("sdkInstallRootv1.1");
            string sdkInstallRoot20 = (string) microsoftRegistryKey.GetValue("sdkInstallRootv2.0");
            
            if(installRoot == null) throw new ExecutionException("NPANDAY-9011-005");

            npandaySettingsVendorsVendor[] vendors = new npandaySettingsVendorsVendor[3];
            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"));
            int vendorCounter = 0;
            if (dirInfo11.Exists)
            {
                npandaySettingsVendorsVendor vendor = new npandaySettingsVendorsVendor();
                vendor.vendorName = "MICROSOFT";
                vendor.vendorVersion = "1.1.4322";
                npandaySettingsVendorsVendorFrameworksFramework[] vendorFrameworks 
                	= new npandaySettingsVendorsVendorFrameworksFramework[1];
                npandaySettingsVendorsVendorFrameworksFramework vf11 
                	= new npandaySettingsVendorsVendorFrameworksFramework();
                vf11.installRoot = dirInfo11.FullName;
                vf11.frameworkVersion = "1.1.4322";
                
                vendorFrameworks[0] = vf11;
                vf11.sdkInstallRoot = sdkInstallRoot11;
                vendor.frameworks = vendorFrameworks;
                
                vendors[vendorCounter++] = vendor;
            }
            if (dirInfo20.Exists)
            {
                npandaySettingsVendorsVendor vendor = new npandaySettingsVendorsVendor();
                vendor.vendorName = "MICROSOFT";
                vendor.vendorVersion = "2.0.50727";
                npandaySettingsVendorsVendorFrameworksFramework[] vendorFrameworks 
                	= new npandaySettingsVendorsVendorFrameworksFramework[1];
                npandaySettingsVendorsVendorFrameworksFramework vf11 = new npandaySettingsVendorsVendorFrameworksFramework();
                vf11.installRoot = dirInfo20.FullName;
                vf11.frameworkVersion = "2.0.50727";
                vendorFrameworks[0] = vf11;
                vf11.sdkInstallRoot = sdkInstallRoot20;
                vendor.frameworks = vendorFrameworks;
                vendors[vendorCounter++] = vendor;
            }
            if (dirInfo30.Exists)
            {
                npandaySettingsVendorsVendor vendor = new npandaySettingsVendorsVendor();
                vendor.vendorName = "MICROSOFT";
                vendor.vendorVersion = "3.0";
                npandaySettingsVendorsVendorFrameworksFramework[] vendorFrameworks = new npandaySettingsVendorsVendorFrameworksFramework[1];
                npandaySettingsVendorsVendorFrameworksFramework vf11 = new npandaySettingsVendorsVendorFrameworksFramework();
                vf11.installRoot = dirInfo30.FullName;
                vf11.frameworkVersion = "3.0";
                vendorFrameworks[0] = vf11;
                vf11.sdkInstallRoot = sdkInstallRoot20;
                vendor.frameworks = vendorFrameworks;
                vendors[vendorCounter++] = vendor;
            }

            return vendors;
        }

        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;

                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;
        }
    }
}