当前位置:   article > 正文

从零实现的Chrome扩展_chrome extension

chrome extension

从零实现的Chrome扩展

Chrome扩展是一种可以在Chrome浏览器中添加新功能和修改浏览器行为的软件程序,例如我们常用的TamperMonkey、Proxy SwitchyOmega、AdGuard等等,这些拓展都是可以通过WebExtensions API来修改、增强浏览器的能力,用来提供一些浏览器本体没有的功能,从而实现一些有趣的事情。

描述

实际上FireFox是才第一个引入浏览器扩展/附加组件的主流浏览器,其在2004年发布了第一个版本的扩展系统,允许开发人员为FireFox编写自定义功能和修改浏览器行为的软件程序。而Chrome浏览器则在2010年支持了扩展系统,同样其也允许开发人员为Chrome编写自定义功能和修改浏览器行为的软件程序。

虽然FireFox是第一个引入浏览器扩展的浏览器,但是Chrome的扩展系统得到了广泛的认可和使用,也已经成为了现代浏览器中最流行的扩展系统之一。目前用于构建FireFox扩展的技术在很大程度上与被基于Chromium内核的浏览器所支持的扩展API所兼容,例如Chrome、Edge、Opera等。在大多数情况下,为基于Chromium内核浏览器而写的插件只需要少许修改就可以在FireFox中运行。那么本文就以Chrome扩展为例,聊聊如何从零实现一个Chrome扩展,本文涉及的相关的代码都在https://github.com/WindrunnerMax/webpack-simple-environment的rspack--chrome-extension分支中。

Manifest

我们可以先来想一下浏览器拓展到底是什么,浏览器本身是支持了非常完备的Web能力的,也就是同时拥有渲染引擎和Js解析引擎,那么浏览器拓展本身就不需要再去实现一套新的可执行能力了,完全复用Web引擎即可。那么问题来了,单纯凭借Js是没有办法做到一些能力的,比如拦截请求、修改请求头等等,这些Native的能力单凭Js肯定是做不到的,起码也得上C++直接运行在浏览器代码中才可以,实际上解决这个问题也很简单,直接通过类似于Js Bridge的方式暴露出一些接口就可以了,这样还可以更方便地做到权限控制,一定程度避免浏览器扩展执行一些恶意的行为导致用户受损。

那么由此看来,浏览器扩展其实就是一个Web应用,只不过其运行在浏览器的上下文中,并且可以调用很多浏览器提供的特殊API来做到一些额外的功能。那么既然是一个Web应用,应该如何让浏览器知道这是一个拓展而非普通的Web应用,那么我们就需要标记和配置文件,这个文件就是manifest.json,通过这个文件我们可以来描述扩展的基本信息,例如扩展的名称、版本、描述、图标、权限等等。

在manifest.json中有一个字段为manifest_version,这个字段标志着当前Chrome的插件版本,现在我们在浏览器安装的大部分都是v2版本的插件,v1版本的插件早已废弃,而v3版本的插件因为存在大量的Breaking Changes,以及诸多原本v2支持的API在v3被限制或移除,导致诸多插件无法无损过渡到v3版本。但是自2022.01.17起,Chrome网上应用店已停止接受新的Manifest V2扩展,所以对于要新开发的拓展来说,我们还是需要使用v3版本的受限能力,而且因为谷歌之前宣布v2版本将在2023初完全废弃,但是又因为不能做到完全兼容v2地能力,现在又延迟到了2024年初。但是无论如何,谷歌都准备逐步废弃v2而使用v3,那么我们在这里也是基于v3来实现Chrome扩展。

