PDFOffice_Windows_exe/PDF Office/ViewModels/Dialog/ToolsDialogs/MergeDialogViewModel.cs

using ComPDFKit.PDFDocument;
using ComPDFKit.PDFPage;
using Microsoft.Win32;
using PDF_Office.Model;
using PDF_Office.Model.Dialog.ToolsDialogs;
using PDF_Office.Views.Dialog;
using Prism.Commands;
using Prism.Mvvm;
using Prism.Services.Dialogs;
using System;
using System.Collections.Generic;
using System.Collections.ObjectModel;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Windows.Controls;

namespace PDF_Office.ViewModels.Dialog.ToolsDialogs
{
    class MergeDialogViewModel : BindableBase, IDialogAware
    {
        enum PageSizeType
        { 
            kDefault=0,
            A4 = 1,
            A3 = 2,
            Letter = 3,
            Legal = 4,
            Customized = 5,
        }

        #region 框架内容

        public string Title => "";

        public event Action<IDialogResult> RequestClose;

        public bool CanCloseDialog()
        {
            return true;
        }

        public void OnDialogClosed()
        {

        }

        public void OnDialogOpened(IDialogParameters parameters)
        {

        }

        #endregion
        #region 定义与初始化

        public ObservableCollection<MergeObject> MergeObjectlist { get; set; }

        public DelegateCommand CancelCommand { get; set; }
        public DelegateCommand MergeCommand { get; set; }
        public DelegateCommand AddFilesCommand { get; set; }
        public DelegateCommand ClearCommand { get; set; }
        public DelegateCommand<object> SetPageSizeTypeCommand { get; set; }

        public IDialogService dialogs;

        private PageSizeType pageSizeType = PageSizeType.kDefault;


        public MergeDialogViewModel(IDialogService dialogService)
        {
            dialogs = dialogService;
            MergeObjectlist = new ObservableCollection<MergeObject>();
            CancelCommand = new DelegateCommand(Cancel);
            AddFilesCommand = new DelegateCommand(AddFiles);
            MergeCommand = new DelegateCommand(Merge);
            ClearCommand = new DelegateCommand(Clear);
            SetPageSizeTypeCommand = new DelegateCommand<object>(SetPageSizeType);
        }

        #endregion

        #region 私有方法
        private void Cancel()
        {
            RequestClose.Invoke(new DialogResult(ButtonResult.Cancel));
        }

        private void AddFiles()
        {
            OpenFileDialog openFile = new OpenFileDialog();
            openFile.Multiselect = true;
            openFile.Filter = "Files(*pdf;*.bmp;*.gif;*.jpeg;*.jpg;*.png;*.tiff)|*.pdf;*.bmp;*.gif;*.jpeg;*.jpg;*.png;*.tiff|" +
                "(*.pdf)|*.pdf|" +
                "(*.bmp)|*.bmp|" +
                "(*.gif)|*.gif|" +
                "(*.jpeg)|*.jpeg|" +
                "(*.jpg)|*.jpg|" +
                "(*.png)|*.png|" +
                "(*.tiff)|*.tiff";
            if (openFile.ShowDialog() == false)
            {
                return;
            }
            bool result = true;
            for (int i = 0; i < openFile.FileNames.Length; i++)
            {
                MergeObject mergeObject = new MergeObject();
                mergeObject.FilePath = openFile.FileNames[i];
                CPDFDocument doc = CPDFDocument.InitWithFilePath(mergeObject.FilePath);

                if (doc.IsLocked)
                {
                    DialogParameters value = new DialogParameters();
                    value.Add(ParameterNames.PDFDocument, doc);
                    dialogs.ShowDialog(DialogNames.VerifyPassWordDialog, value, e =>
                    {

                        if (e.Result != ButtonResult.OK)
                        {
                            result = false;
                        }
                        else
                        {
                            mergeObject.Password = e.Parameters.GetValue<string>(ParameterNames.PassWord);
                        }

                    });
                    if (!result)
                    {
                        doc.Release();
                        continue;
                    }
                }
                mergeObject.DocName = doc.FileName;
                mergeObject.DocPageCount = doc.PageCount.ToString() + "页";
                mergeObject.DocSize = GetFileSize(mergeObject.FilePath);
                MergeObjectlist.Add(mergeObject);
                doc.Release();
            }
        }

