PDFPatcher/App/MainForm.cs

using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Drawing;
using System.Windows.Forms;
using PDFPatcher.Common;
using PDFPatcher.Functions;

namespace PDFPatcher
{
	public partial class MainForm : Form
	{
		static readonly Dictionary<Function, FunctionControl> __FunctionControls = new Dictionary<Function, FunctionControl>();

		ReportControl _LogControl;
		#region 公共功能
		BackgroundWorker _Worker;
		readonly FormState _formState = new FormState();
		bool _FullScreen;
		///<summary>获取或指定全屏显示的值。</summary>
		[Browsable(false)]
		public bool FullScreen {
			get => _FullScreen;
			set {
				if (value == _FullScreen) {
					return;
				}
				_FullScreen = value;
				if (value) {
					_MainMenu.Visible = _GeneralToolbar.Visible = false;
					_formState.Maximize(this);
				}
				else {
					_MainMenu.Visible = true;
					_GeneralToolbar.Visible = AppContext.Toolbar.ShowGeneralToolbar;
					_formState.Restore(this);
				}
			}
		}

		/// <summary>
		/// 设置控件的提示信息。
		/// </summary>
		internal void SetTooltip(Control control, string text) {
			_ToolTip.SetToolTip(control, text);
		}

		/// <summary>
		/// 获取或设置状态栏文本。
		/// </summary>
		internal string StatusText {
			get => _MainStatusLabel.Text;
			set => _MainStatusLabel.Text = value;
		}

		#region Worker
		///<summary>获取或指定后台进程。</summary>
		internal BackgroundWorker GetWorker() {
			if (_Worker == null) {
				_Worker = new BackgroundWorker {
					WorkerReportsProgress = true,
					WorkerSupportsCancellation = true
				};
				_Worker.DoWork += Worker_DoWork;
				_Worker.ProgressChanged += Worker_ProgressChanged;
				_Worker.RunWorkerCompleted += Worker_RunWorkerCompleted;
				Tracker.SetWorker(_Worker);
			}
			return _Worker;
		}
		internal bool IsWorkerBusy => _Worker?.IsBusy == true;

		void Worker_DoWork(object sender, DoWorkEventArgs e) {
			_Worker.ReportProgress(0);
			FileHelper.ResetOverwriteMode();
			AppContext.Abort = false;
		}

		void Worker_RunWorkerCompleted(object sender, RunWorkerCompletedEventArgs e) {
			_MainMenu.Enabled = _GeneralToolbar.Enabled = _FunctionContainer.Enabled = true;
			foreach (TabPage item in _FunctionContainer.TabPages) {
				item.Enabled = true;
			}
			if (e.Error == null || e.Cancelled == false) {
				_LogControl.Complete();
			}
		}

		public void ResetWorker() {
			if (_Worker != null) {
				if (_Worker.IsBusy) {
					throw new InvalidOperationException("Worker is busy. Can't be reset.");
				}
				_Worker.Dispose();
				_Worker = null;
			}
		}

		void Worker_ProgressChanged(object sender, ProgressChangedEventArgs e) {
			var s = e.UserState as string;
			if (s == "INC") {
				_LogControl.IncrementProgress(e.ProgressPercentage);
				return;
			}
			else if (s == "GOAL") {
				_LogControl.SetGoal(e.ProgressPercentage);
				return;
			}
			else if (s == "TINC") {
				_LogControl.IncrementTotalProgress();
			}
			else if (s == "TGOAL") {
				_LogControl.SetTotalGoal(e.ProgressPercentage);
			}
			if (e.ProgressPercentage > 0) {
				_LogControl.SetProgress(e.ProgressPercentage);
			}
			else if (e.ProgressPercentage == 0) {
				_MainMenu.Enabled = _GeneralToolbar.Enabled = _FunctionContainer.Enabled = false;
				foreach (TabPage item in _FunctionContainer.TabPages) {
					item.Enabled = false;
				}
				_LogControl.Reset();
				ShowLogControl();
			}
			else if (s != null) {
				_LogControl.PrintMessage(s, (Tracker.Category)e.ProgressPercentage);
			}
		}
		#endregion

