using Microsoft.CSharp;
using System;
using System.CodeDom.Compiler;
using System.Collections.Concurrent;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Reflection;
using System.Text;

/**
 *
 * @author 失足程序员
 * @Blog http://www.cnblogs.com/ty408/
 * @mail 492794628@qq.com
 * @phone 13882122019
 *
 */
namespace Sz.Network.LoadScriptPool
{

    /// <summary>
    /// 加载脚本文件
    /// </summary>
    public class LoadScriptManager
    {

        private static LoadScriptManager instance = new LoadScriptManager();
        public static LoadScriptManager GetInstance { get { return instance; } }

        HashSet<String> ddlNames = new HashSet<string>();

        LoadScriptManager()
        {
            var asss = AppDomain.CurrentDomain.GetAssemblies();
            foreach (var item in asss)
            {
                foreach (var item222 in item.GetModules(false))
                {
                    ddlNames.Add(item222.FullyQualifiedName);
                }
            }
        }

        #region 返回查找的脚本实例 public IEnumerable<T> GetInstances<T>()
        /// <summary>
        /// 返回查找的脚本实例
        /// </summary>
        /// <typeparam name="T">类型</typeparam>
        /// <returns></returns>
        public IEnumerable<T> GetInstances<T>()
        {
            //使用枚举迭代器，避免了再一次创建对象
            string typeName = typeof(T).Name;
            if (Instances.ContainsKey(typeName))
            {
                foreach (var item in Instances[typeName])
                {
                    if (item.Value is T)
                    {
                        yield return (T)item.Value;
                    }
                }
            }
        }

        ConcurrentDictionary<string, ConcurrentDictionary<string, object>> Instances = new ConcurrentDictionary<string, ConcurrentDictionary<string, object>>();

        #endregion

        #region 根据指定的文件动态编译获取实例 public void LoadCSharpFile(string[] paths, List<String> extensionNames, params string[] dllName)
        /// <summary>
        /// 根据指定的文件动态编译获取实例
        /// <para>如果需要加入调试信息，加入代码 System.Diagnostics.Debugger.Break();</para>
        /// <para>如果传入的是目录。默认只会加载目录中后缀“.cs”文件</para>
        /// </summary>
        /// <param name="paths">
        /// 可以是目录也可以是文件路径
        /// </param>
        /// <param name="dllName">加载的附加DLL文件的路径，绝对路径</param>
        public void LoadCSharpFile(string[] paths, params string[] dllName)
        {
            LoadCSharpFile(paths, null, dllName);
        }

        List<String> csExtensionNames = new List<String>() { ".cs" };

        /// <summary>
        /// 根据指定的文件动态编译获取实例
        /// <para>如果需要加入调试信息，加入代码 System.Diagnostics.Debugger.Break();</para>
        /// <para>如果传入的是目录。默认只会加载目录中后缀“.cs”文件</para>
        /// </summary>
        /// <param name="paths">
        /// 可以是目录也可以是文件路径
        /// </param>
        /// <param name="extensionNames">需要加载目录中的文件后缀</param>
        /// <param name="dllName">加载的附加DLL文件的路径，绝对路径</param>
        public void LoadCSharpFile(string[] paths, List<String> extensionNames, params string[] dllName)
        {
            GC.Collect();
            if (extensionNames == null)
            {
                extensionNames = csExtensionNames;
            }
            foreach (var item in dllName)
            {
                ddlNames.Add(item);
            }
            foreach (var item in ddlNames)
            {
                Console.WriteLine("加载依赖程序集：" + item);
            }
            List<String> fileNames = new List<String>();
            ActionPath(paths, extensionNames, ref fileNames);
            string[] tempDllNames = ddlNames.ToArray();
            foreach (var path in fileNames)
            {
                CSharpCodeProvider provider = new CSharpCodeProvider();
                CompilerParameters parameter = new CompilerParameters();
                //不输出编译文件
                parameter.GenerateExecutable = false;
                //生成调试信息
                parameter.IncludeDebugInformation = true;
                //需要调试必须输出到内存
                parameter.GenerateInMemory = true;
                //添加需要的程序集
                parameter.ReferencedAssemblies.AddRange(tempDllNames);
                //编译文件
                Console.WriteLine("动态加载文件：" + path);
                CompilerResults result = provider.CompileAssemblyFromFile(parameter, path);//根据制定的文件加载脚本
                if (result.Errors.HasErrors)
                {
                    var item = result.Errors.GetEnumerator();
                    while (item.MoveNext())
                    {
                        Console.WriteLine("动态加载文件出错了！" + item.Current.ToString());
                    }
                }
                else
                {
                    ActionAssembly(result.CompiledAssembly);
                }
            }
        }
        #endregion

