PDFOffice_Windows_exe/PDF Office/ViewModels/Dialog/ChatGPTAIDialogs/DocumentaryTranslationDialogViewModel.cs

using ComPDFKit.PDFDocument;
using ComPDFKitViewer;
using ComPDFKitViewer.PdfViewer;
using Microsoft.Win32;
using PDF_Master.CustomControl;
using PDF_Master.Helper;
using PDF_Master.Model;
using PDF_Master.Properties;
using Prism.Commands;
using Prism.Mvvm;
using Prism.Regions;
using Prism.Services.Dialogs;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Threading.Tasks;
using System.Windows;

namespace PDF_Master.ViewModels.Dialog.ChatGPTAIDialogs
{
    public class DocumentaryTranslationDialogViewModel : BindableBase, IDialogAware
    {
        #region 参数和属性

        public string Title => "";

        public IRegionManager regionManager;

        public CPDFViewer PDFViewer;

        public ViewContentViewModel viewContentViewModel;

        private CPDFViewer pdfViewer = null;

        private string fromlanguage = "";

        private string tolanguage = "";

        private int fromlanguageIndex = 0;
        public int FromlanguageIndex
        {
            get { return fromlanguageIndex; }
            set
            {
                SetProperty(ref fromlanguageIndex, value);
                fromlanguage = ChatGTPAIHelper.UpdateLanguagebType(value);
            }
        }

        private int tolanguageIndex = 0;
        public int TolanguageIndex
        {
            get { return tolanguageIndex; }
            set
            {
                SetProperty(ref tolanguageIndex, value);
                tolanguage = ChatGTPAIHelper.UpdateLanguagebType(value + 1);
            }
        }

        /// <summary>
        /// 进度条显示
        /// </summary>
        private Visibility progressVisible = Visibility.Collapsed;
        public Visibility ProgressVisible
        {
            get
            {
                return progressVisible;
            }
            set
            {
                SetProperty(ref progressVisible, value);
            }
        }


        private double processvalue = 0;

        /// <summary>
        /// 进度条当前值
        /// </summary>
        public double Value
        {
            get { return processvalue; }
            set
            {
                SetProperty(ref processvalue, value);
            }
        }

        private double maxValue = 10;

        /// <summary>
        /// 进度条最大值
        /// </summary>
        public double MaxValue
        {
            get { return maxValue; }
            set
            {
                SetProperty(ref maxValue, value);
            }
        }

        /// <summary>
        /// 初始化翻译前语言列表
        /// </summary>

        public static List<string> FromlanguageFamily { set; get; } = new List<string>();

        public string ViewerRegionName { get; set; }

        private void GetFromlanguageOrigin()
        {
            FromlanguageFamily.Clear();
            FromlanguageFamily = ChatGTPAIHelper.SetFromlanguageOrigin();
        }

        /// <summary>
        /// 初始化翻译到语言列表
        /// </summary>
        public List<string> TolanguageFamily { set; get; } = new List<string>();

        private void GetTolanguageOrigin()
        {
            TolanguageFamily.Clear();
            TolanguageFamily = ChatGTPAIHelper.SetTolanguageOrigin();
        }

        public static Action CancelProgress { get; set; }

        #endregion

        #region 委托声明
        public DelegateCommand TranslateCommand { get; set; }
        public DelegateCommand CancelCommand { get; set; }
        public DelegateCommand CancelProgressCommand { get; set; }
        #endregion
        public DocumentaryTranslationDialogViewModel(IRegionManager regionManager)
        {
            this.regionManager = regionManager;
            CancelCommand = new DelegateCommand(cancel);
            TranslateCommand = new DelegateCommand(translate);
            ViewerRegionName = Guid.NewGuid().ToString();
            CancelProgressCommand = new DelegateCommand(cancelprogress);
            init();
        }
        #region 逻辑函数

        /// <summary>
        /// 初始化下拉列表，或其他
        /// </summary>
        private void init()
        {
            GetFromlanguageOrigin();
            GetTolanguageOrigin();
        }
        /// <summary>
        /// 进度条取消
        /// </summary>
        private void cancelprogress()
        {
            if (CancelProgress != null)
            {
                CancelProgress.Invoke();
            }


        }

        /// <summary>
        /// 关闭窗口
        /// </summary>
        private void cancel()
        {
            RequestClose.Invoke(new DialogResult(ButtonResult.Cancel));
        }

