Skip to content

Commit

Permalink
fix(compiler-sfc): make asset url imports stringifiable
Browse files Browse the repository at this point in the history
  • Loading branch information
yyx990803 committed Dec 5, 2021
1 parent 3e5ed6c commit 87c73e9
Show file tree
Hide file tree
Showing 7 changed files with 136 additions and 16 deletions.
19 changes: 17 additions & 2 deletions packages/compiler-dom/src/transforms/stringifyStatic.ts
Expand Up @@ -38,6 +38,12 @@ export const enum StringifyThresholds {

type StringifiableNode = PlainElementNode | TextCallNode

/**
* Regex for replacing placeholders for embedded constant variables
* (e.g. import URL string constants generated by compiler-sfc)
*/
const expReplaceRE = /__VUE_EXP_START__(.*?)__VUE_EXP_END__/g

/**
* Turn eligible hoisted static trees into stringified static nodes, e.g.
*
Expand Down Expand Up @@ -80,7 +86,7 @@ export const stringifyStatic: HoistTransform = (children, context, parent) => {
const staticCall = createCallExpression(context.helper(CREATE_STATIC), [
JSON.stringify(
currentChunk.map(node => stringifyNode(node, context)).join('')
),
).replace(expReplaceRE, `" + $1 + "`),
// the 2nd argument indicates the number of DOM nodes this static vnode
// will insert / hydrate
String(currentChunk.length)
Expand Down Expand Up @@ -273,8 +279,17 @@ function stringifyElement(
res += `="${escapeHtml(p.value.content)}"`
}
} else if (p.type === NodeTypes.DIRECTIVE && p.name === 'bind') {
const exp = p.exp as SimpleExpressionNode
if (exp.content[0] === '_') {
// internally generated string constant references
// e.g. imported URL strings via compiler-sfc transformAssetUrl plugin
res += ` ${(p.arg as SimpleExpressionNode).content}="__VUE_EXP_START__${
exp.content
}__VUE_EXP_END__"`
continue
}
// constant v-bind, e.g. :foo="1"
let evaluated = evaluateConstant(p.exp as SimpleExpressionNode)
let evaluated = evaluateConstant(exp)
if (evaluated != null) {
const arg = p.arg && (p.arg as SimpleExpressionNode).content
if (arg === 'class') {
Expand Down
Expand Up @@ -74,6 +74,22 @@ export function render(_ctx, _cache) {
}"
`;

exports[`compiler sfc: transform asset url transform with stringify 1`] = `
"import { createElementVNode as _createElementVNode, createStaticVNode as _createStaticVNode, openBlock as _openBlock, createElementBlock as _createElementBlock } from \\"vue\\"
import _imports_0 from './bar.png'
import _imports_1 from '/bar.png'
const _hoisted_1 = /*#__PURE__*/_createStaticVNode(\\"<img src=\\\\\\"\\" + _imports_0 + \\"\\\\\\"><img src=\\\\\\"\\" + _imports_1 + \\"\\\\\\"><img src=\\\\\\"https://foo.bar/baz.png\\\\\\"><img src=\\\\\\"//foo.bar/baz.png\\\\\\"><img src=\\\\\\"\\" + _imports_0 + \\"\\\\\\">\\", 5)
const _hoisted_6 = [
_hoisted_1
]
export function render(_ctx, _cache) {
return (_openBlock(), _createElementBlock(\\"div\\", null, _hoisted_6))
}"
`;
exports[`compiler sfc: transform asset url with explicit base 1`] = `
"import { createElementVNode as _createElementVNode, Fragment as _Fragment, openBlock as _openBlock, createElementBlock as _createElementBlock } from \\"vue\\"
import _imports_0 from 'bar.png'
Expand Down
Expand Up @@ -194,3 +194,28 @@ export function render(_ctx, _cache) {
], 64 /* STABLE_FRAGMENT */))
}"
`;

exports[`compiler sfc: transform srcset transform srcset w/ stringify 1`] = `
"import { createElementVNode as _createElementVNode, createStaticVNode as _createStaticVNode, openBlock as _openBlock, createElementBlock as _createElementBlock } from \\"vue\\"
import _imports_0 from './logo.png'
import _imports_1 from '/logo.png'
const _hoisted_1 = _imports_0
const _hoisted_2 = _imports_0 + ' 2x'
const _hoisted_3 = _imports_0 + ' 2x'
const _hoisted_4 = _imports_0 + ', ' + _imports_0 + ' 2x'
const _hoisted_5 = _imports_0 + ' 2x, ' + _imports_0
const _hoisted_6 = _imports_0 + ' 2x, ' + _imports_0 + ' 3x'
const _hoisted_7 = _imports_0 + ', ' + _imports_0 + ' 2x, ' + _imports_0 + ' 3x'
const _hoisted_8 = _imports_1 + ', ' + _imports_1 + ' 2x'
const _hoisted_9 = _imports_1 + ', ' + _imports_0 + ' 2x'
const _hoisted_10 = /*#__PURE__*/_createStaticVNode(\\"<img src=\\\\\\"./logo.png\\\\\\" srcset=\\\\\\"\\\\\\"><img src=\\\\\\"./logo.png\\\\\\" srcset=\\\\\\"\\" + _hoisted_1 + \\"\\\\\\"><img src=\\\\\\"./logo.png\\\\\\" srcset=\\\\\\"\\" + _hoisted_2 + \\"\\\\\\"><img src=\\\\\\"./logo.png\\\\\\" srcset=\\\\\\"\\" + _hoisted_3 + \\"\\\\\\"><img src=\\\\\\"./logo.png\\\\\\" srcset=\\\\\\"\\" + _hoisted_4 + \\"\\\\\\"><img src=\\\\\\"./logo.png\\\\\\" srcset=\\\\\\"\\" + _hoisted_5 + \\"\\\\\\"><img src=\\\\\\"./logo.png\\\\\\" srcset=\\\\\\"\\" + _hoisted_6 + \\"\\\\\\"><img src=\\\\\\"./logo.png\\\\\\" srcset=\\\\\\"\\" + _hoisted_7 + \\"\\\\\\"><img src=\\\\\\"/logo.png\\\\\\" srcset=\\\\\\"\\" + _hoisted_8 + \\"\\\\\\"><img src=\\\\\\"https://example.com/logo.png\\\\\\" srcset=\\\\\\"https://example.com/logo.png, https://example.com/logo.png 2x\\\\\\"><img src=\\\\\\"/logo.png\\\\\\" srcset=\\\\\\"\\" + _hoisted_9 + \\"\\\\\\"><img src=\\\\\\"data:image/png;base64,i\\\\\\" srcset=\\\\\\"data:image/png;base64,i 1x, data:image/png;base64,i 2x\\\\\\">\\", 12)
const _hoisted_22 = [
_hoisted_10
]
export function render(_ctx, _cache) {
return (_openBlock(), _createElementBlock(\\"div\\", null, _hoisted_22))
}"
`;
39 changes: 36 additions & 3 deletions packages/compiler-sfc/__tests__/templateTransformAssetUrl.spec.ts
@@ -1,4 +1,9 @@
import { generate, baseParse, transform } from '@vue/compiler-core'
import {
generate,
baseParse,
transform,
TransformOptions
} from '@vue/compiler-core'
import {
transformAssetUrl,
createAssetUrlTransformWithOptions,
Expand All @@ -7,8 +12,13 @@ import {
} from '../src/templateTransformAssetUrl'
import { transformElement } from '../../compiler-core/src/transforms/transformElement'
import { transformBind } from '../../compiler-core/src/transforms/vBind'
import { stringifyStatic } from '../../compiler-dom/src/transforms/stringifyStatic'

function compileWithAssetUrls(template: string, options?: AssetURLOptions) {
function compileWithAssetUrls(
template: string,
options?: AssetURLOptions,
transformOptions?: TransformOptions
) {
const ast = baseParse(template)
const t = options
? createAssetUrlTransformWithOptions(normalizeOptions(options))
Expand All @@ -17,7 +27,8 @@ function compileWithAssetUrls(template: string, options?: AssetURLOptions) {
nodeTransforms: [t, transformElement],
directiveTransforms: {
bind: transformBind
}
},
...transformOptions
})
return generate(ast, { mode: 'module' })
}
Expand Down Expand Up @@ -131,4 +142,26 @@ describe('compiler sfc: transform asset url', () => {

expect(code).toMatchSnapshot()
})

test('transform with stringify', () => {
const { code } = compileWithAssetUrls(
`<div>` +
`<img src="./bar.png"/>` +
`<img src="/bar.png"/>` +
`<img src="https://foo.bar/baz.png"/>` +
`<img src="//foo.bar/baz.png"/>` +
`<img src="./bar.png"/>` +
`</div>`,
{
includeAbsolute: true
},
{
hoistStatic: true,
transformHoist: stringifyStatic
}
)
console.log(code)

This comment has been minimized.

Copy link
@yuwu9145

yuwu9145 Dec 5, 2021

Contributor

Should this console.log be removed?

Edit: raised a pr for it

expect(code).toMatch(`_createStaticVNode`)
expect(code).toMatchSnapshot()
})
})
32 changes: 29 additions & 3 deletions packages/compiler-sfc/__tests__/templateTransformSrcset.spec.ts
@@ -1,4 +1,9 @@
import { generate, baseParse, transform } from '@vue/compiler-core'
import {
generate,
baseParse,
transform,
TransformOptions
} from '@vue/compiler-core'
import {
transformSrcset,
createSrcsetTransformWithOptions
Expand All @@ -9,8 +14,13 @@ import {
AssetURLOptions,
normalizeOptions
} from '../src/templateTransformAssetUrl'
import { stringifyStatic } from '../../compiler-dom/src/transforms/stringifyStatic'

function compileWithSrcset(template: string, options?: AssetURLOptions) {
function compileWithSrcset(
template: string,
options?: AssetURLOptions,
transformOptions?: TransformOptions
) {
const ast = baseParse(template)
const srcsetTransform = options
? createSrcsetTransformWithOptions(normalizeOptions(options))
Expand All @@ -19,7 +29,8 @@ function compileWithSrcset(template: string, options?: AssetURLOptions) {
nodeTransforms: [srcsetTransform, transformElement],
directiveTransforms: {
bind: transformBind
}
},
...transformOptions
})
return generate(ast, { mode: 'module' })
}
Expand Down Expand Up @@ -59,4 +70,19 @@ describe('compiler sfc: transform srcset', () => {
}).code
).toMatchSnapshot()
})

test('transform srcset w/ stringify', () => {
const code = compileWithSrcset(
`<div>${src}</div>`,
{
includeAbsolute: true
},
{
hoistStatic: true,
transformHoist: stringifyStatic
}
).code
expect(code).toMatch(`_createStaticVNode`)
expect(code).toMatchSnapshot()
})
})
13 changes: 9 additions & 4 deletions packages/compiler-sfc/src/templateTransformAssetUrl.ts
Expand Up @@ -162,7 +162,12 @@ function getImportsExpressionExp(
exp = context.imports[existingIndex].exp as SimpleExpressionNode
} else {
name = `_imports_${context.imports.length}`
exp = createSimpleExpression(name, false, loc, ConstantTypes.CAN_HOIST)
exp = createSimpleExpression(
name,
false,
loc,
ConstantTypes.CAN_STRINGIFY
)
context.imports.push({ exp, path })
}

Expand All @@ -184,13 +189,13 @@ function getImportsExpressionExp(
`_hoisted_${existingHoistIndex + 1}`,
false,
loc,
ConstantTypes.CAN_HOIST
ConstantTypes.CAN_STRINGIFY
)
}
return context.hoist(
createSimpleExpression(hashExp, false, loc, ConstantTypes.CAN_HOIST)
createSimpleExpression(hashExp, false, loc, ConstantTypes.CAN_STRINGIFY)
)
} else {
return createSimpleExpression(`''`, false, loc, ConstantTypes.CAN_HOIST)
return createSimpleExpression(`''`, false, loc, ConstantTypes.CAN_STRINGIFY)
}
}
8 changes: 4 additions & 4 deletions packages/compiler-sfc/src/templateTransformSrcset.ts
Expand Up @@ -113,14 +113,14 @@ export const transformSrcset: NodeTransform = (
`_imports_${existingImportsIndex}`,
false,
attr.loc,
ConstantTypes.CAN_HOIST
ConstantTypes.CAN_STRINGIFY
)
} else {
exp = createSimpleExpression(
`_imports_${context.imports.length}`,
false,
attr.loc,
ConstantTypes.CAN_HOIST
ConstantTypes.CAN_STRINGIFY
)
context.imports.push({ exp, path })
}
Expand All @@ -131,7 +131,7 @@ export const transformSrcset: NodeTransform = (
`"${url}"`,
false,
attr.loc,
ConstantTypes.CAN_HOIST
ConstantTypes.CAN_STRINGIFY
)
compoundExpression.children.push(exp)
}
Expand All @@ -146,7 +146,7 @@ export const transformSrcset: NodeTransform = (
})

const hoisted = context.hoist(compoundExpression)
hoisted.constType = ConstantTypes.CAN_HOIST
hoisted.constType = ConstantTypes.CAN_STRINGIFY

node.props[index] = {
type: NodeTypes.DIRECTIVE,
Expand Down

0 comments on commit 87c73e9

Please sign in to comment.