29WxOrder/Work/_29Tools/CefTools/LocalMenuHandler.cs

using CefSharp;
using NaiBa.Util;
using NaiBa.Util.Extension;
using NaiBa.Util.Helper;
using NaiBa.Util.Model;
using NaiBa.Util.Winform.Forms;
using NaiBa.Util.Winform.Helper;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Reflection;
using System.Text;
using System.Threading.Tasks;
using System.Windows.Forms;

namespace Work._29Tools.CefTools
{
    public class LocalMenuHandler : IContextMenuHandler
    {
        public List<CefContextMenu> ContextMenuList;
        public Func<List<Cookie>, TData> SaveCookie;
        public List<Cookie> CookieList;

        void IContextMenuHandler.OnBeforeContextMenu(IWebBrowser chromiumWebBrowser, IBrowser browser, IFrame frame, IContextMenuParams parameters, IMenuModel model)
        {


            model.Clear();
            bool hasCopyPaste = false;

            if ((parameters.EditStateFlags & ContextMenuEditState.CanCopy).ParseToInt() != 0)
            {
                model.AddItem(CefMenuCommand.Copy, "复制");
                hasCopyPaste = true;
            }
            if ((parameters.EditStateFlags & ContextMenuEditState.CanPaste).ParseToInt() != 0)
            {
                model.AddItem(CefMenuCommand.Paste, "粘贴");
                hasCopyPaste = true;
            }
            if (hasCopyPaste)
            {
                model.AddSeparator();
            }

            model.AddItem(CefMenuCommand.Back, "后退");
            model.AddItem(CefMenuCommand.Forward, "前进");
            model.AddItem(CefMenuCommand.ReloadNoCache, "刷新");
            model.AddSeparator();

            model.AddItem(CefMenuCommand.ViewSource, "查看页面源代码");
            model.AddSeparator();

            model.AddItem((CefMenuCommand)26500, "复制页面地址");

            if (parameters.MediaType == ContextMenuMediaType.Image)
            {
                model.AddSeparator();
                model.AddItem((CefMenuCommand)26600, "复制图片地址");
                model.AddItem((CefMenuCommand)26601, "保存图片");

                if (NaiBaGlobalConfig.IsNewFeature)
                {
                    model.AddSeparator();
                    model.AddItem((CefMenuCommand)26700, "识别图片文字");
                }
            }
            if ((parameters.EditStateFlags & ContextMenuEditState.CanTranslate).ParseToInt() != 0)
            {
                model.AddSeparator();
                model.AddItem((CefMenuCommand)26800, "翻译网页");
            }

            if (SaveCookie != null)
            {
                model.AddSeparator();
                model.AddItem((CefMenuCommand)26900, "保存 Cookie");
            }

            if (ContextMenuList != null && ContextMenuList.Any())
            {
                model.AddSeparator();
                if (ContextMenuList.Where(p => p.CommandCode < 28000 || p.CommandCode > 28500).Any())
                {
                    BaseMsgTipFormHelper.ShowError("Cef右键菜单，无效的 CommandCode，CommandCode 的范围 28000-28500");
                    return;
                }
                foreach (var menu in ContextMenuList)
                {
                    model.AddItem((CefMenuCommand)menu.CommandCode, menu.CommandText);
                }
            }
        }