        private string GetFileSize(string path)
        {
            System.IO.FileInfo fileInfo = null;
            try
            {
                fileInfo = new System.IO.FileInfo(path);
            }
            catch
            {
                return "0KB";
            }
            if (fileInfo != null && fileInfo.Exists)
            {

                var size = Math.Round(fileInfo.Length / 1024.0, 0);
                if (size > 1024)
                {
                    var sizeDouble = Math.Round(size / 1024.0, 2);
                    if (sizeDouble > 1024)
                    {
                        sizeDouble = Math.Round(sizeDouble / 1024.0, 2);
                        return sizeDouble + "G";
                    }
                    else
                        return sizeDouble + "M";
                }
                else
                {

                    return (int)System.Math.Ceiling(size) + "KB";
                }

            }
            else
            {
                return "0KB";
            }
        }

        private void Merge()
        {
            bool result = true;
            CPDFDocument SaveDoc = CPDFDocument.CreateDocument();
            for (int i = 0; i < MergeObjectlist.Count; i++)
            {
                CPDFDocument tempDoc = CPDFDocument.InitWithFilePath(MergeObjectlist[i].FilePath);
                if (!string.IsNullOrEmpty(MergeObjectlist[i].Password))
                {
                    tempDoc.UnlockWithPassword(MergeObjectlist[i].Password);
                }
                result = SaveDoc.ImportPages(tempDoc, "1-1"/*MergeObjectlist[i].SetPageRange*/);
                if (!result)
                {
                    SaveDoc.Release();
                    tempDoc.Release();
                    return;
                }
                tempDoc.Release();
            }

            string path = App.CachePath.MergeFilePath;
            string name = Guid.NewGuid().ToString();
            if (!string.IsNullOrEmpty(path))
            {
                path = System.IO.Path.Combine(path, name);
            }
            else
            {
                return;
            }
            System.Windows.Rect rect = new System.Windows.Rect();

            switch (pageSizeType)
            {
                case PageSizeType.kDefault:
                    break;
                case PageSizeType.A4:
                    break;
                case PageSizeType.A3:
                    break;
                case PageSizeType.Letter:
                    break;
                case PageSizeType.Legal:
                    break;
                case PageSizeType.Customized:
                    break;
                default:
                    break;
            }

            if (rect.Width>0&& rect.Height>0)
            {
                //裁剪
                for (int i = 0; i < SaveDoc.PageCount; i++)
                {
                    CPDFPage page = SaveDoc.PageAtIndex(i);
                    page.CropPage(CPDFDisplayBox.MediaBox, rect);
                }
            }

            bool saveResult = SaveDoc.WriteToFilePath(path);
            SaveDoc.Release();

            DialogParameters valuePairs = new DialogParameters();
            valuePairs.Add(ParameterNames.FilePath, path);
            RequestClose.Invoke(new DialogResult(ButtonResult.OK, valuePairs));
        }

        private void Clear()
        {
            MergeObjectlist.Clear();
        }

        private void SetPageSizeType(object button)
        {
            if (button is RadioButton)
            {
                pageSizeType= (PageSizeType)Convert.ToInt32((button as RadioButton).Tag);
            }
        }

        #endregion

        #region 公开方法

        public void DeleteItem(MergeObject merge)
        {
            MergeObjectlist.Remove(merge);
        }

        public void MoveMerge(MergeObject targetNode, MergeObject soureNode)
        {
            int targetindex = MergeObjectlist.IndexOf(targetNode);
            MergeObjectlist.Remove(soureNode);
            if (targetNode.IsForward)
            {
                if (targetindex+1<MergeObjectlist.Count)
                {
                    MergeObjectlist.Insert(targetindex + 1, soureNode);
                }
                else
                {
                    MergeObjectlist.Add(soureNode);
                }
            }
            else
            {
                MergeObjectlist.Insert(targetindex, soureNode);
            }
        }

        #endregion
    }
}