		internal FunctionControl GetFunctionControl(Function functionName) {
			if (__FunctionControls.TryGetValue(functionName, out FunctionControl f) && f.IsDisposed == false) {
				return f;
			}
			switch (functionName) {
				case Function.FrontPage:
					return __FunctionControls[functionName] = new FrontPageControl();
				case Function.Patcher:
					return __FunctionControls[functionName] = new PatcherControl();
				case Function.Merger:
					return __FunctionControls[functionName] = new MergerControl();
				case Function.BookmarkGenerator:
					return __FunctionControls[functionName] = new AutoBookmarkControl();
				case Function.InfoExchanger:
					return __FunctionControls[functionName] = new InfoExchangerControl();
				case Function.ExtractPages:
					return __FunctionControls[functionName] = new ExtractPageControl();
				case Function.ExtractImages:
					return __FunctionControls[functionName] = new ExtractImageControl();
				case Function.Editor:
					var b = new EditorControl();
					b.DocumentChanged += OnDocumentChanged;
					return b;
				case Function.Ocr:
					return __FunctionControls[functionName] = new OcrControl();
				case Function.RenderPages:
					return __FunctionControls[functionName] = new RenderImageControl();
				case Function.About:
					return __FunctionControls[functionName] = new AboutControl();
				case Function.Inspector:
					var d = new DocumentInspectorControl();
					d.DocumentChanged += OnDocumentChanged;
					return d;
				case Function.Rename:
					return __FunctionControls[functionName] = new RenameControl();
				default:
					return null;
			}
		}

		void OnDocumentChanged(object sender, DocumentChangedEventArgs args) {
			FilePath p = args.Path;
			_MainStatusLabel.Text = p.ToString();
			if (!p.IsValidPath) {
				return;
			}

			var t = p.FileNameWithoutExtension;
			if (t.Length > 20) {
				t = t.Substring(0, 17) + "...";
			}
			if (sender is Control f) {
				f = f.Parent;
				if (f == null) {
					return;
				}
				f.Text = t;
			}
		}

		#endregion

		public MainForm() {
			InitializeComponent();
			this.OnFirstLoad(OnLoad);
		}

		protected override void WndProc(ref Message m) {
			base.WndProc(ref m);
			if (m.Msg == FormHelper.ProcMsg) {
				OpenFiles(FormHelper.GetCopyDataContent(ref m).Split('\n'));
				Activate();
			}
		}

		protected override void OnFormClosing(FormClosingEventArgs e) {
			base.OnFormClosing(e);
			int n = 0;
			foreach (var item in _FunctionContainer.GetPrimaryControlsInTabs<IDocumentEditor>()) {
				if (item.IsDirty) {
					n++;
				}
			}
			if (n != 0
				&& this.ConfirmYesBox(Messages.ConfirmCloseNDirtyDocument.Replace("<N>", n.ToText())) == false) {
				e.Cancel = true;
			}
			else if (FullScreen) {
				FullScreen = false;
				e.Cancel = true;
			}
		}

