PDFOffice_Windows_exe/PDF Office/ViewModels/Dialog/HomePageToolsDialogs/HomePageBatchProcessing/HomePageBates/HomePageBatesDialogViewModel.cs

using ComPDFKit.PDFDocument;
using PDF_Master.CustomControl;
using PDF_Master.Model.Dialog.HomePageToolsDialogs.HomePageBatchProcessing;
using PDF_Master.Model;
using Prism.Commands;
using Prism.Mvvm;
using Prism.Regions;
using System;
using System.Collections.Generic;
using System.Data;
using System.Diagnostics;
using System.IO;
using System.Linq;
using PDF_Master.Helper;
using System.Windows;
using System.Windows.Forms;
using ComPDFKitViewer.PdfViewer;
using Prism.Events;
using PDF_Master.EventAggregators;
using PDFSettings;
using ImTools;
using PDF_Master.Model.EditTools.Watermark;
using PDF_Master.Model.EditTools.Bates;
using PDF_Master.Model.EditTools.Background;
using MessageBox = System.Windows.MessageBox;
using Prism.Services.Dialogs;

namespace PDF_Master.ViewModels.Dialog.HomePageToolsDialogs.HomePageBatchProcessing.HomePageBates
{
    public class HomePageBatesDialogViewModel : BindableBase, INavigationAware
    {
        #region 参数和属性
        private List<string> fileNames;


        private CPDFDocument tempDocument;

        public string PageNumber = "";

        private CPDFBates bates;

        public BatesInfo BatesInfo;

        public IEventAggregator eventAggregator;
        public IRegionManager batesRegion;
        public IDialogService dialogs;

        private CPDFViewer PDFViewer;

        private ViewContentViewModel viewContentViewModel;

        public string TemplateListName = "HomePageBatesTemplateListContent";
        public string CreateName = "HomePageBatesCreateContent";

        public string BatesDocumentName = "HomePageBatesDocumentContent";

        public string BatesType = "HomePageBatesType";

        public List<int> fileNamesView = new List<int>();

        private int fileNamesIndex = 0;

        public int FileNameNumber = 0;

        private string _batesSettingsRegionName;
        /// <summary>
        /// 属性设置Region
        /// </summary>
        public string BatesSettingsRegionName
        {
            get { return _batesSettingsRegionName; }
            set { _batesSettingsRegionName = value; }
        }

        private Visibility _batesSettingsVisible = Visibility.Visible;
        /// <summary>
        /// 属性设置Region可见
        /// </summary>
        public Visibility BatesSettingsVisible
        {
            get { return _batesSettingsVisible; }
            set { _batesSettingsVisible = value; }
        }

        private string batesGridIsEnabled = "True";
        public string BatesGridIsEnabled
        {
            get
            {
                return batesGridIsEnabled;
            }
            set
            {
                SetProperty(ref batesGridIsEnabled, value);
            }
        }

        private string setBatesGridIsEnabled = "True";
        public string SetBatesGridIsEnabled
        {
            get
            {
                return setBatesGridIsEnabled;
            }
            set
            {
                SetProperty(ref setBatesGridIsEnabled, value);
            }
        }

        private DataTable pdfDataTable = new DataTable();
        public DataTable PDFDataTable
        {
            get { return pdfDataTable; }
            set
            {
                SetProperty(ref pdfDataTable, value);
            }
        }

        private DataTable pdfCurrentDataTable = new DataTable();
        public DataTable PDFCurrentDataTable
        {
            get { return pdfCurrentDataTable; }
            set
            {
                SetProperty(ref pdfCurrentDataTable, value);
            }
        }

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

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

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

        private Visibility addFileVisibility = Visibility.Hidden;
        public Visibility AddFileVisibility
        {
            get { return addFileVisibility; }
            set
            {
                SetProperty(ref addFileVisibility, value);
                RaisePropertyChanged();

            }
        }

