{-

 * 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 Main where

import Prelude hiding (catch)

import Control.Monad

import Control.Exception

import System.IO

import Data.Maybe

import Database.MySQL.Simple

import Database.MySQL.Simple.QueryResults

import Database.MySQL.Simple.Result

import Data.List (lookup, elem)

import qualified Data.ByteString.Char8 as B

import Data.Word

import Data.Int

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

import CoreTypes

import Utils

dbQueryAccount =

    "SELECT CASE WHEN users.status = 1 THEN users.pass ELSE '' END, \

    \ (SELECT COUNT(users_roles.rid) FROM users_roles WHERE users.uid = users_roles.uid AND users_roles.rid = 3), \

    \ (SELECT COUNT(users_roles.rid) FROM users_roles WHERE users.uid = users_roles.uid AND users_roles.rid = 13) \

    \ FROM users WHERE users.name = ?"

dbQueryStats =

    "INSERT INTO gameserver_stats (players, rooms, last_update) VALUES (?, ?, UNIX_TIMESTAMP())"

dbQueryAchievement =

    "INSERT INTO achievements (time, typeid, userid, value, filename, location, protocol) \

    \ VALUES (?, (SELECT id FROM achievement_types WHERE name = ?), (SELECT uid FROM users WHERE name = ?), \

    \ ?, ?, ?, ?)"

dbQueryGamesHistory =

    "INSERT INTO rating_games (script, protocol, filename, time) \

    \ VALUES (?, ?, ?, ?)"

dbQueryGamesHistoryPlaces = "INSERT INTO rating_players (userid, gameid, place) \

    \ VALUES ((SELECT uid FROM users WHERE name = ?), LAST_INSERT_ID(), ?)"

dbQueryReplayFilename = "SELECT filename FROM achievements WHERE id = ?"

dbInteractionLoop dbConn = forever $ do

    q <- liftM read getLine

    hPutStrLn stderr $ show q

    case q of

        CheckAccount clId clUid clNick _ -> do

                results <- query dbConn dbQueryAccount $ Only clNick

                let response = case results of

                        [(pass, adm, contr)] ->

                            (

                                clId,

                                clUid,

                                HasAccount

                                    (pass)

                                    (adm == Just (1 :: Int))

                                    (contr == Just (1 :: Int))

                            )

                        _ ->

                            (clId, clUid, Guest)

                print response

                hFlush stdout

        GetReplayName clId clUid fileId -> do

                results <- query dbConn dbQueryReplayFilename $ Only fileId

                let fn = if null results then "" else fromOnly $ head results

                print (clId, clUid, ReplayName fn)

                hFlush stdout

        SendStats clients rooms ->

                void $ execute dbConn dbQueryStats (clients, rooms)

        StoreAchievements p fileName teams script info ->

            sequence_ $ parseStats dbConn p fileName teams script info

--readTime = read . B.unpack . B.take 19 . B.drop 8

readTime = B.take 19 . B.drop 8

parseStats :: 

    Connection

    -> Word16 

    -> B.ByteString 

    -> [(B.ByteString, B.ByteString)] 

    -> B.ByteString

    -> [B.ByteString]

    -> [IO Int64]

parseStats dbConn p fileName teams script = ps

    where

    time = readTime fileName

    ps :: [B.ByteString] -> [IO Int64]

    ps [] = []

    ps ("DRAW" : bs) = execute dbConn dbQueryGamesHistory (script, (fromIntegral p) :: Int, fileName, time)

        : executeMany dbConn dbQueryGamesHistoryPlaces (map drawParams teams)

        : ps bs

    ps ("WINNERS" : n : bs) = let winNum = readInt_ n in execute dbConn dbQueryGamesHistory (script, (fromIntegral p) :: Int, fileName, time)

        : executeMany dbConn dbQueryGamesHistoryPlaces (map (placeParams (take winNum bs)) teams)

        : ps (drop winNum bs)

    ps ("ACHIEVEMENT" : typ : teamname : location : value : bs) = execute dbConn dbQueryAchievement

        ( time

        , typ

        , fromMaybe "" (lookup teamname teams)

        , (readInt_ value) :: Int

        , fileName

        , location

        , (fromIntegral p) :: Int

        ) : ps bs

    ps (b:bs) = ps bs

    drawParams t = (snd t, 0 :: Int)

    placeParams winners t = (snd t, if (fst t) `elem` winners then 1 else 2 :: Int)

dbConnectionLoop mySQLConnectionInfo =

    Control.Exception.handle (\(e :: SomeException) -> hPutStrLn stderr $ show e) $

        bracket

            (connect mySQLConnectionInfo)

            close

            dbInteractionLoop

--processRequest :: DBQuery -> IO String

--processRequest (CheckAccount clId clUid clNick clHost) = return $ show (clclId, clUid, Guest)

main = do

        dbHost <- getLine

        dbName <- getLine

        dbLogin <- getLine

        dbPassword <- getLine

        let mySQLConnectInfo = defaultConnectInfo {

            connectHost = dbHost

            , connectDatabase = dbName

            , connectUser = dbLogin

            , connectPassword = dbPassword

            }

        dbConnectionLoop mySQLConnectInfo