		void OnLoad() {
			Processor.PdfHelper.ToggleReaderDebugMode(true); // 打开容错模式
			Processor.PdfHelper.ToggleUnethicalMode(true); // 打开强制读取加密文档模式

			bool firstLoad;
			try {
				firstLoad = AppContext.Load(null) == false;
			}
			catch (Exception) {
				// ignore loading exception
				firstLoad = true;
			}
			_MainMenu.ScaleIcons(16);
			Text = Constants.AppName + " [" + Application.ProductVersion + "]";
			MinimumSize = Size;
			AllowDrop = true;
			DragEnter += (s, args) => args.FeedbackDragFileOver(Constants.FileExtensions.Pdf);
			DragDrop += (s, args) => OpenFiles(args.DropFileOver(Constants.FileExtensions.Pdf));
			FormClosed += MainForm_FormClosed;

			AppContext.WindowStatus.Position(this);
			SetupCustomizeToolbar();
			if (_GeneralToolbar.Visible = AppContext.Toolbar.ShowGeneralToolbar) {
				ScaleToolbar();
			}

			_OpenPdfDialog.DefaultExt = Constants.FileExtensions.Pdf;
			_OpenPdfDialog.Filter = Constants.FileExtensions.PdfOrXmlFilter;

			_LogControl = new ReportControl {
				Location = _FunctionContainer.Location,
				Size = _FunctionContainer.Size,
				Anchor = AnchorStyles.Top | AnchorStyles.Left | AnchorStyles.Bottom | AnchorStyles.Right
			};
			Controls.Add(_LogControl);
			HideLogControl();
			_LogControl.VisibleChanged += (s, args) => _FunctionContainer.Visible = !_LogControl.Visible;
			_OpenConfigDialog.FileName = _SaveConfigDialog.FileName = Constants.AppName + "配置文件" + Constants.FileExtensions.Json;
			_OpenConfigDialog.Filter = _SaveConfigDialog.Filter = Constants.FileExtensions.JsonFilter;
			_FunctionContainer.Font = new Font(SystemFonts.CaptionFont.FontFamily.Name, 9);
			_FunctionContainer.ImageList = new ImageList();
			_FunctionContainer.DisplayStyleProvider.SelectedTextStyle = FontStyle.Bold;
			_FunctionContainer.AllowDrop = true;
			_FunctionContainer.MiddleClickCloseTab = true;
			_FunctionContainer.DoubleClickCloseTab = true;
			_FunctionContainer.TabClosing += (s, args) => {
				if (_FunctionContainer.SafeCloseTab(args.TabPage)) {
					_MainStatusLabel.Text = String.Empty;
				}
				args.Cancel = true;
			};
			_FunctionContainer.Selected += SelectedFunctionChanged;
			_FunctionContainer.Deselected += FunctionDeselected;

			// 关闭启动屏幕窗口
			using (var closeSplashEvent = new System.Threading.EventWaitHandle(false,
				System.Threading.EventResetMode.ManualReset, "CloseSplashScreenEvent" + Constants.AppEngName)) {
				closeSplashEvent.Set();
			}

			SelectFunctionList(Function.FrontPage);

			_GeneralToolbar.ItemClicked += MenuCommand;
			if (AppContext.CheckUpdateDate < DateTime.Today) {
				CheckUpdate();
				if (AppContext.CheckUpdateInterval != Int32.MaxValue) {
					AppContext.CheckUpdateDate = DateTime.Today + TimeSpan.FromDays(AppContext.CheckUpdateInterval);
				}
			}
			var ca = Environment.GetCommandLineArgs();
			if (ca.HasContent()) {
				OpenFiles(ca);
			}
#if DEBUG
			iTextSharp.text.io.StreamUtil.AddToResourceSearch("iTextAsian.dll");
#endif
		}

		void OpenFiles(string[] files) {
			foreach (var item in files) {
				var p = new FilePath(item);
				if (p.ExistsFile && p.HasExtension(Constants.FileExtensions.Pdf)) {
					OpenFileWithEditor(p.ToFullPath());
				}
			}
		}

		void CheckUpdate() {
			var client = new System.Net.WebClient();
			client.DownloadDataCompleted += (s, args) => {
				if (args.Error != null) {
					goto Exit;
				}
				try {
					var x = new System.Xml.XmlDocument();
					x.Load(new System.IO.MemoryStream(args.Result));
					var r = x.DocumentElement;
					var u = r.GetAttribute("url");
					if (String.IsNullOrEmpty(u) == false
						&& new Version(ProductVersion) < new Version(r.GetAttribute("version"))
						&& this.ConfirmOKBox("发现新版本，是否前往下载？")) {
						this.ShowDialog<UpdateForm>();
					}
				}
				catch (Exception) {
					FormHelper.ErrorBox("版本信息文件格式错误，请稍候重试。");
				}
			Exit:
				client.Dispose();
				client = null;
			};
			client.DownloadDataAsync(new Uri(Constants.AppUpdateFile));
		}

