PDFOffice_Windows_exe/PDF Office/ViewModels/Dialog/ToolsDialogs/SaftyDialogs/SetPasswordDialogViewModel.cs

using ComPDFKit.PDFDocument;
using PDF_Office.CustomControl;
using PDF_Office.Model;
using PDF_Office.Model.Dialog.ToolsDialogs.SaftyDialogs;
using Prism.Commands;
using Prism.Mvvm;
using Prism.Services.Dialogs;
using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Runtime.CompilerServices;
using System.Windows;
using System.Windows.Forms;
using System.Windows.Controls;
using CheckBox = System.Windows.Controls.CheckBox;
using System.Drawing.Printing;
using static PDF_Office.Model.Dialog.ToolsDialogs.SaftyDialogs.SetPasswordDialogModel;

namespace PDF_Office.ViewModels.Dialog.ToolsDialogs.SaftyDialogs
{
    public class SetPasswordDialogViewModel : BindableBase, IDialogAware
    {
        #region 参数和属性
        private SetPasswordDialogModel setPasswordDialogModel = new SetPasswordDialogModel();
        private CPDFDocument document;
        private bool _canOpen;
        public bool CanOpen
        {
            get { return _canOpen; }
            set
            {
                _canOpen = value;
                RaisePropertyChanged("isChecked");
            }
        }

        private bool _canEdit;
        public bool CanEdit
        {
            get { return _canEdit; }
            set
            {
                _canEdit = value;
                RaisePropertyChanged("isChecked");
            }
        }

        private EnumPermissionsMod enumPermissionsMod = EnumPermissionsMod.StatusUnknown|EnumPermissionsMod.StatusRestrictPrinting|EnumPermissionsMod.StatusRestrictCopying;

        private string _changeMod = "0";
        ///<value>
        ///"0"为ChangeMod.None;
        ///"1"为ChangeMod.ChangePage;
        ///"2"为ChangeMod.FormAndSignature;
        ///"3"为ChangeMod.AnnotAndFormAndSignature;
        ///"4"为ChangeMod.ExceptAbstrat;
        ///</value>
        public string ChangeMod
        {
            get { return _changeMod; }
            set
            {
                _changeMod = value;
            }
        }

        private string _printMod = "0";
        ///<value>
        ///"0"为PrintMod.None;
        ///"1"为PrintMod.LowDpi;
        ///"2"为PrintMod.HighDpi;
        ///</value>
        public string PrintMod
        {
            get { return _printMod; }
            set
            {
                _printMod = value;
            }
        }

        private string _isEnableConfirm = "False";
        public string IsEnabledConfirm
        {
            get { return _isEnableConfirm; }
            set
            {
                SetProperty(ref _isEnableConfirm, value);
            }
        }

        private bool _enableConfirm = false;
        public bool EnableConfirm
        {
            get { return _enableConfirm; }
            set { SetProperty(ref _enableConfirm, value); }
        }

        private string _openPassword = "";
        public string OpenPassword
        {
            get { return _openPassword; }
            set
            {
                SetProperty(ref _openPassword, value);
                CheckEnableConfirmEncrypt();
            }
        }

        private string _permissionsPassword = "";
        public string PermissionsPassword
        {
            get { return _permissionsPassword; }
            set
            {
                SetProperty(ref _permissionsPassword, value);
                CheckEnableConfirmEncrypt();
            }
        }

        private string _openPasswordMsg = "Open Password";
        public string OpenPasswordMsg
        {
            get { return _openPasswordMsg; }
            set
            {
                SetProperty(ref _openPasswordMsg, value);
            }
        }
        private string _permissionsPasswordMsg = "Permission Password";
        public string PermissionsPasswordMsg
        {
            get { return _permissionsPasswordMsg; }
            set
            {
                SetProperty(ref _permissionsPasswordMsg, value);
            }
        }

        private Visibility _openTextVisibility = Visibility.Collapsed;
        public Visibility OpenTextVisibility
        {
            get { return _openTextVisibility; }
            set
            {
                SetProperty(ref _openTextVisibility, value);
            }
        }

        private Visibility _openPasswordVisibility = Visibility.Visible;
        public Visibility OpenPasswordVisibility
        {
            get { return _openPasswordVisibility; }
            set
            {
                if (value == Visibility.Visible)
                {
                    OpenTextVisibility = Visibility.Collapsed;
                }
                else
                {
                    OpenTextVisibility = Visibility.Visible;
                }
                SetProperty(ref _openPasswordVisibility, value);
            }
        }

