mirror of https://github.com/voidlizard/hbs2
157 lines
5.4 KiB
Haskell
157 lines
5.4 KiB
Haskell
module HBS2.CLI.Run.RefLog where
|
|
|
|
import HBS2.CLI.Prelude
|
|
import HBS2.CLI.Run.Internal
|
|
import HBS2.CLI.Run.Internal.KeyMan
|
|
|
|
import HBS2.Data.Types.Refs
|
|
import HBS2.Storage
|
|
import HBS2.Peer.CLI.Detect
|
|
import HBS2.Peer.RPC.Client.Unix
|
|
import HBS2.Peer.RPC.API.Peer
|
|
import HBS2.Peer.RPC.API.RefLog
|
|
|
|
import HBS2.Peer.Proto hiding (request)
|
|
import HBS2.Base58
|
|
import HBS2.Net.Auth.Credentials
|
|
import HBS2.Net.Auth.Schema()
|
|
|
|
import HBS2.KeyMan.Keys.Direct
|
|
import HBS2.KeyMan.App.Types
|
|
|
|
import Codec.Serialise
|
|
import Data.Coerce
|
|
import Data.Either
|
|
import Data.ByteString qualified as BS
|
|
import Data.ByteString.Char8 qualified as BS8
|
|
import Data.ByteString (ByteString)
|
|
import Data.ByteString.Lazy.Char8 qualified as LBS8
|
|
import Data.ByteString.Lazy qualified as LBS
|
|
import Data.Text.Encoding qualified as TE
|
|
import Data.Text qualified as Text
|
|
import Control.Monad.Trans.Cont
|
|
|
|
getCredentialsForReflog :: MonadUnliftIO m => RefLogKey 'HBS2Basic -> m (PeerCredentials 'HBS2Basic)
|
|
getCredentialsForReflog reflog = do
|
|
runKeymanClient (loadCredentials reflog)
|
|
>>= orThrowUser "credentials not found"
|
|
|
|
mkRefLogUpdateFrom :: (MonadUnliftIO m) => RefLogKey 'HBS2Basic -> m ByteString -> m (RefLogUpdate L4Proto)
|
|
mkRefLogUpdateFrom reflog mbs = do
|
|
what <- getCredentialsForReflog reflog
|
|
let puk = view peerSignPk what
|
|
let privk = view peerSignSk what
|
|
txraw <- mbs
|
|
makeRefLogUpdate @L4Proto @'HBS2Basic (coerce puk) privk txraw
|
|
|
|
|
|
reflogEntries :: forall c m . (c ~ C, IsContext c, MonadUnliftIO m) => MakeDictM c m ()
|
|
reflogEntries = do
|
|
|
|
entry $ bindMatch "hbs2:reflog:create" $ \case
|
|
[] -> do
|
|
reflog <- keymanNewCredentials (Just "reflog") 0
|
|
|
|
flip runContT pure do
|
|
so <- detectRPC `orDie` "rpc not found"
|
|
api <- ContT $ withRPC2 @PeerAPI @UNIX so
|
|
void $ callService @RpcPollAdd api (reflog, "reflog", 31)
|
|
pure $ mkStr (show $ pretty (AsBase58 reflog))
|
|
|
|
_ -> throwIO (BadFormException @C nil)
|
|
|
|
|
|
entry $ bindMatch "hbs2:reflog:tx:annhashref:create" $ \case
|
|
[StringLike puk, StringLike hash] -> do
|
|
flip runContT pure do
|
|
reflog <- orThrowUser "bad reflog key" (fromStringMay puk)
|
|
sto <- ContT withPeerStorage
|
|
hashref <- orThrowUser "bad hash" (fromStringMay @HashRef hash)
|
|
void $ hasBlock sto (fromHashRef hashref) `orDie` "no block"
|
|
let sref = AnnotatedHashRef Nothing hashref
|
|
rlu <- lift $ mkRefLogUpdateFrom reflog (pure $ LBS.toStrict $ serialise sref) <&> serialise
|
|
pure $ mkForm "blob" [mkStr (LBS8.unpack rlu)]
|
|
|
|
_ -> throwIO (BadFormException @C nil)
|
|
|
|
|
|
entry $ bindMatch "hbs2:reflog:tx:post" $ nil_ \case
|
|
[BlobLike blob] -> do
|
|
so <- detectRPC `orDie` "no rpc found"
|
|
withRPC2 @RefLogAPI so $ \caller -> do
|
|
wtf <- deserialiseOrFail @(RefLogUpdate L4Proto) (LBS.fromStrict blob)
|
|
& orThrowUser "invalid tx"
|
|
void $ callService @RpcRefLogPost caller wtf
|
|
|
|
_ -> throwIO (BadFormException @C nil)
|
|
|
|
entry $ bindMatch "hbs2:reflog:tx:seqref:create" $ \case
|
|
[StringLike puk, LitIntVal sn, StringLike hash] -> do
|
|
flip runContT pure do
|
|
reflog <- orThrowUser "bad reflog key" (fromStringMay puk)
|
|
sto <- ContT withPeerStorage
|
|
hashref <- orThrowUser "bad hash" (fromStringMay @HashRef hash)
|
|
void $ hasBlock sto (fromHashRef hashref) `orDie` "no block"
|
|
let sref = SequentialRef sn (AnnotatedHashRef Nothing hashref)
|
|
rlu <- lift $ mkRefLogUpdateFrom reflog (pure $ LBS.toStrict $ serialise sref) <&> serialise
|
|
pure $ mkForm "blob" [mkStr (LBS8.unpack rlu)]
|
|
|
|
_ -> throwIO (BadFormException @C nil)
|
|
|
|
entry $ bindMatch "hbs2:reflog:tx:create-raw" $ \case
|
|
[SymbolVal "stdin", StringLike rlo] -> do
|
|
reflog <- orThrowUser "bad reflog" (fromStringMay rlo)
|
|
|
|
rlu <- mkRefLogUpdateFrom reflog ( liftIO BS.getContents )
|
|
<&> serialise
|
|
|
|
pure $ mkForm "blob" [mkStr (LBS8.unpack rlu)]
|
|
|
|
[LitStrVal s, StringLike rlo] -> do
|
|
reflog <- orThrowUser "bad reflog" (fromStringMay rlo)
|
|
|
|
rlu <- mkRefLogUpdateFrom reflog ( pure (BS8.pack (Text.unpack s)) )
|
|
<&> serialise
|
|
|
|
pure $ mkForm "blob" [mkStr (LBS8.unpack rlu)]
|
|
|
|
_ -> throwIO (BadFormException @C nil)
|
|
|
|
|
|
entry $ bindMatch "hbs2:reflog:get" $ \case
|
|
[StringLike puk] -> do
|
|
|
|
flip runContT pure do
|
|
reflog <- orThrowUser "bad reflog key" (fromStringMay puk)
|
|
so <- detectRPC `orDie` "rpc not found"
|
|
api <- ContT $ withRPC2 @RefLogAPI @UNIX so
|
|
what <- callService @RpcRefLogGet api reflog
|
|
>>= orThrowUser "can't get reflog"
|
|
pure $ mkStr (show $ pretty what)
|
|
|
|
_ -> throwIO (BadFormException @C nil)
|
|
|
|
entry $ bindMatch "hbs2:reflog:fetch" $ \case
|
|
[StringLike puk] -> do
|
|
flip runContT pure do
|
|
reflog <- orThrowUser "bad reflog key" (fromStringMay puk)
|
|
so <- detectRPC `orDie` "rpc not found"
|
|
api <- ContT $ withRPC2 @RefLogAPI @UNIX so
|
|
void $ callService @RpcRefLogFetch api reflog
|
|
pure $ mkStr "okay"
|
|
|
|
_ -> throwIO (BadFormException @C nil)
|
|
|
|
entry $ bindMatch "hbs2:reflog:list" $ \case
|
|
[] -> do
|
|
flip runContT pure do
|
|
so <- detectRPC `orDie` "rpc not found"
|
|
api <- ContT $ withRPC2 @PeerAPI @UNIX so
|
|
r <- callService @RpcPollList2 api (Just "reflog", Nothing)
|
|
>>= orThrowUser "can't get reflog list"
|
|
pure $ mkList $ fmap (mkStr . show . pretty . AsBase58 . view _1) r
|
|
|
|
_ -> throwIO (BadFormException @C nil)
|
|
|
|
|