From e688c256109e20ed5ea29fc03e97433acf7002bf Mon Sep 17 00:00:00 2001
From: David <david@taiko.xyz>
Date: Fri, 3 Nov 2023 14:45:49 +0800
Subject: [PATCH] feat(prover): improve `/status` API (#444)

---
 prover/server/api.go      | 9 +++++++++
 prover/server/api_test.go | 6 ++++++
 2 files changed, 15 insertions(+)

diff --git a/prover/server/api.go b/prover/server/api.go
index aaacd72fb..ea58eec40 100644
--- a/prover/server/api.go
+++ b/prover/server/api.go
@@ -39,6 +39,8 @@ type Status struct {
 	MinPseZkevmTierFee   uint64 `json:"minPseZkevmTierFee"`
 	MaxExpiry            uint64 `json:"maxExpiry"`
 	CurrentCapacity      uint64 `json:"currentCapacity"`
+	Prover               string `json:"prover"`
+	HeartBeatSignature   []byte `json:"heartBeatSignature"`
 }
 
 // GetStatus handles a query to the current prover server status.
@@ -50,12 +52,19 @@ type Status struct {
 //	@Success		200	{object} Status
 //	@Router			/status [get]
 func (srv *ProverServer) GetStatus(c echo.Context) error {
+	sig, err := crypto.Sign(crypto.Keccak256Hash([]byte("HEART_BEAT")).Bytes(), srv.proverPrivateKey)
+	if err != nil {
+		return echo.NewHTTPError(http.StatusInternalServerError, err)
+	}
+
 	return c.JSON(http.StatusOK, &Status{
 		MinOptimisticTierFee: srv.minOptimisticTierFee.Uint64(),
 		MinSgxTierFee:        srv.minSgxTierFee.Uint64(),
 		MinPseZkevmTierFee:   srv.minPseZkevmTierFee.Uint64(),
 		MaxExpiry:            uint64(srv.maxExpiry.Seconds()),
 		CurrentCapacity:      srv.capacityManager.ReadCapacity(),
+		Prover:               srv.proverAddress.Hex(),
+		HeartBeatSignature:   sig,
 	})
 }
 
diff --git a/prover/server/api_test.go b/prover/server/api_test.go
index d1664b21e..5f86a5b64 100644
--- a/prover/server/api_test.go
+++ b/prover/server/api_test.go
@@ -8,6 +8,7 @@ import (
 	"time"
 
 	"github.com/ethereum/go-ethereum/common"
+	"github.com/ethereum/go-ethereum/crypto"
 	"github.com/taikoxyz/taiko-client/bindings/encoding"
 )
 
@@ -27,6 +28,11 @@ func (s *ProverServerTestSuite) TestGetStatusSuccess() {
 	s.Equal(s.s.minSgxAndPseZkevmTierFee.Uint64(), status.MinSgxTierFee)
 	s.Equal(uint64(s.s.maxExpiry.Seconds()), status.MaxExpiry)
 	s.Greater(status.CurrentCapacity, uint64(0))
+	s.NotEmpty(status.HeartBeatSignature)
+	pubKey, err := crypto.SigToPub(crypto.Keccak256Hash([]byte("HEART_BEAT")).Bytes(), status.HeartBeatSignature)
+	s.Nil(err)
+	s.NotEmpty(status.Prover)
+	s.Equal(status.Prover, crypto.PubkeyToAddress(*pubKey).Hex())
 }
 
 func (s *ProverServerTestSuite) TestProposeBlockSuccess() {