		void MenuCommand(object sender, ToolStripItemClickedEventArgs e) {
			var ci = e.ClickedItem;
			var t = ci.Tag as string;
			if (String.IsNullOrEmpty(t) == false) {
				SelectFunctionList((Function)Enum.Parse(typeof(Function), t));
				return;
			}
			ci.HidePopupMenu();
			if (ci.OwnerItem == _RecentFiles) {
				var f = GetActiveFunctionControl() as FunctionControl;
				f.RecentFileItemClicked?.Invoke(_MainMenu, e);
			}
			else {
				ExecuteCommand(ci.Name);
			}
		}

		internal void ExecuteCommand(string commandName) {
			if (commandName == Commands.ResetOptions) {
				if (GetActiveFunctionControl() is IResettableControl f
					&& FormHelper.YesNoBox("是否将当前功能恢复为默认设置？") == DialogResult.Yes) {
					f.Reset();
				}
			}
			else if (commandName == Commands.RestoreOptions && _OpenConfigDialog.ShowDialog() == DialogResult.OK) {
				if (AppContext.Load(_OpenConfigDialog.FileName) == false) {
					FormHelper.ErrorBox("无法加载指定的配置文件。");
					return;
				}
				foreach (Control item in __FunctionControls.Values) {
					(item as IResettableControl)?.Reload();
				}
				SetupCustomizeToolbar();
			}
			else if (commandName == Commands.SaveOptions && _SaveConfigDialog.ShowDialog() == DialogResult.OK) {
				AppContext.Save(_SaveConfigDialog.FileName, false, false);
			}
			else if (commandName == Commands.LogWindow) {
				ShowLogControl();
			}
			else if (commandName == Commands.CreateShortcut) {
				ShellHelper.CreateShortcut();
			}
			else if (commandName == Commands.VisitHomePage) {
				ShellHelper.VisitHomePage();
			}
			else if (commandName == Commands.CheckUpdate) {
				this.ShowDialog<UpdateForm>();
			}
			else if (commandName == Commands.Close) {
				if (_FunctionContainer.SelectedTab.Tag.CastOrDefault<Function>() == Function.FrontPage) {
					return;
				}
				_FunctionContainer.SelectedTab.Dispose();
			}
			else if (commandName == Commands.CustomizeToolbar || commandName == "_CustomizeToolbarCommand") {
				this.ShowDialog<CustomizeToolbarForm>();
				SetupCustomizeToolbar();
			}
			else if (commandName == Commands.ShowGeneralToolbar) {
				_FunctionContainer.SuspendLayout();
				if (_GeneralToolbar.Visible = AppContext.Toolbar.ShowGeneralToolbar = !AppContext.Toolbar.ShowGeneralToolbar) {
					ScaleToolbar();
				}
				_FunctionContainer.PerformLayout();
			}
			else if (commandName == Commands.Exit) {
				Close();
			}
			else if (commandName == Commands.FullScreenDisplay) {
				FullScreen = !FullScreen;
			}
			else if (GetActiveFunctionControl() is FunctionControl f) {
				if (commandName == Commands.Action && f.DefaultButton != null) {
					f.DefaultButton.PerformClick();
				}
				else {
					f.ExecuteCommand(commandName);
				}
			}
		}

		void SetupCustomizeToolbar() {
			AppContext.Toolbar.RemoveInvalidButtons();
			for (int i = _GeneralToolbar.Items.Count - 1; i > 0; i--) {
				_GeneralToolbar.Items[i].Dispose();
			}
			foreach (var item in AppContext.Toolbar.Buttons) {
				if (item.Visible == false) {
					continue;
				}
				_GeneralToolbar.Items.Add(item.CreateButton());
			}
		}

		void ScaleToolbar() {
			_GeneralToolbar.ScaleIcons(16);
		}

		Control GetActiveFunctionControl() {
			return _FunctionContainer.FirstControlInActiveTab;
		}

