Handle double woth

Author: L-P

tracker/hints.go

@@ -12,12 +12,25 @@ import (
 	"github.com/lithammer/fuzzysearch/fuzzy"
 )
 
+const doubleWOTHMarker = "*"
+
 func (tracker *Tracker) AddWOTH(str string) bool {
 	if len(tracker.woths) >= 5 {
 		log.Printf("warning: WotHs maxed out at 5")
 		return false
 	}
 
+	// Handle Double-WOTH
+	// Each hint is doubled, two WoTH mean you have either seen all of them
+	// _or_ there might be two WoTH hints for the same area.
+	// A third WOTH will add a new line.
+	for k, v := range tracker.woths {
+		if v == str {
+			tracker.woths[k] += doubleWOTHMarker
+			return true
+		}
+	}
+
 	tracker.woths = append(tracker.woths, str)
 	return true
 }

tracker/undo.go

@@ -1,6 +1,9 @@
 package tracker
 
-import "log"
+import (
+	"log"
+	"strings"
+)
 
 // undoStackEntry represents an action (upgrade/downgrade) that happened on an item.
 type undoStackEntry struct {
@@ -47,7 +50,7 @@ func (tracker *Tracker) undo() {
 	if entry.IsHint {
 		switch entry.HintType {
 		case hintTypeWOTH:
-			tracker.woths = tracker.woths[:len(tracker.woths)-1]
+			tracker.undoWOTH(entry)
 		case hintTypeBarren:
 			tracker.barrens = tracker.barrens[:len(tracker.barrens)-1]
 		case hintTypeSometimes:
@@ -66,6 +69,19 @@ func (tracker *Tracker) undo() {
 	}
 }
 
+func (tracker *Tracker) undoWOTH(entry undoStackEntry) {
+	// Clear last added double woth or last woth
+	double := entry.HintText + doubleWOTHMarker
+	for i := len(tracker.woths) - 1; i >= 0; i-- {
+		if tracker.woths[i] == double {
+			tracker.woths[i] = strings.TrimSuffix(tracker.woths[i], doubleWOTHMarker)
+			return
+		}
+	}
+
+	tracker.woths = tracker.woths[:len(tracker.woths)-1]
+}
+
 func (tracker *Tracker) redo() {
 	if len(tracker.redoStack) == 0 {
 		log.Printf("no action to redo")