96 lines
2.7 KiB
TypeScript
96 lines
2.7 KiB
TypeScript
import { DirectiveTransform } from '../transform'
|
|
import {
|
|
createSimpleExpression,
|
|
createObjectProperty,
|
|
createCompoundExpression,
|
|
NodeTypes,
|
|
Property,
|
|
ElementTypes
|
|
} from '../ast'
|
|
import { createCompilerError, ErrorCodes } from '../errors'
|
|
import { isMemberExpression, isSimpleIdentifier, hasScopeRef } from '../utils'
|
|
|
|
export const transformModel: DirectiveTransform = (dir, node, context) => {
|
|
const { exp, arg } = dir
|
|
if (!exp) {
|
|
context.onError(
|
|
createCompilerError(ErrorCodes.X_V_MODEL_NO_EXPRESSION, dir.loc)
|
|
)
|
|
return createTransformProps()
|
|
}
|
|
|
|
const expString =
|
|
exp.type === NodeTypes.SIMPLE_EXPRESSION ? exp.content : exp.loc.source
|
|
if (!isMemberExpression(expString)) {
|
|
context.onError(
|
|
createCompilerError(ErrorCodes.X_V_MODEL_MALFORMED_EXPRESSION, exp.loc)
|
|
)
|
|
return createTransformProps()
|
|
}
|
|
|
|
if (
|
|
!__BROWSER__ &&
|
|
context.prefixIdentifiers &&
|
|
isSimpleIdentifier(expString) &&
|
|
context.identifiers[expString]
|
|
) {
|
|
context.onError(
|
|
createCompilerError(ErrorCodes.X_V_MODEL_ON_SCOPE_VARIABLE, exp.loc)
|
|
)
|
|
return createTransformProps()
|
|
}
|
|
|
|
const propName = arg ? arg : createSimpleExpression('modelValue', true)
|
|
const eventName = arg
|
|
? arg.type === NodeTypes.SIMPLE_EXPRESSION && arg.isStatic
|
|
? createSimpleExpression('onUpdate:' + arg.content, true)
|
|
: createCompoundExpression([
|
|
createSimpleExpression('onUpdate:', true),
|
|
'+',
|
|
...(arg.type === NodeTypes.SIMPLE_EXPRESSION ? [arg] : arg.children)
|
|
])
|
|
: createSimpleExpression('onUpdate:modelValue', true)
|
|
|
|
const props = [
|
|
// modelValue: foo
|
|
createObjectProperty(propName, dir.exp!),
|
|
// "onUpdate:modelValue": $event => (foo = $event)
|
|
createObjectProperty(
|
|
eventName,
|
|
createCompoundExpression([
|
|
`$event => (`,
|
|
...(exp.type === NodeTypes.SIMPLE_EXPRESSION ? [exp] : exp.children),
|
|
` = $event)`
|
|
])
|
|
)
|
|
]
|
|
|
|
// cache v-model handler if applicable (when it doesn't refer any scope vars)
|
|
if (
|
|
!__BROWSER__ &&
|
|
context.prefixIdentifiers &&
|
|
!hasScopeRef(exp, context.identifiers)
|
|
) {
|
|
props[1].value = context.cache(props[1].value)
|
|
}
|
|
|
|
// modelModifiers: { foo: true, "bar-baz": true }
|
|
if (dir.modifiers.length && node.tagType === ElementTypes.COMPONENT) {
|
|
const modifiers = dir.modifiers
|
|
.map(m => (isSimpleIdentifier(m) ? m : JSON.stringify(m)) + `: true`)
|
|
.join(`, `)
|
|
props.push(
|
|
createObjectProperty(
|
|
`modelModifiers`,
|
|
createSimpleExpression(`{ ${modifiers} }`, false, dir.loc, true)
|
|
)
|
|
)
|
|
}
|
|
|
|
return createTransformProps(props)
|
|
}
|
|
|
|
function createTransformProps(props: Property[] = []) {
|
|
return { props, needRuntime: false }
|
|
}
|