TorrentFileEntry.cs

// Copyright © 2017 Paddy Xu
//
// This file is part of QuickLook program.
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program 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 General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.

using System;
using System.Collections.Generic;
using System.Linq;

namespace QuickLook.Plugin.TorrentViewer
{
    public class TorrentFileEntry : IComparable<TorrentFileEntry>
    {
        private readonly TorrentFileEntry _parent;
        private int _cachedDepth = -1;
        private int _cachedLevel = -1;

        public TorrentFileEntry(string name, bool isFolder, TorrentFileEntry parent = null, string md5Sum = null)
        {
            Name = name;
            IsFolder = isFolder;

            _parent = parent;
            _parent?.Children.Add(this);
            this.Md5Sum = md5Sum ?? string.Empty;
        }

        public List<TorrentFileEntry> Children { get; } = new List<TorrentFileEntry>();

        public IList<TorrentFileEntry> VisibileChildren => this.Children.Where(x => !x.IsPaddingFile).ToList();

        public string Name { get; set; }

        public bool Encrypted { get; set; }

        public bool IsFolder { get; set; }

        public bool IsPaddingFile
        {
            get
            {
                return this.Name is string name && name.StartsWith("_____padding_file_");
            }
        }

        public ulong Size { get; set; }

        public string Md5Sum { get; }

        public DateTime ModifiedDate { get; set; }

        /// <summary>
        ///     Returns the maximum depth of all siblings
        /// </summary>
        public int Level
        {
            get
            {
                if (_cachedLevel != -1)
                    return _cachedLevel;

                if (_parent == null)
                    _cachedLevel = GetDepth();
                else
                    _cachedLevel = _parent.Level - 1;

                return _cachedLevel;
            }
        }

        public int CompareTo(TorrentFileEntry other)
        {
            if (IsFolder == other.IsFolder)
                return string.Compare(Name, other.Name, StringComparison.CurrentCulture);

            if (IsFolder)
                return -1;

            return 1;
        }

        /// <summary>
        ///     Returns the number of nodes in the longest path to a leaf
        /// </summary>
        private int GetDepth()
        {
            if (_cachedDepth != -1)
                return _cachedDepth;

            var max = Children.Count == 0 ? 0 : Children.Max(r => r.GetDepth());
            _cachedDepth = max + 1;
            return _cachedDepth;
        }

        public override string ToString()
        {
            if (IsFolder)
                return $"{Name}";

            var en = Encrypted ? "*" : "";
            return $"{Name}{en},{IsFolder},{Size},{ModifiedDate}";
        }
    }
}