PdfPig/tools/UglyToad.PdfPig.ConsoleRunner/Program.cs

using System;
using System.IO;
using System.Text;
using Console = System.Console;

namespace UglyToad.PdfPig.ConsoleRunner
{
    public static class Program
    {
        public static int Main(string[] args)
        {
            if (args.Length == 0)
            {
                Console.WriteLine("At least 1 argument, path to test file directory, must be provided.");
                return 7;
            }

            var path = args[0];

            if (!Directory.Exists(path))
            {
                Console.WriteLine($"The provided path is not a valid directory: {path}.");
                return 7;
            }

            var maxCount = default(int?);

            if (args.Length > 1 && int.TryParse(args[1], out var countIn))
            {
                maxCount = countIn;
            }

            var hasError = false;
            var errorBuilder = new StringBuilder();
            var fileList = Directory.GetFiles(path, "*.pdf", SearchOption.AllDirectories);
            var runningCount = 0;

            Console.WriteLine($"Found {fileList.Length} files.");

            Console.WriteLine($"{GetCleanFilename("File")}| Size\t| Words\t| Pages");

            foreach (var file in fileList)
            {
                if (maxCount.HasValue && runningCount >= maxCount)
                {
                    break;
                }

                try
                {
                    var numWords = 0;
                    var numPages = 0;
                    using (var pdfDocument = PdfDocument.Open(file))
                    {
                        foreach (var page in pdfDocument.GetPages())
                        {
                            numPages++;
                            foreach (var word in page.GetWords())
                            {
                                if (word != null)
                                {
                                    numWords++;
                                }
                            }
                        }
                    }

                    var filename = Path.GetFileName(file);

                    var size = new FileInfo(file);

                    Console.WriteLine($"{GetCleanFilename(filename)}| {size.Length}\t| {numWords}\t| {numPages}");
                }
                catch (Exception ex)
                {
                    hasError = true;
                    errorBuilder.AppendLine($"Parsing document {file} failed due to an error.")
                        .Append(ex)
                        .AppendLine();
                }

                runningCount++;
            }

            if (hasError)
            {
                Console.WriteLine(errorBuilder.ToString());
                return 5;
            }

            Console.WriteLine("Complete! :)");

            return 0;
        }

        private static string GetCleanFilename(string name, int maxLength = 30)
        {
            if (name.Length <= maxLength)
            {
                var fillLength = maxLength - name.Length;

                return name + new string(' ', fillLength);
            }

            return name.Substring(0, maxLength);
        }
    }
}