From 97722fbe9d474adffbba0b92e9727c48a8205234 Mon Sep 17 00:00:00 2001
From: John T <reflectron@outlook.com>
Date: Sat, 11 Nov 2017 08:53:55 -0500
Subject: Only 1/4 broken ShiftFS and WIP File Skimmer

---
 ShiftOS.Engine/ShiftFS/ShiftTree.cs | 66 +++++++++++++++++++++++++++++++++++++
 1 file changed, 66 insertions(+)
 create mode 100644 ShiftOS.Engine/ShiftFS/ShiftTree.cs

(limited to 'ShiftOS.Engine/ShiftFS/ShiftTree.cs')

diff --git a/ShiftOS.Engine/ShiftFS/ShiftTree.cs b/ShiftOS.Engine/ShiftFS/ShiftTree.cs
new file mode 100644
index 0000000..33c300a
--- /dev/null
+++ b/ShiftOS.Engine/ShiftFS/ShiftTree.cs
@@ -0,0 +1,66 @@
+using System;
+using System.Collections.Generic;
+using Whoa;
+
+namespace ShiftOS.Engine.ShiftFS
+{
+	[Serializable]
+	public class ShiftTree : ShiftDirectory, IShiftNode
+	{
+		public ShiftTree(string name, char letter) : base(name)
+		{
+			Name = name;
+			Letter = letter;
+		}
+
+		
+		public new IEnumerable<ShiftFile> Flatten()
+		{
+			foreach (var item in this)
+			{
+				switch (item)
+				{
+					case ShiftFile file:
+						yield return file;
+						break;
+					case ShiftDirectory dir:
+						foreach (var shiftNode in dir.Flatten())
+						{
+							yield return shiftNode;
+						}
+						break;
+				}
+			}
+		}
+
+		public new IEnumerable<ShiftDirectory> FlattenFolders()
+		{
+			foreach (var item in this)
+			{
+				if (!(item is ShiftDirectory dir)) continue;
+				yield return dir;
+				
+				foreach (var subdir in dir.FlattenFolders())
+				{
+					yield return subdir;
+				}
+			}
+		}
+
+		
+		public new string Name { get; set; }
+
+		
+		public char Letter { get; }
+
+		public new string FullName => $@"{Name}:\";
+
+		public new ShiftDirectory Parent
+		{
+			get => null;
+			set => throw new InvalidOperationException("Cannot set parent of ShiftTree");
+		}
+
+		public new ShiftTree Drive => this;
+	}
+}
\ No newline at end of file
-- 
cgit v1.2.3