import html2canvas from 'html2canvas'
import jsPDF from 'jspdf'
// base64转blob
export function toBlob(base64Data) {
let byteString = base64Data
if (base64Data.split(',')[0].indexOf('base64') >= 0) {
byteString = atob(base64Data.split(',')[1]) // base64 解码
} else {
byteString = unescape(base64Data.split(',')[1])
}
// 获取文件类型
const mimeString = base64Data.split(';')[0].split(':')[1] // mime类型
// ArrayBuffer 对象用来表示通用的、固定长度的原始二进制数据缓冲区
// let arrayBuffer = new ArrayBuffer(byteString.length) // 创建缓冲数组
// let uintArr = new Uint8Array(arrayBuffer) // 创建视图
const uintArr = new Uint8Array(byteString.length) // 创建视图
for (let i = 0; i < byteString.length; i += 1) {
uintArr[i] = byteString.charCodeAt(i)
}
// 生成blob
const blob = new Blob([uintArr], {
type: mimeString
})
// 使用 Blob 创建一个指向类型化数组的URL, URL.createObjectURL是new Blob文件的方法,可以生成一个普通的url,可以直接使用,比如用在img.src上
return blob
}
/**
* 输出pdf
* @param {*} idName html元素
* @param {*} pdfName 输出pdf文件名
* @param {*} isDownload 是否直接下载
* @param {*} isPrint 是否直接打印
* @param {*} callback 执行后的回调
*/
export function outPutPdf(idName, pdfName, isDownload = false, isPrint = false, callback) {
const element = document.getElementById(idName) // 这个dom元素是要导出的pdf的div容器
const w = element.offsetWidth // 获得该容器的宽
const h = element.offsetHeight // 获得该容器的高
const offsetTop = element.offsetTop // 获得该容器到文档顶部的距离
const offsetLeft = element.offsetLeft // 获得该容器到文档最左的距离
const canvas = document.createElement('canvas')
let abs = 0
const winI = document.body.clientWidth // 获得当前可视窗口的宽度(不包含滚动条)
const winO = window.innerWidth // 获得当前窗口的宽度(包含滚动条)
if (winO > winI) {
abs = (winO - winI) / 2 // 获得滚动条宽度的一半
}
canvas.width = w * 2 // 将画布宽&&高放大两倍
canvas.height = h * 2
const context = canvas.getContext('2d')
context.scale(2, 2)
context.translate(-offsetLeft - abs, -offsetTop)
// 这里默认横向没有滚动条的情况,因为offset.left(),有无滚动条的时候存在差值,因此translate的时候,要把这个差值去掉
html2canvas(element, {
useCORS: true, // 允许加载跨域的图片
allowTaint: true,
scale: 2 // 提升画面质量,但是会增加文件大小
}).then(async (cs) => {
const contentWidth = cs.width
const contentHeight = cs.height
// 一页pdf显示html页面生成的canvas高度
const pageHeight = (contentWidth / 592.28) * 841.89 - 10
// 未生成pdf的html页面高度
let leftHeight = contentHeight
// 页面偏移
let position = 0
// a4纸的尺寸[595.28,841.89],html页面生成的canvas在pdf中图片的宽高
const imgWidth = 595.28 - 10
const imgHeight = (592.28 / contentWidth) * contentHeight
const pageDate = cs.toDataURL('image/jpeg', 1.0)
const pdf = new jsPDF('', 'pt', 'a4')
// 有两个高度需要区分,一个是html页面的实际高度,和生成pdf的页面的高度(841.89)
// 当内容未超过pdf一页显示的范围,无需分页
if (leftHeight < pageHeight) {
pdf.addImage(pageDate, 'JPEG', 5, 0, imgWidth, imgHeight)
} else {
// 分页
while (leftHeight > 0) {
pdf.addImage(pageDate, 'JPEG', 5, position, imgWidth, imgHeight)
leftHeight -= pageHeight
position -= 841.89
// 避免添加空白页
if (leftHeight > 0) {
pdf.addPage()
}
}
element.style.height = ''
}
if (isDownload) {
await pdf.save(pdfName + '.pdf', { returnPromise: true }).then(() => {
// 去除添加的空div 防止页面混乱
const doms = document.querySelectorAll('.emptyDiv')
for (let i = 0; i < doms.length; i++) {
doms[i].remove()
}
})
}
if (isPrint) {
const link = window.URL.createObjectURL(toBlob(pdf.output('datauristring')))
const iframes = Array.from(document.getElementsByTagName('iframe'))
if (iframes.length > 0) {
iframes.forEach((item) => {
item.remove()
})
}
let iframe = document.createElement('iframe')
iframe.style.display = 'none'
iframe.src = link
iframe.id = 'pdf-iframe'
document.body.appendChild(iframe)
iframe.contentWindow.focus()
iframe.contentWindow.print()
}
callback && callback()
})
}
前端下载、打印pdf
最新推荐文章于 2024-08-25 09:57:22 发布