        private string createGridSpan = "1";
        public string CreateGridSpan
        {
            get { return createGridSpan; }
            set
            {
                SetProperty(ref createGridSpan, value);
                RaisePropertyChanged();

            }
        }

        private Visibility createGridSpanVisibility = Visibility.Visible;
        public Visibility CreateGridSpanVisibility
        {
            get { return createGridSpanVisibility; }
            set
            {
                SetProperty(ref createGridSpanVisibility, value);
                RaisePropertyChanged();

            }
        }

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

        #endregion

        #region 委托声明
        public DelegateCommand ADDPDFCommand { get; set; }

        public DelegateCommand ADDOpenedPDFCommand { get; set; }

        public DelegateCommand RemovePDFFileCommand { get; set; }

        public DelegateCommand CreateCommand { get; set; }

        public DelegateCommand ADDPDFFilesCommand { get; set; }


        #endregion

        public string Unicode = null;

        public HomePageBatesDialogViewModel(IRegionManager batesRegion, IEventAggregator eventAggregator,IDialogService dialogs)
        {
            this.eventAggregator = eventAggregator;
            this.batesRegion = batesRegion;
            this.dialogs = dialogs;
            BatesSettingsRegionName = Guid.NewGuid().ToString();
            Unicode = Guid.NewGuid().ToString();
            PDFCurrentDataTable.Columns.Add("FilePageRangeText");
            PDFCurrentDataTable.Columns.Add("FilePageRangeSelectIndex");
            PDFCurrentDataTable.Columns.Add("IsEvenPageIsEnabled");
            PDFCurrentDataTable.Columns.Add("FileState");
            PDFDataTable.Columns.Add("FileName");
            PDFDataTable.Columns.Add("FilePageRangeText");
            PDFDataTable.Columns.Add("FilePageRangeSelectIndex");
            PDFDataTable.Columns.Add("IsEvenPageIsEnabled");
            PDFDataTable.Columns.Add("MaxPageRange");
            PDFDataTable.Columns.Add("FileSize");
            PDFDataTable.Columns.Add("FileState");
            ADDPDFCommand = new DelegateCommand(addpicture);
            ADDOpenedPDFCommand = new DelegateCommand(addOpenedPDFFiles);
            ADDPDFFilesCommand = new DelegateCommand(addpicturefiles);
            RemovePDFFileCommand = new DelegateCommand(removepdffile);
            CreateCommand = new DelegateCommand(create);
            eventAggregator.GetEvent<EditBatesHeaderFooterTemplateItemEvent>().Subscribe(EditBatesTemplateItem, e => e.Unicode == Unicode && e.Type == BatesType);
            eventAggregator.GetEvent<EnterTemplateListOrCreateEvent>().Subscribe(EnterTemplateListOrCreate, e => e.Unicode == Unicode);
        }
        #region 逻辑函数
        public void EditBatesTemplateItem(BatesHeaderFooterItemUnicode batesItemunicode)
        {
            BatesHeaderFooterItem batesItem = batesItemunicode.Status;
            NavigationParameters param = new NavigationParameters();
            param.Add("Unicode", Unicode);
            param.Add(ParameterNames.PDFViewer, PDFViewer);
            param.Add("BatesItem", batesItem);
            batesRegion.RequestNavigate(BatesSettingsRegionName, CreateName, param);
            BatesSettingsVisible = Visibility.Visible;
            CreateGridSpan = "2";
            CreateGridSpanVisibility = Visibility.Collapsed;
        }

        public void EnterTemplateListOrCreate(EnumTemplateListOrCreateUnicode enumTemplateListOrCreateunicode)
        {
            EnumTemplateListOrCreate enumTemplateListOrCreate = enumTemplateListOrCreateunicode.Status;
            if (enumTemplateListOrCreate == EnumTemplateListOrCreate.StatusTemplate)
            {
                CreateGridSpan = "1";
                CreateGridSpanVisibility = Visibility.Visible;
                EnterSelectedContent(TemplateListName);
            }
            else
            {
                CreateGridSpan = "2";
                CreateGridSpanVisibility = Visibility.Collapsed;
                EnterSelectedContent(CreateName);
            }
        }