那么构建一个扩展应用,你就需要在项目的根目录创建一个manifest.json文件,一个简单的manifest.json的结构如下所示,详细的配置文档可以参考https://developer.mozilla.org/zh-CN/docs/Mozilla/Add-ons/WebExtensions/manifest.json:

  1. json复制代码{
  2. "manifest_version": 3, // 插件版本
  3. "name": "Extension", // 插件名称
  4. "version": "1.0.0", // 插件版本号
  5. "description": "Chrome Extension", // 插件描述信息
  6. "icons": { // 插件在不同位置显示的图标
  7. "16": "icon16.png", // `16x16`像素的图标
  8. "32": "icon32.png", // `32x32`像素的图标
  9. "48": "icon48.png", // `48x48`像素的图标
  10. "128": "icon128.png" // `128x128`像素的图标
  11. },
  12. "action": { // 单击浏览器工具栏按钮时的行为
  13. "default_popup": "popup.html", // 单击按钮时打开的默认弹出窗口
  14. "default_icon": { // 弹出窗口按钮图标 // 可以直接配置为`string`
  15. "16": "icon16.png", // `16x16`像素的图标
  16. "32": "icon32.png", // `32x32`像素的图标
  17. "48": "icon48.png", // `48x48`像素的图标
  18. "128": "icon128.png" // `128x128`像素的图标
  19. }
  20. },
  21. "background": { // 定义后台页面的文件和工作方式
  22. "service_worker": "background.js" // 注册`Service Worker`文件
  23. },
  24. "permissions": [ // 定义插件需要访问的`API`权限
  25. "storage", // 存储访问权限
  26. "activeTab", // 当前选项卡访问权限
  27. "scripting" // 脚本访问权限
  28. ]
  29. }

Bundle

既然在上边我们确定了Chrome扩展实际上还是Web技术,那么我们就完全可以利用Web的相关生态来完成插件的开发,当前实际上是有很多比较成熟的扩展框架的,其中也集合了相当一部分的能力,只不过我们在这里希望从零开始跑通一整套流程,那么我们就自行借助打包工具来完成产物的构建。在这里选用的是Rspack,Rspack是一个于Rust的高性能构建引擎,具备与Webpack生态系统的互操作性,可以被Webpack项目低成本集成,并提供更好的构建性能。选用Rspack的主要原因是其编译速度会快一些,特别是在复杂项目中Webpack特别是CRA创建的项目打包速度简直惨不忍睹,我这边有个项目改造前后的dev速度对比大概是1min35s : 24s,速度提升还是比较明显的,当然在我们这个简单的Chrome扩展场景下实际上是区别不大的,相关的所有代码都在https://github.com/WindrunnerMax/webpack-simple-environment/tree/rspack--chrome-extension下。

那么现在我们先从manifest.json开始,目标是在右上角实现一个弹窗,当前很多扩展程序也都是基于右上角的小弹窗交互来控制相关能力的。首先我们需要在manifest.json配置action,action的配置就是控制单击浏览器工具栏按钮时的行为,因为实际上是web生态,所以我们应该为其配置一个html文件以及icon。

  1. js复制代码"action": {
  2. "default_popup": "popup.html",
  3. "default_icon": "./static/favicon.png"
  4. }

已经有了配置文件,现在我们就需要将HTML生成出来,在这里就需要借助rspack来实现了,实际上跟webpack差不多,整体思路就是先配置一个HTML模版,然后从入口开始打包Js,最后将Js注入到HTML当中就可以了,在这里我们直接配置一个多入口的输出能力,通常一个扩展插件不会是只有一个Js和HTML文件的,所以我们需要配置一个多入口的能力。在这里我们还打包了两个文件,一个是popup.html作为入口,另一个是worker.js作为后台运行的Service Worker独立线程。

  1. js复制代码entry: {
  2. worker: "./src/worker/index.ts",
  3. popup: "./src/popup/index.tsx",
  4. },
  5. plugins: [
  6. new HtmlPlugin({
  7. filename: "popup.html",
  8. template: "./public/popup.html",
  9. inject: false,
  10. }),
  11. ],

实际上我们的dev模式生成的代码都是在内存当中的,而谷歌扩展是基于磁盘的文件的,所以我们需要将生成的相关文件写入到磁盘当中。在这里这个配置是比较简单的,直接在devServer中配置一下就好。

  1. js复制代码devServer: {
  2. devMiddleware: {
  3. writeToDisk: true,
  4. },
  5. },

但是实际上,如果我们是基于磁盘的文件来完成的扩展开发,那么devServer就显得没有那么必要了,我们直接可以通过watch来完成,也就是build --watch,这样就可以实现磁盘文件的实时更新了。我们使用devServer是更希望能够借助于HMR的能力,但是这个能力在Chrome扩展v3上的限制下目前表现的并不好,所以在这里这个能力先暂时放下,毕竟实际上v3当前还是在收集社区意见来更新的。不过我们可以有一些简单的方法,来缓解这个问题,我们在开发扩展的最大的一个问题是需要在更新的时候去手动点击刷新来加载插件,那么针对于这个问题,我们可以借助chrome.runtime.reload()来实现一个简单的插件重新加载能力,让我们在更新代码之后不必要去手动刷新。

