fix #270, support appendTsSuffixTo config

Author: HerringtonDarkholme

src/index.ts

@@ -14,9 +14,9 @@ const definitionFileRegex = /\.d\.ts$/;
 function loader(this: interfaces.Webpack, contents: string) {
     this.cacheable && this.cacheable();
     const callback = this.async();
-    const filePath = path.normalize(this.resourcePath);
-
     const options = makeOptions(this);
+    const rawFilePath = path.normalize(this.resourcePath);
+    const filePath = utils.appendTsSuffixIfMatch(options.appendTsSuffixTo, rawFilePath);
 
     const { instance, error } = instances.ensureTypeScriptInstance(options, this);
 
@@ -58,6 +58,7 @@ function makeOptions(loader: interfaces.Webpack) {
         configFileName: 'tsconfig.json',
         transpileOnly: false,
         compilerOptions: {},
+        appendTsSuffixTo: [],
     }, configFileOptions, queryOptions);
     options.ignoreDiagnostics = arrify(options.ignoreDiagnostics).map(Number);
     options.logLevel = options.logLevel.toUpperCase();

src/instances.ts

@@ -105,7 +105,7 @@ export function ensureTypeScriptInstance(
         modifiedFiles: null,
     };
 
-    const servicesHost = makeServicesHost(scriptRegex, log, loader, instance);
+    const servicesHost = makeServicesHost(scriptRegex, log, loader, instance, loaderOptions.appendTsSuffixTo);
     instance.languageService = compiler.createLanguageService(servicesHost, compiler.createDocumentRegistry());
 
     loader._compiler.plugin("after-compile", afterCompile(instance, configFilePath));

src/interfaces.ts

@@ -85,7 +85,7 @@ export interface Resolve {
     /**
      * The directory (absolute path) that contains your modules.
      * May also be an array of directories.
-     * This setting should be used to add individual directories to the search path. 
+     * This setting should be used to add individual directories to the search path.
      */
     root?: string | string[];
     /**
@@ -153,6 +153,7 @@ export interface LoaderOptions {
     transpileOnly: boolean;
     ignoreDiagnostics: number[];
     compilerOptions: typescript.CompilerOptions;
+    appendTsSuffixTo: RegExp[];
 }
 
 export interface TSFile {

src/servicesHost.ts

@@ -13,7 +13,8 @@ function makeServicesHost(
     scriptRegex: RegExp,
     log: logger.Logger,
     loader: interfaces.Webpack,
-    instance: interfaces.TSInstance
+    instance: interfaces.TSInstance,
+    appendTsSuffixTo: RegExp[]
 ) {
     const { compiler, compilerOptions, files } = instance;
 
@@ -77,6 +78,7 @@ function makeServicesHost(
 
                 try {
                     resolvedFileName = resolver.resolveSync(path.normalize(path.dirname(containingFile)), moduleName);
+                    resolvedFileName = utils.appendTsSuffixIfMatch(appendTsSuffixTo, resolvedFileName);
 
                     if (!resolvedFileName.match(scriptRegex)) {
                         resolvedFileName = null;

src/utils.ts

@@ -71,3 +71,12 @@ export function makeError({ rawMessage, message, location, file }: MakeError): i
 
     return <interfaces.WebpackError> objectAssign(error, { location, file });
 }
+
+export function appendTsSuffixIfMatch(patterns: RegExp[], path: string): string {
+    for (let regexp of patterns) {
+      if (regexp.test(path)) {
+        return path + '.ts';
+      }
+    }
+    return path;
+}