		internal void OpenFileWithEditor(FilePath path) {
			path = path.ToFullPath();
			EditorControl c;
			foreach (var editor in GetOpenedDocumentEditors()) {
				if ((c = editor as EditorControl) != null && path.Equals(c.DocumentPath)) {
					_FunctionContainer.SelectedTab = (TabPage)c.Parent;
					_FunctionContainer.Invalidate();
					return;
				}
			}
			c = SelectFunctionList(Function.Editor) as EditorControl;
			if (String.IsNullOrEmpty(path)) {
				c.ExecuteCommand(Commands.Open);
			}
			else {
				c.ExecuteCommand(Commands.OpenFile, path);
			}
		}

		IEnumerable<IDocumentEditor> GetOpenedDocumentEditors() {
			foreach (TabPage item in _FunctionContainer.TabPages) {
				if (item.Controls.Count != 0 && item.Controls[0] is IDocumentEditor editor) {
					yield return editor;
				}
			}
		}

		internal FunctionControl SelectFunctionList(Function func) {
			if (func == Function.PatcherOptions) {
				this.ShowDialog<PatcherOptionForm>();
			}
			else if (func == Function.MergerOptions) {
				this.ShowDialog<MergerOptionForm>();
			}
			else if (func == Function.InfoFileOptions) {
				this.ShowDialog<InfoFileOptionControl>();
			}
			else if (func == Function.Options) {
				this.ShowDialog<AppOptionForm>();
			}
			else {
				HideLogControl();
				var p = (GetActiveFunctionControl() as IDocumentSource)?.DocumentPath;
				var c = GetFunctionControl(func);
				foreach (TabPage item in _FunctionContainer.TabPages) {
					if (item.Controls.Count > 0 && item.Controls[0] == c) {
						_FunctionContainer.SelectedTab = item;
						if (String.IsNullOrEmpty(p) == false
							&& c is IDocumentSource tc
							&& String.IsNullOrEmpty(tc.DocumentPath)) {
							c.ExecuteCommand(Commands.OpenFile, p);
						}
						return c;
					}
				}
				var t = new TabPage(c.FunctionName) {
					Font = SystemFonts.SmallCaptionFont
				};
				var im = _FunctionContainer.ImageList.Images;
				for (int i = im.Count - 1; i >= 0; i--) {
					if (im[i] == c.IconImage) {
						t.ImageIndex = i;
						break;
					}
				}
				if (t.ImageIndex < 0) {
					im.Add(c.IconImage);
					t.ImageIndex = im.Count - 1;
				}
				t.Tag = func;
				_FunctionContainer.TabPages.Add(t);
				c.Size = t.ClientSize;
				c.Dock = DockStyle.Fill;
				t.Controls.Add(c);
				_FunctionContainer.SelectedTab = t;
				if (func != Function.Editor && String.IsNullOrEmpty(p) == false) {
					c.ExecuteCommand(Commands.OpenFile, p);
				}
				AcceptButton = c.DefaultButton;
				return c;
			}
			return null;
		}

		void FunctionDeselected(object sender, TabControlEventArgs args) {
			if (GetActiveFunctionControl() is FunctionControl c) {
				c.OnDeselected();
			}
		}

		void SelectedFunctionChanged(object sender, TabControlEventArgs args) {
			if (GetActiveFunctionControl() is FunctionControl c) {
				c.OnSelected();
				_MainStatusLabel.Text = c is IDocumentEditor b ? b.DocumentPath : Messages.Welcome;
				AcceptButton = c.DefaultButton;
			}
		}

		internal string ShowPdfFileDialog() {
			return _OpenPdfDialog.ShowDialog() == DialogResult.OK ? _OpenPdfDialog.FileName : null;
		}

		void MainForm_FormClosed(object sender, FormClosedEventArgs e) {
			AppContext.Save(null, true, true);
		}

		void HideLogControl() {
			_LogControl.Hide();
		}
		void ShowLogControl() {
			_LogControl.Show();
		}

		void MenuOpening(object sender, EventArgs e) {
			if (GetActiveFunctionControl() is FunctionControl f) {
				f.SetupMenu(sender as ToolStripMenuItem);
			}
		}

		void RecentFileMenuOpening(object sender, EventArgs e) {
			if (GetActiveFunctionControl() is FunctionControl f && f.ListRecentFiles != null) {
				f.ListRecentFiles(sender, e);
			}
		}
	}
}