在这里主要提供一个思路,我们可以编写一个rspack插件,利用ws.Server启动一个WebSocket服务器,之后在worker.js也就是我们将要启动的Service Worker来链接WebSocket服务器,可以通过new WebSocket来链接并且在监听消息,当收到来自服务端的reload消息之后,我们就可以执行chrome.runtime.reload()来实现插件的重新加载了,那么在开启的WebSocket服务器中需要在每次编译完成之后例如afterDone这个hook向客户端发送reload消息,这样就可以实现一个简单的插件重新加载能力了。但是实际上这引入了另一个问题,在v3版本的Service Worker不会常驻,所以这个WebSocket链接也会随着Service Worker的销毁而销毁,是比较坑的一点,同样也是因为这一点大量的Chrome扩展无法从v2平滑过渡到v3,所以这个能力后续还有可能会被改善。

接下来,开发插件我们肯定是需要使用CSS以及组件库的,在这里我们引入了@arco-design/web-react,并且配置了scss和less的相关样式处理。首先是define,这个能力可以帮助我们借助TreeShaking来在打包的时候将dev模式的代码删除,当然不光是dev模式,我们可以借助这个能力以及配置来区分任意场景的代码打包;接下来pluginImport这个处理引用路径的配置,实际上就相当于babel-plugin-import,用来实现按需加载;最后是CSS以及预处理器相关的配置,用来处理scss module以及组件库的less文件。

  1. js复制代码builtins: {
  2. define: {
  3. "process.env.NODE_ENV": JSON.stringify(process.env.NODE_ENV),
  4. },
  5. pluginImport: [
  6. {
  7. libraryName: "@arco-design/web-react",
  8. style: true,
  9. },
  10. ],
  11. },
  12. module: {
  13. rules: [
  14. {
  15. test: /\.module.scss$/,
  16. use: [{ loader: "sass-loader" }],
  17. type: "css/module",
  18. },
  19. {
  20. test: /\.less$/,
  21. use: [
  22. {
  23. loader: "less-loader",
  24. options: {
  25. lessOptions: {
  26. javascriptEnabled: true,
  27. importLoaders: true,
  28. localIdentName: "[name]__[hash:base64:5]",
  29. },
  30. },
  31. },
  32. ],
  33. type: "css",
  34. },
  35. ],
  36. },

最后,我们需要处理一下资源文件,因为我们在代码中实际上是不会引用manifest.json以及我们配置的资源文件的,所以在这里我们需要通过一个rspack插件来完成相关的功能,因为rspack的相关接口是按照webpack5来做兼容的,所以在编写插件的时候实际跟编写webpack插件差不多。在这里主要是实现两个功能,一个是监听manifest.json配置文件以及资源目录public/static的变化,另一个是将manifest.json文件以及资源文件拷贝到打包目录中。

  1. js复制代码const thread = require("child_process");
  2. const path = require("path");
  3. const exec = command => {
  4. return new Promise((resolve, reject) => {
  5. thread.exec(command, (err, stdout) => {
  6. if (err) reject(err);
  7. resolve(stdout);
  8. });
  9. });
  10. };
  11. class FilesPlugin {
  12. apply(compiler) {
  13. compiler.hooks.make.tap("FilePlugin", compilation => {
  14. const manifest = path.join(__dirname, "../src/manifest.json");
  15. const resources = path.join(__dirname, "../public/static");
  16. !compilation.fileDependencies.has(manifest) && compilation.fileDependencies.add(manifest);
  17. !compilation.contextDependencies.has(resources) &&
  18. compilation.contextDependencies.add(resources);
  19. });
  20. compiler.hooks.done.tapPromise("FilePlugin", () => {
  21. return Promise.all([
  22. exec("cp ./src/manifest.json ./dist/"),
  23. exec("cp -r ./public/static ./dist/static"),
  24. ]);
  25. });
  26. }
  27. }
  28. module.exports = FilesPlugin;

Service Worker

我们在Chrome浏览器中打开chrome://extensions/,可以看到我们浏览器中已经装载的插件,可以看到很多插件都会有一个类似于background.html的文件,这是v2版本的扩展独有的能力,是一个独立的线程,可以用来处理一些后台任务,比如网络请求、消息推送、定时任务等等。那么现在扩展已经发展到了v3版本,在v3版本中一个非常大的区别就是Service Workers不能保证常驻,需要主动唤醒,所以在chrome://extensions/中如果是v3版本的插件,我们会看到一个Service Worker的标识,那么在一段时间不动之后,这个Service Worker就会标记上Idle,在这个时候其就处于休眠状态了,而不再常驻于内存。

