Skip to content

Commit be10678

Browse files
committed
remote: add preStoreConfigToStoreConfig, use in handshake
1 parent 66ad803 commit be10678

File tree

2 files changed

+17
-4
lines changed

2 files changed

+17
-4
lines changed

hnix-store-remote/src/System/Nix/Store/Remote/Client.hs

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ import System.Nix.Store.Remote.Socket (sockPutS, sockGetS)
2626
import System.Nix.Store.Remote.Serializer (bool, enum, int, mapErrorS, protoVersion, text, trustedFlag, workerMagic)
2727
import System.Nix.Store.Remote.Types.Logger (Logger)
2828
import System.Nix.Store.Remote.Types.ProtoVersion (ProtoVersion(..), ourProtoVersion)
29-
import System.Nix.Store.Remote.Types.StoreConfig (PreStoreConfig(..), StoreConfig(..))
29+
import System.Nix.Store.Remote.Types.StoreConfig (PreStoreConfig, preStoreConfigToStoreConfig)
3030
import System.Nix.Store.Remote.Types.WorkerMagic (WorkerMagic(..))
3131
import System.Nix.Store.Remote.Types.WorkerOp (WorkerOp)
3232

@@ -83,7 +83,7 @@ runStoreSocket preStoreConfig code =
8383
runRemoteStoreT preStoreConfig $ do
8484
pv <- greet
8585
mapStoreConfig
86-
(\(PreStoreConfig a b) -> StoreConfig a pv b)
86+
(preStoreConfigToStoreConfig pv)
8787
code
8888

8989
where
@@ -141,9 +141,8 @@ runStoreSocket preStoreConfig code =
141141
return Nothing
142142

143143
-- TODO do something with it
144-
-- TODO patter match better
145144
_ <- mapStoreConfig
146-
(\(PreStoreConfig a b) -> StoreConfig a ourProtoVersion b)
145+
(preStoreConfigToStoreConfig minimumCommonVersion)
147146
processOutput
148147

149148
pure minimumCommonVersion

hnix-store-remote/src/System/Nix/Store/Remote/Types/StoreConfig.hs

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@ module System.Nix.Store.Remote.Types.StoreConfig
44
, StoreConfig(..)
55
, TestStoreConfig(..)
66
, HasStoreSocket(..)
7+
, preStoreConfigToStoreConfig
78
) where
89

910
import GHC.Generics (Generic)
@@ -56,3 +57,16 @@ instance HasProtoVersion TestStoreConfig where
5657

5758
instance HasStoreDir TestStoreConfig where
5859
hasStoreDir = testStoreConfig_dir
60+
61+
-- | Convert @PreStoreConfig@ to @StoreConfig@
62+
-- adding @ProtoVersion@ to latter
63+
preStoreConfigToStoreConfig
64+
:: ProtoVersion
65+
-> PreStoreConfig
66+
-> StoreConfig
67+
preStoreConfigToStoreConfig pv PreStoreConfig{..} =
68+
StoreConfig
69+
{ storeConfig_dir = preStoreConfig_dir
70+
, storeConfig_protoVersion = pv
71+
, storeConfig_socket = preStoreConfig_socket
72+
}

0 commit comments

Comments
 (0)