        private Visibility _permissionsPasswordVisibility = Visibility.Visible;
        public Visibility PermissionsPasswordVisibility
        {
            get { return _permissionsPasswordVisibility; }
            set
            {
                if (value == Visibility.Visible)
                {
                    PermissionsTextVisibility = Visibility.Collapsed;
                }
                else
                {
                    PermissionsTextVisibility = Visibility.Visible;
                }
                SetProperty(ref _permissionsPasswordVisibility, value);
            }
        }

        private Visibility _permissionsTextVisibility = Visibility.Collapsed;
        public Visibility PermissionsTextVisibility
        {
            get { return _permissionsTextVisibility; }
            set { SetProperty(ref _permissionsTextVisibility, value); }
        }

        private bool? _isRestrictPrinting;
        public bool? IsRestrictPrinting
        {
            get { return _isRestrictPrinting; }
            set { SetProperty(ref _isRestrictPrinting, value); }
        }

        private bool? _isRestrictCopying;
        public bool? IsRestrictCopying
        {
            get { return _isRestrictCopying; }
            set { SetProperty(ref _isRestrictCopying, value); }
        }

        private bool _needOpenPassword;
        public bool NeedOpenPassword
        {
            get { return _needOpenPassword; }
            set { SetProperty(ref _needOpenPassword, value); }
        }

        private bool _needPermissionsPassword;
        public bool NeedPermissionsPassword
        {
            get { return _needPermissionsPassword; }
            set { SetProperty(ref _needPermissionsPassword, value); }
        }

        private bool _isOpenPasswordDisplayed;
        public bool IsOpenPasswordDisplayed
        {
            get { return _isOpenPasswordDisplayed; }
            set { SetProperty(ref _isOpenPasswordDisplayed, value); }
        }

        private bool _isPermissionsPasswordDisplayed;
        public bool IsPermissionsPasswordDisplayed
        {
            get { return _isPermissionsPasswordDisplayed; }
            set { SetProperty(ref _isPermissionsPasswordDisplayed, value); }
        }


        #endregion

        #region 委托声明
        public DelegateCommand<object> DisplayPasswordCommand { get; set; }
        public DelegateCommand<object> SetOpenPasswordCommand { get; set; }
        public DelegateCommand<object> SetPermissionsPasswordCommand { get; set; }
        public DelegateCommand<object> SetRestrictCommand { get; set; }
        public DelegateCommand EncryptCommand { get; set; }


        public DelegateCommand DelegateSetOpenCommand { get; set; }
        public DelegateCommand DelegateSetEditCommand { get; set; }
        public DelegateCommand DelegateCanOpenTextChangedCommand { get; set; }
        public DelegateCommand DelegateCanEditTextChangedCommand { get; set; }
        public DelegateCommand DelegateConfirmEncryptCommand { get; set; }
        public DelegateCommand CancelEncryptCommand { get; set; }
        #endregion

        public SetPasswordDialogViewModel()
        {
            SetPasswordDialogModel.PasswordForOpen = "";
            SetPasswordDialogModel.PasswordForEdit = "";
            DisplayPasswordCommand = new DelegateCommand<object>(DisplayPassword);
            SetOpenPasswordCommand = new DelegateCommand<object>(SetOpenPassword);
            SetPermissionsPasswordCommand = new DelegateCommand<object>(SetPermissionsPassword);
            SetRestrictCommand = new DelegateCommand<object>(SetRestrict);
            EncryptCommand = new DelegateCommand(Encrypt);
            CancelEncryptCommand = new DelegateCommand(CancelEncrypt);

            DelegateConfirmEncryptCommand = new DelegateCommand(ConfirmEncrypt);
            DelegateSetOpenCommand = new DelegateCommand(SetCanOpen);
            DelegateSetEditCommand = new DelegateCommand(SetCanEdit);
            DelegateCanOpenTextChangedCommand = new DelegateCommand(CanOpenTextChanged);
            DelegateCanEditTextChangedCommand = new DelegateCommand(CanEditTextChanged);
        }

        #region 检查和初始化
        private void CheckCanConfirmEncrypt()
        {
            if (setPasswordDialogModel.CanOpen)
            {
                if (SetPasswordDialogModel.PasswordForOpen.Length > 0)
                {
                    IsEnabledConfirm = "True";
                    return;
                }
            }
            if (setPasswordDialogModel.CanEdit)
            {
                if (SetPasswordDialogModel.PasswordForEdit.Length > 0)
                {
                    IsEnabledConfirm = "True";
                    return;
                }
            }
            IsEnabledConfirm = "False";
        }

