Refactor ModResolver.directory

Prefer replacing with a new function with_directory rather than
modifying in place.

Author: camsteffen

src/modules.rs

@@ -220,13 +220,11 @@ impl<'ast, 'sess, 'c> ModResolver<'ast, 'sess> {
         sub_mod: Module<'_>,
         from_ast: bool,
     ) -> Result<(), ModuleResolutionError> {
-        let old_directory = self.directory.clone();
         let sub_mod_kind = self.peek_sub_mod(item, &sub_mod)?;
         if let Some(sub_mod_kind) = sub_mod_kind {
             self.insert_sub_mod(sub_mod_kind.clone())?;
             self.visit_sub_mod_inner(sub_mod, sub_mod_kind, from_ast)?;
         }
-        self.directory = old_directory;
         Ok(())
     }
 
@@ -284,39 +282,30 @@ impl<'ast, 'sess, 'c> ModResolver<'ast, 'sess> {
                     path: mod_path.parent().unwrap().to_path_buf(),
                     ownership: directory_ownership,
                 };
-                self.visit_sub_mod_after_directory_update(sub_mod, Some(directory), false)
+                self.with_directory(directory, |this| this.visit_mod_outside_ast(&sub_mod.items))?;
             }
             SubModKind::Internal(item) => {
-                self.push_inline_mod_directory(item.ident, &item.attrs);
-                self.visit_sub_mod_after_directory_update(sub_mod, None, from_ast)
+                let directory = self.inline_mod_directory(item.ident, &item.attrs);
+                self.with_directory(directory, |this| {
+                    if from_ast {
+                        this.visit_mod_from_ast(&sub_mod.items)
+                    } else {
+                        this.visit_mod_outside_ast(&sub_mod.items)
+                    }
+                })?;
             }
             SubModKind::MultiExternal(mods) => {
                 for (mod_path, directory_ownership, sub_mod) in mods {
                     let directory = Directory {
                         path: mod_path.parent().unwrap().to_path_buf(),
                         ownership: directory_ownership,
                     };
-                    self.visit_sub_mod_after_directory_update(sub_mod, Some(directory), false)?;
+                    self.with_directory(directory, |this| {
+                        this.visit_mod_outside_ast(&sub_mod.items)
+                    })?;
                 }
-                Ok(())
             }
         }
-    }
-
-    fn visit_sub_mod_after_directory_update(
-        &mut self,
-        sub_mod: Module<'_>,
-        directory: Option<Directory>,
-        from_ast: bool,
-    ) -> Result<(), ModuleResolutionError> {
-        if let Some(directory) = directory {
-            self.directory = directory;
-        }
-        if from_ast {
-            self.visit_mod_from_ast(&sub_mod.items)?;
-        } else {
-            self.visit_mod_outside_ast(&sub_mod.items)?;
-        }
         Ok(())
     }
 
@@ -449,31 +438,41 @@ impl<'ast, 'sess, 'c> ModResolver<'ast, 'sess> {
         }
     }
 
-    fn push_inline_mod_directory(&mut self, id: symbol::Ident, attrs: &[ast::Attribute]) {
+    fn inline_mod_directory(&mut self, id: symbol::Ident, attrs: &[ast::Attribute]) -> Directory {
         if let Some(path) = find_path_value(attrs) {
-            self.directory.path.push(path.as_str());
-            self.directory.ownership = DirectoryOwnership::Owned { relative: None };
-        } else {
-            let id = id.as_str();
-            // We have to push on the current module name in the case of relative
-            // paths in order to ensure that any additional module paths from inline
-            // `mod x { ... }` come after the relative extension.
-            //
-            // For example, a `mod z { ... }` inside `x/y.rs` should set the current
-            // directory path to `/x/y/z`, not `/x/z` with a relative offset of `y`.
-            if let DirectoryOwnership::Owned { relative } = &mut self.directory.ownership {
-                if let Some(ident) = relative.take() {
-                    // remove the relative offset
-                    self.directory.path.push(ident.as_str());
-
-                    // In the case where there is an x.rs and an ./x directory we want
-                    // to prevent adding x twice. For example, ./x/x
-                    if self.directory.path.exists() && !self.directory.path.join(id).exists() {
-                        return;
-                    }
-                }
+            return Directory {
+                path: self.directory.path.join(path.as_str()),
+                ownership: DirectoryOwnership::Owned { relative: None },
+            };
+        }
+        let id = id.as_str();
+        // We have to push on the current module name in the case of relative
+        // paths in order to ensure that any additional module paths from inline
+        // `mod x { ... }` come after the relative extension.
+        //
+        // For example, a `mod z { ... }` inside `x/y.rs` should set the current
+        // directory path to `/x/y/z`, not `/x/z` with a relative offset of `y`.
+        let new_path = if let DirectoryOwnership::Owned {
+            relative: Some(ident),
+        } = self.directory.ownership
+        {
+            // remove the relative offset
+            let relative = self.directory.path.join(ident.as_str());
+            let nested = relative.join(id);
+
+            // In the case where there is an x.rs and an ./x directory we want
+            // to prevent adding x twice. For example, ./x/x
+            if relative.exists() && !nested.exists() {
+                relative
+            } else {
+                nested
             }
-            self.directory.path.push(id);
+        } else {
+            self.directory.path.join(id)
+        };
+        Directory {
+            path: new_path,
+            ownership: self.directory.ownership,
         }
     }
 
@@ -491,8 +490,7 @@ impl<'ast, 'sess, 'c> ModResolver<'ast, 'sess> {
         }
         let mut result = vec![];
         for path in path_visitor.paths() {
-            let mut actual_path = self.directory.path.clone();
-            actual_path.push(&path);
+            let actual_path = self.directory.path.join(path);
             if !actual_path.exists() {
                 continue;
             }
@@ -520,6 +518,13 @@ impl<'ast, 'sess, 'c> ModResolver<'ast, 'sess> {
         }
         result
     }
+
+    fn with_directory<T>(&mut self, directory: Directory, f: impl FnOnce(&mut Self) -> T) -> T {
+        let old = std::mem::replace(&mut self.directory, directory);
+        let out = f(self);
+        self.directory = old;
+        out
+    }
 }
 
 fn path_value(attr: &ast::Attribute) -> Option<Symbol> {