using System;
using Microsoft.Win32;

namespace MEditor
{
    public class FileTypeRegInfo
    {
        /// <summary>
        ///     目标文件类型说明
        /// </summary>
        public string Description; //"XCodeFactory项目文件"

        /// <summary>
        ///     打开目标类型文件的应用程序
        /// </summary>
        public string ExePath;

        /// <summary>
        ///     目标类型文件的扩展名
        /// </summary>
        public string ExtendName; //".xcf"

        /// <summary>
        ///     目标类型文件关联的图标
        /// </summary>
        public string IcoPath;

        public FileTypeRegInfo()
        {
        }

        public FileTypeRegInfo(string extendName)
        {
            ExtendName = extendName;
        }
    }

    /// <summary>
    ///     FileTypeRegister 用于注册自定义的文件类型。
    ///     zhuweisky 2005.08.31
    /// </summary>
    public class FileTypeRegister
    {
        #region RegisterFileType

        /// <summary>
        ///     RegisterFileType 使文件类型与对应的图标及应用程序关联起来。
        /// </summary>
        public static bool Register(FileTypeRegInfo regInfo)
        {
            if (CheckIfRegistered(regInfo.ExtendName))
            {
                return true;
            }
            try
            {
                string relationName = regInfo.ExtendName.Substring(1, regInfo.ExtendName.Length - 1).ToUpper() +
                                      "_FileType";

                RegistryKey fileTypeKey = Registry.ClassesRoot.CreateSubKey(regInfo.ExtendName);
                fileTypeKey.SetValue("", relationName);
                fileTypeKey.Close();

                RegistryKey relationKey = Registry.ClassesRoot.CreateSubKey(relationName);
                relationKey.SetValue("", regInfo.Description);

                RegistryKey iconKey = relationKey.CreateSubKey("DefaultIcon");
                iconKey.SetValue("", regInfo.IcoPath);

                RegistryKey shellKey = relationKey.CreateSubKey("Shell");
                RegistryKey openKey = shellKey.CreateSubKey("Open");
                RegistryKey commandKey = openKey.CreateSubKey("Command");
                commandKey.SetValue("", regInfo.ExePath + " %1");

                relationKey.Close();
                return true;
            }
            catch
            {
                return false;
            }
        }

        /// <summary>
        ///     RegisterFileType 使文件类型与对应的图标及应用程序关联起来。
        /// </summary>
        public static bool UnRegister(string extendName)
        {
            if (!CheckIfRegistered(extendName))
            {
                return true;
            }

            string relationName = extendName.Substring(1, extendName.Length - 1).ToUpper() + "_FileType";
            try
            {
                Registry.ClassesRoot.DeleteSubKeyTree(extendName);
                Registry.ClassesRoot.DeleteSubKeyTree(relationName);
                return true;
            }
            catch (Exception e)
            {
                //System.Windows.Forms.MessageBox.Show(e.Message);
                return false;
            }
        }


        /// <summary>
        ///     GetFileTypeRegInfo 得到指定文件类型关联信息
        /// </summary>
        public static FileTypeRegInfo GetRegInfo(string extendName)
        {
            if (!CheckIfRegistered(extendName))
            {
                return null;
            }

            var regInfo = new FileTypeRegInfo(extendName);
            try
            {
                string relationName = extendName.Substring(1, extendName.Length - 1).ToUpper() + "_FileType";
                RegistryKey relationKey = Registry.ClassesRoot.OpenSubKey(relationName);
                regInfo.Description = relationKey.GetValue("").ToString();

                RegistryKey iconKey = relationKey.OpenSubKey("DefaultIcon");
                regInfo.IcoPath = iconKey.GetValue("").ToString();

                RegistryKey shellKey = relationKey.OpenSubKey("Shell");
                RegistryKey openKey = shellKey.OpenSubKey("Open");
                RegistryKey commandKey = openKey.OpenSubKey("Command");
                string temp = commandKey.GetValue("").ToString();
                regInfo.ExePath = temp.Substring(0, temp.Length - 3);

                return regInfo;
            }
            catch
            {
                return null;
            }
        }

        /// <summary>
        ///     UpdateFileTypeRegInfo 更新指定文件类型关联信息
        /// </summary>
        public static bool UpdateRegInfo(FileTypeRegInfo regInfo)
        {
            if (!CheckIfRegistered(regInfo.ExtendName))
            {
                return false;
            }

            try
            {
                string extendName = regInfo.ExtendName;
                string relationName = extendName.Substring(1, extendName.Length - 1).ToUpper() + "_FileType";
                RegistryKey relationKey = Registry.ClassesRoot.OpenSubKey(relationName, true);
                relationKey.SetValue("", regInfo.Description);

                RegistryKey iconKey = relationKey.OpenSubKey("DefaultIcon", true);
                iconKey.SetValue("", regInfo.IcoPath);

                RegistryKey shellKey = relationKey.OpenSubKey("Shell");
                RegistryKey openKey = shellKey.OpenSubKey("Open");
                RegistryKey commandKey = openKey.OpenSubKey("Command", true);
                commandKey.SetValue("", regInfo.ExePath + " %1");

                relationKey.Close();

                return true;
            }
            catch
            {
                return false;
            }
        }

        /// <summary>
        ///     FileTypeRegistered 指定文件类型是否已经注册
        /// </summary>
        public static bool CheckIfRegistered(string extendName)
        {
            RegistryKey softwareKey = Registry.ClassesRoot.OpenSubKey(extendName);
            if (softwareKey != null)
            {
                return true;
            }

            return false;
        }

        #endregion
    }
}