yili-onemap/api/QM.KJGH.CGGL/QM.KJGH.CGGL.Repository/Cgzj/CgzjSjbzRepository.cs

using QM.KJGH.CGGL.IRepository;
using QM.KJGH.CGGL.Model.Cggl;
using QM.KJGH.Model;
using QM.KJGH.Model.KJGH;
using QM.OrmSqlSugar;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text.RegularExpressions;
using System.Threading.Tasks;
using WS;

namespace QM.KJGH.CGGL.Repository
{
    /// <summary>
    /// 
    /// </summary>
    public class CgzjSjbzRepository : SqlSugarRepository<CgglSjbz>, ICgzjSjbzRepository
    {
        /// <summary>
        /// 获取目录数
        /// </summary>
        /// <returns></returns>
        public async Task<List<Tree>> GetMlTree(string sjbz_bsm)
        {
            List<Tree> tree = new List<Tree>();
            var list = await Db.Queryable<CgglSjbzmlwj>().Where(t => t.SJBZ_BSM == sjbz_bsm).OrderBy(t => t.PX).ToListAsync();
            //根目录
            var roots = list.Where(t => string.IsNullOrWhiteSpace(t.SJML)).ToList();
            if (roots == null || roots.Count < 1)
                return tree;

            tree = ParseTree(list, null);
            return tree;
        }

        /// <summary>
        /// 
        /// </summary>
        /// <param name="list"></param>
        /// <param name="sjml"></param>
        private List<Tree> ParseTree(List<CgglSjbzmlwj> list, string sjml)
        {
            List<Tree> tree = new List<Tree>();
            List<CgglSjbzmlwj> result = new List<CgglSjbzmlwj>();
            if (string.IsNullOrWhiteSpace(sjml))
            {
                result = list.Where(t => string.IsNullOrWhiteSpace(t.SJML)).OrderBy(t => t.PX).ToList();
            }
            else {
                result = list.Where(t => t.SJML == sjml).OrderBy(t => t.PX).ToList();
            }

            foreach (var item in result)
            {
                var treeModel = new Tree() {
                    value = item.BSM,
                    label = item.MLMC,
                    parent = item.SJML,
                    IsParent = item.LX == 1,
                    Add = item
                };
                if (treeModel.IsParent)
                {
                    treeModel.children = ParseTree(list, treeModel.value);
                }
                tree.Add(treeModel);
            }
            return tree;
        }


        /// <summary>
        /// 检查数据目录
        /// </summary>
        /// <param name="sjbz_bsm">数据标准</param>
        /// <param name="xzqdm">行政区代码</param>
        /// <param name="root">检查路径</param>
        /// <returns></returns>
        public async Task<CgglSjbzCheck> FileCheck(string sjbz_bsm, string xzqdm, string xzqmc, string root)
        {
            //150925凉城县县级行政区县级国土空间总体规划电子成果数据/2栅格图件/150925凉城县县级行政区13综合防灾减灾规划图.jpg
            CgglSjbzCheck errors = new CgglSjbzCheck();
            var sjbz = await Db.Queryable<CgglSjbz>().FirstAsync(t => t.BSM == sjbz_bsm);
            var list = await Db.Queryable<CgglSjbzmlwj>().Where(t => t.SJBZ_BSM == sjbz_bsm).OrderBy(t => t.PX).ToListAsync();
            //根目录
            var roots = list.Where(t => string.IsNullOrWhiteSpace(t.SJML)).ToList();
            if (sjbz == null || roots == null || roots.Count < 1)
                throw new UseArgumentException($"未授权的数据，请联系管理人员，{sjbz_bsm}，{sjbz == null }，{ roots == null}，{ roots.Count < 1}");

            FileCheckSub(null, root, root, xzqdm, xzqmc, list, errors);
            return errors;
        }

