Support for type declaration moves in tree file diff view

(issue #703)
tsantalis · May 2, 2024 · 25959ec · 25959ec
1 parent 2b828fa
commit 25959ec
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/src/main/java/org/refactoringminer/astDiff/matchers/ProjectASTDiffer.java b/src/main/java/org/refactoringminer/astDiff/matchers/ProjectASTDiffer.java
@@ -85,6 +85,7 @@ private void createASTDiffsForMovedDeclarations() {
         for(ASTDiff diff : projectASTDiff.getDiffSet()) {
         	Map<Tree, List<Mapping>> methodDeclarationMappings = new LinkedHashMap<Tree, List<Mapping>>();
             Map<Tree, List<Mapping>> fieldDeclarationMappings = new LinkedHashMap<Tree, List<Mapping>>();
+            Map<Tree, List<Mapping>> typeDeclarationMappings = new LinkedHashMap<Tree, List<Mapping>>();
             Map<Tree, Action> actionMap = new LinkedHashMap<Tree, Action>();
         	ExtendedOnlyRootsClassifier classifier = (ExtendedOnlyRootsClassifier) diff.createRootNodesClassifier();
         	Map<Tree, Action> map = classifier.getSrcMoveOutTreeMap();
@@ -117,6 +118,14 @@ private void createASTDiffsForMovedDeclarations() {
         				}
         			}
         		}
+        		if(src.getType().name.equals(Constants.TYPE_DECLARATION) ||
+        				src.getType().name.equals(Constants.ENUM_DECLARATION) ||
+        				src.getType().name.equals(Constants.RECORD_DECLARATION)) {
+        			actionMap.put(src, map.get(src));
+        			List<Mapping> mappings = new ArrayList<Mapping>();
+    				mappings.addAll(getMappingForLeft(diff, src));
+    				typeDeclarationMappings.put(src, mappings);
+        		}
         	}
         	//group the mappings based on the pair of src and dst files.
         	String srcPath = diff.getSrcPath();
@@ -156,6 +165,24 @@ private void createASTDiffsForMovedDeclarations() {
         			}
             	}
         	}
+        	for(Tree key : typeDeclarationMappings.keySet()) {
+        		if(actionMap.containsKey(key)) {
+        			Action action = actionMap.get(key);
+        			if(action instanceof MoveOut) {
+        				MoveOut moveOut = (MoveOut)action;
+        				String dstPath = moveOut.getDstFile();
+        				Pair<String, String> pair = new Pair<String, String>(srcPath, dstPath);
+        				if(filePairMappings.containsKey(pair)) {
+        					filePairMappings.get(pair).addAll(typeDeclarationMappings.get(key));
+        				}
+        				else {
+        					List<Mapping> mappings = new ArrayList<Mapping>();
+        					mappings.addAll(typeDeclarationMappings.get(key));
+        					filePairMappings.put(pair, mappings);
+        				}
+        			}
+            	}
+        	}
         }
         for(Pair<String, String> pair : filePairMappings.keySet()) {
         	Pair<TreeContext, TreeContext> treeContextPairs = findTreeContexts(pair.first, pair.second);