438 lines
13 KiB
C#

using System;
using System.Collections.Generic;
using System.Text;
using UniversalEditor.Accessors;
namespace UniversalEditor.ObjectModels.FileSystem
{
public class FileSystemObjectModel : ObjectModel, IFileSystemContainer
{
private ObjectModelReference _omr = null;
protected override ObjectModelReference MakeReferenceInternal()
{
if (_omr == null)
{
_omr = new ObjectModelReference(GetType(), new Guid("{A23026E9-DFE1-4090-AF35-8B916D3F1FCD}"));
_omr.Title = "File system/archive";
_omr.Path = new string[] { "General", "File system/archive" };
}
return _omr;
}
public override void Clear()
{
mvarFiles.Clear();
mvarFolders.Clear();
mvarID = Guid.Empty;
mvarTitle = String.Empty;
mvarPathSeparators = new string[] { System.IO.Path.DirectorySeparatorChar.ToString(), System.IO.Path.AltDirectorySeparatorChar.ToString() };
}
public override void CopyTo(ObjectModel where)
{
FileSystemObjectModel clone = (where as FileSystemObjectModel);
clone.ID = mvarID;
for (int i = 0; i < mvarFiles.Count; i++)
{
File file = mvarFiles[i];
clone.Files.Add(file.Clone() as File);
}
for (int i = 0; i < mvarFolders.Count; i++)
{
Folder folder = mvarFolders[i];
clone.Folders.Add(folder.Clone() as Folder);
}
}
public static FileSystemObjectModel FromFiles(string[] fileNames)
{
// TODO: This doesn't work because GetAvailableObjectModel returns an
// ObjectModel but automatically closes the file after reading... deferred
// FileSystemObjectModels associated files need to remain open in order to
// read the file data. Consider using a Document-based approach which provides
// more control over closing files as needed?
FileSystemObjectModel fsom = new FileSystemObjectModel();
foreach (string fileName in fileNames)
{
FileAccessor accessor = new FileAccessor(fileName);
FileSystemObjectModel fsom1 = UniversalEditor.Common.Reflection.GetAvailableObjectModel<FileSystemObjectModel>(accessor);
if (fsom1 == null) continue;
fsom1.CopyTo(fsom);
}
return fsom;
}
public static FileSystemObjectModel FromDirectory(string path, string searchPattern = "*.*", System.IO.SearchOption searchOption = System.IO.SearchOption.TopDirectoryOnly)
{
string[] folders = System.IO.Directory.GetDirectories(path);
string[] files = System.IO.Directory.GetFiles(path);
FileSystemObjectModel fsom = new FileSystemObjectModel();
foreach (string folder in folders)
{
string title = System.IO.Path.GetFileName(folder);
fsom.Folders.Add(title);
}
foreach (string fileName in files)
{
string title = System.IO.Path.GetFileName(fileName);
fsom.Files.Add(title, fileName);
}
return fsom;
// string[] files = System.IO.Directory.GetFiles(path, searchPattern, searchOption);
// return FromFiles(files);
}
private File.FileCollection mvarFiles = new File.FileCollection();
public File.FileCollection Files { get { return mvarFiles; } }
private Folder.FolderCollection mvarFolders = new Folder.FolderCollection();
public Folder.FolderCollection Folders { get { return mvarFolders; } }
private Guid mvarID = Guid.Empty;
/// <summary>
/// The unique ID associated with this file system. Not supported by all data formats.
/// </summary>
public Guid ID { get { return mvarID; } set { mvarID = value; } }
private string mvarTitle = String.Empty;
/// <summary>
/// The title associated with this file system. Not supported by all data formats.
/// </summary>
public string Title { get { return mvarTitle; } set { mvarTitle = value; } }
private string[] mvarPathSeparators = new string[] { System.IO.Path.DirectorySeparatorChar.ToString(), System.IO.Path.AltDirectorySeparatorChar.ToString() };
public string[] PathSeparators { get { return mvarPathSeparators; } set { mvarPathSeparators = value; } }
public bool ContainsFile(string path)
{
return (FindFile(path) != null);
}
public File FindFile(string path)
{
string[] pathParts = path.Split(new char[] { System.IO.Path.DirectorySeparatorChar, System.IO.Path.AltDirectorySeparatorChar });
if (pathParts.Length == 1)
{
File file = mvarFiles[pathParts[0]];
if (file != null) return file;
}
else
{
Folder parentFolder = mvarFolders[pathParts[0]];
if (parentFolder == null) return null;
for (int i = 1; i < pathParts.Length; i++)
{
if (i < pathParts.Length - 1)
{
parentFolder = parentFolder.Folders[pathParts[i]];
if (parentFolder == null) return null;
}
else
{
return parentFolder.Files[pathParts[i]];
}
}
}
return null;
}
public Folder FindFolder(string name)
{
string[] path = name.Split(new char[] { '/' });
Folder parent = null;
for (int i = 0; i < path.Length - 1; i++)
{
if (parent == null)
{
parent = mvarFolders[path[i]];
}
else
{
parent = parent.Folders[path[i]];
}
}
if (parent == null)
{
return mvarFolders[path[path.Length - 1]];
}
else
{
return parent.Folders[path[path.Length - 1]];
}
}
public object FindObject(string name)
{
string[] path = name.Split(new char[] { '/' });
Folder parent = null;
for (int i = 0; i < path.Length - 1; i++)
{
if (parent == null)
{
parent = mvarFolders[path[i]];
}
else
{
parent = parent.Folders[path[i]];
}
}
if (parent == null)
{
File file = mvarFiles[path[path.Length - 1]];
Folder folder = mvarFolders[path[path.Length - 1]];
if (folder == null) return file;
return folder;
}
else
{
File file = parent.Files[path[path.Length - 1]];
Folder folder = parent.Folders[path[path.Length - 1]];
if (folder == null) return file;
return folder;
}
}
public Folder AddFolder(string name)
{
string[] path = name.Split(mvarPathSeparators, StringSplitOptions.None);
Folder parent = null;
for (int i = 0; i < path.Length - 1; i++)
{
if (parent == null)
{
parent = mvarFolders[path[i]];
}
else
{
parent = parent.Folders[path[i]];
}
if (parent == null) throw new System.IO.DirectoryNotFoundException();
}
if (parent == null)
{
return mvarFolders.Add(path[path.Length - 1]);
}
return parent.Folders.Add(path[path.Length - 1]);
}
/// <summary>
/// Adds a <see cref="File" /> to this <see cref="FileSystemObjectModel" />, building the parent directory hierarchy as appropriate.
/// </summary>
/// <param name="name">The full path of the <see cref="File" /> to create, including any parent directories.</param>
/// <param name="fileData"></param>
/// <returns></returns>
public File AddFile(string name, byte[] fileData = null)
{
string[] path = name.Split(mvarPathSeparators, StringSplitOptions.None);
Folder parent = null;
for (int i = 0; i < path.Length - 1; i++)
{
if (parent == null)
{
if (mvarFolders.Contains(path[i]))
{
parent = mvarFolders[path[i]];
}
else
{
parent = mvarFolders.Add(path[i]);
}
}
else
{
if (parent.Folders.Contains(path[i]))
{
parent = parent.Folders[path[i]];
}
else
{
parent = parent.Folders.Add(path[i]);
}
}
if (parent == null)
{
throw new System.IO.DirectoryNotFoundException();
}
}
File file = new File();
file.Name = path[path.Length - 1];
if (fileData != null)
{
file.SetDataAsByteArray(fileData);
}
if (parent == null)
{
mvarFiles.Add(file);
}
else
{
parent.Files.Add(file);
}
return file;
}
/// <summary>
/// Gets all files in all folders of the <see cref="FileSystemObjectModel" /> with file names that
/// match the <see cref="searchPattern"/>, and assigns the file names separated by the
/// <see cref="pathSeparator"/>.
/// </summary>
/// <param name="searchPattern">The string by which to filter the retrieved file names.</param>
/// <param name="pathSeparator">The string by which to separate directory and file names.</param>
/// <returns></returns>
public File[] GetFiles(string searchPattern = null, string pathSeparator = null)
{
if (pathSeparator == null) pathSeparator = "/";
List<File> files = new List<File>();
for (int i = 0; i < mvarFiles.Count; i++)
{
File file = mvarFiles[i];
if (searchPattern != null && !file.Name.Match(searchPattern)) continue;
files.Add(file);
}
for (int i = 0; i < mvarFolders.Count; i++)
{
Folder folder = mvarFolders[i];
GetAllFilesRecursively(folder, ref files, folder.Name, pathSeparator, searchPattern);
}
return files.ToArray();
}
/// <summary>
/// Gets all files in all folders of the <see cref="FileSystemObjectModel" />, and assigns the file names
/// separated by the default path separator.
/// </summary>
/// <returns></returns>
public File[] GetAllFiles(string pathSeparator = null)
{
if (pathSeparator == null) pathSeparator = "/";
List<File> files = new List<File>();
for (int i = 0; i < mvarFiles.Count; i++)
{
File file = mvarFiles[i];
files.Add(file);
}
for (int i = 0; i < mvarFolders.Count; i++ )
{
Folder folder = mvarFolders[i];
GetAllFilesRecursively(folder, ref files, folder.Name, pathSeparator);
}
return files.ToArray();
}
private void GetAllFilesRecursively(Folder folder, ref List<File> files, string parentPath, string pathSeparator, string searchPattern = null)
{
for (int i = 0; i < folder.Files.Count; i++)
{
File file = folder.Files[i];
if (searchPattern != null && !file.Name.Match(searchPattern)) continue;
File file2 = (file.Clone() as File);
file2.Name = parentPath + pathSeparator + file.Name;
files.Add(file2);
}
for (int i = 0; i < folder.Folders.Count; i++)
{
Folder folder1 = folder.Folders[i];
GetAllFilesRecursively(folder1, ref files, parentPath + pathSeparator + folder1.Name, pathSeparator, searchPattern);
}
}
public IFileSystemObject[] GetAllObjects(string pathSeparator = null, System.IO.SearchOption option = System.IO.SearchOption.AllDirectories, IFileSystemObjectType objectTypes = IFileSystemObjectType.All)
{
if (pathSeparator == null) pathSeparator = "/";
List<IFileSystemObject> files = new List<IFileSystemObject>();
if ((objectTypes & IFileSystemObjectType.File) == IFileSystemObjectType.File)
{
for (int i = 0; i < mvarFiles.Count; i++)
{
File file = mvarFiles[i];
files.Add(file);
}
}
for (int i = 0; i < mvarFolders.Count; i++)
{
Folder folder = mvarFolders[i];
if ((objectTypes & IFileSystemObjectType.Folder) == IFileSystemObjectType.Folder)
{
files.Add(folder);
}
if (option == System.IO.SearchOption.AllDirectories) GetAllObjectsRecursively(folder, ref files, folder.Name, pathSeparator, null, objectTypes);
}
return files.ToArray();
}
private void GetAllObjectsRecursively(Folder folder, ref List<IFileSystemObject> files, string parentPath, string pathSeparator, string searchPattern = null, IFileSystemObjectType objectTypes = IFileSystemObjectType.All)
{
if ((objectTypes & IFileSystemObjectType.File) == IFileSystemObjectType.File)
{
for (int i = 0; i < folder.Files.Count; i++)
{
File file = folder.Files[i];
if (searchPattern != null && !file.Name.Match(searchPattern)) continue;
File file2 = (file.Clone() as File);
file2.Name = parentPath + pathSeparator + file.Name;
files.Add(file2);
}
}
for (int i = 0; i < folder.Folders.Count; i++)
{
Folder folder1 = folder.Folders[i];
if ((objectTypes & IFileSystemObjectType.Folder) == IFileSystemObjectType.Folder)
{
files.Add(folder1);
}
GetAllObjectsRecursively(folder1, ref files, parentPath + pathSeparator + folder1.Name, pathSeparator, searchPattern);
}
}
/// <summary>
/// Gets the next available "New Folder" name for the given <see cref="IFileSystemContainer" />.
/// </summary>
/// <returns>A string "New Folder" if there are no other "New Folder"s in the given <see cref="IFileSystemContainer" />; otherwise, a string "New Folder (n)" where N is the number of "New Folder"s in the given <see cref="IFileSystemContainer" /> plus one.</returns>
public static string GetNewFolderName(IFileSystemContainer container)
{
int count = 0;
foreach (Folder f in container.Folders)
{
if (f.Name.StartsWith("New Folder (") && f.Name.EndsWith(")"))
{
string strIntPart = f.Name.Substring("New Folder (".Length, f.Name.Length - "New Folder (".Length - 1);
int intPart = 0;
if (Int32.TryParse(strIntPart, out intPart))
{
if (intPart > count) count = intPart;
}
}
else if (f.Name == "New Folder")
{
count++;
}
}
if (count == 0) return "New Folder";
return "New Folder (" + (count + 1).ToString() + ")";
}
/// <summary>
/// Gets the next available "New Folder" name for this <see cref="FileSystemObjectModel" />.
/// </summary>
/// <returns>A string "New Folder" if there are no other "New Folder"s in this <see cref="FileSystemObjectModel" />; otherwise, a string "New Folder (n)" where N is the number of "New Folder"s in this <see cref="FileSystemObjectModel" /> plus one.</returns>
public string GetNewFolderName()
{
return FileSystemObjectModel.GetNewFolderName(this);
}
}
}