        public void EnterSelectedContent(string SelectedContentName)
        {
            NavigationParameters param = new NavigationParameters();
            param.Add("Unicode", Unicode);
            param.Add(ParameterNames.PDFViewer, PDFViewer);
            param.Add("ViewModel", this);
            batesRegion.RequestNavigate(BatesSettingsRegionName, SelectedContentName, param);
            BatesSettingsVisible = Visibility.Visible;
        }

        private void create()
        {
            FolderBrowserDialog dlg = new FolderBrowserDialog();
            string saveSelectedPath = "";
            if (dlg.ShowDialog() == System.Windows.Forms.DialogResult.OK)
            {
                saveSelectedPath = dlg.SelectedPath.Trim();
                saveSelectedPath = saveSelectedPath + "\\";
            }
            else { return; }
            HomePageBatchProcessingDialogModel.closeDialog = true;
            updateListview("wait");
            fileNamesIndex = 0;
            BatesGridIsEnabled = "False";
            foreach (var filename in fileNames)
            {
                FileInfo fileinfo = new FileInfo(filename);
                string OutputPath = fileinfo.DirectoryName;
                char[] enumerationSeparator = new char[] { ',' };
                char[] rangeSeparator = new char[] { '-' };
                List<int> PageIndexLists = new List<int>();
                CPDFDocument document = CPDFDocument.InitWithFilePath(filename);
                if (document == null)
                {
                    PDFDataTable.Rows[fileNamesIndex]["FileState"] = "error";
                    PDFCurrentDataTable.Rows[fileNamesIndex]["FileState"] = "error";
                    fileNamesIndex++;
                    continue;
                }
                if (document.IsLocked)
                {
                    DialogParameters value = new DialogParameters();
                    value.Add(ParameterNames.PDFDocument, document);
                    dialogs.ShowDialog(DialogNames.VerifyPassWordDialog, value, e =>
                    {
                        if (e.Result == ButtonResult.OK)
                        {
                            if (e.Parameters.ContainsKey(ParameterNames.PassWord) && e.Parameters.GetValue<string>(ParameterNames.PassWord) != null)
                            {
                                document.UnlockWithPassword(e.Parameters.GetValue<string>(ParameterNames.PassWord).ToString());
                            }
                        }
                    });

                    if (document.IsLocked)
                    {
                        //未成功解密文档时，释放Document对象，返回
                        PDFDataTable.Rows[fileNamesIndex]["FileState"] = "error";
                        PDFCurrentDataTable.Rows[fileNamesIndex]["FileState"] = "error";
                        fileNamesIndex++;
                        continue;
                    }
                }
                // MessageBox.Show(filename);
                CreateBates(document);
                document.WriteToFilePath(CommonHelper.CreateFilePath(saveSelectedPath + document.FileName  + "_Bates.pdf"));
                //if (!CommonHelper.GetPagesInRange(ref PageIndexLists, "1-" + document.PageCount.ToString(), document.PageCount, enumerationSeparator, rangeSeparator))
                //{ //TODO
                //    Trace.WriteLine("输入不对");
                //    MessageBoxEx.Show("输入不对");
                //    return;
                //}
                document.Release();
                PDFDataTable.Rows[fileNamesIndex]["FileState"] = "complete";
                PDFCurrentDataTable.Rows[fileNamesIndex]["FileState"] = "complete";
                fileNamesIndex++;
            }
            BatesGridIsEnabled = "True";
            HomePageBatchProcessingDialogModel.closeDialog = false ;
        }