        /// <summary>
        /// 翻译当前文档
        /// </summary>
        public async void translate()
        {
            //添加付费拦截锁
            if (!ServiceHelper.IAPBeforeFunction())
            {
                return;
            }
            if (Settings.Default.UserDate.subscribestatus != 1)
            {
                App.mainWindowViewModel.dialogs.ShowDialog(DialogNames.IAPCompareDialog);
                return;
            }
            ProgressVisible = Visibility.Visible;
            Value = 1;
            string newfile = "";
            FileInfo fileInfo = new FileInfo(pdfViewer.Document.FilePath);
            Value = 3;
            //判断文件大小
            if ((((float)fileInfo.Length) / 1024 / 1024) > 30)
            {
                Value = 9;
                AlertsMessage alertsMessage = new AlertsMessage();
                alertsMessage.ShowDialog("", "The uploaded file cannot exceed 10MB", App.ServiceLoader.GetString("Text_ok"));
                ProgressVisible = Visibility.Collapsed;
                return;
            }
            //判断文件页数
            if (fileInfo.Extension.ToLower() == ".pdf")
            {
                CPDFDocument cPDFDocument = CPDFDocument.InitWithFilePath(pdfViewer.Document.FilePath);
                if (cPDFDocument.PageCount > 30)
                {
                    Value = 9;
                    AlertsMessage alertsMessage = new AlertsMessage();
                    alertsMessage.ShowDialog("", "The uploaded file cannot exceed 30Pages", App.ServiceLoader.GetString("Text_ok"));
                    ProgressVisible = Visibility.Collapsed;
                    return;
                }
            }
            Value = 4;
            await Task.Run(async delegate
            {
                newfile = await ChatGTPAIHelper.fileTranslate(pdfViewer.Document.FilePath, fromlanguage, tolanguage, false);
            });
            Value = 8;
            if (!string.IsNullOrEmpty(newfile))
            {
                if (File.Exists(newfile))
                {
                    if (viewContentViewModel != null)
                    {
                        viewContentViewModel.OpenFile(new string[] { newfile });
                        RequestClose.Invoke(new DialogResult(ButtonResult.Cancel));
                    }
                }
                else
                {
                    //newfile code
                    AlertsMessage alertsMessage = new AlertsMessage();
                    alertsMessage.ShowDialog("", newfile, App.ServiceLoader.GetString("Text_ok"));
                }
            }

            Value = 9;
            ProgressVisible = Visibility.Collapsed;

        }
        #endregion

        #region 构架行为
        public event Action<IDialogResult> RequestClose;

        public bool CanCloseDialog()
        {
            return true;
        }

        public void OnDialogClosed()
        {
        }

        public void OnDialogOpened(IDialogParameters parameters)
        {

            parameters.TryGetValue<CPDFViewer>(ParameterNames.PDFViewer, out pdfViewer);
            parameters.TryGetValue<ViewContentViewModel>(ParameterNames.ViewContentViewModel, out viewContentViewModel);

            if (pdfViewer != null)
            {
                if (!regionManager.Regions[ViewerRegionName].Views.Contains(pdfViewer))
                {
                    //设置PDFview为预览
                    PDFViewer = new CPDFViewer();
                    PDFViewer.InitDocument(pdfViewer.Document);
                    regionManager.AddToRegion(ViewerRegionName, PDFViewer);
                    PDFViewer.SetAnnotInteraction(!PDFViewer.GetAnnotInteraction());
                    PDFViewer.Load();
                    if (PDFViewer.Document.PageCount == 1)
                    {
                        //防止单页pdfview闪烁问题
                        PDFViewer.ChangeViewMode(ViewMode.Single);
                    }
                    else
                    {
                        PDFViewer.ChangeViewMode(ViewMode.SingleContinuous);
                    }
                    PDFViewer.ChangeFitMode(FitMode.FitHeight);
                    PDFViewer.SetMouseMode(MouseModes.PanTool);
                    PDFViewer.Document.ReleasePages();
                    PDFViewer.ReloadDocument();
                    fromlanguage = ChatGTPAIHelper.UpdateLanguagebType(0);
                    tolanguage = ChatGTPAIHelper.UpdateLanguagebType(1);
                    //PageRangeNumber = PDFViewer.Document.PageCount;
                    //PDFViewer.Zoom(0.1);
                }
            }
        }
        #endregion
    }
}