对于这个Service Worker,Chrome会每5分钟清理所有扩展Service Workers,也就是说扩展的Worker最多存活5分钟,然后等待用户下次激活,但是激活方式没有明确的表述,那假如我们的拓展要做的工作没做完,要接上次的工作怎么办,Google答复是用chrome.storage类似存储来暂存工作任务,等待下次激活。为了对抗随机的清理事件,出现了很多肮脏的手段,甚至有的为了保持持续后台,做两个扩展然后相互唤醒。除了这方面还有一些类似于webRequest -> declarativeNetRequest、setTimeout/setInterval、DOM解析、window/document等等的限制,会影响大部分的插件能力。

当然如果我们想在用户主观运行时实现相关能力的常驻,就可以直接chrome.tabs.create在浏览器Tab中打开扩展程序的HTML页面,这样就可以作为前台运行,同样这个扩展程序的代码就会一直运行着。

Chrome官方博客发布了一个声明More details on the transition to Manifest V3,将Manifest V2的废除时间从2023年1月向后推迟了一年:

  1. css复制代码Starting in June in Chrome 115, Chrome may run experiments to turn off support for Manifest V2 extensions in all channels, including stable channel.
  2. In January 2024, following the expiration of the Manifest V2 enterprise policy, the Chrome Web Store will remove all remaining Manifest V2 items from the store.

再来看看两年前对废除Manifest V2的声明:

perl复制代码January 2023: The Chrome browser will no longer run Manifest V2 extensions. Developers may no longer push updates to existing Manifest V2 extensions.

从原本的斩钉截铁,变成现在的含糊和留有余地,看来强如Google想要执行一个影响全世界65%互联网用户的Breaking Change,也不是那么容易的。但v3实际上并不全是缺点,在用户隐私上面,v3绝对是一个提升,v3增加了很多在隐私方面的限制,非常重要的一点是不允许引用外部资源。Chrome扩展能做的东西实在是太多了,如果不了解或者不开源的话根本不敢安装,因为扩展权限太高可能会造成很严重的例如用户信息泄漏等问题,即使是比如像Firefox那样必须要上传源代码的方式来加强审核,也很难杜绝所有的隐患。

通信方案

Chrome扩展在设计上有非常多的模块和能力,我们常见的模块有background/worker、popup、content、inject、devtools等,不同的模块对应着不同的作用,协作构成了插件的扩展功能。

  • background/worker: 这个模块负责在后台运行扩展,可以实现一些需要长期运行的操作,例如与服务器通信、定时任务等。
  • popup: 这个模块是扩展的弹出层界面,可以通过点击扩展图标在浏览器中弹出,用于显示扩展的一些信息或操作界面。
  • content: 这个模块可以访问当前页面的DOM结构和样式,可以实现一些与页面交互的操作,但该模块的window与页面的window是隔离的。
  • inject: 这个模块可以向当前页面注入自定义的JavaScript或CSS代码,可以实现一些与页面交互的操作,例如修改页面行为、添加样式等。
  • devtools: 这个模块可以扩展Chrome开发者工具的功能,可以添加新的面板、修改现有面板的行为等。
  1. bash复制代码https://developer.mozilla.org/zh-CN/docs/Mozilla/Add-ons/WebExtensions/Content_scripts
  2. https://developer.mozilla.org/en-US/docs/Mozilla/Add-ons/WebExtensions/Background_scripts
  3. https://developer.mozilla.org/en-US/docs/Mozilla/Add-ons/WebExtensions/user_interface/Popups
  4. https://developer.mozilla.org/zh-CN/docs/Mozilla/Add-ons/WebExtensions/user_interface/devtools_panels
  5. https://developer.mozilla.org/zh-CN/docs/Mozilla/Add-ons/WebExtensions/manifest.json/web_accessible_resources

在插件的能力上,不同的模块也有着不同的区别,这个能力主要在于Chrome API、DOM访问、跨域访问、页面Window对象访问等。

模块

Chrome API

DOM访问

跨域访问

页面Window对象访问

background/worker

绝大部分API,除了devtools系列

不可直接访问页面DOM

