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;
using PDF_Office.Model.Dialog.HomePageToolsDialogs.HomePageBatchProcessing;

namespace PDF_Office.ViewModels.Dialog.ToolsDialogs.SaftyDialogs
{
    public class SetPasswordDialogViewModel : BindableBase, IDialogAware
    {
        #region 参数和属性
        private SetPasswordDialogModel setPasswordDialogModel = new SetPasswordDialogModel();
        private CPDFDocument document;
        public IDialogService dialogs;

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

        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 = false;
        public bool IsOpenPasswordDisplayed
        {
            get { return _isOpenPasswordDisplayed; }
            set { SetProperty(ref _isOpenPasswordDisplayed, value); }
        }

        private bool _isPermissionsPasswordDisplayed = false;
        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 CancelEncryptCommand { get; set; }
        public DelegateCommand BatchSetPasswordCommand { get; set; }

        #endregion

        public SetPasswordDialogViewModel(IDialogService dialogService)
        {
            dialogs = dialogService;
            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);
            BatchSetPasswordCommand = new DelegateCommand(BatchSetPassword);
        }

        #region 检查和初始化

        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;
            }
        }
        #endregion
        #region 逻辑函数

        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();
        }

        private void BatchSetPassword()
        {
            DialogParameters setPasswordpdf = new DialogParameters();
            setPasswordpdf.Add(ParameterNames.BatchProcessing_Name, "2");
            HomePageBatchProcessingDialogModel.FilePaths = new List<string> { document.FilePath.ToString() };
            HomePageBatchProcessingDialogModel.BatchProcessingIndex = 2;
            setPasswordpdf.Add(ParameterNames.FilePath, new string[] { document.FilePath.ToString() });
            dialogs.ShowDialog(DialogNames.HomePageBatchProcessingDialog, setPasswordpdf, e => { });
        }


        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 = null;

                if (NeedOpenPassword)
                {
                    if (!string.IsNullOrEmpty(OpenPassword))
                    {
                        openPassword = OpenPassword;
                    }
                }
                if (NeedPermissionsPassword && (!string.IsNullOrEmpty(PermissionsPassword)))
                {
                    permissionsPassword = PermissionsPassword;
                    permissionsInfo = CreatePermissionsInfo(enumPermissionsMod);
                }
                else
                {
                    permissionsInfo = CreateDefaultPermissionsInfo();
                }

                document.Encrypt(openPassword, permissionsPassword, permissionsInfo);
                document.WriteToFilePath(sfd.FileName);
                MessageBoxEx.Show(sfd.FileName + " 保存成功");
                RequestClose?.Invoke(new Prism.Services.Dialogs.DialogResult(ButtonResult.OK));
            }
        }

        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
    }
}