        public void CreateBates(CPDFDocument document)
        {
            if (BatesInfo != null)
            {
                if (bates != null) { bates.Release(); }
                
                    bates = document.GetBates();
                
                
                for (int i = 0; i < 6; i++)
                {
                    bates.SetText(i, BatesInfo.TextData[i].text);

                    bates.SetFontName(i, BatesInfo.TextData[i].fontName);
                    bates.SetFontSize(i, BatesInfo.TextData[i].fontSize);

                    bates.SetTextColor(i, BatesInfo.TextData[i].Color);
                }
                bates.SetPageOffset(BatesInfo.StarPagetNumber + 1);

                bates.SetPages(getFilePageRange(document, PDFCurrentDataTable.Rows[fileNamesIndex]["FilePageRangeText"].ToString(), PDFCurrentDataTable.Rows[fileNamesIndex]["FilePageRangeSelectIndex"].ToString()));

                bates.SetMargin(BatesInfo.margin);
                bates.Update();

            }
        }
        #endregion

        #region 批量处理逻辑函数
        /// <summary>
        /// 添加PDF文件
        /// </summary>
        private void addpicture()
        {
            FileNameNumber = fileNames.Count;
            System.Windows.Forms.OpenFileDialog dlg = new System.Windows.Forms.OpenFileDialog();
            dlg.Multiselect = true;
            dlg.Filter = "PDF|*.pdf;*.PDF;";
            if (dlg.ShowDialog() == System.Windows.Forms.DialogResult.OK)
            {
                fileNames.AddRange(dlg.FileNames.ToList());
                RemoveExcess(ref fileNames);
                SetBatesGridIsEnabled = "True";
                AddFileVisibility = Visibility.Collapsed;
                updateListview("wait");
            }
        }
        /// <summary>
        /// 拖拽添加PDF文件
        /// </summary>
        public void addPDFFiles(string filename)
        {
            FileNameNumber = fileNames.Count;
            fileNames.Add(filename);
            RemoveExcess(ref fileNames);
            SetBatesGridIsEnabled = "True";
            AddFileVisibility = Visibility.Collapsed;
            updateListview("wait");
        }

        /// <summary>
        /// 添加已打开PDF文件
        /// </summary>
        public void addOpenedPDFFiles()
        {
            FileNameNumber = fileNames.Count;
            foreach (var filename in App.OpenedFileList) {
                fileNames.Add(filename);
            }
            RemoveExcess(ref fileNames);
            SetBatesGridIsEnabled = "True";
            AddFileVisibility = Visibility.Collapsed;
            updateListview("wait");
        }

        /// <summary>
        /// 删除重复的文件
        /// </summary>
        public void RemoveExcess(ref List<string> Filenames)
        {
            List<string> filenames = new List<string>();
            foreach (var fileName in Filenames)
            {
                if (!filenames.Contains(fileName))
                {
                    filenames.Add(fileName);
                }
            }
            Filenames.Clear();
            Filenames = filenames;
        }