        #region 根据指定的文件动态编译获取实例 public void LoadDll(string[] paths)

        List<String> dllExtensionNames = new List<String>() { ".dll", ".DLL" };

        /// <summary>
        /// 根据指定的文件动态编译获取实例
        /// <para>如果需要加入调试信息，加入代码 System.Diagnostics.Debugger.Break();</para>
        /// </summary>
        /// <param name="paths">
        /// 可以是目录也可以是文件路径
        /// <para>如果传入的是目录。只会加载目录中后缀“.dll”,“.DLL”文件</para>
        /// </param>
        public void LoadDll(string[] paths)
        {
            GC.Collect();
            List<String> fileNames = new List<String>();
            ActionPath(paths, dllExtensionNames, ref fileNames);
            byte[] bFile = null;
            foreach (var path in fileNames)
            {
                try
                {
                    Console.WriteLine("动态加载文件：" + path);
                    using (FileStream fs = new FileStream(path, FileMode.Open, FileAccess.Read))
                    {
                        using (BinaryReader br = new BinaryReader(fs))
                        {
                            bFile = br.ReadBytes((int)fs.Length);
                            ActionAssembly(Assembly.Load(bFile));
                        }
                    }
                }
                catch (Exception ex)
                {
                    Console.WriteLine("动态加载文件：" + ex);
                }
            }
        }
        #endregion

        #region 处理加载出来的实例 void ActionAssembly(Assembly assembly)
        /// <summary>
        /// 处理加载出来的实例
        /// </summary>
        /// <param name="assembly"></param>
        void ActionAssembly(Assembly assembly)
        {
            ConcurrentDictionary<string, ConcurrentDictionary<string, object>> tempInstances = new ConcurrentDictionary<string, ConcurrentDictionary<string, object>>();
            //获取加载的所有对象模型
            Type[] instances = assembly.GetExportedTypes();
            foreach (var itemType in instances)
            {
                //获取单个模型的所有继承关系和接口关系
                Type[] interfaces = itemType.GetInterfaces();
                //生成实例
                object obj = Activator.CreateInstance(itemType);
                foreach (var iteminterface in interfaces)
                {
                    //加入对象集合
                    if (!Instances.ContainsKey(iteminterface.Name))
                    {
                        tempInstances[iteminterface.Name] = new ConcurrentDictionary<string, object>();
                    }
                    tempInstances[iteminterface.Name][itemType.Name] = obj;
                }
            }
            Instances = tempInstances;
        }
        #endregion

        #region 处理传入的路径 void ActionPath(string[] paths, List<String> extensionNames, ref List<String> fileNames)
        /// <summary>
        /// 处理传入的路径，
        /// </summary>
        /// <param name="paths"></param>
        /// <param name="extensionNames"></param>
        /// <param name="fileNames"></param>
        void ActionPath(string[] paths, List<String> extensionNames, ref List<String> fileNames)
        {
            foreach (var path in paths)
            {
                if (System.IO.Path.HasExtension(path))
                {
                    if (System.IO.File.Exists(path))
                    {
                        fileNames.Add(path);
                    }
                    else
                    {
                        Console.WriteLine("动态加载 无法找到文件：" + path);
                    }
                }
                else
                {
                    GetFiles(path, extensionNames, ref fileNames);
                }
            }
        }
        #endregion

        #region 根据指定文件夹获取指定路径里面全部文件 void GetFiles(string sourceDirectory, List<String> extensionNames, ref  List<String> fileNames)
        /// <summary>
        /// 根据指定文件夹获取指定路径里面全部文件
        /// </summary>
        /// <param name="sourceDirectory">目录</param>
        /// <param name="extensionNames">需要获取的文件扩展名</param>
        /// <param name="fileNames">返回文件名</param>
        void GetFiles(string sourceDirectory, List<String> extensionNames, ref  List<String> fileNames)
        {
            if (!Directory.Exists(sourceDirectory))
            {
                return;
            }
            {
                //获取所有文件名称
                string[] fileName = Directory.GetFiles(sourceDirectory);
                foreach (string path in fileName)
                {
                    if (System.IO.File.Exists(path))
                    {
                        string extName = System.IO.Path.GetExtension(path);
                        if (extensionNames.Contains(extName))
                        {
                            fileNames.Add(path);
                        }
                        else
                        {
                            Console.WriteLine("无法识别文件：" + path);
                        }
                    }
                    else
                    {
                        Console.WriteLine("动态加载 无法找到文件：" + path);
                    }
                }
            }
            //拷贝子目录
            //获取所有子目录名称
            string[] directionName = Directory.GetDirectories(sourceDirectory);
            foreach (string directionPath in directionName)
            {
                //递归下去
                GetFiles(directionPath, extensionNames, ref fileNames);
            }
        }
        #endregion
    }
}