        private void CheckEnableConfirmEncrypt()
        {
            EnableConfirm = true;
            if (!(NeedOpenPassword || NeedPermissionsPassword))
            {
                EnableConfirm = false;
            }
            if (NeedOpenPassword && string.IsNullOrEmpty(OpenPassword))
            {
                EnableConfirm = false;
            }
            if (NeedPermissionsPassword && string.IsNullOrEmpty(PermissionsPassword))
            {
                EnableConfirm = false;
            }
            if (NeedPermissionsPassword && (!((bool)IsRestrictCopying || (bool)IsRestrictPrinting)))
            {
                EnableConfirm = false;
            }
        }

        private void InitPermissionsDictionary(ref Dictionary<string, SetPasswordDialogModel.PrintMod> GetPrintMod, ref Dictionary<string, SetPasswordDialogModel.ChangeMod> GetChangeMod)
        {
            GetPrintMod.Add("0", SetPasswordDialogModel.PrintMod.None);
            GetPrintMod.Add("1", SetPasswordDialogModel.PrintMod.LowDpi);
            GetPrintMod.Add("2", SetPasswordDialogModel.PrintMod.HighDpi);

            GetChangeMod.Add("0", SetPasswordDialogModel.ChangeMod.None);
            GetChangeMod.Add("1", SetPasswordDialogModel.ChangeMod.ChangePage);
            GetChangeMod.Add("2", SetPasswordDialogModel.ChangeMod.FormAndSignature);
            GetChangeMod.Add("3", SetPasswordDialogModel.ChangeMod.AnnotAndFormAndSignature);
            GetChangeMod.Add("4", SetPasswordDialogModel.ChangeMod.ExceptAbstrat);
        }
        #endregion

        #region 逻辑函数
        public void SetCanOpen()
        {
            setPasswordDialogModel.CanOpen = CanOpen;
            CheckCanConfirmEncrypt();
        }

        public void SetCanEdit()
        {
            setPasswordDialogModel.CanEdit = CanEdit;
            CheckCanConfirmEncrypt();
        }

        public void CanOpenTextChanged()
        {
            CheckCanConfirmEncrypt();
        }

        public void CanEditTextChanged()
        {
            CheckCanConfirmEncrypt();
        }

        public void DisplayPassword(object e)
        {
            var chk = e as CheckBox;
            if (chk != null)
            {
                if (chk.Name == "DisplayOpenPasswordChk")
                {
                    if (chk.IsChecked == true)
                    {
                        OpenPasswordVisibility = Visibility.Collapsed;
                    }
                    else
                    {
                        OpenPasswordVisibility = Visibility.Visible;
                    }
                }
                else
                {
                    if (chk.IsChecked == true)
                    {
                        PermissionsPasswordVisibility = Visibility.Collapsed;
                    }
                    else
                    {
                        PermissionsPasswordVisibility = Visibility.Visible;
                    }
                }
            }
        }

        public void SetOpenPassword(object e)
        {
            var chk = e as CheckBox;
            if (chk != null)
            {
                IsOpenPasswordDisplayed = false;
                OpenPasswordVisibility = Visibility.Visible;
            }
            CheckEnableConfirmEncrypt();
        }

        public void SetPermissionsPassword(object e)
        {
            var chk = e as CheckBox;
            if (chk != null)
            {
                IsPermissionsPasswordDisplayed = false;
                PermissionsPasswordVisibility = Visibility.Visible;
                if (chk.IsChecked == true)
                {
                    IsRestrictPrinting = true;
                    IsRestrictCopying = true;
                }
                else
                {
                    IsRestrictPrinting = false;
                    IsRestrictCopying = false;
                }
            }
            CheckEnableConfirmEncrypt();
        }

        public void SetRestrict(object e)
        {
            var chk = e as CheckBox;
            if (chk != null)
            {

                if ((bool)chk.IsChecked)
                {
                    if (chk.Tag.ToString() == "Printing")
                    {
                        enumPermissionsMod |= EnumPermissionsMod.StatusRestrictPrinting;
                    }
                    else if (chk.Tag.ToString() == "Copying")
                    {
                        enumPermissionsMod |= EnumPermissionsMod.StatusRestrictCopying;
                    }
                }
                else
                {
                    if (chk.Tag.ToString() == "Printing")
                    {
                        enumPermissionsMod -= EnumPermissionsMod.StatusRestrictPrinting;
                    }
                    else if (chk.Tag.ToString() == "Copying")
                    {
                        enumPermissionsMod -= EnumPermissionsMod.StatusRestrictCopying;
                    }
                }
            }
            CheckEnableConfirmEncrypt();
        }

