juvix/app/Commands/MicroJuvix.hs

module Commands.MicroJuvix where

import MiniJuvix.Prelude hiding (Doc)
import Options.Applicative

data MicroJuvixCommand
  = Pretty
  | TypeCheck MicroJuvixTypeOptions
  | Arity

newtype MicroJuvixTypeOptions = MicroJuvixTypeOptions
  { _microJuvixTypePrint :: Bool
  }

makeLenses ''MicroJuvixTypeOptions

parseMicroJuvixCommand :: Parser MicroJuvixCommand
parseMicroJuvixCommand =
  hsubparser $
    mconcat
      [ commandPretty,
        commandArity,
        commandTypeCheck
      ]
  where
    commandArity :: Mod CommandFields MicroJuvixCommand
    commandArity = command "arity" arityInfo

    commandPretty :: Mod CommandFields MicroJuvixCommand
    commandPretty = command "pretty" prettyInfo

    commandTypeCheck :: Mod CommandFields MicroJuvixCommand
    commandTypeCheck = command "typecheck" typeCheckInfo

    arityInfo :: ParserInfo MicroJuvixCommand
    arityInfo =
      info
        (pure Arity)
        (progDesc "Translate a MiniJuvix file to MicroJuvix and insert holes")

    prettyInfo :: ParserInfo MicroJuvixCommand
    prettyInfo =
      info
        (pure Pretty)
        (progDesc "Translate a MiniJuvix file to MicroJuvix and pretty print the result")

    typeCheckInfo :: ParserInfo MicroJuvixCommand
    typeCheckInfo =
      info
        (TypeCheck <$> parseMicroJuvixType)
        (progDesc "Translate a MiniJuvix file to MicroJuvix and typecheck the result")

parseMicroJuvixType :: Parser MicroJuvixTypeOptions
parseMicroJuvixType = do
  _microJuvixTypePrint <-
    switch
      ( long "print-result"
          <> help "Print the type checked module if successful"
      )
  pure MicroJuvixTypeOptions {..}