        public bool OnContextMenuCommand(IWebBrowser browserControl, IBrowser browser, IFrame frame, IContextMenuParams parameters, CefMenuCommand commandId, CefEventFlags eventFlags)
        {
            if (commandId == (CefMenuCommand)26500)
            {
                string url = parameters.PageUrl;
                if (!string.IsNullOrEmpty(url))
                {
                    Clipboard.SetText(url);
                    BaseMsgTipFormHelper.ShowInfo("复制页面地址到剪贴板成功");
                }
                else
                {
                    BaseMsgTipFormHelper.ShowError("复制页面地址到剪贴板失败");
                }
                return true;
            }

            if (commandId == (CefMenuCommand)26600)
            {
                string url = parameters.SourceUrl;
                if (!string.IsNullOrEmpty(url))
                {
                    Clipboard.SetText(url);
                    BaseMsgTipFormHelper.ShowInfo("复制图片地址到剪贴板成功");
                }
                else
                {
                    BaseMsgTipFormHelper.ShowError("复制图片地址到剪贴板失败");
                }
                return true;
            }

            if (commandId == (CefMenuCommand)26601)
            {
                string url = parameters.SourceUrl;
                SaveFileDialog dialog = FormDialogHelper.SaveFileDialogCustom(Path.GetFileNameWithoutExtension(url), Path.GetExtension(url), false, false);
                if (dialog != null)
                {
                    try
                    {
                        HttpHelper.DownloadFile(new HttpParam { Url = url }, dialog.FileName);
                    }
                    catch (Exception ex)
                    {
                        LogHelper.WriteWithTime(ex);
                    }
                }
                return true;
            }

            if (commandId == (CefMenuCommand)26700)
            {
                string url = parameters.SourceUrl;
                try
                {
                    if (NaiBaGlobalConfig.ImageOCR != null)
                    {
                        var byteArr = HttpHelper.GetByte(new HttpParam { Url = url });
                        var obj = NaiBaGlobalConfig.ImageOCR(Convert.ToBase64String(byteArr));
                        if (obj.Tag == 1)
                        {
                            FileHelper.OpenNotepad(string.Join(Environment.NewLine, obj.Data));
                        }
                        else
                        {
                            BaseMsgFormHelper.ShowError(obj.Message);
                        }
                    }
                }
                catch (Exception ex)
                {
                    LogHelper.WriteWithTime(ex);
                }
                return true;
            }

            if (commandId == (CefMenuCommand)26800)
            {
                try
                {
                    string fileName = "NaiBa.CefHelper.Resources.google_translate.js";
                    string js = FileHelper.GetResourceFileContent(Assembly.GetExecutingAssembly(), fileName);

                    browser.MainFrame.ExecuteJavaScriptAsync(js);
                }
                catch (Exception ex)
                {
                    LogHelper.WriteWithTime(ex);
                }
                return true;
            }

            if (commandId == (CefMenuCommand)26900)
            {
                try
                {
                    if (CookieList == null)
                    {
                        BaseMsgTipFormHelper.ShowError("未获取到 Cookie，请等待页面加载完后再试");
                    }
                    else
                    {
                        var obj = SaveCookie(CookieList);
                        if (obj.Tag == 1)
                        {
                            BaseMsgTipFormHelper.ShowInfo("保存 Cookie 成功");
                        }
                        else
                        {
                            BaseMsgTipFormHelper.ShowError("保存 Cookie 失败：" + obj.Message);
                        }
                    }
                }
                catch (Exception ex)
                {
                    LogHelper.WriteWithTime(ex);
                }
                return true;
            }

            if ((int)commandId >= 28000 && (int)commandId <= 28500)
            {
                if (ContextMenuList != null)
                {
                    var contextMenu = ContextMenuList.Where(p => p.CommandCode == (int)commandId).FirstOrDefault();
                    if (contextMenu != null)
                    {
                        try
                        {
                            contextMenu.CommandAction?.Invoke(parameters);
                        }
                        catch (Exception ex)
                        {
                            LogHelper.WriteWithTime(ex);
                        }
                        return true;
                    }
                }
            }

            // Return false should ignore the selected option of the user !
            return false;
        }

        void IContextMenuHandler.OnContextMenuDismissed(IWebBrowser chromiumWebBrowser, IBrowser browser, IFrame frame)
        {

        }

        bool IContextMenuHandler.RunContextMenu(IWebBrowser chromiumWebBrowser, IBrowser browser, IFrame frame, IContextMenuParams parameters, IMenuModel model, IRunContextMenuCallback callback)
        {
            return false;
        }
    }

    public class CefContextMenu
    {
        public int CommandCode { get; set; }

        public string CommandText { get; set; }

        public Action<IContextMenuParams> CommandAction { get; set; }
    }
}