-
Notifications
You must be signed in to change notification settings - Fork 149
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
fe6b689
commit 9bd1a88
Showing
6 changed files
with
254 additions
and
401 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,236 @@ | ||
| import { existsSync } from 'fs'; | ||
| import { dirname, basename, resolve } from 'path'; | ||
|
|
||
| import ts from 'typescript'; | ||
|
|
||
| import { Options } from '../../types'; | ||
|
|
||
| type CompilerOptions = Options.Typescript['compilerOptions']; | ||
|
|
||
| function getFilenameExtension(filename: string) { | ||
| filename = basename(filename); | ||
| const lastDotIndex = filename.lastIndexOf('.'); | ||
|
|
||
| if (lastDotIndex <= 0) return ''; | ||
|
|
||
| return filename.substr(lastDotIndex + 1); | ||
| } | ||
|
|
||
| function isSvelteFile(filename: string) { | ||
| const importExtension = getFilenameExtension(filename); | ||
|
|
||
| return importExtension === 'svelte' || importExtension === 'html'; | ||
| } | ||
|
|
||
| const IMPORTEE_PATTERN = /['"](.*?)['"]/; | ||
|
|
||
| function isValidSvelteImportDiagnostic(filename: string, diagnostic: any) { | ||
| // TS2307: 'cannot find module' | ||
| if (diagnostic.code !== 2307) return true; | ||
|
|
||
| const importeeMatch = diagnostic.messageText.match(IMPORTEE_PATTERN); | ||
|
|
||
| // istanbul ignore if | ||
| if (!importeeMatch) return true; | ||
|
|
||
| let [, importeePath] = importeeMatch; | ||
|
|
||
| /** if we're not dealing with a relative path, assume the file exists */ | ||
| if (importeePath[0] !== '.') return false; | ||
|
|
||
| /** if the importee is not a svelte file, do nothing */ | ||
| if (!isSvelteFile(importeePath)) return true; | ||
|
|
||
| importeePath = resolve(dirname(filename), importeePath); | ||
|
|
||
| return existsSync(importeePath) === false; | ||
| } | ||
|
|
||
| const TS2552_REGEX = /Cannot find name '\$([a-zA-Z0-9_]+)'. Did you mean '([a-zA-Z0-9_]+)'\?/i; | ||
|
|
||
| function isValidSvelteReactiveValueDiagnostic( | ||
| filename: string, | ||
| diagnostic: any, | ||
| ): boolean { | ||
| if (diagnostic.code !== 2552) return true; | ||
|
|
||
| /** if the importee is not a svelte file, do nothing */ | ||
| if (!isSvelteFile(filename)) return true; | ||
|
|
||
| /** if error message doesn't contain a reactive value, do nothing */ | ||
| if (!diagnostic.messageText.includes('$')) return true; | ||
|
|
||
| const [, usedVar, proposedVar] = | ||
| diagnostic.messageText.match(TS2552_REGEX) || []; | ||
|
|
||
| return !(usedVar && proposedVar && usedVar === proposedVar); | ||
| } | ||
|
|
||
| function createImportTransformerFromProgram(program: ts.Program) { | ||
| const checker = program.getTypeChecker(); | ||
|
|
||
| const importedTypeRemoverTransformer: ts.TransformerFactory<ts.SourceFile> = ( | ||
| context, | ||
| ) => { | ||
| const visit: ts.Visitor = (node) => { | ||
| if (!ts.isImportDeclaration(node)) { | ||
| return ts.visitEachChild(node, (child) => visit(child), context); | ||
| } | ||
|
|
||
| let newImportClause: ts.ImportClause = node.importClause; | ||
|
|
||
| if (node.importClause) { | ||
| // import {...} from './blah' | ||
| if (node.importClause?.isTypeOnly) { | ||
| return ts.createEmptyStatement(); | ||
| } | ||
|
|
||
| // import Blah, { blah } from './blah' | ||
| newImportClause = ts.getMutableClone(node.importClause); | ||
|
|
||
| // types can't be default exports, so we just worry about { blah } and { blah as name } exports | ||
| if ( | ||
| newImportClause.namedBindings && | ||
| ts.isNamedImports(newImportClause.namedBindings) | ||
| ) { | ||
| const newBindings = ts.getMutableClone(newImportClause.namedBindings); | ||
| const newElements = []; | ||
|
|
||
| newImportClause.namedBindings = undefined; | ||
|
|
||
| for (const spec of newBindings.elements) { | ||
| const ident = spec.name; | ||
|
|
||
| const symbol = checker.getSymbolAtLocation(ident); | ||
| const aliased = checker.getAliasedSymbol(symbol); | ||
|
|
||
| if (aliased) { | ||
| if ( | ||
| (aliased.flags & | ||
| (ts.SymbolFlags.TypeAlias | ts.SymbolFlags.Interface)) > | ||
| 0 | ||
| ) { | ||
| // We found an imported type, don't add to our new import clause | ||
| continue; | ||
| } | ||
| } | ||
| newElements.push(spec); | ||
| } | ||
|
|
||
| if (newElements.length > 0) { | ||
| newBindings.elements = ts.createNodeArray( | ||
| newElements, | ||
| newBindings.elements.hasTrailingComma, | ||
| ); | ||
| newImportClause.namedBindings = newBindings; | ||
| } | ||
| } | ||
|
|
||
| // we ended up removing all named bindings and we didn't have a name? nothing left to import. | ||
| if ( | ||
| newImportClause.namedBindings == null && | ||
| newImportClause.name == null | ||
| ) { | ||
| return ts.createEmptyStatement(); | ||
| } | ||
| } | ||
|
|
||
| return ts.createImportDeclaration( | ||
| node.decorators, | ||
| node.modifiers, | ||
| newImportClause, | ||
| node.moduleSpecifier, | ||
| ); | ||
| }; | ||
|
|
||
| return (node) => ts.visitNode(node, visit); | ||
| }; | ||
|
|
||
| return importedTypeRemoverTransformer; | ||
| } | ||
|
|
||
| export function compileFileFromMemory( | ||
| compilerOptions: CompilerOptions, | ||
| { filename, content }: { filename: string; content: string }, | ||
| ) { | ||
| let code = content; | ||
| let map; | ||
|
|
||
| const realHost = ts.createCompilerHost(compilerOptions, true); | ||
| const dummyFileName = ts.sys.resolvePath(filename); | ||
|
|
||
| const isDummyFile = (fileName: string) => | ||
| ts.sys.resolvePath(fileName) === dummyFileName; | ||
|
|
||
| const host: ts.CompilerHost = { | ||
| fileExists: (fileName) => | ||
| isDummyFile(fileName) || realHost.fileExists(fileName), | ||
| getCanonicalFileName: (fileName) => | ||
| isDummyFile(fileName) | ||
| ? ts.sys.useCaseSensitiveFileNames | ||
| ? fileName | ||
| : fileName.toLowerCase() | ||
| : realHost.getCanonicalFileName(fileName), | ||
| getSourceFile: ( | ||
| fileName, | ||
| languageVersion, | ||
| onError, | ||
| shouldCreateNewSourceFile, | ||
| // eslint-disable-next-line max-params | ||
| ) => | ||
| isDummyFile(fileName) | ||
| ? ts.createSourceFile(dummyFileName, code, languageVersion) | ||
| : realHost.getSourceFile( | ||
| fileName, | ||
| languageVersion, | ||
| onError, | ||
| shouldCreateNewSourceFile, | ||
| ), | ||
| readFile: (fileName) => | ||
| isDummyFile(fileName) ? content : realHost.readFile(fileName), | ||
| writeFile: (fileName, data) => { | ||
| if (fileName.endsWith('.map')) { | ||
| map = data; | ||
| } else { | ||
| code = data; | ||
| } | ||
| }, | ||
| directoryExists: | ||
| realHost.directoryExists && realHost.directoryExists.bind(realHost), | ||
| getCurrentDirectory: realHost.getCurrentDirectory.bind(realHost), | ||
| getDirectories: realHost.getDirectories.bind(realHost), | ||
| getNewLine: realHost.getNewLine.bind(realHost), | ||
| getDefaultLibFileName: realHost.getDefaultLibFileName.bind(realHost), | ||
| resolveModuleNames: | ||
| realHost.resolveModuleNames && realHost.resolveModuleNames.bind(realHost), | ||
| useCaseSensitiveFileNames: realHost.useCaseSensitiveFileNames.bind( | ||
| realHost, | ||
| ), | ||
| }; | ||
|
|
||
| const program = ts.createProgram([dummyFileName], compilerOptions, host); | ||
|
|
||
| const transformers = { | ||
| before: [createImportTransformerFromProgram(program)], | ||
| }; | ||
|
|
||
| const emitResult = program.emit( | ||
| program.getSourceFile(dummyFileName), | ||
| undefined, | ||
| undefined, | ||
| undefined, | ||
| transformers, | ||
| ); | ||
|
|
||
| // collect diagnostics without svelte import errors | ||
| const diagnostics = [ | ||
| ...emitResult.diagnostics, | ||
| ...ts.getPreEmitDiagnostics(program), | ||
| ].filter( | ||
| (diagnostic) => | ||
| isValidSvelteImportDiagnostic(filename, diagnostic) && | ||
| isValidSvelteReactiveValueDiagnostic(filename, diagnostic), | ||
| ); | ||
|
|
||
| return { code, map, diagnostics }; | ||
| } |
Oops, something went wrong.