1
0
mirror of https://gitee.com/bitdance-team/chrome-extension synced 2025-01-12 22:58:14 +08:00
Code Issues Projects Releases Wiki Activity GitHub Gitee
chrome-extension/packages/shell-chrome/assets/html/screenshot/background.js

109 lines
3.7 KiB
JavaScript
Raw Normal View History

// 设置截图方法和存储方法。
chrome.storage.sync.get((config) => {
chrome.storage.sync.set({ method: 'crop' });
chrome.storage.sync.set({ format: 'png' });
chrome.storage.sync.set({ save: 'file' });
chrome.storage.sync.set({ dpr: true });
if (config.save === 'clipboard') {
config.save = 'url'
chrome.storage.sync.set({ save: 'url' })
}
})
// 定义inject函数点击图标就执行
function inject(tab) {
chrome.tabs.sendMessage(tab.id, { message: 'init' }, (res) => {
if (res) {
// 停止植入css和js文件。
clearTimeout(timeout)
}
})
// 植入css和js文件
var timeout = setTimeout(() => {
var relativePath = 'assets/html/screenshot/'
chrome.tabs.insertCSS(tab.id, { file: relativePath + 'css/jquery.Jcrop.min.css', runAt: 'document_start' })
chrome.tabs.insertCSS(tab.id, { file: relativePath + 'css/content.css', runAt: 'document_start' })
chrome.tabs.executeScript(tab.id, { file: relativePath + 'js/jquery.min.js', runAt: 'document_start' })
chrome.tabs.executeScript(tab.id, { file: relativePath + 'js/jquery.Jcrop.min.js', runAt: 'document_start' })
chrome.tabs.executeScript(tab.id, { file: relativePath + 'js/content.js', runAt: 'document_start' })
// 间隔发送"init"message
setTimeout(() => {
chrome.tabs.sendMessage(tab.id, { message: 'init' })
}, 100)
}, 100)
}
var takeScreenshot = inject
chrome.runtime.onMessage.addListener((req, sender, res) => {
console.log(`进入 assets\html\screenshot\background.js 中的onMessage Listener`)
if (req.senderId !== "screenshot") {
// 抛给下一个Listener
res();
}
if (req.message === 'capture') {
console.log("capture开始")
chrome.storage.sync.get((config) => {
chrome.tabs.getSelected(null, (tab) => {
chrome.tabs.captureVisibleTab(tab.windowId, { format: config.format }, (image) => {
// 整个网页截图
// console.log("image", image);
crop(image, req.area, req.dpr, config.dpr, config.format, (cropped) => {
// 裁剪后
// console.log("cropped", cropped)
console.log("capture结束")
res({ message: 'image', image: cropped })
// 回调有问题,参数么有传回去,使用以下变通方式直接转换为可下载的文件
2022-02-06 02:42:55 +08:00
var link = document.createElement('a'); link.download = "学生助手截图-" + Date.now(); link.href = cropped; link.click();
})
})
})
})
}
else if (req.message === 'active') {
if (req.active) {
chrome.browserAction.setTitle({ tabId: sender.tab.id, title: 'Crop and Save' })
// chrome.browserAction.setBadgeText({tabId: sender.tab.id, text: '◩'})
}
else {
chrome.browserAction.setTitle({ tabId: sender.tab.id, title: 'Screenshot Capture' })
// chrome.browserAction.setBadgeText({tabId: sender.tab.id, text: ''})
}
}
console.log(`离开 assets\html\screenshot\background.js 中的onMessage Listener`)
return true
})
function crop(image, area, dpr, preserve, format, done) {
var top = area.y * dpr
var left = area.x * dpr
var width = area.w * dpr
var height = area.h * dpr
var w = (dpr !== 1 && preserve) ? width : area.w
var h = (dpr !== 1 && preserve) ? height : area.h
var canvas = null
if (!canvas) {
canvas = document.createElement('canvas')
document.body.appendChild(canvas)
}
canvas.width = w
canvas.height = h
var img = new Image()
img.onload = () => {
var context = canvas.getContext('2d')
context.drawImage(img,
left, top,
width, height,
0, 0,
w, h
)
var cropped = canvas.toDataURL(`image/${format}`)
done(cropped)
}
img.src = image
}