Rust codegen

lambda-buffers-codegen/app/LambdaBuffers/Codegen/Cli/GenRust.hs

@@ -0,0 +1,49 @@
+module LambdaBuffers.Codegen.Cli.GenRust (GenOpts (..), gen) where
+
+import Control.Lens (makeLenses, (^.))
+import Control.Monad (unless)
+import Data.Aeson (decodeFileStrict')
+import LambdaBuffers.Codegen.Cli.Gen (logError)
+import LambdaBuffers.Codegen.Cli.Gen qualified as Gen
+import LambdaBuffers.Codegen.Rust (runPrint)
+import LambdaBuffers.Codegen.Rust.Config qualified as R
+import Paths_lambda_buffers_codegen qualified as Paths
+import System.Directory (doesFileExist)
+import System.Directory.Internal.Prelude (exitFailure)
+
+data GenOpts = MkGenOpts
+  { _config :: [FilePath]
+  , _common :: Gen.GenOpts
+  }
+
+makeLenses 'MkGenOpts
+
+gen :: GenOpts -> IO ()
+gen opts = do
+  cfg <- case opts ^. config of
+    [] -> do
+      fp <- Paths.getDataFileName "data/rust-prelude-base.json"
+      readRustConfig fp
+    fps -> do
+      cfgs <- traverse readRustConfig fps
+      return (mconcat cfgs)
+
+  Gen.gen
+    (opts ^. common)
+    (\ci -> fmap (\(fp, code, deps) -> Gen.Generated fp code deps) . runPrint cfg ci <$> (ci ^. #modules))
+
+readRustConfig :: FilePath -> IO R.Config
+readRustConfig f = do
+  fExists <- doesFileExist f
+  unless
+    fExists
+    ( do
+        logError "" $ "Provided Rust Codegen configuration file doesn't exists: " <> f
+        exitFailure
+    )
+  mayCfg <- decodeFileStrict' f
+  case mayCfg of
+    Nothing -> do
+      logError "" $ "Invalid Rust configuration file " <> f
+      exitFailure
+    Just cfg -> return cfg

lambda-buffers-codegen/app/Main.hs

@@ -6,6 +6,7 @@ import LambdaBuffers.Codegen.Cli.Gen (GenOpts (GenOpts))
 import LambdaBuffers.Codegen.Cli.GenHaskell qualified as Haskell
 import LambdaBuffers.Codegen.Cli.GenPlutarch qualified as Plutarch
 import LambdaBuffers.Codegen.Cli.GenPurescript qualified as Purescript
+import LambdaBuffers.Codegen.Cli.GenRust qualified as Rust
 import Options.Applicative (
   InfoMod,
   Parser,
@@ -36,6 +37,7 @@ data Command
   = GenHaskell Haskell.GenOpts
   | GenPurescript Purescript.GenOpts
   | GenPlutarch Plutarch.GenOpts
+  | GenRust Rust.GenOpts
 
 genOptsP :: Parser GenOpts
 genOptsP =
@@ -115,6 +117,19 @@ plutarchGenOptsP =
       )
     <*> genOptsP
 
+rustGenOptsP :: Parser Rust.GenOpts
+rustGenOptsP =
+  Rust.MkGenOpts
+    <$> many
+      ( strOption
+          ( long "config"
+              <> short 'c'
+              <> metavar "FILEPATH"
+              <> help "Configuration file for the Rust Codegen module (multiple `config`s are merged with left first merge conflict strategy)"
+          )
+      )
+    <*> genOptsP
+
 mkProgDesc :: forall {a}. String -> InfoMod a
 mkProgDesc backend =
   progDesc $
@@ -144,6 +159,12 @@ commandP =
             (GenPlutarch <$> (helper *> plutarchGenOptsP))
             (mkProgDesc "Plutarch")
         )
+      <> command
+        "gen-rust"
+        ( info
+            (GenRust <$> (helper *> rustGenOptsP))
+            (mkProgDesc "Rust")
+        )
 
 parserInfo :: ParserInfo Command
 parserInfo = info (commandP <**> helper) (fullDesc <> progDesc "LambdaBuffers Codegen command-line interface tool")
@@ -156,3 +177,4 @@ main = do
     GenHaskell opts -> Haskell.gen opts
     GenPurescript opts -> Purescript.gen opts
     GenPlutarch opts -> Plutarch.gen opts
+    GenRust opts -> Rust.gen opts

lambda-buffers-codegen/build.nix

@@ -45,6 +45,9 @@ _:
         lbg-plutarch = pkgs.writeShellScriptBin "lbg-plutarch" ''
           ${config.packages.lbg}/bin/lbg gen-plutarch $@
         '';
+        lbg-rust = pkgs.writeShellScriptBin "lbg-rust" ''
+          ${config.packages.lbg}/bin/lbg gen-rust $@
+        '';
 
         codegen-configs = pkgs.stdenv.mkDerivation {
           name = "codegen-configs";

lambda-buffers-codegen/data/rust-plutus-pla.json

@@ -0,0 +1,185 @@
+{
+  "opaquesConfig": {
+    "Plutus.V1.PlutusData": [
+      "plutus-ledger-api",
+      "plutus_data",
+      "PlutusData"
+    ],
+    "Plutus.V1.Address": [
+      "plutus-ledger-api",
+      "v1",
+      "address",
+      "Address"
+    ],
+    "Plutus.V1.Credential": [
+      "plutus-ledger-api",
+      "v1",
+      "address",
+      "Credential"
+    ],
+    "Plutus.V1.StakingCredential": [
+      "plutus-ledger-api",
+      "v1",
+      "address",
+      "StakingCredential"
+    ],
+    "Plutus.V1.PubKeyHash": [
+      "plutus-ledger-api",
+      "v1",
+      "crypto",
+      "Ed25519PubKeyHash"
+    ],
+    "Plutus.V1.Bytes": [
+      "plutus-ledger-api",
+      "v1",
+      "crypto",
+      "LedgerBytes"
+    ],
+    "Plutus.V1.Interval": [
+      "plutus-ledger-api",
+      "v1",
+      "interval",
+      "Interval"
+    ],
+    "Plutus.V1.Extended": [
+      "plutus-ledger-api",
+      "v1",
+      "interval",
+      "Extended"
+    ],
+    "Plutus.V1.LowerBound": [
+      "plutus-ledger-api",
+      "v1",
+      "interval",
+      "LowerBound"
+    ],
+    "Plutus.V1.UpperBound": [
+      "plutus-ledger-api",
+      "v1",
+      "interval",
+      "UpperBound"
+    ],
+    "Plutus.V1.POSIXTime": [
+      "plutus-ledger-api",
+      "v1",
+      "transaction",
+      "POSIXTime"
+    ],
+    "Plutus.V1.POSIXTimeRange": [
+      "plutus-ledger-api",
+      "v1",
+      "transaction",
+      "POSIXTimeRange"
+    ],
+    "Plutus.V1.Value": [
+      "plutus-ledger-api",
+      "v1",
+      "value",
+      "Value"
+    ],
+    "Plutus.V1.CurrencySymbol": [
+      "plutus-ledger-api",
+      "v1",
+      "value",
+      "CurrencySymbol"
+    ],
+    "Plutus.V1.AssetClass": [
+      "plutus-ledger-api",
+      "v1",
+      "value",
+      "AssetClass"
+    ],
+    "Plutus.V1.TokenName": [
+      "plutus-ledger-api",
+      "v1",
+      "value",
+      "TokenName"
+    ],
+    "Plutus.V1.Redeemer": [
+      "plutus-ledger-api",
+      "v1",
+      "redeemer",
+      "Redeemer"
+    ],
+    "Plutus.V1.Datum": [
+      "plutus-ledger-api",
+      "v1",
+      "datum",
+      "Datum"
+    ],
+    "Plutus.V1.DatumHash": [
+      "plutus-ledger-api",
+      "v1",
+      "datum",
+      "DatumHash"
+    ],
+    "Plutus.V1.RedeemerHash": [
+      "plutus-ledger-api",
+      "v1",
+      "redeemer",
+      "RedeemerHash"
+    ],
+    "Plutus.V1.ScriptHash": [
+      "plutus-ledger-api",
+      "v1",
+      "script",
+      "ScriptHash"
+    ],
+    "Plutus.V1.TxInInfo": [
+      "plutus-ledger-api",
+      "v1",
+      "transaction",
+      "TxInInfo"
+    ],
+    "Plutus.V1.TxId": [
+      "plutus-ledger-api",
+      "v1",
+      "transaction",
+      "TransactionHash"
+    ],
+    "Plutus.V1.TxOut": [
+      "plutus-ledger-api",
+      "v1",
+      "transaction",
+      "TransactionOutput"
+    ],
+    "Plutus.V1.TxOutRef": [
+      "plutus-ledger-api",
+      "v1",
+      "transaction",
+      "TransactionInput"
+    ],
+    "Plutus.V1.Map": [
+      "std",
+      "collections",
+      "BTreeMap"
+    ],
+    "Plutus.V2.TxInInfo": [
+      "plutus-ledger-api",
+      "v2",
+      "transaction",
+      "TxInInfo"
+    ],
+    "Plutus.V2.OutputDatum": [
+      "plutus-ledger-api",
+      "v2",
+      "datum",
+      "OutputDatum"
+    ],
+    "Plutus.V2.TxOut": [
+      "plutus-ledger-api",
+      "v2",
+      "transaction",
+      "TransactionOutput"
+    ]
+  },
+  "classesConfig": {
+    "Plutus.V1.PlutusData": [
+      [
+        "plutus-ledger-api",
+        "plutus_data",
+        "IsPlutusData"
+      ]
+    ]
+  }
+}

lambda-buffers-codegen/data/rust-prelude-base.json

@@ -0,0 +1,72 @@
+{
+  "opaquesConfig": {
+    "Prelude.Map": [
+      "std",
+      "collections",
+      "BTreeMap"
+    ],
+    "Prelude.Set": [
+      "std",
+      "collections",
+      "BTreeSet"
+    ],
+    "Prelude.List": [
+      "std",
+      "vec",
+      "Vec"
+    ],
+    "Prelude.Either": [
+      "std",
+      "result",
+      "Result"
+    ],
+    "Prelude.Maybe": [
+      "std",
+      "option",
+      "Option"
+    ],
+    "Prelude.Bytes": [
+      "std",
+      "vec",
+      "Vec<u8>"
+    ],
+    "Prelude.Text": [
+      "std",
+      "string",
+      "String"
+    ],
+    "Prelude.Char": [
+      "char"
+    ],
+    "Prelude.Integer": [
+      "num_bigint",
+      "BigInt"
+    ],
+    "Prelude.Bool": [
+      "std",
+      "primitive",
+      "bool"
+    ]
+  },
+  "classesConfig": {
+    "Prelude.Eq": [
+      [
+        "std",
+        "cmp",
+        "PartialEq"
+      ],
+      [
+        "std",
+        "cmp",
+        "Eq"
+      ]
+    ],
+    "Prelude.Json": [
+      [
+        "lbr-prelude",
+        "json",
+        "Json"
+      ]
+    ]
+  }
+}