本身动手实现一个html2canvas

前言

昨天写了新手引导动画的4种实现方式, 里面用到了 html2canvas 因而就顺便了解了一下实现思路.

大概就是 利用 svg 的 foreignObject 标签, 嵌入 dom, 最后再利用 canvas 绘制 svg. 从而实现最终目的.

先让你们看看效果

MDN示例

var canvas = document.getElementById('canvas');
var ctx = canvas.getContext('2d');

var data = '<svg xmlns="http://www.w3.org/2000/svg" width="200" height="200">' +
           '<foreignObject width="100%" height="100%">' +
           '<div xmlns="http://www.w3.org/1999/xhtml" style="font-size:40px">' +
             '<em>I</em> like' +
             '<span style="color:white; text-shadow:0 0 2px blue;">' +
             'cheese</span>' +
           '</div>' +
           '</foreignObject>' +
           '</svg>';

var DOMURL = window.URL || window.webkitURL || window;

var img = new Image();
var svg = new Blob([data], {type: 'image/svg+xml;charset=utf-8'});
var url = DOMURL.createObjectURL(svg);

img.onload = function () {
  ctx.drawImage(img, 0, 0);
  DOMURL.revokeObjectURL(url);
}

img.src = url;
复制代码

MDN示例其实写的很清楚,不过也相对比较简单一点, dom 是已经构建好的字符串, 其实我以为整个过程里面最麻烦的就是构建 dom. 因此接下来,咱们就来看看具体怎么实现吧

第一步 遍历目标节点的全部子元素,并构建对应的字符串

/** * 递归遍历全部子节点 * @param element Document Element 要计算的元素 * @param isTop Boolean 是不是最外层元素 **/
function renderDom (element, isTop) {
	let tag = element.tagName.toLowerCase()
	let str = `<${tag} `
    // 最外层的节点,须要加 xmlns 命名空间
	isTop && (str += `xmlns="http://www.w3.org/1999/xhtml" `)
	str += ` style="${getElementStyles(element)}">\n`

	if (element.children.length) {
	    // 递归子元素
		for (let el of element.children) {
			str += renderDom(el)
		}
	} else {
		str += element.innerHTML
	}
	str += `</${tag}>\n`
	return str
}
复制代码

这里只作了一个最简单的处理,因为是简单实现,不少特殊状况没考虑进去(如:单标签, img等),有兴趣的童鞋能够本身尝试实现看看.

最外层的元素, 须要加命名空间,不然没法识别

这里用到的 getElementStyles 就是获取元素的最终渲染样式,下一步会实现.

第二步, 获取元素的最终渲染样式,并拼接成行内样式

正常的 dom 元素, 是没法直接放在 foreignObject 里面准确地渲染的, 由于还要涉及到父子元素直接的属性继承, 元素默认属性, 非行内样式没法渲染等问题. 因此咱们要获取每一个元素的最终渲染样式, 而后拼接成行内样式.

如何获取元素的最终渲染样式呢? 恰好,浏览器有提供一个 window.getComputedStyle() 方法能够作到.

// 计算每一个 dom 的样式
// 这里原本应该直接用 Object.keys + forEach 遍历取出的
// 可是不知道为何,遍历取出的,会渲染不出来,应该是某些属性有问题
// 暂时没空去排查那些有问题,因此目前先把经常使用的直接写死.
function getElementStyles (el) {
	let css = window.getComputedStyle(el)
	let style = ''
	// 尺寸相关
	style += `width:${css.width};`
	style += `height: ${css.height};`
	style += `line-height: ${css.lineHeight};`
	style += `max-height: ${css.maxHeight};`
	style += `min-height: ${css.minHeight};`
	style += `max-width: ${css.maxWidth};`
	style += `min-width: ${css.minWidth};`

	style += `font-size: ${css.fontSize};`
	// 颜色相关
	style += `color: ${css.color};`
	style += `background: ${css.background};`
	// 边框相关
	style += `border: ${css.border};`
	style += `box-sizing: ${css.boxSizing};`
	// 位置相关
	style += `margin: ${css.margin};`
	style += `padding: ${css.padding};`
	style += `position: ${css.position};`
	style += `left: ${css.left};`
	style += `right: ${css.right};`
	style += `top: ${css.top};`
	style += `bottom: ${css.bottom};`
	// 布局相关
	style += `display: ${css.display};`
	style += `flex: ${css.flex};`
	return style
}
复制代码

第三步, 渲染 svg

把拼接好的 svg 字符串用 Blob 对象 new 出来(Blob真的是个很强大的对象啊), 而后用 DOMURL.createObjectURL() 转换为 url, 有了url, 接下来就看你们自由发挥了. 能够直接下载,也能够在 canvas 里绘制. 或者看成图片直接插入到文档...

// 主入口函数
function shotScreen () {
    let target = document.querySelector('.content')
    let data = getSvgDomString(target)

    let DOMURL = window.URL || window.webkitURL || window;

    let img = new Image();
    let svg = new Blob([data], {type: 'image/svg+xml;charset=utf-8'});
    let url = DOMURL.createObjectURL(svg);

    img.src = url;
    document.body.appendChild(img)
}

// 计算 svg 的字符串
function getSvgDomString (element) {
	return ` <svg xmlns="http://www.w3.org/2000/svg" width="200" height="200">\n <foreignObject width="100%" height="100%">\n ${renderDom(element, 1)} </foreignObject>\n </svg>`
}

复制代码

这里顺便给个绘制到 canvas 里的代码

// 若是想画到 canvas 里面
let canvas = document.getElementById('canvas');
let ctx = canvas.getContext('2d');
let img = new Image();

img.onload = function () {
   ctx.drawImage(img, 0, 0);
   DOMURL.revokeObjectURL(url);
}
复制代码

最后

参考文档:

MDN: 将 DOM 对象绘制到 canvas 中

MDN: foreignObject

完整的代码在这里,能够直接运行看效果.

本文地址在->我的技术帖合集, 欢迎给个 start 或 follow