feat: support resolving jsxImportSource by referrer

Author: dsherret

internal/api/server.go

@@ -68,6 +68,7 @@ const (
 	CallbackGetAccessibleEntries
 	CallbackReadFile
 	CallbackRealpath
+	CallbackResolveJsxImportSource
 	CallbackResolveModuleName
 	CallbackResolveTypeReferenceDirective
 	CallbackGetPackageJsonScopeIfApplicable
@@ -290,6 +291,24 @@ func (r *resolverWrapper) GetPackageScopeForPath(directory string) *packagejson.
 	return r.inner.GetPackageScopeForPath(directory)
 }
 
+// ResolveJsxImportSource implements module.ResolverInterface.
+func (r *resolverWrapper) ResolveJsxImportSource(referrerPath string) string {
+	if r.server.CallbackEnabled(CallbackResolveJsxImportSource) {
+		result, err := r.server.call("resolveJsxImportSource", referrerPath)
+		if err != nil {
+			panic(err)
+		}
+		if len(result) > 0 {
+			var res string
+			if err := json.Unmarshal(result, &res); err != nil {
+				panic(err)
+			}
+			return res
+		}
+	}
+	return r.inner.ResolveJsxImportSource(referrerPath)
+}
+
 // ResolveModuleName implements module.ResolverInterface.
 func (r *resolverWrapper) ResolveModuleName(moduleName string, containingFile string, importAttributeType *string, resolutionMode core.ResolutionMode, redirectedReference module.ResolvedProjectReference) (*module.ResolvedModule, []module.DiagAndArgs) {
 	if r.server.CallbackEnabled(CallbackResolveModuleName) {
@@ -534,6 +553,8 @@ func (s *Server) enableCallback(callback string) error {
 		s.enabledCallbacks |= CallbackReadFile
 	case "realpath":
 		s.enabledCallbacks |= CallbackRealpath
+	case "resolveJsxImportSource":
+		s.enabledCallbacks |= CallbackResolveJsxImportSource
 	case "resolveModuleName":
 		s.enabledCallbacks |= CallbackResolveModuleName
 	case "resolveTypeReferenceDirective":

internal/ast/utilities.go

@@ -2657,12 +2657,19 @@ func IsRequireVariableStatement(node *Node) bool {
 	return false
 }
 
-func GetJSXImplicitImportBase(compilerOptions *core.CompilerOptions, file *SourceFile) string {
+func GetJSXImplicitImportBase(compilerOptions *core.CompilerOptions, file *SourceFile, resolveJsxImportSource func(referrer string) string) string {
 	jsxImportSourcePragma := GetPragmaFromSourceFile(file, "jsximportsource")
 	jsxRuntimePragma := GetPragmaFromSourceFile(file, "jsxruntime")
 	if GetPragmaArgument(jsxRuntimePragma, "factory") == "classic" {
 		return ""
 	}
+	// deno: resolve the jsxImportSource based on the referrer
+	if file != nil && resolveJsxImportSource != nil {
+		resolvedJsxImportSource := resolveJsxImportSource(file.FileName())
+		if resolvedJsxImportSource != "" {
+			return resolvedJsxImportSource
+		}
+	}
 	if compilerOptions.Jsx == core.JsxEmitReactJSX ||
 		compilerOptions.Jsx == core.JsxEmitReactJSXDev ||
 		compilerOptions.JsxImportSource != "" ||

internal/compiler/fileloader.go

@@ -530,7 +530,7 @@ func (p *fileLoader) resolveImportsAndModuleAugmentations(t *parseTask) {
 			t.importHelpersImportSpecifier = specifier
 		}
 
-		jsxImport := ast.GetJSXRuntimeImport(ast.GetJSXImplicitImportBase(optionsForFile, file), optionsForFile)
+		jsxImport := ast.GetJSXRuntimeImport(ast.GetJSXImplicitImportBase(optionsForFile, file, p.resolver.ResolveJsxImportSource), optionsForFile)
 		if jsxImport != "" {
 			specifier := p.createSyntheticImport(jsxImport, file)
 			moduleNames = append(moduleNames, specifier)

internal/module/resolver.go

@@ -150,6 +150,7 @@ func GetCompilerOptionsWithRedirect(compilerOptions *core.CompilerOptions, redir
 type ResolverInterface interface {
 	ResolveModuleName(moduleName string, containingFile string, importAttributeType *string, resolutionMode core.ResolutionMode, redirectedReference ResolvedProjectReference) (*ResolvedModule, []DiagAndArgs)
 	ResolveTypeReferenceDirective(typeReferenceDirectiveName string, containingFile string, resolutionMode core.ResolutionMode, redirectedReference ResolvedProjectReference) (*ResolvedTypeReferenceDirective, []DiagAndArgs)
+	ResolveJsxImportSource(referrer string) string
 	GetPackageJsonScopeIfApplicable(path string) *packagejson.InfoCacheEntry
 	GetPackageScopeForPath(directory string) *packagejson.InfoCacheEntry
 	GetImpliedNodeFormatForFile(path string, packageJsonType string) core.ModuleKind
@@ -285,6 +286,10 @@ func (r *Resolver) GetImpliedNodeFormatForFile(path string, packageJsonType stri
 	return ast.GetImpliedNodeFormatForFile(path, packageJsonType)
 }
 
+func (r *Resolver) ResolveJsxImportSource(referrer string) string {
+	return ""
+}
+
 func (r *Resolver) tryResolveFromTypingsLocation(moduleName string, containingDirectory string, originalResult *ResolvedModule, traceBuilder *tracer) *ResolvedModule {
 	if r.typingsLocation == "" ||
 		tspath.IsExternalModuleNameRelative(moduleName) ||

internal/transformers/jsxtransforms/jsx.go

@@ -211,7 +211,7 @@ func (tx *JSXTransformer) visitSourceFile(file *ast.SourceFile) *ast.Node {
 	}
 
 	tx.currentSourceFile = file
-	tx.importSpecifier = ast.GetJSXImplicitImportBase(tx.compilerOptions, file)
+	tx.importSpecifier = ast.GetJSXImplicitImportBase(tx.compilerOptions, file, nil)
 	tx.filenameDeclaration = nil
 	tx.utilizedImplicitRuntimeImports = make(map[string]map[string]*ast.Node)