gameServer/OfficialServer/GameReplayStore.hs
changeset 9662 47dbd9601342
parent 8511 4f899fbce66d
child 10051 cc6f62d7aea2
--- a/gameServer/OfficialServer/GameReplayStore.hs	Thu Oct 31 23:27:06 2013 +0400
+++ b/gameServer/OfficialServer/GameReplayStore.hs	Thu Oct 31 23:27:46 2013 +0400
@@ -17,13 +17,16 @@
 import EngineInteraction
 
 
-pickReplayFile :: Int -> IO String
-pickReplayFile p = do
-    files <- liftM (filter (isSuffixOf ('.' : show p))) $ getDirectoryContents "replays"
+pickReplayFile :: Int -> [String] -> IO String
+pickReplayFile p blackList = do
+    files <- liftM (filter (\f -> sameProto f && notBlacklisted f)) $ getDirectoryContents "replays"
     if (not $ null files) then
         return $ "replays/" ++ head files
         else
         return ""
+    where
+        sameProto = (isSuffixOf ('.' : show p))
+        notBlacklisted = flip notElem blackList
 
 saveReplay :: RoomInfo -> IO ()
 saveReplay r = do
@@ -38,9 +41,9 @@
             (\(e :: IOException) -> warningM "REPLAYS" $ "Couldn't write to " ++ fileName ++ ": " ++ show e)
 
 
-loadReplay :: Int -> IO (Maybe CheckInfo, [B.ByteString])
-loadReplay p = E.handle (\(e :: SomeException) -> warningM "REPLAYS" "Problems reading replay" >> return (Nothing, [])) $ do
-    fileName <- pickReplayFile p
+loadReplay :: Int -> [String] -> IO (Maybe CheckInfo, [B.ByteString])
+loadReplay p blackList = E.handle (\(e :: SomeException) -> warningM "REPLAYS" "Problems reading replay" >> return (Nothing, [])) $ do
+    fileName <- pickReplayFile p blackList
     if (not $ null fileName) then
         loadFile fileName
         else