daily update on 20240422 by nitload youmu, ssd structure improve

源代码/StgSharp/Data/SSD/SSDVirtualFolder.cs

@@ -31,43 +31,99 @@
 using StgSharp.Data.FileIO;
 
 using System;
+using System.Collections;
 using System.Collections.Generic;
 using System.Text;
 
 namespace StgSharp.Data
 {
-    public class SSDFolder: SSDSerializable
+    public class SSDFolder : SSDSerializable, IEnumerable<SSDSegmentHead>
     {
-        internal string _name;
+        private SSD local;
+        private SSDrawHead _head;
         internal int _folderID;
-        internal List<SSDraw> files;
-        internal List<SSDFolder> folders;
+        internal string _name;
+        internal Dictionary<int, SSDraw> files;
+        internal Dictionary<int, SSDFolder> folders;
+        internal Dictionary<string, int> idIndex;
 
-        public SSDFolder(string name)
+        internal SSDFolder()
+        {
+            folders = new Dictionary<int, SSDFolder>();
+            files = new Dictionary<int, SSDraw>();
+        }
+
+        internal SSDFolder(string name)
         {
             if (name == "root")
             {
-                throw new ArgumentNullException("Name cannot be root");
+                throw new ArgumentNullException(
+                    "Name cannot be root," +
+                    "it is preserved for root file indexer.");
+            }
+            if (name == "default")
+            {
+                throw new ArgumentNullException(
+                    "Name cannot be default," +
+                    "it is preserved for default file id provider.");
             }
             _name = name;
-            folders = new List<SSDFolder>();
-
+            folders = new Dictionary<int, SSDFolder>();
+            files = new Dictionary<int, SSDraw>();
         }
 
         public override byte SSDTypeCode => SerializableTypeCode.VirtualFolder;
 
+        /// <summary>
+        /// Check out if this virtual folder contains certain file.
+        /// </summary>
+        /// <param name="name">Name of the file</param>
+        /// <param name="type">Type code defined in <see cref="SerializableTypeCode"/></param>
+        /// <returns>Id of the file in its virtual disc</returns>
+        public int FindFileId(string name, byte type)
+        {
+            int fileid = 0;
+            if (idIndex.TryGetValue($"{name}{SerializableTypeCode.GetNameTail(type)}"
+                , out fileid) && !(type == SerializableTypeCode.VirtualFolder))
+            {
+                return fileid;
+            }
+            return 0;
+        }
+
+
+        /// <summary>
+        /// Check out if this virtual folder contains certain file.
+        /// </summary>
+        /// <param name="name">Name of the file</param>
+        /// <returns>Id of the folder in its virtual disc</returns>
+        public int FindFolderId(string name)
+        {
+            int folderid = 0;
+            if (idIndex.TryGetValue($"{name}_f", out folderid))
+            {
+                return folderid;
+            }
+            return 0;
+        }
+
+        public SSDraw CreateFile(string name, byte type)
+        {
+            SSDraw ret = new SSDraw(name + SerializableTypeCode.GetNameTail(type));
+            ret.ID = local.
+        }
+
+        //TODO 虚拟文件夹的GetBytes好难写
         public override byte[] GetBytes()
         {
-            List<byte> rawBytes =new List<byte>();
+            List<byte> rawBytes = new List<byte>();
             byte[] size = new byte[5];
             List<int> fileid = new List<int>();
             fileid.Add(4);
             foreach (var item in files)
             {
-                fileid.Add(item.)
+                fileid.Add(item.);
             }
-
-            
             return rawBytes.ToArray();
         }
 
@@ -80,5 +136,30 @@ namespace StgSharp.Data
         {
             throw new NotImplementedException();
         }
+
+        IEnumerator IEnumerable.GetEnumerator()
+        {
+            foreach (var f in files)
+            {
+                yield return f.Value.Head._head;
+            }
+            foreach (var f in folders)
+            {
+                yield return f.Value._head._head;
+            }
+        }
+
+        IEnumerator<SSDSegmentHead> IEnumerable<SSDSegmentHead>.GetEnumerator()
+        {
+            foreach (var f in files)
+            {
+                yield return f.Value.Head._head;
+            }
+            foreach (var f in folders)
+            {
+                yield return f.Value._head._head;
+            }
+        }
+
     }
 }