Add package action to change

src/codemodder/change.py

@@ -1,20 +1,46 @@
 from dataclasses import dataclass, field
+from enum import Enum
+
+
+class Action(Enum):
+    ADD = "add"
+    REMOVE = "remove"
+
+
+class Result(Enum):
+    COMPLETED = "completed"
+    FAILED = "failed"
+    SKIPPED = "skipped"
+
+
+@dataclass
+class PackageAction:
+    action: Action
+    result: Result
+    package: str
+
+    def to_json(self):
+        return {
+            "action": self.action.value,
+            "result": self.result.value,
+            "package": self.package,
+        }
 
 
 @dataclass
 class Change:
     lineNumber: int
     description: str
     properties: dict = field(default_factory=dict)
-    packageActions: list = field(default_factory=list)
+    packageActions: list[PackageAction] = field(default_factory=list)
 
     def to_json(self):
         return {
             # Not sure why this is a string but it's in the spec
             "lineNumber": str(self.lineNumber),
             "description": self.description,
             "properties": self.properties,
-            "packageActions": self.packageActions,
+            "packageActions": [pa.to_json() for pa in self.packageActions],
         }
 
 

src/codemodder/dependency_manager.py

@@ -5,7 +5,7 @@
 import difflib
 from packaging.requirements import Requirement
 
-from codemodder.change import Change, ChangeSet
+from codemodder.change import Action, Change, ChangeSet, PackageAction, Result
 from codemodder.dependency import Dependency
 
 
@@ -48,6 +48,9 @@ def write(self, dry_run: bool = False) -> Optional[ChangeSet]:
                 lineNumber=len(self._lines) + i + 1,
                 description=dep.build_description(),
                 properties={"contextual_description": True},
+                packageActions=[
+                    PackageAction(Action.ADD, Result.COMPLETED, str(dep.requirement))
+                ],
             )
             for i, dep in enumerate(self._new_requirements)
         ]