fix issues

Author: eryue0220

server/utils/import-resolver.ts

@@ -103,6 +103,37 @@ function getExtensionPriority(sourceFile: string): string[][] {
   return [[], ['.ts', '.js'], ['.d.ts'], ['.json']]
 }
 
+/**
+ * Resolve an alias specifier to the directory path within a file path.
+ * Supports #, ~, and @ prefixes (e.g. #app, ~/app, @/app).
+ * The alias must match a path segment exactly (no partial matches).
+ */
+export function resolveAliasToDir(aliasSpec: string, filePath?: string | null): string | null {
+  if (
+    (!aliasSpec.startsWith('#') && !aliasSpec.startsWith('~') && !aliasSpec.startsWith('@')) ||
+    !filePath
+  ) {
+    return null
+  }
+
+  // Support #app, #/app, ~app, ~/app, @app, @/app
+  const alias = aliasSpec.replace(/^[#~@]\/?/, '')
+  const segments = filePath.split('/')
+
+  let lastMatchIndex = -1
+  for (let i = 0; i < segments.length; i++) {
+    if (segments[i] === alias) {
+      lastMatchIndex = i
+    }
+  }
+
+  if (lastMatchIndex === -1) {
+    return null
+  }
+
+  return segments.slice(0, lastMatchIndex + 1).join('/')
+}
+
 /**
  * Get index file extensions to try for directory imports.
  */
@@ -222,6 +253,60 @@ function normalizeInternalImportTarget(target: InternalImportTarget): string | n
   return null
 }
 
+function guessInternalImportTarget(
+  imports: InternalImportsMap,
+  specifier: string,
+  files: FileSet,
+  currentFile: string,
+): string | null {
+  for (const [key, value] of Object.entries(imports)) {
+    if (specifier.startsWith(key)) {
+      const basePath = resolveAliasToDir(key, normalizeInternalImportTarget(value))
+      if (!basePath) continue
+
+      const suffix = specifier.substring(key.length).trim().replace(/^\//, '')
+      const pathWithoutExt = suffix ? `${basePath}/${suffix}` : basePath
+
+      const toCheckPath = (p: string) => files.has(normalizePath(p)) || files.has(p)
+
+      // Path already has an extension-like suffix on the last segment - return as is if exists
+      const filename = pathWithoutExt.split('/').pop() ?? ''
+      if (filename.includes('.') && !filename.endsWith('.')) {
+        if (toCheckPath(pathWithoutExt)) {
+          return pathWithoutExt.startsWith('./') ? pathWithoutExt : `./${pathWithoutExt}`
+        }
+        return null
+      }
+
+      // Try adding extensions based on currentFile type
+      const extensionGroups = getExtensionPriority(currentFile)
+      for (const extensions of extensionGroups) {
+        if (extensions.length === 0) {
+          if (toCheckPath(pathWithoutExt)) {
+            return pathWithoutExt.startsWith('./') ? pathWithoutExt : `./${pathWithoutExt}`
+          }
+        } else {
+          for (const ext of extensions) {
+            const pathWithExt = pathWithoutExt + ext
+            if (toCheckPath(pathWithExt)) {
+              return pathWithExt.startsWith('./') ? pathWithExt : `./${pathWithExt}`
+            }
+          }
+        }
+      }
+
+      // Try as directory with index file
+      for (const indexFile of getIndexExtensions(currentFile)) {
+        const indexPath = `${pathWithoutExt}/${indexFile}`
+        if (toCheckPath(indexPath)) {
+          return indexPath.startsWith('./') ? indexPath : `./${indexPath}`
+        }
+      }
+    }
+  }
+  return null
+}
+
 /**
  * import ... from '#components/Button.vue'
  * import ... from '#/components/Button.vue'
@@ -230,6 +315,7 @@ function normalizeInternalImportTarget(target: InternalImportTarget): string | n
  */
 export function resolveInternalImport(
   specifier: string,
+  currentFile: string,
   imports: InternalImportsMap | undefined,
   files: FileSet,
 ): ResolvedImport | null {
@@ -239,8 +325,12 @@ export function resolveInternalImport(
     return null
   }
 
-  const target = normalizeInternalImportTarget(imports[cleanSpecifier])
-  console.log('resolved internal import', imports, cleanSpecifier, target)
+  const importTarget = normalizeInternalImportTarget(imports[cleanSpecifier])
+  const target =
+    importTarget != null
+      ? importTarget
+      : guessInternalImportTarget(imports, cleanSpecifier, files, currentFile)
+
   if (!target || !target.startsWith('./')) {
     return null
   }
@@ -265,8 +355,8 @@ export function createImportResolver(
 ): (specifier: string) => string | null {
   return (specifier: string) => {
     const relativeResolved = resolveRelativeImport(specifier, currentFile, files)
-    const internalResolved = resolveInternalImport(specifier, internalImports, files)
-    const resolved = relativeResolved ?? internalResolved
+    const internalResolved = resolveInternalImport(specifier, currentFile, internalImports, files)
+    const resolved = relativeResolved != null ? relativeResolved : internalResolved
 
     if (resolved) {
       return `/package-code/${packageName}/v/${version}/${resolved.path}`

test/unit/server/utils/import-resolver.spec.ts

@@ -197,6 +197,7 @@ describe('resolveInternalSpecifier', () => {
 
     const resolved = resolveInternalImport(
       '#app/nuxt',
+      'dist/index.js',
       {
         '#app/nuxt': './dist/app/nuxt.js',
       },
@@ -211,6 +212,7 @@ describe('resolveInternalSpecifier', () => {
 
     const resolved = resolveInternalImport(
       '#app/nuxt',
+      'dist/index.js',
       {
         '#app/nuxt': { import: './dist/app/nuxt.js' },
       },
@@ -225,6 +227,7 @@ describe('resolveInternalSpecifier', () => {
 
     const resolved = resolveInternalImport(
       '#app/nuxt',
+      'dist/index.js',
       {
         '#app/nuxt': './dist/app/nuxt.js',
       },
@@ -234,17 +237,18 @@ describe('resolveInternalSpecifier', () => {
     expect(resolved).toBeNull()
   })
 
-  it('returns null for unknown internal specifiers', () => {
+  it('resolves prefix matches with extension resolution via guessInternalImportTarget', () => {
     const files = new Set<string>(['dist/app/nuxt.js'])
 
     const resolved = resolveInternalImport(
       '#app/nuxt',
+      'dist/index.js',
       {
         '#app': './dist/app/index.js',
       },
       files,
     )
 
-    expect(resolved).toBeNull()
+    expect(resolved?.path).toBe('dist/app/nuxt.js')
   })
 })