refactor: improve type resolve error output

packages/compiler-sfc/__tests__/compileScript/resolveType.spec.ts

@@ -327,16 +327,34 @@ describe('resolveType', () => {
   })
 
   describe('errors', () => {
-    test('error on computed keys', () => {
+    test('failed type reference', () => {
-      expect(() => resolve(`type Target = { [Foo]: string }`)).toThrow(
+      expect(() => resolve(`type Target = X`)).toThrow(
-        `computed keys are not supported in types referenced by SFC macros`
+        `Unresolvable type reference`
       )
     })
+
+    test('unsupported computed keys', () => {
+      expect(() => resolve(`type Target = { [Foo]: string }`)).toThrow(
+        `Unsupported computed key in type referenced by a macro`
+      )
+    })
+
+    test('unsupported index type', () => {
+      expect(() => resolve(`type Target = X[K]`)).toThrow(
+        `Unsupported index type`
+      )
+    })
+
+    test('failed improt source resolve', () => {
+      expect(() =>
+        resolve(`import { X } from './foo'; type Target = X`)
+      ).toThrow(`Failed to resolve import source "./foo" for type X`)
+    })
   })
 })
 
 function resolve(code: string, files: Record<string, string> = {}) {
-  const { descriptor } = parse(`<script setup lang="ts">${code}</script>`, {
+  const { descriptor } = parse(`<script setup lang="ts">\n${code}\n</script>`, {
     filename: 'Test.vue'
   })
   const ctx = new ScriptCompileContext(descriptor, {

packages/compiler-sfc/src/script/context.ts

@@ -126,13 +126,14 @@ export class ScriptCompileContext {
   }
 
   error(msg: string, node: Node & WithScope, scope?: TypeScope): never {
+    const offset = scope ? scope.offset || 0 : this.startOffset!
     throw new Error(
       `[@vue/compiler-sfc] ${msg}\n\n${
-        this.descriptor.filename
+        (scope || this.descriptor).filename
       }\n${generateCodeFrame(
-        this.descriptor.source,
+        (scope || this.descriptor).source,
-        node.start! + this.startOffset!,
+        node.start! + offset,
-        node.end! + this.startOffset!
+        node.end! + offset
       )}`
     )
   }

packages/compiler-sfc/src/script/resolveType.ts

@@ -33,6 +33,7 @@ type Import = Pick<ImportBinding, 'source' | 'imported'>
 export interface TypeScope {
   filename: string
   source: string
+  offset: number
   imports: Record<string, Import>
   types: Record<
     string,
@@ -128,7 +129,8 @@ function innerResolveTypeElements(
       } else {
         ctx.error(
           `Unsupported index type: ${node.indexType.type}`,
-          node.indexType
+          node.indexType,
+          scope
         )
       }
     }
@@ -144,16 +146,17 @@ function innerResolveTypeElements(
           // @ts-ignore
           SupportedBuiltinsSet.has(typeName)
         ) {
-          return resolveBuiltin(ctx, node, typeName as any)
+          return resolveBuiltin(ctx, node, typeName as any, scope)
         }
         ctx.error(
-          `Failed to resolve type reference, or unsupported built-in utlility type.`,
+          `Unresolvable type reference or unsupported built-in utlility type`,
-          node
+          node,
+          scope
         )
       }
     }
   }
-  ctx.error(`Unresolvable type in SFC macro: ${node.type}`, node)
+  ctx.error(`Unresolvable type: ${node.type}`, node, scope)
 }
 
 function typeElementsToMap(
@@ -169,13 +172,14 @@ function typeElementsToMap(
       if (name && !e.computed) {
         res.props[name] = e as ResolvedElements['props'][string]
       } else if (e.key.type === 'TemplateLiteral') {
-        for (const key of resolveTemplateKeys(ctx, e.key)) {
+        for (const key of resolveTemplateKeys(ctx, e.key, scope)) {
           res.props[key] = e as ResolvedElements['props'][string]
         }
       } else {
         ctx.error(
-          `computed keys are not supported in types referenced by SFC macros.`,
+          `Unsupported computed key in type referenced by a macro`,
-          e
+          e.key,
+          scope
         )
       }
     } else if (e.type === 'TSCallSignatureDeclaration') {
@@ -256,10 +260,7 @@ function resolveMappedType(
   scope: TypeScope
 ): ResolvedElements {
   const res: ResolvedElements = { props: {} }
-  if (!node.typeParameter.constraint) {
+  const keys = resolveStringType(ctx, node.typeParameter.constraint!, scope)
-    ctx.error(`mapped type used in macros must have a finite constraint.`, node)
-  }
-  const keys = resolveStringType(ctx, node.typeParameter.constraint)
   for (const key of keys) {
     res.props[key] = createProperty(
       {
@@ -273,25 +274,29 @@ function resolveMappedType(
   return res
 }
 
-function resolveStringType(ctx: ScriptCompileContext, node: Node): string[] {
+function resolveStringType(
+  ctx: ScriptCompileContext,
+  node: Node,
+  scope: TypeScope
+): string[] {
   switch (node.type) {
     case 'StringLiteral':
       return [node.value]
     case 'TSLiteralType':
-      return resolveStringType(ctx, node.literal)
+      return resolveStringType(ctx, node.literal, scope)
     case 'TSUnionType':
-      return node.types.map(t => resolveStringType(ctx, t)).flat()
+      return node.types.map(t => resolveStringType(ctx, t, scope)).flat()
     case 'TemplateLiteral': {
-      return resolveTemplateKeys(ctx, node)
+      return resolveTemplateKeys(ctx, node, scope)
     }
     case 'TSTypeReference': {
-      const resolved = resolveTypeReference(ctx, node)
+      const resolved = resolveTypeReference(ctx, node, scope)
       if (resolved) {
-        return resolveStringType(ctx, resolved)
+        return resolveStringType(ctx, resolved, scope)
       }
       if (node.typeName.type === 'Identifier') {
         const getParam = (index = 0) =>
-          resolveStringType(ctx, node.typeParameters!.params[index])
+          resolveStringType(ctx, node.typeParameters!.params[index], scope)
         switch (node.typeName.name) {
           case 'Extract':
             return getParam(1)
@@ -308,17 +313,18 @@ function resolveStringType(ctx: ScriptCompileContext, node: Node): string[] {
           case 'Uncapitalize':
             return getParam().map(s => s[0].toLowerCase() + s.slice(1))
           default:
-            ctx.error('Failed to resolve type reference', node)
+            ctx.error('Failed to resolve type reference', node, scope)
         }
       }
     }
   }
-  ctx.error('Failed to resolve string type into finite keys', node)
+  ctx.error('Failed to resolve string type into finite keys', node, scope)
 }
 
 function resolveTemplateKeys(
   ctx: ScriptCompileContext,
-  node: TemplateLiteral
+  node: TemplateLiteral,
+  scope: TypeScope
 ): string[] {
   if (!node.expressions.length) {
     return [node.quasis[0].value.raw]
@@ -328,12 +334,16 @@ function resolveTemplateKeys(
   const e = node.expressions[0]
   const q = node.quasis[0]
   const leading = q ? q.value.raw : ``
-  const resolved = resolveStringType(ctx, e)
+  const resolved = resolveStringType(ctx, e, scope)
-  const restResolved = resolveTemplateKeys(ctx, {
+  const restResolved = resolveTemplateKeys(
-    ...node,
+    ctx,
-    expressions: node.expressions.slice(1),
+    {
-    quasis: q ? node.quasis.slice(1) : node.quasis
+      ...node,
-  })
+      expressions: node.expressions.slice(1),
+      quasis: q ? node.quasis.slice(1) : node.quasis
+    },
+    scope
+  )
 
   for (const r of resolved) {
     for (const rr of restResolved) {
@@ -357,7 +367,8 @@ type GetSetType<T> = T extends Set<infer V> ? V : never
 function resolveBuiltin(
   ctx: ScriptCompileContext,
   node: TSTypeReference | TSExpressionWithTypeArguments,
-  name: GetSetType<typeof SupportedBuiltinsSet>
+  name: GetSetType<typeof SupportedBuiltinsSet>,
+  scope: TypeScope
 ): ResolvedElements {
   const t = resolveTypeElements(ctx, node.typeParameters!.params[0])
   switch (name) {
@@ -366,7 +377,11 @@ function resolveBuiltin(
     case 'Readonly':
       return t
     case 'Pick': {
-      const picked = resolveStringType(ctx, node.typeParameters!.params[1])
+      const picked = resolveStringType(
+        ctx,
+        node.typeParameters!.params[1],
+        scope
+      )
       const res: ResolvedElements = { props: {}, calls: t.calls }
       for (const key of picked) {
         res.props[key] = t.props[key]
@@ -374,7 +389,11 @@ function resolveBuiltin(
       return res
     }
     case 'Omit':
-      const omitted = resolveStringType(ctx, node.typeParameters!.params[1])
+      const omitted = resolveStringType(
+        ctx,
+        node.typeParameters!.params[1],
+        scope
+      )
       const res: ResolvedElements = { props: {}, calls: t.calls }
       for (const key in t.props) {
         if (!omitted.includes(key)) {
@@ -415,7 +434,7 @@ function innerResolveTypeReference(
 ): Node | undefined {
   if (typeof name === 'string') {
     if (scope.imports[name]) {
-      return resolveTypeFromImport(ctx, scope, scope.imports[name], node)
+      return resolveTypeFromImport(ctx, node, name, scope)
     } else {
       const types = onlyExported ? scope.exportedTypes : scope.types
       return types[name]
@@ -462,19 +481,21 @@ function qualifiedNameToPath(node: Identifier | TSQualifiedName): string[] {
 
 function resolveTypeFromImport(
   ctx: ScriptCompileContext,
-  scope: TypeScope,
+  node: TSTypeReference | TSExpressionWithTypeArguments,
-  { source, imported }: Import,
+  name: string,
-  node: TSTypeReference | TSExpressionWithTypeArguments
+  scope: TypeScope
 ): Node | undefined {
   const fs = ctx.options.fs
   if (!fs) {
     ctx.error(
       `fs options for compileScript are required for resolving imported types`,
-      node
+      node,
+      scope
     )
   }
   // TODO (hmr) register dependency file on ctx
   const containingFile = scope.filename
+  const { source, imported } = scope.imports[name]
   if (source.startsWith('.')) {
     // relative import - fast path
     const filename = path.join(containingFile, '..', source)
@@ -488,7 +509,13 @@ function resolveTypeFromImport(
         true
       )
     } else {
-      ctx.error(`Failed to resolve import source for type`, node)
+      ctx.error(
+        `Failed to resolve import source ${JSON.stringify(
+          source
+        )} for type ${name}`,
+        node,
+        scope
+      )
     }
   } else {
     // TODO module or aliased import - use full TS resolution
@@ -519,10 +546,11 @@ function fileToScope(
 ): TypeScope {
   // TODO cache
   const source = fs.readFile(filename)
-  const body = parseFile(ctx, filename, source)
+  const [body, offset] = parseFile(ctx, filename, source)
   const scope: TypeScope = {
     filename,
     source,
+    offset,
     types: Object.create(null),
     exportedTypes: Object.create(null),
     imports: recordImports(body)
@@ -535,10 +563,12 @@ function parseFile(
   ctx: ScriptCompileContext,
   filename: string,
   content: string
-): Statement[] {
+): [Statement[], number] {
+  let body: Statement[] = []
+  let offset = 0
   const ext = path.extname(filename)
   if (ext === '.ts' || ext === '.tsx') {
-    return babelParse(content, {
+    body = babelParse(content, {
       plugins: resolveParserPlugins(
         ext.slice(1),
         ctx.options.babelParserPlugins
@@ -551,12 +581,13 @@ function parseFile(
     } = parse(content)
     const scriptContent = (script?.content || '') + (scriptSetup?.content || '')
     const lang = script?.lang || scriptSetup?.lang
-    return babelParse(scriptContent, {
+    body = babelParse(scriptContent, {
       plugins: resolveParserPlugins(lang!, ctx.options.babelParserPlugins),
       sourceType: 'module'
     }).program.body
+    offset = scriptSetup ? scriptSetup.loc.start.offset : 0
   }
-  return []
+  return [body, offset]
 }
 
 function ctxToScope(ctx: ScriptCompileContext): TypeScope {
@@ -567,6 +598,7 @@ function ctxToScope(ctx: ScriptCompileContext): TypeScope {
   const scope: TypeScope = {
     filename: ctx.descriptor.filename,
     source: ctx.descriptor.source,
+    offset: ctx.startOffset!,
     imports: Object.create(ctx.userImports),
     types: Object.create(null),
     exportedTypes: Object.create(null)