{-

 * Hedgewars, a free turn based strategy game

 * Copyright (c) 2004-2015 Andrey Korotaev <unC0Rr@gmail.com>

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation; version 2 of the License

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

 \-}

{-# LANGUAGE ScopedTypeVariables, OverloadedStrings #-}

module OfficialServer.GameReplayStore where

import Data.Time

import Control.Exception as E

import qualified Data.Map as Map

import Data.Sequence()

import System.Log.Logger

import Data.Maybe

import Data.Unique

import Control.Monad

import Data.List

import qualified Data.ByteString as B

import System.Directory

import Control.DeepSeq

---------------

import CoreTypes

import EngineInteraction

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

    let gi = fromJust $ gameInfo r

    when (allPlayersHaveRegisteredAccounts gi) $ do

        time <- getCurrentTime

        u <- liftM hashUnique newUnique

        let fileName = "replays/" ++ show time ++ "-" ++ show u ++ "." ++ show (roomProto r)

        let replayInfo = (teamsAtStart gi, Map.toList $ mapParams r, Map.toList $ params r, roundMsgs gi)

        E.catch

            (writeFile fileName (show replayInfo))

            (\(e :: IOException) -> warningM "REPLAYS" $ "Couldn't write to " ++ fileName ++ ": " ++ show e)

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

        return (Nothing, [])

    where

        loadFile :: String -> IO (Maybe CheckInfo, [B.ByteString])

        loadFile fileName = E.handle (\(e :: SomeException) ->

                    warningM "REPLAYS" ("Problems reading " ++ fileName ++ ": " ++ show e) >> return (Just $ CheckInfo fileName [] "", [])) $ do

            (teams, params1, params2, roundMsgs) <- liftM read $ readFile fileName

            let d = replayToDemo teams (Map.fromList params1) (Map.fromList params2) (reverse roundMsgs)

            d `deepseq` return $ (

                Just (CheckInfo fileName teams (head $ fst d))

                , snd d

                )

moveFailedRecord :: String -> IO ()

moveFailedRecord fn = E.handle (\(e :: SomeException) -> warningM "REPLAYS" $ show e) $

    renameFile fn ("failed/" ++ drop 8 fn)

moveCheckedRecord :: String -> IO ()

moveCheckedRecord fn = E.handle (\(e :: SomeException) -> warningM "REPLAYS" $ show e) $

    renameFile fn ("checked/" ++ drop 8 fn)