可跨域访问

不可直接访问页面Window

popup

绝大部分API,除了devtools系列

能直接访问自身的DOM

可跨域访问

能直接访问自身的Window

content

有限制,只能访问runtime和extension等部分API

可以访问页面DOM

不可跨域访问

不可直接访问页面Window

inject

不能访问Chrome API

可以访问页面DOM

不可跨域访问

可直接访问页面Window

devtools

有限制,只能访问devtools、runtime和extension等部分API

可以访问页面DOM

不可跨域访问

可直接访问页面Window

对于消息通信,在不同的模块需要配合三种API来实现,短链接chrome.runtime.onMessage + chrome.runtime/tabs.sendMessage、长链接chrome.runtime.connect + port.postMessage + port.onMessage + chrome.runtime/tabs.onConnect,原生消息window.postMessage + window.addEventListener,下边的表格中展示的是直接通信的情况,我们可以根据实际的业务来完成间接通信方案。

background/worker

popup

content

inject

devtools

background/worker

/

chrome.extension.getViews

chrome.tabs.sendMessage / chrome.tabs.connect

/

/

popup

chrome.extension.getBackgroundPage

/

chrome.tabs.sendMessage / chrome.tabs.connect

/

/

content

chrome.runtime.sendMessage / chrome.runtime.connect

chrome.runtime.sendMessage / chrome.runtime.connect

/

window.postMessage

/

inject

/

/

window.postMessage

/

/

devtools

chrome.runtime.sendMessage

chrome.runtime.sendMessage

/

chrome.devtools.inspectedWindow.eval

/

实例

接下来我们来实现一个实例,主要的功能是解除浏览器复制限制的通用方案,具体可以参考https://github.com/WindrunnerMax/TKScript文本选中复制-通用这部分,完整的操作实例都在https://github.com/WindrunnerMax/webpack-simple-environment/tree/rspack--chrome-extension中。此外注册Chrome扩展的开发者价格是5$,注册之后才能在谷歌商店发布扩展。那么首先,我们先在popup中绘制一个界面,用来展示当前的扩展状态,以及提供一些操作按钮。

  1. js复制代码export const App: FC = () => {
  2. const [copyState, setCopyState] = useState(false);
  3. const [copyStateOnce, setCopyStateOnce] = useState(false);
  4. const [menuState, setMenuState] = useState(false);
  5. const [menuStateOnce, setMenuStateOnce] = useState(false);
  6. const [keydownState, setKeydownState] = useState(false);
  7. const [keydownStateOnce, setKeydownStateOnce] = useState(false);
  8. // 与`content`通信 操作事件与`DOM`
  9. const onSwitchChange = (
  10. type:
  11. | typeof POPUP_CONTENT_ACTION.MENU
  12. | typeof POPUP_CONTENT_ACTION.KEYDOWN
  13. | typeof POPUP_CONTENT_ACTION.COPY,
  14. checked: boolean,
  15. once = false
  16. ) => {
  17. PopupContentBridge.postMessage({ type: type, payload: { checked, once } });
  18. };
  19. // 与`content`通信 查询开启状态
  20. useLayoutEffect(() => {
  21. const queue = [
  22. { key: QUERY_STATE_KEY.STORAGE_COPY, state: setCopyState },
  23. { key: QUERY_STATE_KEY.STORAGE_MENU, state: setMenuState },
  24. { key: QUERY_STATE_KEY.STORAGE_KEYDOWN, state: setKeydownState },
  25. { key: QUERY_STATE_KEY.SESSION_COPY, state: setCopyStateOnce },
  26. { key: QUERY_STATE_KEY.SESSION_MENU, state: setMenuStateOnce },
  27. { key: QUERY_STATE_KEY.SESSION_KEYDOWN, state: setKeydownStateOnce },
  28. ];
  29. queue.forEach(item => {
  30. PopupContentBridge.postMessage({
  31. type: POPUP_CONTENT_ACTION.QUERY_STATE,
  32. payload: item.key,
  33. }).then(r => {
  34. r && item.state(r.payload);
  35. });
  36. });
  37. }, []);
  38. return (
  39. <div className={cs(style.container)}>
  40. { /* xxx */ }
  41. </div>
  42. );
  43. };