        /// <summary>
        /// 子目录
        /// </summary>
        /// <param name="pid"></param>
        /// <param name="list"></param>
        /// <param name="errors"></param>
        private void FileCheckSub(string pid, string path, string root, string xzqdm, string xzqmc, List<CgglSjbzmlwj> list, CgglSjbzCheck errors) {
            List<CgglSjbzmlwj> subList = new List<CgglSjbzmlwj>();
            if(string.IsNullOrWhiteSpace(pid))
                subList = list.Where(t => string.IsNullOrWhiteSpace(t.SJML)).ToList();
            else
                subList = list.Where(t => t.SJML == pid).OrderBy(t => t.PX).ToList();

            //标准文档要求
            var bzfiles = new List<string>();
            foreach (var item in subList)
            {
                string pathFull = string.Empty;
                if (string.IsNullOrWhiteSpace(item.GZMC))
                    pathFull = Path.Combine(path, item.MLMC);
                else
                {
                    pathFull = FormatPath(path, xzqdm, xzqmc, item.GZMC);
                    bzfiles.Add(Path.GetFileName(pathFull));
                }

                if (item.YSTJ == "M")
                {
                    if (pathFull.Contains("|"))
                    {
                        var keylist = GetTokens(pathFull).Where(t=>t.Contains("|")).ToList();
                        //是否全通过
                        var fullOk = true;
                        foreach (var keys in keylist)
                        {
                            var values = keys.Split("|");
                            //是否有一个通过
                            var isOk = false;
                            foreach (var value in values)
                            {
                                var value2 = value.Replace("{", "").Replace("}", "");
                                var temp = pathFull.Replace(keys, value2);
                                //类型（0目录）
                                if (item.LX == 0 && Directory.Exists(temp))
                                {
                                    isOk = true;
                                    if (temp.EndsWith(".gdb"))
                                    {
                                        break;
                                    }
                                    FileCheckSub(item.BSM, temp, root, xzqdm, xzqmc, list, errors);
                                    break;
                                }
                                //类型（1文件）
                                if (item.LX == 1 && File.Exists(temp))
                                {
                                    isOk = true;
                                    break;
                                }
                            }
                            //如果都没有过就要报错，并跳出整个循环
                            if(!isOk)
                            {
                                fullOk = false; //因为报错所以fullOk就不能为true
                                break;
                            }
                        }
                        //如果都没有过，报错
                        if (!fullOk)
                        {
                            errors.lack.Add(pathFull.Replace(root, ""));
                        }
                        else
                        {
                            continue;
                        }
                    }
                    else
                    {
                        //类型（0目录）
                        if (item.LX == 0 && Directory.Exists(pathFull))
                        {
                            if (pathFull.EndsWith(".gdb"))
                                continue;
                            FileCheckSub(item.BSM, pathFull, root, xzqdm, xzqmc, list, errors);
                            continue;
                        }
                        //类型（1文件）
                        if (item.LX == 1 && File.Exists(pathFull))
                        {
                            continue;
                        }
                        errors.lack.Add(pathFull.Replace(root, ""));
                    }
                }
            }
            //检查文件数量是否相同（可选）
            var files = Directory.GetFiles(path);
            foreach (var item in files)
            {
                if (!bzfiles.Contains(Path.GetFileName(item)))
                {
                    errors.many.Add(item.Replace(root, ""));
                }
            }
        }

        /// <summary>
        /// 检查数据目录
        /// </summary>
        /// <param name="root">根目录</param>
        /// <param name="xzqdm">行政区代码</param>
        /// <param name="xzqdm">行政区名称</param>
        /// <param name="path">检查路径</param>
        /// <returns></returns>
        private string FormatPath(string root, string xzqdm, string xzqmc, string path)
        {
            var keys = GetTokens(path);
            foreach (var item in keys)
            {
                if (item.Contains("行政区代码"))
                {
                    var index = item.IndexOf(":");
                    var len = xzqdm.Length;
                    if(index >= 0)
                    {
                        len = (item.Split(':')[1].Replace("}", "").ToInt());
                    }
                    if(xzqdm.Length < len)
                    {
                        throw new UseArgumentException($"基本信息文件中的行政区代码长度与数据标准不符");
                    }
                    xzqdm = xzqdm.Substring(0, len);
                    path = path.Replace(item, xzqdm);
                    continue;
                }
                if (item.Contains("行政区名称"))
                {
                    path = path.Replace(item, xzqmc);
                    continue;
                }
            }
            var fullPath = Path.Combine(root, path);
            return fullPath;
        }

        /// <summary>
        /// 匹配花括号
        /// </summary>
        /// <param name="str">需要匹配的字符串</param>
        /// <returns></returns>
        private List<string> GetTokens(string str)
        {
            Regex regex = new Regex(@"\{.*?\}", RegexOptions.IgnoreCase);
            MatchCollection matches = regex.Matches(str);
            var list = matches.Cast<Match>().Select(m => m.Value).Distinct().ToList();
            return list;
        }
    }
}