        public void ConfirmEncrypt()
        {
            string openPassword = "";
            string editPassword = "";
            FolderBrowserDialog folderDialog = new FolderBrowserDialog();
            System.Windows.Forms.SaveFileDialog sfd = new System.Windows.Forms.SaveFileDialog();
            /*
             *设置这个对话框的起始保存路径
             */
            sfd.InitialDirectory = document.FilePath;
            /*
             *设置保存的文件的类型，注意过滤器的语法 例子：“文件类型|*.后缀名；*.后缀名；”
             */
            sfd.Filter = "PDF|*.pdf;";
            /*
             *调用ShowDialog()方法显示该对话框，该方法的返回值代表用户是否点击了确定按钮
             **/
            sfd.FileName = document.FileName + "_SetPassword.pdf";
            if (sfd.ShowDialog() == System.Windows.Forms.DialogResult.OK)
            {
                /*
                 * 做一些工作
                 */
                CPDFPermissionsInfo permissionsInfo = new CPDFPermissionsInfo();
                if (setPasswordDialogModel.CanOpen)
                {
                    if (!string.IsNullOrEmpty(SetPasswordDialogModel.PasswordForOpen))
                    {
                        openPassword = SetPasswordDialogModel.PasswordForOpen;
                    }
                }
                if (setPasswordDialogModel.CanEdit)
                {
                    if (!string.IsNullOrEmpty(SetPasswordDialogModel.PasswordForEdit))
                    {
                        editPassword = SetPasswordDialogModel.PasswordForEdit;
                        Dictionary<string, SetPasswordDialogModel.PrintMod> GetPrintMod = new Dictionary<string, SetPasswordDialogModel.PrintMod>();
                        Dictionary<string, SetPasswordDialogModel.ChangeMod> GetChangeMod = new Dictionary<string, SetPasswordDialogModel.ChangeMod>(); ;
                        InitPermissionsDictionary(ref GetPrintMod, ref GetChangeMod);
                        permissionsInfo = setPasswordDialogModel.CreatePermissionsInfo(GetPrintMod[PrintMod], GetChangeMod[ChangeMod]);
                    }
                }
                document.Encrypt(openPassword, editPassword, permissionsInfo);

                document.WriteToFilePath(sfd.FileName);
                MessageBoxEx.Show(sfd.FileName + " 保存成功");
                RequestClose?.Invoke(new Prism.Services.Dialogs.DialogResult(ButtonResult.Cancel));
            }
        }

        public void Encrypt()
        {
            string openPassword = "";
            string permissionsPassword = "";
            FolderBrowserDialog folderDialog = new FolderBrowserDialog();
            System.Windows.Forms.SaveFileDialog sfd = new System.Windows.Forms.SaveFileDialog();
            /*
             *设置这个对话框的起始保存路径
             */
            sfd.InitialDirectory = document.FilePath;
            /*
             *设置保存的文件的类型，注意过滤器的语法 例子：“文件类型|*.后缀名；*.后缀名；”
             */
            sfd.Filter = "PDF|*.pdf;";
            /*
             *调用ShowDialog()方法显示该对话框，该方法的返回值代表用户是否点击了确定按钮
             **/
            sfd.FileName = document.FileName + "_SetPassword.pdf";
            if (sfd.ShowDialog() == System.Windows.Forms.DialogResult.OK)
            {
                CPDFPermissionsInfo permissionsInfo = new CPDFPermissionsInfo();
                if (NeedOpenPassword)
                {
                    if (!string.IsNullOrEmpty(OpenPassword))
                    {
                        openPassword= OpenPassword;
                    }
                }
                if (NeedPermissionsPassword){
                    if (!string.IsNullOrEmpty(PermissionsPassword))
                    {
                        permissionsPassword = PermissionsPassword;
                        permissionsInfo = CreatePermissionsInfo(enumPermissionsMod);
                    }
                }
                document.Encrypt(openPassword, permissionsPassword, permissionsInfo);

                document.WriteToFilePath(sfd.FileName);
                MessageBoxEx.Show(sfd.FileName + " 保存成功");
                RequestClose?.Invoke(new Prism.Services.Dialogs.DialogResult(ButtonResult.Cancel));
            }
        }

        public void CancelEncrypt() =>
            RequestClose?.Invoke(new Prism.Services.Dialogs.DialogResult(ButtonResult.Cancel));
        #endregion

        #region 框架相关
        public string Title => "";
        public event Action<IDialogResult> RequestClose;

        public bool CanCloseDialog()
        {
            return true;
        }

        public void OnDialogClosed()
        {
        }

        public void OnDialogOpened(IDialogParameters parameters)
        {
            CPDFDocument doc = null;
            parameters.TryGetValue<CPDFDocument>(ParameterNames.PDFDocument, out doc);
            if (doc != null)
            {
                document = doc;
            }
        }
        #endregion
    }
}