可以看到我们实际上主要是通过bridge与content script进行了通信,在前边我们也描述了如何进行通信,在这里我们可以通过设计一个通信类来完成相关操作,同时为了保持完整的TS类型,在这里定义了很多通信时的标志。实际上在这里我们选择了一个相对麻烦的操作,所有的操作都必须要要通信到content script中完成,因为事件与DOM操作都必须要在content script或者inject script中才可以完成,但是实际上chrome.scripting.executeScript也可以完成类似的操作,但是在这里为了演示通信能力所以采用了比较麻烦的操作,另外如果要保持下次打开该页面的状态依旧是保持Hook状态的话,也必须要用content script。

  1. js复制代码export const POPUP_CONTENT_ACTION = {
  2. COPY: "___COPY",
  3. MENU: "___MENU",
  4. KEYDOWN: "___KEYDOWN",
  5. QUERY_STATE: "___QUERY_STATE",
  6. } as const;
  7. export const QUERY_STATE_KEY = {
  8. STORAGE_COPY: "___STORAGE_COPY",
  9. STORAGE_MENU: "___STORAGE_MENU",
  10. STORAGE_KEYDOWN: "___STORAGE_KEYDOWN",
  11. SESSION_COPY: "___SESSION_COPY",
  12. SESSION_MENU: "___SESSION_MENU",
  13. SESSION_KEYDOWN: "___SESSION_KEYDOWN",
  14. } as const;
  15. export const POPUP_CONTENT_RTN = {
  16. STATE: "___STATE",
  17. } as const;
  18. export type PopupContentAction =
  19. | {
  20. type:
  21. | typeof POPUP_CONTENT_ACTION.MENU
  22. | typeof POPUP_CONTENT_ACTION.KEYDOWN
  23. | typeof POPUP_CONTENT_ACTION.COPY;
  24. payload: { checked: boolean; once: boolean };
  25. }
  26. | {
  27. type: typeof POPUP_CONTENT_ACTION.QUERY_STATE;
  28. payload: (typeof QUERY_STATE_KEY)[keyof typeof QUERY_STATE_KEY];
  29. };
  30. type PopupContentRTN = {
  31. type: (typeof POPUP_CONTENT_RTN)[keyof typeof POPUP_CONTENT_RTN];
  32. payload: boolean;
  33. };
  34. export class PopupContentBridge {
  35. static async postMessage(data: PopupContentAction) {
  36. return new Promise<PopupContentRTN | null>(resolve => {
  37. chrome.tabs.query({ active: true, currentWindow: true }, tabs => {
  38. const tabId = tabs[0] && tabs[0].id;
  39. if (tabId) {
  40. chrome.tabs.sendMessage(tabId, data).then(resolve);
  41. // https://developer.chrome.com/docs/extensions/reference/scripting/#runtime-functions
  42. // chrome.scripting.executeScript;
  43. } else {
  44. resolve(null);
  45. }
  46. });
  47. });
  48. }
  49. static onMessage(cb: (data: PopupContentAction) => void | PopupContentRTN) {
  50. const handler = (
  51. message: PopupContentAction,
  52. sender: chrome.runtime.MessageSender,
  53. sendResponse: (response?: PopupContentRTN | null) => void
  54. ) => {
  55. const rtn = cb(message);
  56. sendResponse(rtn || null);
  57. };
  58. chrome.runtime.onMessage.addListener(handler);
  59. return () => {
  60. chrome.runtime.onMessage.removeListener(handler);
  61. };
  62. }
  63. }

