{- * Hedgewars, a free turn based strategy game * Copyright (c) 2004-2014 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 whereimport Prelude hiding (catch)import Control.Monadimport Control.Exceptionimport System.IOimport Data.Maybeimport Database.HDBCimport Database.HDBC.MySQLimport Data.List (lookup)import qualified Data.ByteString.Char8 as Bimport Data.Word--------------------------import CoreTypesimport UtilsdbQueryAccount = "SELECT users.pass, \ \ (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 = ?), \ \ ?, ?, ?, ?)"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 statement <- prepare dbConn dbQueryAccount execute statement [SqlByteString clNick] result <- fetchRow statement finish statement let response = if isJust result then let [pass, adm, contr] = fromJust result in ( clId, clUid, HasAccount (fromSql pass) (fromSql adm == Just (1 :: Int)) (fromSql contr == Just (1 :: Int)) ) else (clId, clUid, Guest) print response hFlush stdout GetReplayName clId clUid fileId -> do statement <- prepare dbConn dbQueryReplayFilename execute statement [SqlByteString fileId] result <- fetchRow statement finish statement let fn = if (isJust result) then fromJust . fromSql . head . fromJust $ result else "" print (clId, clUid, ReplayName fn) hFlush stdout SendStats clients rooms -> run dbConn dbQueryStats [SqlInt32 $ fromIntegral clients, SqlInt32 $ fromIntegral rooms] >> return ()--StoreAchievements (B.pack fileName) (map toPair teams) info StoreAchievements p fileName teams info -> mapM_ (run dbConn dbQueryAchievement) $ (parseStats p fileName teams) inforeadTime = read . B.unpack . B.take 19 . B.drop 8parseStats :: Word16 -> B.ByteString -> [(B.ByteString, B.ByteString)] -> [B.ByteString] -> [[SqlValue]]parseStats p fileName teams = ps where time = readTime fileName ps [] = [] ps ("DRAW" : bs) = ps bs ps ("WINNERS" : n : bs) = ps $ drop (readInt_ n) bs ps ("ACHIEVEMENT" : typ : teamname : location : value : bs) = [ SqlUTCTime time , SqlByteString typ , SqlByteString $ fromMaybe "" (lookup teamname teams) , SqlInt32 (readInt_ value) , SqlByteString fileName , SqlByteString location , SqlInt32 $ fromIntegral p ] : ps bs ps (b:bs) = ps bsdbConnectionLoop mySQLConnectionInfo = Control.Exception.handle (\(e :: IOException) -> hPutStrLn stderr $ show e) $ handleSqlError $ bracket (connectMySQL mySQLConnectionInfo) disconnect 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 = defaultMySQLConnectInfo {mysqlHost = dbHost, mysqlDatabase = dbName, mysqlUser = dbLogin, mysqlPassword = dbPassword} dbConnectionLoop mySQLConnectInfo