//  Copyright (C) 2015, The Duplicati Team

//  http://www.duplicati.com, info@duplicati.com
//
//  This library is free software; you can redistribute it and/or modify
//  it under the terms of the GNU Lesser General Public License as
//  published by the Free Software Foundation; either version 2.1 of the
//  License, or (at your option) any later version.
//
//  This library is distributed in the hope that it will be useful, but
//  WITHOUT ANY WARRANTY; without even the implied warranty of
//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
//  Lesser General Public License for more details.
//
//  You should have received a copy of the GNU Lesser General Public
//  License along with this library; if not, write to the Free Software
//  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text.RegularExpressions;

namespace Duplicati.Library.AutoUpdater
{
    public static class AutoUpdateSettings
    {
        private static readonly Dictionary<string, string> _cache = new Dictionary<string, string>();
        private const string APP_NAME = "AutoUpdateAppName.txt";
        private const string UPDATE_URL = "AutoUpdateURL.txt";
        private const string UPDATE_KEY = "AutoUpdateSignKey.txt";
        private const string UPDATE_CHANNEL = "AutoUpdateBuildChannel.txt";
        private const string UPDATE_README = "AutoUpdateFolderReadme.txt";
        private const string UPDATE_INSTALL_FILE = "AutoUpdateInstallIDTemplate.txt";

        private const string OEM_APP_NAME = "oem-app-name.txt";
        private const string OEM_UPDATE_URL = "oem-update-url.txt";
        private const string OEM_UPDATE_KEY = "oem-update-key.txt";
        private const string OEM_UPDATE_README = "oem-update-readme.txt";
        private const string OEM_UPDATE_INSTALL_FILE = "oem-update-installid.txt";

        internal const string UPDATEURL_ENVNAME_TEMPLATE = "AUTOUPDATER_{0}_URLS";
        internal const string UPDATECHANNEL_ENVNAME_TEMPLATE = "AUTOUPDATER_{0}_CHANNEL";

        internal const string MATCH_UPDATE_URL_PREFIX_GROUP = "prefix";
        internal const string MATCH_UPDATE_URL_CHANNEL_GROUP = "channel";
        internal const string MATCH_UPDATE_URL_FILENAME_GROUP = "filename";

        internal static readonly Regex MATCH_AUTOUPDATE_URL = 
            new Regex(string.Format(
                "(?<{0}>.+)(?<{1}>{3})(?<{2}>/([^/]+).manifest)", 
                MATCH_UPDATE_URL_PREFIX_GROUP,
                MATCH_UPDATE_URL_CHANNEL_GROUP,
                MATCH_UPDATE_URL_FILENAME_GROUP,
                string.Join("|", Enum.GetNames(typeof(ReleaseType)).Union(new [] { "preview", "rene" }) )), RegexOptions.Compiled | RegexOptions.IgnoreCase);


        static AutoUpdateSettings()
        {
            ReadResourceText(APP_NAME, OEM_APP_NAME);
            ReadResourceText(UPDATE_URL, OEM_UPDATE_URL);
            ReadResourceText(UPDATE_KEY, OEM_UPDATE_KEY);
            ReadResourceText(UPDATE_README, OEM_UPDATE_README);
            ReadResourceText(UPDATE_INSTALL_FILE, OEM_UPDATE_INSTALL_FILE);
            ReadResourceText(UPDATE_CHANNEL, null);
        }

        private static string ReadResourceText(string name, string oemname)
        {
            // First try to read from _cache
            if (_cache.TryGetValue(name, out string result))
                return result;

            try
            {
                using (var rd = new System.IO.StreamReader(System.Reflection.Assembly.GetExecutingAssembly().GetManifestResourceStream(typeof(AutoUpdateSettings), name)))
                    result = rd.ReadToEnd();
            }
            catch
            {
            }

            try
            {
                // Check for OEM override
                if (!string.IsNullOrWhiteSpace(oemname) && System.IO.File.Exists(oemname))
                    result = System.IO.File.ReadAllText(oemname);
            }
            catch
            {
            }

            if (string.IsNullOrWhiteSpace(result))
                result = "";
            else
                result = result.Trim();

            _cache[name] = result;
            return result;
        }

        public static string[] URLs
        {
            get 
            { 
                if (UsesAlternateURLs)
                    return Environment.GetEnvironmentVariable(string.Format(UPDATEURL_ENVNAME_TEMPLATE, AppName)).Split(new char[] { ';' }, StringSplitOptions.RemoveEmptyEntries);
                else
                    return ReadResourceText(UPDATE_URL, OEM_UPDATE_URL).Split(new char[] { ';' }, StringSplitOptions.RemoveEmptyEntries);
            }
        }

        public static bool UsesAlternateURLs
        {
            get 
            {
                return !string.IsNullOrWhiteSpace(Environment.GetEnvironmentVariable(string.Format(UPDATEURL_ENVNAME_TEMPLATE, AppName)));
            }
        }

        public static ReleaseType DefaultUpdateChannel
        {
            get
            {
                var channelstring = Environment.GetEnvironmentVariable(string.Format(UPDATECHANNEL_ENVNAME_TEMPLATE, AppName));

                if (UsesAlternateURLs && string.IsNullOrWhiteSpace(channelstring))
                {
                    foreach(var url in URLs)
                    {
                        var match = AutoUpdateSettings.MATCH_AUTOUPDATE_URL.Match(url);
                        if (match.Success)
                        {
                            channelstring = match.Groups[AutoUpdateSettings.MATCH_UPDATE_URL_CHANNEL_GROUP].Value;
                            break;
                        }   
                    }
                }

                if (string.IsNullOrWhiteSpace(channelstring))
                    channelstring = BuildUpdateChannel;

                if (string.IsNullOrWhiteSpace(channelstring))
                    channelstring = UpdaterManager.BaseVersion.ReleaseType;


                // Update from older builds
                if (string.Equals(channelstring, "preview", StringComparison.OrdinalIgnoreCase))
                    channelstring = ReleaseType.Experimental.ToString();
                if (string.Equals(channelstring, "rene", StringComparison.OrdinalIgnoreCase))
                    channelstring = ReleaseType.Canary.ToString();
                
                ReleaseType rt;
                if (!Enum.TryParse<ReleaseType>(channelstring, true, out rt))
                    rt = ReleaseType.Stable;

                return rt;
            }
        }

        public static string AppName
        {
            get { return ReadResourceText(APP_NAME, OEM_APP_NAME); }
        }

        public static string BuildUpdateChannel
        {
            get { return ReadResourceText(UPDATE_CHANNEL, null); }
        }

        public static string UpdateFolderReadme
        {
            get { return ReadResourceText(UPDATE_README, OEM_UPDATE_README); }
        }

        public static string UpdateInstallFileText
        {
            get { return string.Format(ReadResourceText(UPDATE_INSTALL_FILE, OEM_UPDATE_INSTALL_FILE), Guid.NewGuid().ToString("N")); }
        }

        public static System.Security.Cryptography.RSACryptoServiceProvider SignKey
        {
            get
            { 
                try
                {
                    var key = System.Security.Cryptography.RSA.Create();
                    key.FromXmlString(ReadResourceText(UPDATE_KEY, OEM_UPDATE_KEY)); 
                    return (System.Security.Cryptography.RSACryptoServiceProvider)key;
                }
                catch
                {
                }

                return null;
            }
        }
    }
}