FileSqlServerSync/Server/Common/Dir.cs

namespace Common;

/// <summary>
/// 文件夹结构，它包含文件和文件夹
/// </summary>
/// <param name="path">绝对路径</param>
/// <param name="children">子文件或文件夹</param>
public class Dir(string path, List<AFileOrDir>? children = null, NextOpType? nextOp = null)
    : AFileOrDir(path, DirOrFile.Dir, nextOp)
{
    public List<AFileOrDir> Children { get; set; } = children ?? [];

    public override bool IsEqual(AFileOrDir other)
    {
        if (other is not Dir otherDir)
        {
            return false;
        }
        else
        {
            if (this.FormatedPath != otherDir.FormatedPath || this.NextOp != otherDir.NextOp)
            {
                return false;
            }
            if (this.Children.Count != otherDir.Children.Count)
            {
                return false;
            }
            this.Children.Sort(AFileOrDir.Compare);
            otherDir.Children.Sort(AFileOrDir.Compare);
            for (int i = 0; i < this.Children.Count; i++)
            {
                if (!this.Children[i].IsEqual(otherDir.Children[i]))
                {
                    return false;
                }
            }
            return true;
        }
    }

    /// <summary>
    /// clone, 但是更改根目录
    /// </summary>
    /// <param name="optype">操作步骤</param>
    /// <param name="oldRootPath">旧根路径</param>
    /// <param name="newRootPath">新根路径</param>
    /// <param name="IsResetNextOpType">是否重置下步操作</param>
    /// <returns></returns>
    public Dir Clone(
        NextOpType? optype,
        string oldRootPath,
        string newRootPath,
        bool IsResetNextOpType = false
    )
    {
        var ndir = this.Clone(optype, IsResetNextOpType);
        ndir.ResetRootPath(oldRootPath, newRootPath);
        return ndir;
    }

    /// <summary>
    /// clone
    /// </summary>
    /// <param name="optype"></param>
    /// <param name="IsResetNextOpType"></param>
    /// <returns></returns>
    /// <exception cref="Exception"></exception>
    public Dir Clone(NextOpType? optype = null, bool IsResetNextOpType = false)
    {
        var ndir = new Dir(this.FormatedPath, [], IsResetNextOpType ? optype : this.NextOp);

        var nchildren = this
            .Children.AsEnumerable()
            .Select(x =>
            {
                if (x is File file)
                {
                    return new File(
                            file.FormatedPath,
                            file.MTime,
                            IsResetNextOpType ? optype : file.NextOp
                        ) as AFileOrDir;
                }
                else if (x is Dir dir)
                {
                    return dir.Clone(optype, IsResetNextOpType);
                }
                else
                {
                    throw new Exception("cannot be here!");
                }
            })
            .ToList();
        ndir.Children = nchildren;

        return ndir;
    }

    /// <summary>
    /// 重设置根目录
    /// </summary>
    /// <param name="oldPath"></param>
    /// <param name="newPath"></param>
    public void ResetRootPath(string oldPath, string newPath)
    {
        this.FormatedPath = this.FormatedPath.Replace(oldPath, newPath);
        this.Children.ForEach(e =>
        {
            if (e is File file)
            {
                file.FormatedPath = file.FormatedPath.Replace(oldPath, newPath);
            }
            else if (e is Dir dir)
            {
                dir.ResetRootPath(oldPath, newPath);
            }
        });
    }

    /// <summary>
    /// 文件夹合并
    /// </summary>
    /// <param name="fileDirOp">具体操作步骤</param>
    /// <param name="diffdir">将要更新的内容</param>
    /// <param name="IsUpdateObject"> 是否更新Object对象</param>
    /// <param name="IsUpdateDirFile">是否更新文件目录树</param>
    /// <returns></returns>
    public void Combine(
        FileDirOpStra? fileDirOp,
        Dir diffdir,
        bool IsUpdateObject = true,
        bool IsUpdateDirFile = false
    )
    {
        ///验证 target 文件或者文件夹是否具备相关文件权限。
        if (IsUpdateDirFile)
        {
            diffdir.AccessCheck();
        }
        if (this.FormatedPath != diffdir.FormatedPath)
        {
            throw new ArgumentException("their path is not same");
        }
        else
        {
            var ldir = this;
            var rdir = diffdir;

            foreach (var oc in diffdir.Children)
            {
                if (oc is File rfile)
                {
                    if (rfile.NextOp != null)
                    {
                        if (oc.NextOp == NextOpType.Add)
                        {
                            if (IsUpdateObject)
                            {
                                ldir.AddChild(new File(rfile.FormatedPath, rfile.MTime));
                            }
                            if (IsUpdateDirFile)
                            {
                                fileDirOp?.FileCreate(rfile.FormatedPath, rfile.MTime);
                            }
                        }
                        else
                        {
                            var n = ldir
                                .Children.Where(x =>
                                    x.FormatedPath == oc.FormatedPath && x.Type == DirOrFile.File
                                )
                                .FirstOrDefault();
                            if (n is not null)
                            {
                                if (oc.NextOp == NextOpType.Del)
                                {
                                    if (IsUpdateObject)
                                    {
                                        ldir.Children.Remove(n);
                                    }
                                    if (IsUpdateDirFile)
                                    {
                                        fileDirOp?.FileDel(rfile.FormatedPath);
                                    }
                                }
                                else if (oc.NextOp == NextOpType.Modify)
                                {
                                    if (n is File lfile)
                                    {
                                        if (IsUpdateObject)
                                        {
                                            lfile.MTime = rfile.MTime;
                                        }
                                        if (IsUpdateDirFile)
                                        {
                                            fileDirOp?.FileModify(rfile.FormatedPath, rfile.MTime);
                                        }
                                    }
                                }
                            }
                        }
                    }
                }
                else if (oc is Dir rrdir)
                {
                    //新增和删除意味着整个文件夹都被新增和删除
                    if (rrdir.NextOp == NextOpType.Add)
                    {
                        if (IsUpdateDirFile)
                        {
                            fileDirOp?.DirCreate(rrdir);
                        }
                        if (IsUpdateObject)
                        {
                            ldir.AddChild(rrdir.Clone(null, true));
                        }
                    }
                    else if (rrdir.NextOp == NextOpType.Del)
                    {
                        if (IsUpdateDirFile)
                        {
                            fileDirOp?.DirDel(rrdir, false);
                        }
                        if (IsUpdateObject)
                        {
                            ldir.Children.RemoveAt(
                                ldir.Children.FindIndex(x => x.FormatedPath == rrdir.FormatedPath)
                            );
                        }
                    }
                    //当子文件夹和文件不确定时
                    else
                    {
                        var n = ldir
                            .Children.Where(x =>
                                x.FormatedPath == rrdir.FormatedPath && x.Type == DirOrFile.Dir
                            )
                            .FirstOrDefault();
                        if (n is Dir lldir)
                        {
                            lldir.Combine(fileDirOp, rrdir, IsUpdateObject, IsUpdateDirFile);
                        }
                    }
                }
            }
        }
    }

    /// <summary>
    /// 合并两个文件夹,other不会发生改变，this将合并一个副本，这不会改变文件结构
    /// </summary>
    /// <param name="other">它的一个clone将被合并的dir,它的NextOp 不应该是空，否则什么都不会发生</param>
    /// <returns></returns>
    public void CombineJustObject(Dir other)
    {
        Combine(null, other, true, false);
    }

    /// <summary>
    /// 合并两个文件夹,other不会发生改变，this将不会改变，而文件结构会改变
    /// </summary>
    /// <param name="other">它的一个clone将被合并的dir,它的NextOp 不应该是空，否则什么都不会发生</param>
    /// <returns></returns>
    public void CombineJustDirFile(FileDirOpStra fileDirOp, Dir diffDir)
    {
        Combine(fileDirOp, diffDir, false, true);
    }

    /// <summary>
    /// 添加子节点,根目录相同,才会被添加进去
    /// </summary>
    /// <param name="child"></param>
    /// <returns></returns>/
    protected void AddChild(AFileOrDir child)
    {
        if (child.FormatedPath[..this.FormatedPath.Length] != this.FormatedPath)
        {
            throw new ArgumentException("their rootpath are not same!");
        }
        var filtedChildren = this.Children.Where(x => x.Type == child.Type);

        var mached = filtedChildren.Where(x => x.FormatedPath == child.FormatedPath);

        if (mached.Any())
        {
            if (child is File)
            {
                throw new ArgumentException(
                    $"there are same path in the children:{child.FormatedPath}"
                );
            }
            else if (child is Dir dir)
            {
                var tdir = mached.FirstOrDefault();
                if (tdir is Dir ndir)
                {
                    foreach (var d in dir.Children)
                    {
                        ndir.AddChild(d);
                    }
                }
            }
        }
        else
        {
            this.Children.Add(child);
        }
    }

    /// <summary>
    /// 从文件目录结构提起文件信息，注意，此目录文件树不包含文件内容，仅有修改时间mtime
    /// </summary>
    /// <returns></returns>
    public void ExtractInfo()
    {
        if (this.Children.Count != 0)
        {
            throw new NotSupportedException("this dir is not empty.");
        }
        string[] files = Directory.GetFiles(this.FormatedPath);
        string[] dirs = Directory.GetDirectories(this.FormatedPath);
        foreach (var file in files)
        {
            this.Children.Add(new File(file, System.IO.File.GetLastWriteTime($"{file}")));
        }
        foreach (var dir in dirs)
        {
            var ndir = new Dir(dir);
            ndir.ExtractInfo();
            this.Children.Add(ndir);
        }
    }

    /// <summary>
    ///  写入目录文件树，首先必须定义写入文件的策略，此目录结构不包含文件内容,但有一个
    ///  文件的修改时间，是否修改文件的修改时间，需要定义文件的写入策略 WriteFileStrageFunc
    /// </summary>
    /// <returns></returns>
    public void WriteByThisInfo(FileDirOpStra fileDirOp)
    {
        static void f(Dir dir, FileDirOpStra fileDirOp)
        {
            foreach (var child in dir.Children)
            {
                if (child.Type == DirOrFile.Dir)
                {
                    if (child is Dir childDir)
                    {
                        fileDirOp.DirCreate(childDir, false);
                        f(childDir, fileDirOp);
                    }
                }
                else
                {
                    if (child is File childFile)
                    {
                        fileDirOp.FileCreate(child.FormatedPath, childFile.MTime);
                    }
                    else
                    {
                        throw new ArgumentException("child is not File!");
                    }
                }
            }
        }
        f(this, fileDirOp);
    }

    /// <summary>
    /// 校验文件夹和文件权限
    /// </summary>
    private void AccessCheck()
    {
        this.Children.ForEach(e =>
        {
            if (e is File file)
            {
                if (file.NextOp == null) { }
                else if (file.NextOp == NextOpType.Add)
                {
                    if (
                        !AccessWrapper.CheckDirAccess(
                            Path.GetDirectoryName(file.FormatedPath)
                                ?? throw new DirectoryNotFoundException(
                                    $"{file.FormatedPath} 此父路径不存在"
                                ),
                            [DirAcess.CreateFiles]
                        )
                    )
                    {
                        throw new UnauthorizedAccessException($"{file.FormatedPath} 无权限创建文件");
                    }
                }
                else if (file.NextOp == NextOpType.Modify)
                {
                    if (
                        !(
                            AccessWrapper.CheckFileAccess(file.FormatedPath, [FileAccess.Delete])
                            && AccessWrapper.CheckDirAccess(
                                Path.GetDirectoryName(file.FormatedPath)
                                    ?? throw new DirectoryNotFoundException(
                                        $"{file.FormatedPath} 此父路径不存在"
                                    ),
                                [DirAcess.CreateFiles]
                            )
                        )
                    )
                    {
                        throw new UnauthorizedAccessException(
                            $"{file.FormatedPath} 无权限删除源文件或者创建新文件"
                        );
                    }
                }
                else if (file.NextOp == NextOpType.Del)
                {
                    if (!AccessWrapper.CheckFileAccess(file.FormatedPath, [FileAccess.Delete]))
                    {
                        throw new UnauthorizedAccessException($"{file.FormatedPath} 无权限删除源文件");
                    }
                }
            }
            else if (e is Dir dir)
            {
                if (dir.NextOp == null) { }
                else if (dir.NextOp == NextOpType.Add)
                {
                    if (
                        !AccessWrapper.CheckDirAccess(
                            Path.GetDirectoryName(dir.FormatedPath)
                                ?? throw new DirectoryNotFoundException(
                                    $"{dir.FormatedPath} 此父路径不存在"
                                ),
                            [DirAcess.CreateDirectories, DirAcess.CreateFiles]
                        )
                    )
                    {
                        throw new UnauthorizedAccessException($"{dir.FormatedPath} 无权限创建文件夹或者文件");
                    }
                }
                else if (dir.NextOp == NextOpType.Del)
                {
                    if (!AccessWrapper.CheckDirAccess(dir.FormatedPath, [DirAcess.Delete]))
                    {
                        throw new UnauthorizedAccessException($"{dir.FormatedPath} 无权限删除文件夹");
                    }
                    else
                    {
                        //校验是否拥有子文件或者文件夹的删除权限，
                        dir.AccessCheck();
                    }
                }
            }
        });
    }

    /// <summary>
    /// 比较两个目录文件树是否相同,不相同返回差异部分,左侧是右侧的下一个版本,任何一个节点的nextop != null，即所有
    /// 节点都会打上标记
    /// 文件夹的 NextOp 只有新增和删除
    /// </summary>
    /// <param name="otherRootDir"></param>
    /// <returns></returns>
    public Dir Diff(Dir other)
    {
        var ldir = this;
        var rdir = other;
        Dir? cDir = new Dir(rdir.FormatedPath);
        //分别对文件和文件夹分组
        List<File> lFiles = [];
        List<File> rFiles = [];
        List<Dir> lDirs = [];
        List<Dir> rDirs = [];
        var lGroups = ldir.Children.GroupBy(x => x.Type);
        var rGroups = rdir.Children.GroupBy(x => x.Type);
        foreach (var g in lGroups)
        {
            if (g.Key == DirOrFile.Dir)
            {
                lDirs = g.AsEnumerable()
                    .Select(n =>
                    {
                        if (n is Dir dir)
                        {
                            return dir;
                        }
                        throw new Exception("cannot be here");
                    })
                    .ToList();
            }
            else
            {
                lFiles = g.AsEnumerable()
                    .Select(n =>
                    {
                        if (n is File file)
                        {
                            return file;
                        }
                        throw new Exception("cannot be here");
                    })
                    .ToList();
            }
        }

        foreach (var g in rGroups)
        {
            if (g.Key == DirOrFile.Dir)
            {
                rDirs = g.AsEnumerable()
                    .Select(n =>
                    {
                        if (n is Dir dir)
                        {
                            return dir;
                        }
                        throw new Exception("cannot be here");
                    })
                    .ToList();
            }
            else
            {
                rFiles = g.AsEnumerable()
                    .Select(n =>
                    {
                        if (n is File file)
                        {
                            return file;
                        }
                        throw new Exception("cannot be here");
                    })
                    .ToList();
            }
        }

        //排序，然后对比
        int lIndex_f = 0;
        int rIndex_f = 0;
        int lIndex_d = 0;
        int rIndex_d = 0;
        lFiles.Sort(Compare);
        rFiles.Sort(Compare);
        lDirs.Sort(Compare);
        rDirs.Sort(Compare);
        //对比文件
        while (true)
        {
            //当两个线性表都走到最后时，退出循环
            if (lIndex_f == lFiles.Count && rIndex_f == rFiles.Count)
            {
                break;
            }
            //左侧先到底，右侧都是将要删除的
            if (lIndex_f == lFiles.Count)
            {
                var er = rFiles[rIndex_f];
                cDir.Children.Add(new File(er.FormatedPath, er.MTime, NextOpType.Del));
                rIndex_f++;
                continue;
            }
            //右侧先到底，左侧都是要添加的
            if (rIndex_f == rFiles.Count)
            {
                var el = lFiles[lIndex_f];

                cDir.Children.Add(
                    new File(
                        el.FormatedPath.Replace(ldir.FormatedPath, rdir.FormatedPath),
                        el.MTime,
                        NextOpType.Add
                    )
                );
                lIndex_f++;
                continue;
            }
            var l = lFiles[lIndex_f];
            var r = rFiles[rIndex_f];
            //将根路径差异抹平
            var lreativePath = l.FormatedPath.Replace(ldir.FormatedPath, "");
            var rreativePath = r.FormatedPath.Replace(rdir.FormatedPath, "");
            //两文件相同，对比文件修改时间，不同增加到diff内容
            if (lreativePath == rreativePath)
            {
                lIndex_f++;
                rIndex_f++;
                if (l.MTime != r.MTime)
                {
                    cDir.Children.Add(new File(r.FormatedPath, l.MTime, NextOpType.Modify));
                }
            }
            else
            {
                //因为已经按照文件路径排过序了，当左侧文件名大于右侧，那么将根据右侧,添加一个删除diff
                if (lreativePath.CompareTo(rreativePath) > 0)
                {
                    rIndex_f++;
                    cDir.Children.Add(new File(r.FormatedPath, r.MTime, NextOpType.Del));
                }
                //相反，根据左侧，添加一个新增diff
                else
                {
                    lIndex_f++;
                    cDir.Children.Add(
                        new File(
                            l.FormatedPath.Replace(ldir.FormatedPath, rdir.FormatedPath),
                            l.MTime,
                            NextOpType.Add
                        )
                    );
                }
            }
        }

        //文件夹的比较和文件类似，但是他会递归调用文件夹的diff函数，直至文件停止

        while (true)
        {
            if (lIndex_d == lDirs.Count && rIndex_d == rDirs.Count)
            {
                break;
            }
            if (lIndex_d == lDirs.Count)
            {
                var er = rDirs[rIndex_d];
                cDir.Children.Add(er.Clone(NextOpType.Del, true));
                rIndex_d++;
                continue;
            }
            if (rIndex_d == rDirs.Count)
            {
                var el = lDirs[lIndex_d];
                cDir.Children.Add(
                    el.Clone(NextOpType.Add, ldir.FormatedPath, rdir.FormatedPath, true)
                );
                lIndex_d++;
                continue;
            }
            var l = lDirs[lIndex_d];
            var r = rDirs[rIndex_d];
            var lreativePath = l.FormatedPath.Replace(ldir.FormatedPath, "");
            var rreativePath = r.FormatedPath.Replace(rdir.FormatedPath, "");
            if (lreativePath == rreativePath)
            {
                lIndex_d++;
                rIndex_d++;
                var rDir = l.Diff(r);
                //而等于0，这表示此文件夹的内容没有变化
                if (rDir.Children.Count != 0)
                {
                    cDir.Children.Add(rDir);
                }
            }
            else
            {
                //文件夹重命名将会触发整个文件夹的删除和新增操作，这里没有办法定位到操作是修改文件夹(?) 和git类似。
                //潜在的问题是，修改文件夹名，此文件夹包含大量的文件，将触发大量操作。

                if (lreativePath.CompareTo(rreativePath) > 0)
                {
                    cDir.Children.Add(r.Clone(NextOpType.Del, true));
                    rIndex_d++;
                }
                else
                {
                    cDir.Children.Add(
                        l.Clone(NextOpType.Add, ldir.FormatedPath, rdir.FormatedPath, true)
                    );
                    lIndex_d++;
                }
            }
        }
        return cDir;
    }
}