fuzz tester: avoid deleting entire repository (had to happen eventually, right?)

Command/FuzzTest.hs

@@ -211,31 +211,37 @@ genFuzzAction = do
 
 existingFile :: Int -> FilePath -> IO (Maybe FuzzFile)
 existingFile 0 _ = return Nothing
-existingFile n top = maybe (return Nothing) (go . toFilePath) =<< existingDir
+existingFile n top = do
-  where
+	dir <- existingDirIncludingTop
-  	go dir = do
+	contents <- catchDefaultIO [] (getDirectoryContents dir)
-		contents <- catchDefaultIO [] (getDirectoryContents dir)
+	let files = filter isFuzzFile contents
-		let files = filter isFuzzFile contents
+	if null files
-		if null files
+		then do
-			then do
+			let dirs = filter isFuzzDir contents
-				let dirs = filter isFuzzDir contents
+			if null dirs
-				if null dirs
+				then return Nothing
-					then return Nothing
+				else do
-					else do
+					i <- getStdRandom $ randomR (0, length dirs - 1)
-						i <- getStdRandom $ randomR (0, length dirs - 1)
+					existingFile (n - 1) (top </> dirs !! i)
-						existingFile (n - 1) (top </> dirs !! i)
+		else do
-			else do
+			i <- getStdRandom $ randomR (0, length files - 1)
-				i <- getStdRandom $ randomR (0, length files - 1)
+			return $ Just $ FuzzFile $ top </> dir </> files !! i
-				return $ Just $ FuzzFile $ top </> dir </> files !! i
+
+existingDirIncludingTop :: IO FilePath
+existingDirIncludingTop = do
+	dirs <- filter isFuzzDir <$> getDirectoryContents "."
+	if null dirs
+		then return "."
+		else do
+			n <- getStdRandom $ randomR (0, length dirs)
+			return $ ("." : dirs) !! n
 
 existingDir :: IO (Maybe FuzzDir)
 existingDir = do
-	dirs <- filter isFuzzDir <$> getDirectoryContents "."
+	d <- existingDirIncludingTop
-	if null dirs
+	return $ if isFuzzDir d
-		then return Nothing
+		then Just $ FuzzDir d
-		else do
+		else Nothing
-			n <- getStdRandom $ randomR (0, length dirs)
-			return $ Just $ FuzzDir $ ("." : dirs) !! n
 
 newFile :: IO (Maybe FuzzFile)
 newFile = go (100 :: Int)