最后,我们在content script中之行了实际上的操作,复制行为的Hook在这里抹除了细节,如果感兴趣可以直接看上边的仓库地址,在content script主要实现的操作就是接收popup发送过来的消息执行操作,并且根据存储在storage中的数据来做一些初始化的行为。

  1. js复制代码let DOMLoaded = false;
  2. const collector: (() => void)[] = [];
  3. // Equivalent to content_scripts document_end
  4. window.addEventListener("DOMContentLoaded", () => {
  5. DOMLoaded = true;
  6. collector.forEach(fn => fn());
  7. });
  8. const withDOMReady = (fn: () => void) => {
  9. if (DOMLoaded) {
  10. fn();
  11. } else {
  12. collector.push(fn);
  13. }
  14. };
  15. const onMessage = (data: PopupContentAction) => {
  16. switch (data.type) {
  17. case ACTION.COPY: {
  18. if (data.payload.checked) withDOMReady(enableCopyHook);
  19. else withDOMReady(disableCopyHook);
  20. const key = STORAGE_KEY_PREFIX + ACTION.COPY;
  21. if (!data.payload.once) {
  22. localStorage.setItem(key, data.payload.checked ? "true" : "");
  23. } else {
  24. console.log("111", 111);
  25. sessionStorage.setItem(key, data.payload.checked ? "true" : "");
  26. }
  27. break;
  28. }
  29. case ACTION.MENU: {
  30. if (data.payload.checked) enableContextMenuHook();
  31. else disableContextMenuHook();
  32. const key = STORAGE_KEY_PREFIX + ACTION.MENU;
  33. if (!data.payload.once) {
  34. localStorage.setItem(key, data.payload.checked ? "true" : "");
  35. } else {
  36. sessionStorage.setItem(key, data.payload.checked ? "true" : "");
  37. }
  38. break;
  39. }
  40. case ACTION.KEYDOWN: {
  41. if (data.payload.checked) enableKeydownHook();
  42. else disableKeydownHook();
  43. const key = STORAGE_KEY_PREFIX + ACTION.KEYDOWN;
  44. if (!data.payload.once) {
  45. localStorage.setItem(key, data.payload.checked ? "true" : "");
  46. } else {
  47. sessionStorage.setItem(key, data.payload.checked ? "true" : "");
  48. }
  49. break;
  50. }
  51. case ACTION.QUERY_STATE: {
  52. const STATE_MAP = {
  53. [QUERY_STATE_KEY.STORAGE_COPY]: { key: ACTION.COPY, storage: localStorage },
  54. [QUERY_STATE_KEY.STORAGE_MENU]: { key: ACTION.MENU, storage: localStorage },
  55. [QUERY_STATE_KEY.STORAGE_KEYDOWN]: { key: ACTION.KEYDOWN, storage: localStorage },
  56. [QUERY_STATE_KEY.SESSION_COPY]: { key: ACTION.COPY, storage: sessionStorage },
  57. [QUERY_STATE_KEY.SESSION_MENU]: { key: ACTION.MENU, storage: sessionStorage },
  58. [QUERY_STATE_KEY.SESSION_KEYDOWN]: { key: ACTION.KEYDOWN, storage: sessionStorage },
  59. };
  60. for (const [key, value] of Object.entries(STATE_MAP)) {
  61. if (key === data.payload)
  62. return {
  63. type: POPUP_CONTENT_RTN.STATE,
  64. payload: !!value.storage[STORAGE_KEY_PREFIX + value.key],
  65. };
  66. }
  67. }
  68. }
  69. };
  70. PopupContentBridge.onMessage(onMessage);
  71. if (
  72. localStorage.getItem(STORAGE_KEY_PREFIX + ACTION.COPY) ||
  73. sessionStorage.getItem(STORAGE_KEY_PREFIX + ACTION.COPY)
  74. ) {
  75. withDOMReady(enableCopyHook);
  76. }
  77. if (
  78. localStorage.getItem(STORAGE_KEY_PREFIX + ACTION.MENU) ||
  79. sessionStorage.getItem(STORAGE_KEY_PREFIX + ACTION.MENU)
  80. ) {
  81. enableContextMenuHook();
  82. }
  83. if (
  84. localStorage.getItem(STORAGE_KEY_PREFIX + ACTION.KEYDOWN) ||
  85. sessionStorage.getItem(STORAGE_KEY_PREFIX + ACTION.KEYDOWN)
  86. ) {
  87. enableKeydownHook();
  88. }

因为在这里这个插件并没有发布到Chrome的应用市场,所以如果想检验效果只能本地处理,在run dev后可以发现打包出来的产物已经在dist文件夹下了,接下来我们在chrome://extensions/打开开发者模式,然后点击加载已解压的扩展程序,选择dist文件夹,这样就可以看到我们的插件了。之后我在百度搜索了"实习报告"关键词,出现了很多文档,随便打开一个在复制的时候就会出现付费的行为,此时我们点击插件,启动Hook复制行为,再复制文本内容就会发现不会弹出付费框了,内容也是成功复制了。请注意在这里我们实现的是一个通用的复制能力,对于百度文库、腾讯文档这类的canvas绘制的文档站是需要单独处理的,关于这些可以参考https://github.com/WindrunnerMax/TKScript。

声明:本文内容由网友自发贡献,转载请注明出处:【wpsshop博客】
推荐阅读
相关标签
  

闽ICP备14008679号