nixos-compose/src/Options.hs at 277f727eeabb8e5283ce2801defefaba3365f849 · garnix-io/nixos-compose · GitHub

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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
module Options
  ( parser,
    Options (..),
    Verbosity (..),
    Command (..),
    StartOptions (..),
    VmName (..),
  )
where

import Data.List.NonEmpty (NonEmpty)
import Data.List.NonEmpty qualified as NonEmpty
import Data.Text
import Options.Applicative
import StdLib

newtype Options = Options Command
  deriving stock (Show)

data Command
  = List
  | Start {verbosity :: Verbosity, options :: StartOptions}
  | Ssh {vmName :: VmName, sshCommand :: [Text]}
  | Status {vmNames :: [VmName]}
  | Stop {vmName :: VmName}
  deriving stock (Show, Generic)

data Verbosity
  = DefaultVerbosity
  | Verbose
  deriving stock (Show)

parseVerbosity :: Parser Verbosity
parseVerbosity =
  flag
    DefaultVerbosity
    Verbose
    ( long "verbose"
        <> short 'v'
        <> help "increase verbosity"
    )

data StartOptions
  = StartAll
  | StartSome (NonEmpty VmName)
  deriving stock (Show, Generic)

parseStartOptions :: Parser StartOptions
parseStartOptions =
  flag' StartAll (long "all")
    <|> (StartSome . NonEmpty.fromList <$> some parseVmName)

parser :: ParserInfo Options
parser =
  info p mempty
  where
    p =
      Options
        <$> hsubparser
          ( command
              "list"
              ( info
                  (pure List)
                  (fullDesc <> progDesc "List all configured vms")
              )
              <> command
                "start"
                ( info
                    (Start <$> parseVerbosity <*> parseStartOptions)
                    (fullDesc <> progDesc "Start a development vm")
                )
              <> command
                "ssh"
                ( info
                    (Ssh <$> parseVmName <*> many (argument str (metavar "SSH_COMMAND")))
                    (fullDesc <> progDesc "`ssh` into a running vm")
                )
              <> command
                "status"
                ( info
                    (Status <$> many parseVmName)
                    (fullDesc <> progDesc "Show the status of running vms")
                )
              <> command
                "stop"
                ( info
                    (Stop <$> parseVmName)
                    (progDesc "Stop a running vm")
                )
          )

newtype VmName = VmName {vmNameToText :: Text}
  deriving stock (Eq, Show, Ord)

parseVmName :: Parser VmName
parseVmName = VmName <$> argument str (metavar "VM_NAME")