UNPKG

@94ai/unplugin-vue-define-options

Version:

Add defineOptions macro for Vue <script setup>.

107 lines (104 loc) 3.17 kB
// src/core/utils.ts import { DEFINE_OPTIONS, isCallOf } from "@vue-macros/common"; function filterMacro(stmts) { return stmts.map((raw) => { let node = raw; if (raw.type === "ExpressionStatement") node = raw.expression; return isCallOf(node, DEFINE_OPTIONS) ? node : void 0; }).filter((node) => !!node); } function hasPropsOrEmits(node) { return node.properties.some( (prop) => (prop.type === "ObjectProperty" || prop.type === "ObjectMethod") && prop.key.type === "Identifier" && (prop.key.name === "props" || prop.key.name === "emits" || prop.key.name === "expose" || prop.key.name === "slots") ); } // src/core/index.ts import { DEFINE_OPTIONS as DEFINE_OPTIONS2, MagicString, addNormalScript, checkInvalidScopeReference, generateTransform, importHelperFn, parseSFC } from "@vue-macros/common"; import { walkAST } from "ast-walker-scope"; function transformDefineOptions(code, id) { if (!code.includes(DEFINE_OPTIONS2)) return; const sfc = parseSFC(code, id); if (!sfc.scriptSetup) return; const { scriptSetup, getSetupAst, getScriptAst } = sfc; const setupOffset = scriptSetup.loc.start.offset; const setupAst = getSetupAst(); const nodes = filterMacro(setupAst.body); if (nodes.length === 0) { return; } else if (nodes.length > 1) throw new SyntaxError(`duplicate ${DEFINE_OPTIONS2}() call`); const scriptAst = getScriptAst(); if (scriptAst) checkDefaultExport(scriptAst.body); const setupBindings = getRootScopeIds(setupAst); const s = new MagicString(code); const [node] = nodes; const [arg] = node.arguments; if (arg) { const normalScript = addNormalScript(sfc, s); const scriptOffset = normalScript.start(); s.appendLeft( scriptOffset, ` export default /*#__PURE__*/ ${importHelperFn( s, scriptOffset, "defineComponent" )}(` ); if (arg.type === "ObjectExpression" && hasPropsOrEmits(arg)) throw new SyntaxError( `${DEFINE_OPTIONS2}() please use defineProps, defineEmits, defineExpose, or defineSlots instead.` ); checkInvalidScopeReference(arg, DEFINE_OPTIONS2, setupBindings); s.moveNode(arg, scriptOffset, { offset: setupOffset }); s.remove(setupOffset + node.start, setupOffset + arg.start); s.remove(setupOffset + arg.end, setupOffset + node.end); s.appendRight(scriptOffset, ");"); normalScript.end(); } else { s.removeNode(node, { offset: setupOffset }); } return generateTransform(s, id); } function checkDefaultExport(stmts) { const hasDefaultExport = stmts.some( (node) => node.type === "ExportDefaultDeclaration" ); if (hasDefaultExport) throw new SyntaxError( `${DEFINE_OPTIONS2} cannot be used with default export within <script>.` ); } function getRootScopeIds(program) { let ids = []; walkAST(program, { enter(node) { if (node.type === "BlockStatement") { this.skip(); } }, leave(node) { if (node.type !== "Program") return; ids = Object.keys(this.scope); } }); return ids; } export { filterMacro, hasPropsOrEmits, transformDefineOptions };