        /// <summary>
        /// 添加PDF文件夹
        /// </summary>
        private void addpicturefiles()
        {
            FileNameNumber = fileNames.Count;
            FolderBrowserDialog dialog = new FolderBrowserDialog();
            dialog.Description = "请选择文件路径";
            if (dialog.ShowDialog() == System.Windows.Forms.DialogResult.OK)
            {

                string foldPath = dialog.SelectedPath;
                var apps = System.IO.Directory.GetFiles(foldPath);
                foreach (string app in apps)
                {
                    var fi = new FileInfo(app);
                    if (fi.Extension == ".pdf" || fi.Extension == ".PDF")
                    {
                        fileNames.Add(app);
                    }
                }
                RemoveExcess(ref fileNames);
                updateListview("wait");
                SetBatesGridIsEnabled = "True";
                AddFileVisibility = Visibility.Collapsed;
            }

        }
        /// <summary>
        /// 更新listview显示
        /// state 状态显示字符串
        /// </summary>
        public void updateListview(string state)
        {
            updateCurrentListview();
            DataTable pdfdatatable = new DataTable();
            pdfdatatable.Columns.Add("FileName");
            pdfdatatable.Columns.Add("FilePageRangeText");
            pdfdatatable.Columns.Add("FilePageRangeSelectIndex");
            pdfdatatable.Columns.Add("IsEvenPageIsEnabled");
            pdfdatatable.Columns.Add("MaxPageRange");
            pdfdatatable.Columns.Add("FileSize");
            pdfdatatable.Columns.Add("FileState");
            int datatableindex = 0;
            List<int> PageIndexLists = new List<int>();
            foreach (var fileName in fileNames)
            {
                string file_all = fileName;
                FileInfo f = new FileInfo(file_all);
                string file_size = (((float)f.Length) / 1024).ToString() + " K";
                if (!isEvenPage(fileName)&& PDFCurrentDataTable.Rows[datatableindex]["FilePageRangeSelectIndex"].ToString()=="2") {

                    PDFCurrentDataTable.Rows[datatableindex]["FilePageRangeSelectIndex"] = "0";
                }
                int pagecount = CPDFDocument.InitWithFilePath(fileName).PageCount;
                if (PDFCurrentDataTable.Rows[datatableindex]["FilePageRangeSelectIndex"].ToString() == "3" && !CommonHelper.GetPagesInRange(ref PageIndexLists, PDFCurrentDataTable.Rows[datatableindex]["FilePageRangeText"].ToString(), pagecount, new char[] { ',' }, new char[] { '-' },true)) {
                    PDFCurrentDataTable.Rows[datatableindex]["FilePageRangeSelectIndex"] = "0";

                }
                pdfdatatable.Rows.Add(f.Name, PDFCurrentDataTable.Rows[datatableindex]["FilePageRangeText"], PDFCurrentDataTable.Rows[datatableindex]["FilePageRangeSelectIndex"], isEvenPage(fileName), pagecount, file_size, PDFCurrentDataTable.Rows[datatableindex]["FileState"]);
                datatableindex++;
            }
            PDFDataTable = pdfdatatable;
            HomePageBatchProcessingDialogModel.FilePaths = fileNames;

        }
        /// <summary>
        /// 更新Currentlistview显示
        /// pagerangetext 自定义页面范围字符串  pagerangeselectindex combobox下拉索引（0全部页面 1奇数页 2偶数页 3自定义页面）isevenpageisenabled 偶数页状态 state 状态显示字符串
        /// </summary>
        public void updateCurrentListview(string pagerangetext = "1", string pagerangeselectindex = "0", bool isevenpageisenabled = true, string state = "wait")
        {
            if (fileNames.Count >= FileNameNumber)
            {
                for (int i = 0; fileNames.Count - FileNameNumber > i; i++)
                {
                    PDFCurrentDataTable.Rows.Add(pagerangetext, pagerangeselectindex, isevenpageisenabled, state);
                }
            }
            else
            {
                Reverseorder(ref fileNamesView);
                foreach (int filenamesview in fileNamesView)
                {
                    PDFCurrentDataTable.Rows.RemoveAt(filenamesview);
                }
            }
        }

