1 files changed, 94 insertions, 0 deletions
diff --git a/lib/pama/remove.go b/lib/pama/remove.go
new file mode 100644
index 00000000..d74874a1
--- /dev/null
+++ b/lib/pama/remove.go
@@ -0,0 +1,94 @@
+package pama
+
+import (
+	"fmt"
+
+	"git.sr.ht/~rjarry/aerc/lib/pama/models"
+	"git.sr.ht/~rjarry/aerc/log"
+)
+
+func (m PatchManager) RemovePatch(patch string) error {
+	p, err := m.CurrentProject()
+	if err != nil {
+		return err
+	}
+
+	if !models.Commits(p.Commits).HasTag(patch) {
+		return fmt.Errorf("Patch '%s' not found in project '%s'", patch, p.Name)
+	}
+
+	rc, err := m.rc(p.RevctrlID, p.Root)
+	if err != nil {
+		return revErr(err)
+	}
+
+	if !rc.Clean() {
+		return fmt.Errorf("Aborting... There are unstaged changes " +
+			"or a rebase in progress")
+	}
+
+	toRemove := make([]models.Commit, 0)
+	for _, c := range p.Commits {
+		if !rc.Exists(c.ID) {
+			log.Errorf("failed to find commit. %v", c)
+			return fmt.Errorf("Cannot remove patch. " +
+				"Please rebase first with ':patch rebase'")
+		}
+		if c.Tag == patch {
+			toRemove = append(toRemove, c)
+		}
+	}
+
+	removed := make(map[string]struct{})
+	for i := len(toRemove) - 1; i >= 0; i-- {
+		commitID := toRemove[i].ID
+		beforeIDs, err := rc.History(commitID)
+		if err != nil {
+			log.Errorf("failed to remove %v (commits before): %v", toRemove[i], err)
+			continue
+		}
+		err = rc.Remove(commitID)
+		if err != nil {
+			log.Errorf("failed to remove %v (remove): %v", toRemove[i], err)
+			continue
+		}
+		removed[commitID] = struct{}{}
+		afterIDs, err := rc.History(p.Base.ID)
+		if err != nil {
+			log.Errorf("failed to remove %v (commits after): %v", toRemove[i], err)
+			continue
+		}
+		afterIDs = afterIDs[len(afterIDs)-len(beforeIDs):]
+		transform := make(map[string]string)
+		for j := 0; j < len(beforeIDs); j++ {
+			transform[beforeIDs[j]] = afterIDs[j]
+		}
+		for j, c := range p.Commits {
+			if newId, ok := transform[c.ID]; ok {
+				msgid := p.Commits[j].MessageId
+				p.Commits[j] = models.NewCommit(
+					rc,
+					newId,
+					p.Commits[j].Tag,
+				)
+				p.Commits[j].MessageId = msgid
+			}
+		}
+	}
+
+	if len(removed) < len(toRemove) {
+		return fmt.Errorf("Failed to remove commits. Removed %d of %d.",
+			len(removed), len(toRemove))
+	}
+
+	commits := make([]models.Commit, 0, len(p.Commits))
+	for _, c := range p.Commits {
+		if _, ok := removed[c.ID]; ok {
+			continue
+		}
+		commits = append(commits, c)
+	}
+	p.Commits = commits
+
+	return storeErr(m.store().StoreProject(p, true))
+}