Merge pull request #110 from zkFold/vlasin-add-proveAPI

Add proveAPI
zkFold · May 14, 2024 · bd28079 · bd28079
2 parents a6f76af + 7e4c20d
commit bd28079
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/src/ZkFold/Base/Protocol/NonInteractiveProof.hs b/src/ZkFold/Base/Protocol/NonInteractiveProof.hs
@@ -1,11 +1,14 @@
 {-# LANGUAGE AllowAmbiguousTypes #-}
+{-# LANGUAGE DeriveAnyClass      #-}
 {-# LANGUAGE TypeApplications    #-}
 
 module ZkFold.Base.Protocol.NonInteractiveProof where
 
+import           Control.DeepSeq             (NFData)
 import           Crypto.Hash.SHA256          (hash)
 import           Data.ByteString             (ByteString, cons)
 import           Data.Maybe                  (fromJust)
+import           GHC.Generics                (Generic)
 import           Numeric.Natural             (Natural)
 import           Prelude
 
@@ -58,3 +61,23 @@ class NonInteractiveProof a where
 
     verify :: Setup a -> Input a -> Proof a -> Bool
 
+data ProveAPIResult = ProveAPISuccess ByteString | ProveAPIErrorSetup | ProveAPIErrorWitness
+    deriving (Show, Eq, Generic, NFData)
+
+proveAPI
+    :: forall a
+    . (NonInteractiveProof a
+    , Binary (Setup a)
+    , Binary (Witness a)
+    , Binary (Input a)
+    , Binary (Proof a))
+    => ByteString
+    -> ByteString
+    -> ProveAPIResult
+proveAPI bsS bsW =
+    let mS = fromByteString bsS
+        mW = fromByteString bsW
+    in case (mS, mW) of
+        (Nothing, _)     -> ProveAPIErrorSetup
+        (_, Nothing)     -> ProveAPIErrorWitness
+        (Just s, Just w) -> ProveAPISuccess $ toByteString $ prove @a s w