add post-repair actions

1 files changed, 19 insertions, 8 deletions

diff --git a/Assistant/RepoProblem.hs b/Assistant/RepoProblem.hs
index d2e5a5cf1..6913fefc6 100644
--- a/Assistant/RepoProblem.hs
+++ b/Assistant/RepoProblem.hs
@@ -8,16 +8,27 @@
 module Assistant.RepoProblem where
 
 import Assistant.Common
+import Assistant.Types.RepoProblem
 import Utility.TList
 
 import Control.Concurrent.STM
 
-{- Gets all repositories that have problems. 
- - Blocks until there is at least one. -}
-getRepoProblems :: Assistant [UUID]
-getRepoProblems = (atomically . getTList) <<~ repoProblemChan
+{- Gets all repositories that have problems. Blocks until there is at
+ - least one. -}
+getRepoProblems :: Assistant [RepoProblem]
+getRepoProblems = nubBy sameRepoProblem
+	<$> (atomically . getTList) <<~ repoProblemChan
 
-{- Indicates that there was a problem accessing a repo, and the problem
- - appears to not be a transient (eg network connection) problem. -}
-repoHasProblem :: UUID -> Assistant ()
-repoHasProblem r  = (atomically . flip consTList r) <<~ repoProblemChan
+{- Indicates that there was a problem with a repository, and the problem
+ - appears to not be a transient (eg network connection) problem.
+ -
+ - If the problem is able to be repaired, the passed action will be run.
+ - (However, if multiple problems are reported with a single repository,
+ - only a single action will be run.)
+ -}
+repoHasProblem :: UUID -> Assistant () -> Assistant ()
+repoHasProblem u afterrepair = do
+	rp <- RepoProblem
+		<$> pure u
+		<*> asIO afterrepair
+	(atomically . flip consTList rp) <<~ repoProblemChan