private static void BuildTree(string rootFolderPath, Folder root)
{
string[] files = Directory.GetFiles(rootFolderPath);
foreach (var file in files)
{
var fileInfo = new FileInfo(file);
root.Files.Add(new File(fileInfo.Name, fileInfo.Length));
}
try
{
string[] directories = Directory.GetDirectories(rootFolderPath);
foreach (var directory in directories)
{
var directoryInfo = new DirectoryInfo(directory);
var newDir = new Folder(directoryInfo.Name);
BuildTree(directory, newDir);
root.Folders.Add(newDir);
}
}
catch (UnauthorizedAccessException)
{
// Skipping folders in which we don't have access
}
}
public static void Main()
{
string rootFolderPath = @"C:\Windows";
var root = new Folder(GetName(rootFolderPath));
BuildTree(rootFolderPath, root);
long size = 0;
GetSumOfFileSizesInFolder(root, ref size);
Console.WriteLine("Size of all files in {0}: {1}", root.Name, size);
}
// DFS Algorithm
private static void GetSumOfFileSizesInFolder(Folder root, ref long size)
{
foreach (var file in root.Files)
{
size += file.Size;
}
foreach (var folder in root.Folders)
{
GetSumOfFileSizesInFolder(folder, ref size);
}
}