        /// <summary>
        /// 更新Currentlistview显示
        /// pagerangetext 自定义页面范围字符串  pagerangeselectindex combobox下拉索引（0全部页面 1奇数页 2偶数页 3自定义页面）
        /// </summary>
        public void updateFilesPageRange(string pagerangetext = "1", string pagerangeselectindex = "0")
        {
            for (int i = 0; fileNames.Count > i; i++)
            {

                PDFDataTable.Rows[i]["FilePageRangeText"] = pagerangetext;
                PDFDataTable.Rows[i]["FilePageRangeSelectIndex"] = pagerangeselectindex;
                PDFCurrentDataTable.Rows[i]["FilePageRangeText"] = pagerangetext;
                PDFCurrentDataTable.Rows[i]["FilePageRangeSelectIndex"] = pagerangeselectindex;
            }
        }
        /// <summary>
        /// 获取文件列表单个文件对应的页码范围
        /// document 文档对象 pagerangetext 自定义页面范围字符串  pagerangeselectindex combobox下拉索引（0全部页面 1奇数页 2偶数页 3自定义页面）
        /// </summary>
        public string getFilePageRange(CPDFDocument document, string pagerangetext = "1", string pagerangeselectindex = "0")
        {
            string filePageRange = "1";
            EditToolsHelper.GetPageRange(int.Parse(pagerangeselectindex), document, ref filePageRange, pagerangetext);
            return filePageRange;
        }

        /// <summary>
        /// 判断是否关闭偶数页
        /// document 文档对象 ）
        /// </summary>
        public bool isEvenPage(string filename)
        {
            CPDFDocument document = CPDFDocument.InitWithFilePath(filename);
            bool isevenpage = false;
            if (document.PageCount > 1)
            {
                isevenpage = true;
            }
            document.Release();
            return isevenpage;
        }

        /// <summary>
        /// 逆序int类型集合
        /// </summary>
        public void Reverseorder(ref List<int> Numbers)
        {
            Numbers = Numbers.OrderBy(a => a).ToList();
            Numbers.Reverse();
        }

        /// <summary>
        /// 打开文件PDF
        /// </summary>
        public void openfiledialog()
        {

            foreach (int filenamesview in fileNamesView)
            {
                System.Diagnostics.Process.Start("Explorer", "/select," + fileNames[filenamesview]);
            }
        }

        /// <summary>
        /// 删除文件PDF
        /// </summary>
        public void removepdffile()
        {
            FileNameNumber = fileNames.Count;
            Reverseorder(ref fileNamesView);
            foreach (int filenamesview in fileNamesView)
            {
                //Trace.WriteLine(filenamesview);
                fileNames.Remove(fileNames[filenamesview]);
            }
            if (fileNames.Count < 1)
            {
                SetBatesGridIsEnabled = "False";
                AddFileVisibility = Visibility.Visible;
            }

            updateListview("wait");
        }
        public void removepdffile(int index)
        {

            PDFCurrentDataTable.Rows.RemoveAt(index);
            fileNames.Remove(fileNames[index]);
            
            if (fileNames.Count < 1)
            {
                SetBatesGridIsEnabled = "False";
                AddFileVisibility = Visibility.Visible;
            }

            updateListview("wait");
            
        }

        public void PDFFileCount()
        {
            if (fileNames.Count == 0)
            {
                SetBatesGridIsEnabled = "False";
                AddFileVisibility = Visibility.Visible;
            }
            else
            {
                SetBatesGridIsEnabled = "True";
                AddFileVisibility = Visibility.Collapsed;
            }
        }



        #endregion

        #region 构架行为
        public void OnNavigatedTo(NavigationContext navigationContext)
        {
            List<string> filepath = new List<string>();
            navigationContext.Parameters.TryGetValue<ViewContentViewModel>(ParameterNames.ViewContentViewModel, out viewContentViewModel);
            navigationContext.Parameters.TryGetValue<CPDFViewer>(ParameterNames.PDFViewer, out PDFViewer);

            navigationContext.Parameters.TryGetValue<List<string>>(ParameterNames.FilePath, out filepath);
            if (filepath != null)
            {
                fileNames = filepath;
                CreateGridSpan = "1";
                CreateGridSpanVisibility = Visibility.Visible;
                PDFFileCount();
                updateListview("wait");
                EnterSelectedContent(TemplateListName);
            }
        }


        public bool IsNavigationTarget(NavigationContext navigationContext)
        {
            return true;
        }

        public void OnNavigatedFrom(NavigationContext navigationContext)
        {
        }
        #endregion
    }
}