1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
module Network (
Network(..),
Node(..),
NodeName(..), textNodeName, unpackNodeName,
nextNodeName,
) where
import Control.Arrow
import Control.Concurrent
import Data.Text (Text)
import Data.Text qualified as T
import {-# SOURCE #-} Process
import Test
data Network = Network
{ netNodes :: MVar [Node]
, netProcesses :: MVar [Process]
, netDir :: FilePath
}
data Node = Node
{ nodeName :: NodeName
, nodeIp :: Text
, nodeNetwork :: Network
, nodeDir :: FilePath
}
data NodeName = NodeName Text Word
deriving (Eq, Ord)
textNodeName :: NodeName -> Text
textNodeName (NodeName name 0) = name
textNodeName (NodeName name num) = name <> T.pack "~" <> T.pack (show num)
unpackNodeName :: NodeName -> String
unpackNodeName = T.unpack . textNodeName
nextNodeName :: VarName -> [NodeName] -> NodeName
nextNodeName (VarName tname) = go 0
where
go n [] = NodeName tname n
go n (NodeName tname' m : ns) | tname == tname' = go (max n m + 1) ns
| otherwise = go n ns
instance ExprType Network where
textExprType _ = T.pack "network"
textExprValue _ = T.pack "s:0"
emptyVarValue = Network undefined undefined undefined
instance ExprType Node where
textExprType _ = T.pack "node"
textExprValue n = T.pack "n:" <> textNodeName (nodeName n)
emptyVarValue = Node (NodeName T.empty 0) T.empty undefined undefined
recordMembers = map (first T.pack)
[ ("ip", RecordSelector $ nodeIp)
]
|