fix(Schematics): remove ts extension when importing reducer in container (#1061)

Closes #1056

Author: timdeschryver

modules/effects/schematics-core/utility/find-module.ts

@@ -98,7 +98,7 @@ export function buildRelativePath(from: string, to: string): string {
 
   // Remove file names (preserving destination)
   fromParts.pop();
-  const toFileName = toParts.pop();
+  const toFileName = convertToTypeScriptFileName(toParts.pop());
 
   const relativePath = relative(
     normalize(fromParts.join('/')),
@@ -116,5 +116,16 @@ export function buildRelativePath(from: string, to: string): string {
     pathPrefix += '/';
   }
 
-  return pathPrefix + (relativePath ? relativePath + '/' : '') + toFileName;
+  return toFileName
+    ? pathPrefix + (relativePath ? relativePath + '/' : '') + toFileName
+    : pathPrefix + relativePath;
+}
+
+/**
+ * Strips the typescript extension and clears index filenames
+ * foo.ts -> foo
+ * index.ts -> empty
+ */
+function convertToTypeScriptFileName(filename: string | undefined) {
+  return filename ? filename.replace(/(\.ts)|(index\.ts)$/, '') : '';
 }

modules/entity/schematics-core/utility/find-module.ts

@@ -98,7 +98,7 @@ export function buildRelativePath(from: string, to: string): string {
 
   // Remove file names (preserving destination)
   fromParts.pop();
-  const toFileName = toParts.pop();
+  const toFileName = convertToTypeScriptFileName(toParts.pop());
 
   const relativePath = relative(
     normalize(fromParts.join('/')),
@@ -116,5 +116,16 @@ export function buildRelativePath(from: string, to: string): string {
     pathPrefix += '/';
   }
 
-  return pathPrefix + (relativePath ? relativePath + '/' : '') + toFileName;
+  return toFileName
+    ? pathPrefix + (relativePath ? relativePath + '/' : '') + toFileName
+    : pathPrefix + relativePath;
+}
+
+/**
+ * Strips the typescript extension and clears index filenames
+ * foo.ts -> foo
+ * index.ts -> empty
+ */
+function convertToTypeScriptFileName(filename: string | undefined) {
+  return filename ? filename.replace(/(\.ts)|(index\.ts)$/, '') : '';
 }

modules/router-store/schematics-core/utility/find-module.ts

@@ -98,7 +98,7 @@ export function buildRelativePath(from: string, to: string): string {
 
   // Remove file names (preserving destination)
   fromParts.pop();
-  const toFileName = toParts.pop();
+  const toFileName = convertToTypeScriptFileName(toParts.pop());
 
   const relativePath = relative(
     normalize(fromParts.join('/')),
@@ -116,5 +116,16 @@ export function buildRelativePath(from: string, to: string): string {
     pathPrefix += '/';
   }
 
-  return pathPrefix + (relativePath ? relativePath + '/' : '') + toFileName;
+  return toFileName
+    ? pathPrefix + (relativePath ? relativePath + '/' : '') + toFileName
+    : pathPrefix + relativePath;
+}
+
+/**
+ * Strips the typescript extension and clears index filenames
+ * foo.ts -> foo
+ * index.ts -> empty
+ */
+function convertToTypeScriptFileName(filename: string | undefined) {
+  return filename ? filename.replace(/(\.ts)|(index\.ts)$/, '') : '';
 }

modules/schematics-core/utility/find-module.ts

@@ -98,7 +98,7 @@ export function buildRelativePath(from: string, to: string): string {
 
   // Remove file names (preserving destination)
   fromParts.pop();
-  const toFileName = toParts.pop();
+  const toFileName = convertToTypeScriptFileName(toParts.pop());
 
   const relativePath = relative(
     normalize(fromParts.join('/')),
@@ -116,5 +116,16 @@ export function buildRelativePath(from: string, to: string): string {
     pathPrefix += '/';
   }
 
-  return pathPrefix + (relativePath ? relativePath + '/' : '') + toFileName;
+  return toFileName
+    ? pathPrefix + (relativePath ? relativePath + '/' : '') + toFileName
+    : pathPrefix + relativePath;
+}
+
+/**
+ * Strips the typescript extension and clears index filenames
+ * foo.ts -> foo
+ * index.ts -> empty
+ */
+function convertToTypeScriptFileName(filename: string | undefined) {
+  return filename ? filename.replace(/(\.ts)|(index\.ts)$/, '') : '';
 }

modules/schematics/schematics-core/utility/find-module.ts

@@ -98,7 +98,7 @@ export function buildRelativePath(from: string, to: string): string {
 
   // Remove file names (preserving destination)
   fromParts.pop();
-  const toFileName = toParts.pop();
+  const toFileName = convertToTypeScriptFileName(toParts.pop());
 
   const relativePath = relative(
     normalize(fromParts.join('/')),
@@ -116,5 +116,16 @@ export function buildRelativePath(from: string, to: string): string {
     pathPrefix += '/';
   }
 
-  return pathPrefix + (relativePath ? relativePath + '/' : '') + toFileName;
+  return toFileName
+    ? pathPrefix + (relativePath ? relativePath + '/' : '') + toFileName
+    : pathPrefix + relativePath;
+}
+
+/**
+ * Strips the typescript extension and clears index filenames
+ * foo.ts -> foo
+ * index.ts -> empty
+ */
+function convertToTypeScriptFileName(filename: string | undefined) {
+  return filename ? filename.replace(/(\.ts)|(index\.ts)$/, '') : '';
 }

modules/schematics/src/container/index.spec.ts

@@ -57,6 +57,26 @@ describe('Container Schematic', () => {
     expect(content).toMatch(/import \* as fromStore from '..\/reducers';/);
   });
 
+  it('should remove .ts from the state path if provided', () => {
+    const options = { ...defaultOptions, state: 'reducers/foo.ts' };
+    appTree.create(`${projectPath}/src/app/reducers/foo.ts`, '');
+    const tree = schematicRunner.runSchematic('container', options, appTree);
+    const content = tree.readContent(
+      `${projectPath}/src/app/foo/foo.component.ts`
+    );
+    expect(content).toMatch(/import \* as fromStore from '..\/reducers\/foo';/);
+  });
+
+  it('should remove index.ts from the state path if provided', () => {
+    const options = { ...defaultOptions, state: 'reducers/index.ts' };
+    appTree.create(`${projectPath}/src/app/reducers/index.ts`, '');
+    const tree = schematicRunner.runSchematic('container', options, appTree);
+    const content = tree.readContent(
+      `${projectPath}/src/app/foo/foo.component.ts`
+    );
+    expect(content).toMatch(/import \* as fromStore from '..\/reducers';/);
+  });
+
   it('should import Store into the component', () => {
     const options = { ...defaultOptions, state: 'reducers' };
     appTree.create(`${projectPath}/src/app/reducers`, '');

modules/store-devtools/schematics-core/utility/find-module.ts

@@ -98,7 +98,7 @@ export function buildRelativePath(from: string, to: string): string {
 
   // Remove file names (preserving destination)
   fromParts.pop();
-  const toFileName = toParts.pop();
+  const toFileName = convertToTypeScriptFileName(toParts.pop());
 
   const relativePath = relative(
     normalize(fromParts.join('/')),
@@ -116,5 +116,16 @@ export function buildRelativePath(from: string, to: string): string {
     pathPrefix += '/';
   }
 
-  return pathPrefix + (relativePath ? relativePath + '/' : '') + toFileName;
+  return toFileName
+    ? pathPrefix + (relativePath ? relativePath + '/' : '') + toFileName
+    : pathPrefix + relativePath;
+}
+
+/**
+ * Strips the typescript extension and clears index filenames
+ * foo.ts -> foo
+ * index.ts -> empty
+ */
+function convertToTypeScriptFileName(filename: string | undefined) {
+  return filename ? filename.replace(/(\.ts)|(index\.ts)$/, '') : '';
 }

modules/store/schematics-core/utility/find-module.ts

@@ -98,7 +98,7 @@ export function buildRelativePath(from: string, to: string): string {
 
   // Remove file names (preserving destination)
   fromParts.pop();
-  const toFileName = toParts.pop();
+  const toFileName = convertToTypeScriptFileName(toParts.pop());
 
   const relativePath = relative(
     normalize(fromParts.join('/')),
@@ -116,5 +116,16 @@ export function buildRelativePath(from: string, to: string): string {
     pathPrefix += '/';
   }
 
-  return pathPrefix + (relativePath ? relativePath + '/' : '') + toFileName;
+  return toFileName
+    ? pathPrefix + (relativePath ? relativePath + '/' : '') + toFileName
+    : pathPrefix + relativePath;
+}
+
+/**
+ * Strips the typescript extension and clears index filenames
+ * foo.ts -> foo
+ * index.ts -> empty
+ */
+function convertToTypeScriptFileName(filename: string | undefined) {
+  return filename ? filename.replace(/(\.ts)|(index\.ts)$/, '') : '';
 }