mirror of https://github.com/vuejs/core.git
84 lines
2.2 KiB
TypeScript
84 lines
2.2 KiB
TypeScript
import {
|
|
transformModel as baseTransform,
|
|
DirectiveTransform,
|
|
ElementTypes,
|
|
findProp,
|
|
NodeTypes
|
|
} from '@vue/compiler-core'
|
|
import { createDOMCompilerError, DOMErrorCodes } from '../errors'
|
|
import {
|
|
V_MODEL_CHECKBOX,
|
|
V_MODEL_RADIO,
|
|
V_MODEL_SELECT,
|
|
V_MODEL_TEXT,
|
|
V_MODEL_DYNAMIC
|
|
} from '../runtimeHelpers'
|
|
|
|
export const transformModel: DirectiveTransform = (dir, node, context) => {
|
|
const baseResult = baseTransform(dir, node, context)
|
|
// base transform has errors
|
|
if (!baseResult.props.length) {
|
|
return baseResult
|
|
}
|
|
|
|
const { tag, tagType } = node
|
|
if (tagType === ElementTypes.ELEMENT) {
|
|
if (dir.arg) {
|
|
context.onError(
|
|
createDOMCompilerError(
|
|
DOMErrorCodes.X_V_MODEL_ARG_ON_ELEMENT,
|
|
dir.arg.loc
|
|
)
|
|
)
|
|
}
|
|
|
|
if (tag === 'input' || tag === 'textarea' || tag === 'select') {
|
|
let directiveToUse = V_MODEL_TEXT
|
|
let isInvalidType = false
|
|
if (tag === 'input') {
|
|
const type = findProp(node, `type`)
|
|
if (type) {
|
|
if (type.type === NodeTypes.DIRECTIVE) {
|
|
// :type="foo"
|
|
directiveToUse = V_MODEL_DYNAMIC
|
|
} else if (type.value) {
|
|
switch (type.value.content) {
|
|
case 'radio':
|
|
directiveToUse = V_MODEL_RADIO
|
|
break
|
|
case 'checkbox':
|
|
directiveToUse = V_MODEL_CHECKBOX
|
|
break
|
|
case 'file':
|
|
isInvalidType = true
|
|
context.onError(
|
|
createDOMCompilerError(
|
|
DOMErrorCodes.X_V_MODEL_ON_FILE_INPUT_ELEMENT,
|
|
dir.loc
|
|
)
|
|
)
|
|
break
|
|
}
|
|
}
|
|
}
|
|
} else if (tag === 'select') {
|
|
directiveToUse = V_MODEL_SELECT
|
|
}
|
|
// inject runtime directive
|
|
// by returning the helper symbol via needRuntime
|
|
// the import will replaced a resolveDirective call.
|
|
if (!isInvalidType) {
|
|
baseResult.needRuntime = context.helper(directiveToUse)
|
|
}
|
|
} else {
|
|
context.onError(
|
|
createDOMCompilerError(
|
|
DOMErrorCodes.X_V_MODEL_ON_INVALID_ELEMENT,
|
|
dir.loc
|
|
)
|
|
)
|
|
}
|
|
}
|
|
return baseResult
|
|
}
|