refactor(@angular-devkit/schematics): improve performance of move()

FrozenPandaz · FrozenPandaz · commit 4f6b98d51088 · 2018-11-02T17:44:43.000-04:00
diff --git a/packages/angular_devkit/schematics/src/rules/move.ts b/packages/angular_devkit/schematics/src/rules/move.ts
@@ -23,9 +23,17 @@ export function move(from: string, to?: string): Rule {
     return noop;
   }
 
-  return tree => tree.visit(path => {
-    if (path.startsWith(fromPath)) {
-      tree.rename(path, toPath + '/' + path.substr(fromPath.length));
+  return tree => {
+    if (tree.exists(fromPath)) {
+      // fromPath is a file
+      tree.rename(fromPath, toPath);
+    } else {
+      // fromPath is a directory
+      tree.getDir(fromPath).visit(path => {
+        tree.rename(path, toPath + '/' + path.substr(fromPath.length));
+      });
     }
-  });
+
+    return tree;
+  };
 }
