@nestia/core 12.0.0-dev.20260601.1 → 12.0.0-dev.20260612.2

package/native/transform/transform.go

@@ -1,403 +1,336 @@
-package transform
-
-import (
-	"bytes"
-	"encoding/json"
-	"flag"
-	"fmt"
-	"os"
-	"path/filepath"
-	"regexp"
-	"sort"
-	"strings"
-
-	"github.com/samchon/nestia/packages/core/native/plugin"
-	"github.com/samchon/ttsc/packages/ttsc/driver"
-	typiaadapter "github.com/samchon/typia/packages/typia/native/adapter"
-)
-
-type transformProjectOutput struct {
-	Diagnostics []transformCompilerDiagnostic `json:"diagnostics,omitempty"`
-	TypeScript  map[string]string             `json:"typescript"`
-}
-
-type transformCompilerDiagnostic struct {
-	File        *string `json:"file"`
-	Category    string  `json:"category"`
-	Code        string  `json:"code"`
-	Line        int     `json:"line,omitempty"`
-	Character   int     `json:"character,omitempty"`
-	MessageText string  `json:"messageText"`
-}
-
-func runTransform(args []string) int {
-	fs := flag.NewFlagSet("transform", flag.ContinueOnError)
-	fs.SetOutput(stderr)
-	file := fs.String("file", "", "absolute or cwd-relative path of the .ts file to transform")
-	tsconfigPath := fs.String("tsconfig", "tsconfig.json", "tsconfig.json owning --file")
-	cwdOverride := fs.String("cwd", "", "override the working directory")
-	out := fs.String("out", "", "write output to PATH")
-	output := fs.String("output", "ts", "transform output kind: ts")
-	pluginsJSON := fs.String("plugins-json", "", "ordered ttsc plugin payload")
-	if err := fs.Parse(args); err != nil {
-		return 2
-	}
-	if *output != "ts" {
-		fmt.Fprintf(stderr, "ttsc-nestia transform: unknown --output value %q\n", *output)
-		return 2
-	}
-	plan, err := plugin.ParsePlan(*pluginsJSON)
-	if err != nil {
-		fmt.Fprintf(stderr, "ttsc-nestia transform: %v\n", err)
-		return 2
-	}
-
-	cwd, ok := resolveCWD("ttsc-nestia transform", *cwdOverride)
-	if !ok {
-		return 2
-	}
-	prog, diags, err := driver.LoadProgram(cwd, *tsconfigPath, driver.LoadProgramOptions{
-		ForceNoEmit: true,
-	})
-	if err != nil {
-		fmt.Fprintf(stderr, "ttsc-nestia transform: %v\n", err)
-		return 2
-	}
-	if len(diags) > 0 {
-		driver.WritePrettyDiagnostics(stderr, diags, cwd)
-		return 2
-	}
-	defer prog.Close()
-
-	if *file == "" {
-		if *out != "" {
-			fmt.Fprintln(stderr, "ttsc-nestia transform: --out requires --file")
-			return 2
-		}
-		return runTransformProject(prog, cwd, *tsconfigPath, plan)
-	}
-
-	absFile := *file
-	if !filepath.IsAbs(absFile) {
-		absFile = filepath.Join(cwd, absFile)
-	}
-	absFile = filepath.ToSlash(absFile)
-	target := prog.SourceFile(absFile)
-	if target == nil {
-		fmt.Fprintf(stderr, "ttsc-nestia transform: source file is not in program: %s\n", absFile)
-		return 2
-	}
-	rewrites, diagnostics := collectTypiaSourceRewrites(
-		prog,
-		cwd,
-		absFile,
-		readTypiaPluginOptions(cwd, *tsconfigPath),
-	)
-	if len(diagnostics) > 0 {
-		WriteTypiaTransformDiagnostics(stderr, diagnostics, cwd)
-		return 3
-	}
-	coreRewriteMap, coreDiagnostics := collectNestiaCoreSourceRewriteMap(prog, plan, absFile)
-	if len(coreDiagnostics) > 0 {
-		WriteTypiaTransformDiagnostics(stderr, coreDiagnostics, cwd)
-		return 3
-	}
-	rewrites = append(rewrites, coreRewriteMap[filepath.ToSlash(absFile)]...)
-	contributorRewriteMap, contributorDiagnostics := collectContributorSourceRewriteMap(prog, plan, absFile)
-	if len(contributorDiagnostics) > 0 {
-		WriteTypiaTransformDiagnostics(stderr, contributorDiagnostics, cwd)
-		return 3
-	}
-	rewrites = append(rewrites, contributorRewriteMap[filepath.ToSlash(absFile)]...)
-	source, ok := SourceFileText(target)
-	if !ok {
-		fmt.Fprintf(stderr, "ttsc-nestia transform: source text is unavailable for %s\n", absFile)
-		return 3
-	}
-	source, err = ApplySourceRewrites(source, rewrites)
-	if err != nil {
-		fmt.Fprintf(stderr, "ttsc-nestia transform: source rewrite: %v\n", err)
-		return 3
-	}
-	source = cleanupTypeScriptTransformText(source)
-	if *out == "" {
-		if _, err := bytes.NewBufferString(source).WriteTo(stdout); err != nil {
-			fmt.Fprintf(stderr, "ttsc-nestia transform: write stdout: %v\n", err)
-			return 3
-		}
-		return 0
-	}
-	if dir := filepath.Dir(*out); dir != "" {
-		if err := os.MkdirAll(dir, 0o755); err != nil {
-			fmt.Fprintf(stderr, "ttsc-nestia transform: mkdir: %v\n", err)
-			return 3
-		}
-	}
-	if err := os.WriteFile(*out, []byte(source), 0o644); err != nil {
-		fmt.Fprintf(stderr, "ttsc-nestia transform: write %s: %v\n", *out, err)
-		return 3
-	}
-	return 0
-}
-
-func runTransformProject(prog *driver.Program, cwd string, tsconfigPath string, plan plugin.Plan) int {
-	rewrites, diags := collectTypiaSourceRewriteMap(
-		prog,
-		readTypiaPluginOptions(cwd, tsconfigPath),
-	)
-	coreRewriteMap, coreDiags := collectNestiaCoreSourceRewriteMap(prog, plan, "")
-	for file, entries := range coreRewriteMap {
-		rewrites[file] = append(rewrites[file], entries...)
-	}
-	contributorRewriteMap, contributorDiags := collectContributorSourceRewriteMap(prog, plan, "")
-	for file, entries := range contributorRewriteMap {
-		rewrites[file] = append(rewrites[file], entries...)
-	}
-	diags = append(diags, coreDiags...)
-	diags = append(diags, contributorDiags...)
-	output := transformProjectOutput{
-		Diagnostics: make([]transformCompilerDiagnostic, 0, len(diags)),
-		TypeScript:  map[string]string{},
-	}
-	for _, diag := range diags {
-		output.Diagnostics = append(output.Diagnostics, transformDiagnosticToCompilerDiagnostic(diag))
-	}
-	for _, file := range prog.SourceFiles() {
-		filename := filepath.ToSlash(file.FileName())
-		source, ok := SourceFileText(file)
-		if !ok {
-			output.Diagnostics = append(
-				output.Diagnostics,
-				newTransformCompilerDiagnostic(filename, "nestia.transform", "source text is unavailable"),
-			)
-			continue
-		}
-		transformed, err := ApplySourceRewrites(source, rewrites[filename])
-		if err != nil {
-			output.Diagnostics = append(
-				output.Diagnostics,
-				newTransformCompilerDiagnostic(filename, "typia.transform", err.Error()),
-			)
-			continue
-		}
-		output.TypeScript[sourceFileKey(cwd, filename)] = cleanupTypeScriptTransformText(transformed)
-	}
-	if err := json.NewEncoder(stdout).Encode(output); err != nil {
-		fmt.Fprintf(stderr, "ttsc-nestia transform: encode output: %v\n", err)
-		return 3
-	}
-	if len(output.Diagnostics) > 0 {
-		return 3
-	}
-	return 0
-}
-
-type SourceRewrite struct {
-	start       int
-	end         int
-	replacement string
-}
-
-func collectTypiaSourceRewrites(
-	prog *driver.Program,
-	cwd string,
-	onlyFile string,
-	pluginOptions typiaadapter.PluginOptions,
-) ([]SourceRewrite, []Diagnostic) {
-	sites := collectNestiaTypiaCallSites(prog.SourceFiles(), prog.Checker)
-	rewrites := []SourceRewrite{}
-	diagnostics := []Diagnostic{}
-	for _, site := range sites {
-		if filepath.ToSlash(site.FilePath) != filepath.ToSlash(onlyFile) {
-			continue
-		}
-		if reason := typiaadapter.UnsupportedReason(site); reason != "" {
-			diagnostics = append(diagnostics, NewDiagnostic(site, reason))
-			continue
-		}
-		expr, handled, err := typiaadapter.EmitCallWithOptionsPreservingTypes(prog, site, pluginOptions)
-		if !handled {
-			diagnostics = append(diagnostics, NewDiagnostic(site, "method not covered"))
-			continue
-		}
-		if err != nil {
-			diagnostics = append(diagnostics, NewDiagnostic(site, err.Error()))
-			continue
-		}
-		expr = parenthesizeTypiaReplacement(site, expr)
-		node := site.Call.AsNode()
-		rewrites = append(rewrites, SourceRewrite{
-			start:       node.Pos(),
-			end:         node.End(),
-			replacement: expr,
-		})
-		_ = cwd
-	}
-	return rewrites, diagnostics
-}
-
-func collectTypiaSourceRewriteMap(
-	prog *driver.Program,
-	pluginOptions typiaadapter.PluginOptions,
-) (map[string][]SourceRewrite, []Diagnostic) {
-	sites := collectNestiaTypiaCallSites(prog.SourceFiles(), prog.Checker)
-	rewrites := map[string][]SourceRewrite{}
-	diagnostics := []Diagnostic{}
-	for _, site := range sites {
-		file := filepath.ToSlash(site.FilePath)
-		if reason := typiaadapter.UnsupportedReason(site); reason != "" {
-			diagnostics = append(diagnostics, NewDiagnostic(site, reason))
-			continue
-		}
-		expr, handled, err := typiaadapter.EmitCallWithOptionsPreservingTypes(prog, site, pluginOptions)
-		if !handled {
-			diagnostics = append(diagnostics, NewDiagnostic(site, "method not covered"))
-			continue
-		}
-		if err != nil {
-			diagnostics = append(diagnostics, NewDiagnostic(site, err.Error()))
-			continue
-		}
-		expr = parenthesizeTypiaReplacement(site, expr)
-		node := site.Call.AsNode()
-		rewrites[file] = append(rewrites[file], SourceRewrite{
-			start:       node.Pos(),
-			end:         node.End(),
-			replacement: expr,
-		})
-	}
-	return rewrites, diagnostics
-}
-
-func ApplySourceRewrites(source string, rewrites []SourceRewrite) (string, error) {
-	sort.SliceStable(rewrites, func(i, j int) bool {
-		return rewrites[i].start > rewrites[j].start
-	})
-	for i := 0; i < len(rewrites)-1; i++ {
-		if rewrites[i+1].end > rewrites[i].start {
-			return "", fmt.Errorf("overlapping rewrites: [%d,%d) vs [%d,%d)",
-				rewrites[i+1].start, rewrites[i+1].end, rewrites[i].start, rewrites[i].end)
-		}
-	}
-	output := source
-	for _, rewrite := range rewrites {
-		if rewrite.start < 0 || rewrite.end < rewrite.start || rewrite.end > len(output) {
-			return "", fmt.Errorf("invalid rewrite range [%d,%d)", rewrite.start, rewrite.end)
-		}
-		output = output[:rewrite.start] + rewrite.replacement + output[rewrite.end:]
-	}
-	return output, nil
-}
-
-func SourceFileText(target any) (string, bool) {
-	type sourceText interface {
-		Text() string
-	}
-	file, ok := target.(sourceText)
-	if !ok {
-		return "", false
-	}
-	return file.Text(), true
-}
-
-// These patterns run once per transformed file in cleanupTypeScriptTransformText
-// / normalizeParenthesizedTypeAnnotations. Compiling them at package scope keeps
-// the per-file cost to a match instead of a recompile (the SDK `transform` path
-// runs this for every emitted source file).
-var (
-	cleanupImportTypeBlockPattern   = regexp.MustCompile(`(?m)^import type \{([^{}\n]+)\} from`)
-	cleanupImportTypeLinePattern    = regexp.MustCompile(`^import type \{\s*([^{}\n]+?)\s*\} from`)
-	cleanupImportBlankLinePattern   = regexp.MustCompile(`(?m)(^import [^\n]+;\n)\n+(const |let |var |export )`)
-	cleanupInputIsParenPattern      = regexp.MustCompile(`input is \(([A-Za-z_$][A-Za-z0-9_$.]*)\)`)
-	cleanupCollapseBlankCallPattern = regexp.MustCompile(`\n\n([A-Za-z_$][A-Za-z0-9_$]*\([^;\n]*\);?)`)
-)
-
-func cleanupTypeScriptTransformText(text string) string {
-	text = cleanupTransformedText(text)
-	text = normalizeParenthesizedTypeAnnotations(text)
-	text = cleanupImportTypeBlockPattern.ReplaceAllStringFunc(text, func(line string) string {
-		return cleanupImportTypeLinePattern.ReplaceAllString(line, "import type { $1 } from")
-	})
-	text = cleanupImportBlankLinePattern.ReplaceAllString(text, "$1$2")
-	text = strings.ReplaceAll(text, "=(() =>", "= (() =>")
-	text = strings.ReplaceAll(text, ": (any) =>", ": any =>")
-	text = strings.ReplaceAll(text, ": (boolean) =>", ": boolean =>")
-	text = cleanupInputIsParenPattern.ReplaceAllString(text, "input is $1")
-	text = strings.ReplaceAll(text, "return (success ? ", "return success ? ")
-	text = strings.ReplaceAll(text, "}) as any;", "} as any;")
-	text = strings.ReplaceAll(text, "(() => {\n    const ", "(() => { const ")
-	text = strings.ReplaceAll(text, "(() => {\n    let ", "(() => { let ")
-	text = strings.ReplaceAll(text, "(() => {\n    return ", "(() => { return ")
-	text = strings.ReplaceAll(text, ";\n    const ", "; const ")
-	text = strings.ReplaceAll(text, ";\n    let ", "; let ")
-	text = strings.ReplaceAll(text, ";\n    return ", "; return ")
-	text = strings.ReplaceAll(text, "\n    };\n})()", "\n}; })()")
-	text = strings.ReplaceAll(text, "\n    });\n})()", "\n}); })()")
-	text = strings.ReplaceAll(text, "\n    }); let ", "\n}); let ")
-	text = strings.ReplaceAll(text, ";\n})()", "; })()")
-	text = strings.ReplaceAll(text, "\n        ", "\n    ")
-	text = cleanupCollapseBlankCallPattern.ReplaceAllString(text, "\n$1")
-	trimmed := strings.TrimRight(text, " \t\r\n")
-	if strings.HasSuffix(trimmed, ")") && !strings.HasSuffix(trimmed, ";") {
-		return trimmed + ";\n"
-	}
-	if text != "" && !strings.HasSuffix(text, "\n") {
-		return text + "\n"
-	}
-	return text
-}
-
-var (
-	normalizeParenArrowTypePattern = regexp.MustCompile(`: \(([A-Za-z_$][A-Za-z0-9_$.]*(<[^()\n;{}]*>)?)\)(\s*=>)`)
-	normalizeParenNullishPattern   = regexp.MustCompile(`\| \((null|undefined)\)`)
-)
-
-func normalizeParenthesizedTypeAnnotations(text string) string {
-	text = normalizeParenArrowTypePattern.ReplaceAllString(text, ": $1$3")
-	text = normalizeParenNullishPattern.ReplaceAllString(text, "| $1")
-	return text
-}
-
-func sourceFileKey(cwd string, file string) string {
-	rel, err := filepath.Rel(cwd, filepath.FromSlash(file))
-	if err != nil || rel == ".." || strings.HasPrefix(rel, ".."+string(os.PathSeparator)) {
-		return filepath.ToSlash(file)
-	}
-	return filepath.ToSlash(rel)
-}
-
-func newTransformCompilerDiagnostic(
-	file string,
-	code string,
-	message string,
-) transformCompilerDiagnostic {
-	var ptr *string
-	if file != "" {
-		normalized := filepath.ToSlash(file)
-		ptr = &normalized
-	}
-	return transformCompilerDiagnostic{
-		File:        ptr,
-		Category:    "error",
-		Code:        code,
-		MessageText: message,
-	}
-}
-
-func transformDiagnosticToCompilerDiagnostic(
-	diag Diagnostic,
-) transformCompilerDiagnostic {
-	var ptr *string
-	if diag.File != "" {
-		normalized := filepath.ToSlash(diag.File)
-		ptr = &normalized
-	}
-	return transformCompilerDiagnostic{
-		File:        ptr,
-		Category:    "error",
-		Code:        diag.Code,
-		Line:        diag.Line,
-		Character:   diag.Column,
-		MessageText: diag.Message,
-	}
-}
+package transform
+
+import (
+	"bytes"
+	"encoding/json"
+	"flag"
+	"fmt"
+	"os"
+	"path/filepath"
+	"regexp"
+	"sort"
+	"strings"
+
+	shimast "github.com/microsoft/typescript-go/shim/ast"
+	shimprinter "github.com/microsoft/typescript-go/shim/printer"
+	"github.com/samchon/nestia/packages/core/native/plugin"
+	"github.com/samchon/ttsc/packages/ttsc/driver"
+)
+
+type transformProjectOutput struct {
+	Diagnostics []transformCompilerDiagnostic `json:"diagnostics,omitempty"`
+	TypeScript  map[string]string             `json:"typescript"`
+}
+
+type transformCompilerDiagnostic struct {
+	File        *string `json:"file"`
+	Category    string  `json:"category"`
+	Code        string  `json:"code"`
+	Line        int     `json:"line,omitempty"`
+	Character   int     `json:"character,omitempty"`
+	MessageText string  `json:"messageText"`
+}
+
+func runTransform(args []string) int {
+	fs := flag.NewFlagSet("transform", flag.ContinueOnError)
+	fs.SetOutput(stderr)
+	file := fs.String("file", "", "absolute or cwd-relative path of the .ts file to transform")
+	tsconfigPath := fs.String("tsconfig", "tsconfig.json", "tsconfig.json owning --file")
+	cwdOverride := fs.String("cwd", "", "override the working directory")
+	out := fs.String("out", "", "write output to PATH")
+	output := fs.String("output", "ts", "transform output kind: ts")
+	pluginsJSON := fs.String("plugins-json", "", "ordered ttsc plugin payload")
+	if err := fs.Parse(args); err != nil {
+		return 2
+	}
+	if *output != "ts" {
+		fmt.Fprintf(stderr, "ttsc-nestia transform: unknown --output value %q\n", *output)
+		return 2
+	}
+	plan, err := plugin.ParsePlan(*pluginsJSON)
+	if err != nil {
+		fmt.Fprintf(stderr, "ttsc-nestia transform: %v\n", err)
+		return 2
+	}
+
+	cwd, ok := resolveCWD("ttsc-nestia transform", *cwdOverride)
+	if !ok {
+		return 2
+	}
+	prog, diags, err := driver.LoadProgram(cwd, *tsconfigPath, driver.LoadProgramOptions{
+		ForceEmit: true,
+	})
+	if err != nil {
+		fmt.Fprintf(stderr, "ttsc-nestia transform: %v\n", err)
+		return 2
+	}
+	if len(diags) > 0 {
+		driver.WritePrettyDiagnostics(stderr, diags, cwd)
+		return 2
+	}
+	defer prog.Close()
+
+	// AST-integration source-to-source: typia's, core's, and any linked
+	// contributor's per-file node transformers run inside one shared EmitContext
+	// and the result SourceFile is printed back as TypeScript (no JS script
+	// transformers), mirroring typia's `transform` subcommand. Injected namespace
+	// imports stay as ES imports the caller can type-strip per file, so there is
+	// no text-splice RewriteSet.
+	transformDiags := []Diagnostic{}
+	addDiagnostic := func(diag Diagnostic) {
+		transformDiags = append(transformDiags, diag)
+	}
+	typiaTransform := nestiaTypiaNodeTransform(prog, readTypiaPluginOptions(cwd, *tsconfigPath), addDiagnostic)
+	coreTransform := nestiaCoreNodeTransform(prog, plan, addDiagnostic)
+	contributorTransforms, contributorDiags := collectContributorEmitTransforms(prog, plan)
+	if len(contributorDiags) > 0 {
+		WriteTypiaTransformDiagnostics(stderr, contributorDiags, cwd)
+		return 3
+	}
+	transforms := append([]driver.PluginTransform{typiaTransform, coreTransform}, contributorTransforms...)
+
+	if *file == "" {
+		if *out != "" {
+			fmt.Fprintln(stderr, "ttsc-nestia transform: --out requires --file")
+			return 2
+		}
+		return runTransformProject(prog, cwd, transforms, &transformDiags)
+	}
+
+	absFile := *file
+	if !filepath.IsAbs(absFile) {
+		absFile = filepath.Join(cwd, absFile)
+	}
+	absFile = filepath.ToSlash(absFile)
+	target := prog.SourceFile(absFile)
+	if target == nil {
+		fmt.Fprintf(stderr, "ttsc-nestia transform: source file is not in program: %s\n", absFile)
+		return 2
+	}
+	text := transformFileToTypeScript(prog, transforms, target)
+	if len(transformDiags) > 0 {
+		WriteTypiaTransformDiagnostics(stderr, transformDiags, cwd)
+		return 3
+	}
+	return writeSingleOutput(text, *out)
+}
+
+func runTransformProject(
+	prog *driver.Program,
+	cwd string,
+	transforms []driver.PluginTransform,
+	transformDiags *[]Diagnostic,
+) int {
+	output := transformProjectOutput{
+		Diagnostics: []transformCompilerDiagnostic{},
+		TypeScript:  map[string]string{},
+	}
+	for _, sf := range prog.SourceFiles() {
+		if sf.IsDeclarationFile {
+			continue
+		}
+		key := sourceFileKey(cwd, filepath.ToSlash(sf.FileName()))
+		if filepath.IsAbs(key) || key == ".." || strings.HasPrefix(key, "../") {
+			continue
+		}
+		output.TypeScript[key] = transformFileToTypeScript(prog, transforms, sf)
+	}
+	for _, diag := range *transformDiags {
+		output.Diagnostics = append(output.Diagnostics, transformDiagnosticToCompilerDiagnostic(diag))
+	}
+	if err := json.NewEncoder(stdout).Encode(output); err != nil {
+		fmt.Fprintf(stderr, "ttsc-nestia transform: encode output: %v\n", err)
+		return 3
+	}
+	if len(output.Diagnostics) > 0 {
+		return 3
+	}
+	return 0
+}
+
+// transformFileToTypeScript runs nestia's node transformers on one source file
+// in a fresh EmitContext and prints the result as TypeScript. It deliberately
+// skips the JS script transformers (type-erase, module-transform): the caller
+// wants TS, so the namespace imports the transformers inject stay as ES imports.
+func transformFileToTypeScript(
+	prog *driver.Program,
+	transforms []driver.PluginTransform,
+	sf *shimast.SourceFile,
+) string {
+	options := prog.TSProgram.Options()
+	ec := shimprinter.NewEmitContext()
+	result := sf
+	for _, t := range transforms {
+		if t == nil {
+			continue
+		}
+		if next := t(ec, result); next != nil {
+			result = next
+		}
+	}
+	shimast.SetParentInChildrenUnset(result.AsNode())
+	writer := shimprinter.NewTextWriter(options.NewLine.GetNewLineCharacter(), 0)
+	printer := shimprinter.NewPrinter(shimprinter.PrinterOptions{NewLine: options.NewLine}, shimprinter.PrintHandlers{}, ec)
+	printer.Write(result.AsNode(), result, writer, nil)
+	return writer.String()
+}
+
+func writeSingleOutput(text, outPath string) int {
+	if outPath == "" {
+		if _, err := bytes.NewReader([]byte(text)).WriteTo(stdout); err != nil {
+			fmt.Fprintf(stderr, "ttsc-nestia transform: write stdout: %v\n", err)
+			return 3
+		}
+		return 0
+	}
+	if dir := filepath.Dir(outPath); dir != "" {
+		if err := os.MkdirAll(dir, 0o755); err != nil {
+			fmt.Fprintf(stderr, "ttsc-nestia transform: mkdir: %v\n", err)
+			return 3
+		}
+	}
+	if err := os.WriteFile(outPath, []byte(text), 0o644); err != nil {
+		fmt.Fprintf(stderr, "ttsc-nestia transform: write %s: %v\n", outPath, err)
+		return 3
+	}
+	return 0
+}
+
+type SourceRewrite struct {
+	start       int
+	end         int
+	replacement string
+}
+
+func ApplySourceRewrites(source string, rewrites []SourceRewrite) (string, error) {
+	sort.SliceStable(rewrites, func(i, j int) bool {
+		return rewrites[i].start > rewrites[j].start
+	})
+	for i := 0; i < len(rewrites)-1; i++ {
+		if rewrites[i+1].end > rewrites[i].start {
+			return "", fmt.Errorf("overlapping rewrites: [%d,%d) vs [%d,%d)",
+				rewrites[i+1].start, rewrites[i+1].end, rewrites[i].start, rewrites[i].end)
+		}
+	}
+	output := source
+	for _, rewrite := range rewrites {
+		if rewrite.start < 0 || rewrite.end < rewrite.start || rewrite.end > len(output) {
+			return "", fmt.Errorf("invalid rewrite range [%d,%d)", rewrite.start, rewrite.end)
+		}
+		output = output[:rewrite.start] + rewrite.replacement + output[rewrite.end:]
+	}
+	return output, nil
+}
+
+func SourceFileText(target any) (string, bool) {
+	type sourceText interface {
+		Text() string
+	}
+	file, ok := target.(sourceText)
+	if !ok {
+		return "", false
+	}
+	return file.Text(), true
+}
+
+// These patterns run once per transformed file in cleanupTypeScriptTransformText
+// / normalizeParenthesizedTypeAnnotations. Compiling them at package scope keeps
+// the per-file cost to a match instead of a recompile (the SDK `transform` path
+// runs this for every emitted source file).
+var (
+	cleanupImportTypeBlockPattern   = regexp.MustCompile(`(?m)^import type \{([^{}\n]+)\} from`)
+	cleanupImportTypeLinePattern    = regexp.MustCompile(`^import type \{\s*([^{}\n]+?)\s*\} from`)
+	cleanupImportBlankLinePattern   = regexp.MustCompile(`(?m)(^import [^\n]+;\n)\n+(const |let |var |export )`)
+	cleanupInputIsParenPattern      = regexp.MustCompile(`input is \(([A-Za-z_$][A-Za-z0-9_$.]*)\)`)
+	cleanupCollapseBlankCallPattern = regexp.MustCompile(`\n\n([A-Za-z_$][A-Za-z0-9_$]*\([^;\n]*\);?)`)
+)
+
+func cleanupTypeScriptTransformText(text string) string {
+	text = cleanupTransformedText(text)
+	text = normalizeParenthesizedTypeAnnotations(text)
+	text = cleanupImportTypeBlockPattern.ReplaceAllStringFunc(text, func(line string) string {
+		return cleanupImportTypeLinePattern.ReplaceAllString(line, "import type { $1 } from")
+	})
+	text = cleanupImportBlankLinePattern.ReplaceAllString(text, "$1$2")
+	text = strings.ReplaceAll(text, "=(() =>", "= (() =>")
+	text = strings.ReplaceAll(text, ": (any) =>", ": any =>")
+	text = strings.ReplaceAll(text, ": (boolean) =>", ": boolean =>")
+	text = cleanupInputIsParenPattern.ReplaceAllString(text, "input is $1")
+	text = strings.ReplaceAll(text, "return (success ? ", "return success ? ")
+	text = strings.ReplaceAll(text, "}) as any;", "} as any;")
+	text = strings.ReplaceAll(text, "(() => {\n    const ", "(() => { const ")
+	text = strings.ReplaceAll(text, "(() => {\n    let ", "(() => { let ")
+	text = strings.ReplaceAll(text, "(() => {\n    return ", "(() => { return ")
+	text = strings.ReplaceAll(text, ";\n    const ", "; const ")
+	text = strings.ReplaceAll(text, ";\n    let ", "; let ")
+	text = strings.ReplaceAll(text, ";\n    return ", "; return ")
+	text = strings.ReplaceAll(text, "\n    };\n})()", "\n}; })()")
+	text = strings.ReplaceAll(text, "\n    });\n})()", "\n}); })()")
+	text = strings.ReplaceAll(text, "\n    }); let ", "\n}); let ")
+	text = strings.ReplaceAll(text, ";\n})()", "; })()")
+	text = strings.ReplaceAll(text, "\n        ", "\n    ")
+	text = cleanupCollapseBlankCallPattern.ReplaceAllString(text, "\n$1")
+	trimmed := strings.TrimRight(text, " \t\r\n")
+	if strings.HasSuffix(trimmed, ")") && !strings.HasSuffix(trimmed, ";") {
+		return trimmed + ";\n"
+	}
+	if text != "" && !strings.HasSuffix(text, "\n") {
+		return text + "\n"
+	}
+	return text
+}
+
+var (
+	normalizeParenArrowTypePattern = regexp.MustCompile(`: \(([A-Za-z_$][A-Za-z0-9_$.]*(<[^()\n;{}]*>)?)\)(\s*=>)`)
+	normalizeParenNullishPattern   = regexp.MustCompile(`\| \((null|undefined)\)`)
+)
+
+func normalizeParenthesizedTypeAnnotations(text string) string {
+	text = normalizeParenArrowTypePattern.ReplaceAllString(text, ": $1$3")
+	text = normalizeParenNullishPattern.ReplaceAllString(text, "| $1")
+	return text
+}
+
+func sourceFileKey(cwd string, file string) string {
+	rel, err := filepath.Rel(cwd, filepath.FromSlash(file))
+	if err != nil || rel == ".." || strings.HasPrefix(rel, ".."+string(os.PathSeparator)) {
+		return filepath.ToSlash(file)
+	}
+	return filepath.ToSlash(rel)
+}
+
+func newTransformCompilerDiagnostic(
+	file string,
+	code string,
+	message string,
+) transformCompilerDiagnostic {
+	var ptr *string
+	if file != "" {
+		normalized := filepath.ToSlash(file)
+		ptr = &normalized
+	}
+	return transformCompilerDiagnostic{
+		File:        ptr,
+		Category:    "error",
+		Code:        code,
+		MessageText: message,
+	}
+}
+
+func transformDiagnosticToCompilerDiagnostic(
+	diag Diagnostic,
+) transformCompilerDiagnostic {
+	var ptr *string
+	if diag.File != "" {
+		normalized := filepath.ToSlash(diag.File)
+		ptr = &normalized
+	}
+	return transformCompilerDiagnostic{
+		File:        ptr,
+		Category:    "error",
+		Code:        diag.Code,
+		Line:        diag.Line,
+		Character:   diag.Column,
+		MessageText: diag.Message,
+	}
+}