Terminal.Gui/Terminal.Gui/FileServices/FileSystemTreeBuilder.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.IO.Abstractions;
using System.Linq;

namespace Terminal.Gui {

	/// <summary>
	/// TreeView builder for creating file system based trees.
	/// </summary>
	public class FileSystemTreeBuilder : ITreeBuilder<IFileSystemInfo>, IComparer<IFileSystemInfo> {


		/// <summary>
		/// Creates a new instance of the <see cref="FileSystemTreeBuilder"/> class.
		/// </summary>
		public FileSystemTreeBuilder ()
		{
			Sorter = this;
		}

		/// <inheritdoc/>
		public bool SupportsCanExpand => true;

		/// <summary>
		/// Gets or sets a flag indicating whether to show files as leaf elements
		/// in the tree. Defaults to true.
		/// </summary>
		public bool IncludeFiles { get; } = true;

		/// <summary>
		/// Gets or sets the order of directory children.  Defaults to <see langword="this"/>.
		/// </summary>
		public IComparer<IFileSystemInfo> Sorter { get; set; }

		/// <inheritdoc/>
		public bool CanExpand (IFileSystemInfo toExpand)
		{
			return this.TryGetChildren (toExpand).Any ();
		}

		/// <inheritdoc/>
		public IEnumerable<IFileSystemInfo> GetChildren (IFileSystemInfo forObject)
		{
			return this.TryGetChildren (forObject).OrderBy(k=>k,Sorter);
		}

		private IEnumerable<IFileSystemInfo> TryGetChildren (IFileSystemInfo entry)
		{
			if (entry is IFileInfo) {
				return Enumerable.Empty<IFileSystemInfo> ();
			}

			var dir = (IDirectoryInfo)entry;

			try {
				return dir.GetFileSystemInfos ().Where (e => IncludeFiles || e is IDirectoryInfo);

			} catch (Exception) {

				return Enumerable.Empty<IFileSystemInfo> ();
			}
		}

		/// <inheritdoc/>
		public int Compare (IFileSystemInfo x, IFileSystemInfo y)
		{
			if (x is IDirectoryInfo && y is not IDirectoryInfo) {
				return -1;
			}

			if (x is not IDirectoryInfo && y is IDirectoryInfo) {
				return 1;
			}

			return x.Name.CompareTo (y.Name);
		}
	}
}