author | unc0rr |
Thu, 11 Nov 2010 22:17:54 +0300 | |
branch | 0.9.14 |
changeset 4242 | 5e3c5fe2cb14 |
parent 3671 | a94d1dc4a8d9 |
child 4266 | bc6e9859f142 |
child 4295 | 1f5604cd99be |
permissions | -rw-r--r-- |
1804 | 1 |
module Utils where |
2 |
||
3 |
import Control.Concurrent |
|
4 |
import Control.Concurrent.STM |
|
5 |
import Data.Char |
|
6 |
import Data.Word |
|
7 |
import qualified Data.Map as Map |
|
8 |
import qualified Data.IntMap as IntMap |
|
2304 | 9 |
import qualified Data.Set as Set |
2310 | 10 |
import Data.ByteString.Internal (w2c) |
1917 | 11 |
import Numeric |
12 |
import Network.Socket |
|
1964 | 13 |
import System.IO |
1917 | 14 |
import qualified Data.List as List |
2349 | 15 |
import Control.Monad |
4242 | 16 |
import Maybe |
1804 | 17 |
------------------------------------------------- |
18 |
import qualified Codec.Binary.Base64 as Base64 |
|
4242 | 19 |
import qualified Data.ByteString.UTF8 as BUTF8 |
20 |
import qualified Data.ByteString as B |
|
1804 | 21 |
import CoreTypes |
22 |
||
1917 | 23 |
|
4242 | 24 |
sockAddr2String :: SockAddr -> IO String |
25 |
sockAddr2String (SockAddrInet _ hostAddr) = inet_ntoa hostAddr |
|
1917 | 26 |
sockAddr2String (SockAddrInet6 _ _ (a, b, c, d) _) = |
4242 | 27 |
return $ (foldr1 (.) |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2747
diff
changeset
|
28 |
$ List.intersperse (\a -> ':':a) |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2747
diff
changeset
|
29 |
$ concatMap (\n -> (\(a, b) -> [showHex a, showHex b]) $ divMod n 65536) [a, b, c, d]) [] |
1917 | 30 |
|
4242 | 31 |
toEngineMsg :: String -> String |
32 |
toEngineMsg msg = Base64.encode (fromIntegral (B.length encodedMsg) : (B.unpack encodedMsg)) |
|
33 |
where |
|
34 |
encodedMsg = BUTF8.fromString msg |
|
1804 | 35 |
|
4242 | 36 |
fromEngineMsg :: String -> Maybe String |
37 |
fromEngineMsg msg = liftM (map w2c) (Base64.decode msg >>= removeLength) |
|
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2747
diff
changeset
|
38 |
where |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2747
diff
changeset
|
39 |
removeLength (x:xs) = if length xs == fromIntegral x then Just xs else Nothing |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2747
diff
changeset
|
40 |
removeLength _ = Nothing |
2304 | 41 |
|
4242 | 42 |
checkNetCmd :: String -> (Bool, Bool) |
43 |
checkNetCmd msg = check decoded |
|
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2747
diff
changeset
|
44 |
where |
4242 | 45 |
decoded = fromEngineMsg msg |
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2747
diff
changeset
|
46 |
check Nothing = (False, False) |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2747
diff
changeset
|
47 |
check (Just (m:ms)) = (m `Set.member` legalMessages, m == '+') |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2747
diff
changeset
|
48 |
check _ = (False, False) |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2747
diff
changeset
|
49 |
legalMessages = Set.fromList $ "M#+LlRrUuDdZzAaSjJ,sFNpPwtghb12345" ++ slotMessages |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2747
diff
changeset
|
50 |
slotMessages = "\128\129\130\131\132\133\134\135\136\137\138" |
1804 | 51 |
|
52 |
maybeRead :: Read a => String -> Maybe a |
|
53 |
maybeRead s = case reads s of |
|
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2747
diff
changeset
|
54 |
[(x, rest)] | all isSpace rest -> Just x |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2747
diff
changeset
|
55 |
_ -> Nothing |
1804 | 56 |
|
4242 | 57 |
teamToNet :: Word16 -> TeamInfo -> [String] |
58 |
teamToNet protocol team |
|
59 |
| protocol < 30 = [ |
|
60 |
"ADD_TEAM", |
|
61 |
teamname team, |
|
62 |
teamgrave team, |
|
63 |
teamfort team, |
|
64 |
teamvoicepack team, |
|
65 |
teamowner team, |
|
66 |
show $ difficulty team |
|
67 |
] |
|
68 |
++ hhsInfo |
|
69 |
| otherwise = [ |
|
70 |
"ADD_TEAM", |
|
71 |
teamname team, |
|
72 |
teamgrave team, |
|
73 |
teamfort team, |
|
74 |
teamvoicepack team, |
|
75 |
teamflag team, |
|
76 |
teamowner team, |
|
77 |
show $ difficulty team |
|
78 |
] |
|
79 |
++ hhsInfo |
|
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2747
diff
changeset
|
80 |
where |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2747
diff
changeset
|
81 |
hhsInfo = concatMap (\(HedgehogInfo name hat) -> [name, hat]) $ hedgehogs team |
1804 | 82 |
|
83 |
modifyTeam :: TeamInfo -> RoomInfo -> RoomInfo |
|
84 |
modifyTeam team room = room{teams = replaceTeam team $ teams room} |
|
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2747
diff
changeset
|
85 |
where |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2747
diff
changeset
|
86 |
replaceTeam _ [] = error "modifyTeam: no such team" |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2747
diff
changeset
|
87 |
replaceTeam team (t:teams) = |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2747
diff
changeset
|
88 |
if teamname team == teamname t then |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2747
diff
changeset
|
89 |
team : teams |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2747
diff
changeset
|
90 |
else |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2747
diff
changeset
|
91 |
t : replaceTeam team teams |
1804 | 92 |
|
4242 | 93 |
illegalName :: String -> Bool |
94 |
illegalName = all isSpace |
|
2150
45b695f3a7b9
Forbid room names and nicknames consisting only of space characters
unc0rr
parents:
2113
diff
changeset
|
95 |
|
4242 | 96 |
protoNumber2ver :: Word16 -> String |
1804 | 97 |
protoNumber2ver 17 = "0.9.7-dev" |
98 |
protoNumber2ver 19 = "0.9.7" |
|
99 |
protoNumber2ver 20 = "0.9.8-dev" |
|
100 |
protoNumber2ver 21 = "0.9.8" |
|
101 |
protoNumber2ver 22 = "0.9.9-dev" |
|
102 |
protoNumber2ver 23 = "0.9.9" |
|
103 |
protoNumber2ver 24 = "0.9.10-dev" |
|
104 |
protoNumber2ver 25 = "0.9.10" |
|
1953 | 105 |
protoNumber2ver 26 = "0.9.11-dev" |
2113 | 106 |
protoNumber2ver 27 = "0.9.11" |
107 |
protoNumber2ver 28 = "0.9.12-dev" |
|
2448 | 108 |
protoNumber2ver 29 = "0.9.12" |
109 |
protoNumber2ver 30 = "0.9.13-dev" |
|
3297 | 110 |
protoNumber2ver 31 = "0.9.13" |
111 |
protoNumber2ver 32 = "0.9.14-dev" |
|
1804 | 112 |
protoNumber2ver _ = "Unknown" |
113 |
||
1964 | 114 |
askFromConsole :: String -> IO String |
115 |
askFromConsole msg = do |
|
2867
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2747
diff
changeset
|
116 |
putStr msg |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2747
diff
changeset
|
117 |
hFlush stdout |
9be6693c78cb
- Unbreak support for client versions prior to 0.9.13-dev
unc0rr
parents:
2747
diff
changeset
|
118 |
getLine |