--server needs a port

1 files changed, 4 insertions, 2 deletions

diff --git a/CmdLine.hs b/CmdLine.hs
index 283e768..c15ca83 100644
--- a/CmdLine.hs
+++ b/CmdLine.hs
@@ -11,6 +11,7 @@ import qualified Gpg
 import Options.Applicative
 import qualified Data.ByteString.UTF8 as BU8
 import System.Directory
+import Network.Wai.Handler.Warp (Port)
 
 data CmdLine = CmdLine
 	{ mode :: Maybe Mode
@@ -21,7 +22,7 @@ data CmdLine = CmdLine
 	, customShareParams :: Maybe ShareParams
 	}
 
-data Mode = Backup | Restore | UploadQueued | Server | Benchmark
+data Mode = Backup | Restore | UploadQueued | Server Port | Benchmark
 	deriving (Show)
 
 parse :: Parser CmdLine
@@ -45,8 +46,9 @@ parse = CmdLine
 		( long "uploadqueued"
 		<> help "Upload any data to servers that was queued by a previous --backup run."
 		)
-	server = flag' Server
+	server = Server <$> option auto
 		( long "server"
+		<> metavar "PORT"
 		<> help "Run as a keysafe server, accepting objects and storing them to ~/.keysafe/objects/local/"
 		)
 	benchmark = flag' Benchmark