Example #1
0
        static int Main(string[] args)
        {
            var directories = new List <DirectoryInfo>();
            var switches    = new List <string>();

            var maybeSwitch = true;

            foreach (var arg in args)
            {
                if (maybeSwitch)
                {
                    if (arg.StartsWithAny("-", "/"))
                    {
                        switches.Add(arg);
                        continue;
                    }

                    maybeSwitch = false;
                }

                var directory = new DirectoryInfo(arg);

                if (!directory.Exists)
                {
                    Console.WriteLine($"[Error] Directory {arg} not found - Aborting!");
                    return((int)CLI.ExitCode.DirectoryNotFound);
                }

                directories.Add(directory);
            }

            // add current directory if none passed
            if (!directories.Any())
            {
                directories.Add(new DirectoryInfo(Environment.CurrentDirectory));
            }

            var configuration = new Configuration();

            CLI.ProcessCommandLine(switches, configuration);
            var stats = new RuntimeStats();

            DuplicateFileMerger.ProcessFolders(directories, configuration, stats);
            Console.WriteLine();
            Console.WriteLine("Statistics");
            Console.WriteLine();
            Console.WriteLine("         HardLinks  SymbolicLinks");
            Console.WriteLine($"Created  {stats.HardLinkStats.Created,9:N0}  {stats.SymbolicLinkStats.Created,13:N0}");
            Console.WriteLine($"Removed  {stats.HardLinkStats.Removed,9:N0}  {stats.SymbolicLinkStats.Removed,13:N0}");
            Console.WriteLine($"Deleted  {stats.HardLinkStats.Deleted,9:N0}  {stats.SymbolicLinkStats.Deleted,13:N0}");
            Console.WriteLine($"Seen     {stats.HardLinkStats.Seen,9:N0}  {stats.SymbolicLinkStats.Seen,13:N0}");
            Console.WriteLine();
            Console.WriteLine($"Folders Total : {stats.FolderCount:N0}");
            Console.WriteLine($"Files Total   : {stats.FileCount:N0}");
            Console.WriteLine($"Bytes Total   : {stats.BytesTotal:N0} ({FilesizeFormatter.FormatIEC(stats.BytesTotal,"N1")})");


#if DEBUG
            Console.WriteLine("READY.");
            Console.ReadKey();
#endif

            return((int)CLI.ExitCode.Success);
        }
Example #2
0
        /// <summary>
        /// Handles a specific file along with a group of files already seen with the same size.
        /// </summary>
        /// <param name="item">The item.</param>
        /// <param name="configuration">The configuration.</param>
        /// <param name="knownWithThisLength">Length of the known with this.</param>
        private static void _HandleFileWithGivenSize(
            FileInfo item,
            Configuration configuration,
            ConcurrentDictionary <string, FileEntry> knownWithThisLength)
        {
            var myKey    = _GenerateKey(item);
            var checksum = knownWithThisLength.GetOrAdd(myKey, new FileEntry(item));

            var isHardLink = false;
            IEnumerable <FileInfo> hardlinks;

            try {
                hardlinks = item.GetHardLinkTargets();
            } catch (Exception e) {
                _RemoveFileEntry(item, knownWithThisLength);
                Console.WriteLine($"[Error] Could not enumerate HardLinks {item.FullName}: {e.Message}");
                return;
            }

            foreach (var target in hardlinks)
            {
                isHardLink = true;
                knownWithThisLength.TryAdd(_GenerateKey(target), new FileEntry(target));
            }

            if (isHardLink)
            {
                _HandleExistingHardLink(item, configuration, knownWithThisLength);
                return;
            }

            string symlink;

            try {
                symlink = item.GetSymbolicLinkTarget();
            } catch (Exception e) {
                _RemoveFileEntry(item, knownWithThisLength);
                Console.WriteLine($"[Error] Could not enumerate SymLink {item.FullName}: {e.Message}");
                return;
            }

            if (symlink != null)
            {
                knownWithThisLength.TryAdd(symlink, new FileEntry(new FileInfo(symlink)));
                _HandleExistingSymbolicLink(item, configuration, knownWithThisLength);
                return;
            }

            // find matching file in seen list and try to hard or symlink
            var sameFiles =
                knownWithThisLength
                .Where(kvp => kvp.Key != myKey)
                .Where(kvp => kvp.Value.Equals(checksum))
                .Select(kvp => kvp.Key)
            ;

            foreach (var sameFile in sameFiles)
            {
                var temporaryFile = _CreateTemporaryFileInSameDirectory(item);
                temporaryFile.Delete();

                var isSymlink = false;

                try {
                    temporaryFile.CreateHardLinkFrom(sameFile);
                } catch (Exception e1) {
                    if (configuration.AlsoTrySymbolicLinks)
                    {
                        isSymlink = true;
                        try {
                            temporaryFile.CreateSymbolicLinkFrom(sameFile);
                        } catch (Exception e2) {
                            Console.WriteLine(
                                $"[Warning] Could not Symlink {item.FullName}({FilesizeFormatter.FormatUnit(item.Length, true)}) --> {sameFile}: {e2.Message}");
                            continue;
                        }
                    }
                    else
                    {
                        Console.WriteLine(
                            $"[Warning] Could not Hardlink {item.FullName}({FilesizeFormatter.FormatUnit(item.Length, true)}) --> {sameFile}: {e1.Message}");
                        continue;
                    }
                }

                var isAlreadyDeleted = false;

                try {
                    item.Attributes &= ~FileAttributes.ReadOnly;
                    item.Delete();
                    isAlreadyDeleted = true;
                    File.Move(temporaryFile.FullName, item.FullName);
                } catch {
                    if (isAlreadyDeleted)
                    {
                        // undo file deletion
                        temporaryFile.CopyTo(item.FullName, true);
                    }
                    else
                    {
                        // undo temp file creation
                        temporaryFile.Delete();
                    }
                    throw;
                }

                if (isSymlink)
                {
                    if (configuration.SetReadOnlyAttributeOnNewSymbolicLinks)
                    {
                        item.Attributes |= FileAttributes.ReadOnly;
                    }
                }
                else
                {
                    if (configuration.SetReadOnlyAttributeOnNewHardLinks)
                    {
                        item.Attributes |= FileAttributes.ReadOnly;
                    }
                }

                Console.WriteLine($"[Info] Created {(isSymlink ? "Symlink" : "Hardlink")} for {item.FullName}({FilesizeFormatter.FormatUnit(item.Length, true)}) --> {sameFile}");
                return;
            }
        }