fix: file extension is not added to imported files w/ NodeNext - CodeRabbit suggestions

Author: rkday-pro

packages/core/src/utils/tsconfig.ts

@@ -13,6 +13,13 @@ export function isSyntheticDefaultImportsAllow(config?: Tsconfig) {
 
 const NODE_NEXT_MODULES = new Set(['nodenext', 'node16']);
 
+const NODE_NEXT_EXTENSION_MAP: ReadonlyArray<readonly [string, string]> = [
+  ['.tsx', '.jsx'],
+  ['.mts', '.mjs'],
+  ['.cts', '.cjs'],
+  ['.ts', '.js'],
+];
+
 export function getImportExtension(
   fileExtension: string,
   tsconfig?: Tsconfig,
@@ -29,9 +36,12 @@ export function getImportExtension(
     (module && NODE_NEXT_MODULES.has(module)) ||
     (moduleResolution && NODE_NEXT_MODULES.has(moduleResolution))
   ) {
-    return fileExtension.endsWith('.ts')
-      ? `${fileExtension.slice(0, -3)}.js`
-      : fileExtension;
+    for (const [from, to] of NODE_NEXT_EXTENSION_MAP) {
+      if (fileExtension.endsWith(from)) {
+        return `${fileExtension.slice(0, -from.length)}${to}`;
+      }
+    }
+    return fileExtension;
   }
 
   return fileExtension.replace(/\.ts$/, '') || '';

packages/core/src/writers/schemas.ts

@@ -222,6 +222,10 @@ function normalizeCanonicalImportPaths(
         schemaPath,
         canonical.importPath.replaceAll('\\', '/'),
       );
+      // `relative` derives from canonical.importPath (built via getPath) so it
+      // normally ends with `fileExtension`; the `.ts$` branch is a defensive
+      // fallback for legacy/hardcoded `.ts` paths that don't match the
+      // configured fileExtension (e.g. `.gen.ts`).
       const withoutFileExtension = relative.endsWith(fileExtension)
         ? relative.slice(0, -fileExtension.length)
         : relative.replace(/\.ts$/, '');