From 4d5087df14e79338727097f3fc6ca04777ef1b2e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Joosep=20J=C3=A4=C3=A4ger?= <joosep.jaager@iohk.io>
Date: Thu, 27 Mar 2025 18:34:10 +0200
Subject: [PATCH] Added prelude to gen, added help option

---
 bin/Main.hs | 70 ++++++++++++++++++++++++++++++++---------------------
 1 file changed, 43 insertions(+), 27 deletions(-)

diff --git a/bin/Main.hs b/bin/Main.hs
index 7e019f9..9270ca5 100644
--- a/bin/Main.hs
+++ b/bin/Main.hs
@@ -28,7 +28,7 @@ import Text.Megaparsec (ParseErrorBundle, Parsec, errorBundlePretty, runParser)
 
 data Opts = Opts Command String
 
-newtype ValidateOpts = ValidateOpts {noPrelude :: Bool}
+newtype ValidateOpts = ValidateOpts {vNoPrelude :: Bool}
 
 data Command
   = Format FormatOpts
@@ -53,6 +53,7 @@ pCBOROutputFormat = eitherReader $ \case
 data GenOpts = GenOpts
   { itemName :: T.Text
   , outputFormat :: CBOROutputFormat
+  , gNoPrelude :: Bool
   }
 
 pGenOpts :: Parser GenOpts
@@ -71,6 +72,10 @@ pGenOpts =
           <> help "Output format"
           <> value AsCBOR
       )
+    <*> switch
+      ( long "no-prelude"
+          <> help "Do not include the CDDL prelude."
+      )
 
 newtype FormatOpts = FormatOpts
   {sort :: Bool}
@@ -98,19 +103,19 @@ opts =
       ( command
           "format"
           ( info
-              (Format <$> pFormatOpts)
+              (Format <$> pFormatOpts <**> helper)
               (progDesc "Format the provided CDDL file")
           )
           <> command
             "validate"
             ( info
-                (Validate <$> pValidateOpts)
+                (Validate <$> pValidateOpts <**> helper)
                 (progDesc "Validate the provided CDDL file")
             )
           <> command
             "gen"
             ( info
-                (GenerateCBOR <$> pGenOpts)
+                (GenerateCBOR <$> pGenOpts <**> helper)
                 (progDesc "Generate a CBOR term matching the schema")
             )
       )
@@ -134,29 +139,40 @@ run (Opts cmd cddlFile) = do
     Left err -> do
       putStrLnErr $ errorBundlePretty err
       exitFailure
-    Right res -> case cmd of
-      Format fOpts ->
-        let defs = if sort fOpts then sortCDDL res else res
-         in putDocW 80 $ pretty defs
-      Validate vOpts ->
-        let
-          resWithPrelude
-            | noPrelude vOpts = res
-            | otherwise = prependPrelude res
-         in
-          case fullResolveCDDL resWithPrelude of
-            Left err -> putStrLnErr (show err) >> exitFailure
-            Right _ -> exitSuccess
-      (GenerateCBOR x) -> case fullResolveCDDL res of
-        Left err -> putStrLnErr (show err) >> exitFailure
-        Right mt -> do
-          stdGen <- getStdGen
-          let term = generateCBORTerm mt (Name $ itemName x) stdGen
-           in case outputFormat x of
-                AsTerm -> print term
-                AsFlatTerm -> print $ toFlatTerm (encodeTerm term)
-                AsCBOR -> BSC.putStrLn . Base16.encode . toStrictByteString $ encodeTerm term
-                AsPrettyCBOR -> putStrLn . prettyHexEnc $ encodeTerm term
+    Right res ->
+      case cmd of
+        Format fOpts ->
+          let
+            defs
+              | sort fOpts = sortCDDL res
+              | otherwise = res
+           in
+            putDocW 80 $ pretty defs
+        Validate vOpts ->
+          let
+            res'
+              | vNoPrelude vOpts = res
+              | otherwise = prependPrelude res
+           in
+            case fullResolveCDDL res' of
+              Left err -> putStrLnErr (show err) >> exitFailure
+              Right _ -> exitSuccess
+        (GenerateCBOR gOpts) ->
+          let
+            res'
+              | gNoPrelude gOpts = res
+              | otherwise = prependPrelude res
+           in
+            case fullResolveCDDL res' of
+              Left err -> putStrLnErr (show err) >> exitFailure
+              Right mt -> do
+                stdGen <- getStdGen
+                let term = generateCBORTerm mt (Name $ itemName gOpts) stdGen
+                 in case outputFormat gOpts of
+                      AsTerm -> print term
+                      AsFlatTerm -> print $ toFlatTerm (encodeTerm term)
+                      AsCBOR -> BSC.putStrLn . Base16.encode . toStrictByteString $ encodeTerm term
+                      AsPrettyCBOR -> putStrLn . prettyHexEnc $ encodeTerm term
 
 putStrLnErr :: String -> IO ()
 putStrLnErr = hPutStrLn stderr