Worker.cs

using System;
using System.Threading;
using System.Diagnostics;
using System.IO;
using System.Text.RegularExpressions;

namespace fmdevtools
{
    class Worker
    {
        public string file { get; private set; }
        private string source;
        private string target;

        private string username;
        private string password;

        private string dmtPath;
        private string extraArgs;

        private bool running;

        private Thread t;

        private Process proc;

        public DateTime startTime { get; private set; }

        public DateTime endTime { get; private set; }

        public long fileSize { get; private set; }

        public int fileNumber { get; private set; }

        public Worker(string file, string source, string target, string username, string password, string dmtPath, string extraArgs, int fileNumber) {
            this.dmtPath = dmtPath;
            this.file = escape(file);
            this.source = escape(source);
            this.target = escape(target);
            this.username = escape(username);
            this.password = escape(password);
            this.extraArgs = extraArgs;
            this.fileNumber = fileNumber;
        }

        private static string escape(string raw) {
            Regex pattern = new Regex("[\"]");
            return pattern.Replace(raw, "\"\"\"");
        }

        static String BytesToString(long byteCount) {
            string[] suf = { "B", "KB", "MB", "GB", "TB", "PB", "EB" }; //Longs run out around EB
            if (byteCount == 0)
                return "0" + suf[0];
            long bytes = Math.Abs(byteCount);
            int place = Convert.ToInt32(Math.Floor(Math.Log(bytes, 1024)));
            double num = Math.Round(bytes / Math.Pow(1024, place), 1);
            return (Math.Sign(byteCount) * num).ToString() + suf[place];
        }

        public void ThreadProc() {
            this.startTime = DateTime.Now;

            String outFileName = file + ".log";
            try {
                if (!File.Exists(outFileName)) {
                    File.Create(outFileName).Dispose();
                }

                if (!Directory.Exists("target"))
                {
                    Directory.CreateDirectory("target");
                }

                proc = new Process();
                proc.StartInfo.UseShellExecute = false;
                proc.StartInfo.FileName = this.dmtPath;
                proc.StartInfo.CreateNoWindow = true;
                proc.StartInfo.RedirectStandardOutput = true;
                proc.StartInfo.RedirectStandardError = true;
                proc.StartInfo.Arguments = "--recover \""+source+"\" -target_filename \""+target+"\"" +
                    " -u \""+username+"\" -p \""+password+"\" "+extraArgs;
                proc.Start();
                Console.WriteLine(file + " - " + DateTime.Now.ToString("HH:mm:ss") + " - Started");
                StreamReader reader = proc.StandardOutput;
                string line = null;

                while (!proc.HasExited) {
                    //Thread.Sleep(100);
                    line = reader.ReadLine();
                    String currentTimestamp = DateTime.Now.ToString("HH:mm:ss");
                    Console.WriteLine(file + " - " + currentTimestamp + " - " + line);
                    File.AppendAllText(outFileName, currentTimestamp + "\t" + line + "\r\n");
                }
                line = reader.ReadToEnd();
                Console.WriteLine(line);
                File.AppendAllText(outFileName, line + "\r\n");
            } catch (Exception e) {
                running = false;
                Console.Beep(800, 200);
                throw e;
            }
            this.endTime = DateTime.Now;

            this.fileSize = -1;
            try {
                FileInfo info = new FileInfo(this.target);
                this.fileSize = info.Length;
            } catch (Exception) {

            }

            running = false;
            Console.Beep(800, 200);
        }

        public void Dispose()
        {
            proc.Kill();
        }

        public bool isRunning() {
            return running;
        }

        public void start() {
            this.t = new Thread(new ThreadStart(ThreadProc));
            this.t.Start();
            this.running = true;
        }
    }
}