protected override bool RunInternal(object options)
{
PristineVerbOptions localOptions = options as PristineVerbOptions;
var records = Workspace.GetRecords(Workspace.Version);
HashSet <string> recordNames = new HashSet <string>();
HashSet <string> lowercaseRecordNames = new HashSet <string>();
foreach (var x in records)
{
recordNames.Add(x.CanonicalName);
lowercaseRecordNames.Add(x.CanonicalName.ToLower());
}
List <string> files = new List <string>();
List <string> directories = new List <string>();
PopulateFilesystem(files, directories, Workspace.RootDirectory, localOptions);
Printer.PrintMessage("#b#{0}## objects in vault, #b#{1}## objects in filesystem.", records.Count, files.Count + directories.Count);
System.Collections.Concurrent.ConcurrentBag <string> filesToDelete = new System.Collections.Concurrent.ConcurrentBag <string>();
System.Collections.Concurrent.ConcurrentBag <string> directoriesToDelete = new System.Collections.Concurrent.ConcurrentBag <string>();
Parallel.ForEach(files, (string fn) =>
{
string localName = Workspace.GetLocalPath(fn);
if (!lowercaseRecordNames.Contains(localName.ToLower()))
{
filesToDelete.Add(fn);
}
else if (!recordNames.Contains(localName))
{
Printer.PrintMessage("Incorrect record name - wrong case for object #b#{0}##", localName);
}
});
Parallel.ForEach(directories, (string fn) =>
{
string localName = Workspace.GetLocalPath(fn + "/");
if (!lowercaseRecordNames.Contains(localName.ToLower()))
{
directoriesToDelete.Add(fn);
}
else if (!recordNames.Contains(localName))
{
Printer.PrintMessage("Incorrect record name - wrong case for object #b#{0}##", localName);
}
});
if (filesToDelete.Count == 0 && directoriesToDelete.Count == 0)
{
Printer.PrintMessage("No changes found.");
return(true);
}
if (localOptions.List)
{
Printer.PrintMessage("Identified #e#{0}## files and #e#{1}## directories to remove.", filesToDelete.Count, directoriesToDelete.Count);
foreach (var x in filesToDelete)
{
Printer.PrintMessage(x);
}
foreach (var x in directoriesToDelete)
{
Printer.PrintMessage(x);
}
}
Printer.PrintMessage("Identified #e#{0}## files and #e#{1}## directories which will be removed.", filesToDelete.Count, directoriesToDelete.Count);
if (localOptions.Force || Printer.Prompt("Restore to pristine state?"))
{
foreach (var x in filesToDelete)
{
try
{
System.IO.File.SetAttributes(x, FileAttributes.Normal);
System.IO.File.Delete(x);
if (!localOptions.Quiet)
{
Printer.PrintMessage("#e#Purging:## {0}", x);
}
}
catch
{
Printer.PrintMessage("#x#Couldn't delete: {0}##", x);
}
}
Printer.PrintMessage("#e#Removing Directories...##");
foreach (var x in directoriesToDelete.ToArray().OrderByDescending(x => x.Length))
{
try
{
System.IO.Directory.Delete(x, true);
}
catch
{
Printer.PrintMessage("#x#Couldn't delete: {0}##", x);
}
}
}
return(true);
}
private void PopulateFilesystem(List <string> files, List <string> directories, DirectoryInfo rootDirectory, PristineVerbOptions options)
{
foreach (var x in rootDirectory.GetFiles())
{
if (!options.Hard && x.Name.StartsWith(".") && options.ProtectedHiddenFiles.Contains(x.Name))
{
continue;
}
files.Add(x.FullName);
}
foreach (var x in rootDirectory.GetDirectories())
{
if (x.Name.StartsWith(".") && (x.Name == ".versionr" || !options.Hard && options.ProtectedHiddenDirectories.Contains(x.Name)))
{
continue;
}
directories.Add(x.FullName);
PopulateFilesystem(files, directories, x, options);
}
}