Add lighter protocol to the API

Author: mpetrun5

api/handlers/signing.go

@@ -11,6 +11,7 @@ import (
 	"github.com/ethereum/go-ethereum/common"
 	"github.com/gorilla/mux"
 	evmMessage "github.com/sprintertech/sprinter-signing/chains/evm/message"
+	lighterMessage "github.com/sprintertech/sprinter-signing/chains/lighter/message"
 	"github.com/sygmaprotocol/sygma-core/relayer/message"
 )
 
@@ -20,8 +21,9 @@ const (
 	AcrossProtocol     ProtocolType = "across"
 	MayanProtocol      ProtocolType = "mayan"
 	RhinestoneProtocol ProtocolType = "rhinestone"
-	LifiEscrow         ProtocolType = "lifi-escrow"
+	LifiEscrowProtocol ProtocolType = "lifi-escrow"
 	LifiProtocol       ProtocolType = "lifi"
+	LighterProtocol    ProtocolType = "lighter"
 )
 
 type SigningBody struct {
@@ -113,7 +115,7 @@ func (h *SigningHandler) HandleSigning(w http.ResponseWriter, r *http.Request) {
 				BorrowAmount:  b.BorrowAmount.Int,
 			})
 		}
-	case LifiEscrow:
+	case LifiEscrowProtocol:
 		{
 			m = evmMessage.NewLifiEscrowData(0, b.ChainId, &evmMessage.LifiEscrowData{
 				OrderID:       b.DepositId,
@@ -126,6 +128,18 @@ func (h *SigningHandler) HandleSigning(w http.ResponseWriter, r *http.Request) {
 				BorrowAmount:  b.BorrowAmount.Int,
 			})
 		}
+	case LighterProtocol:
+		{
+			m = lighterMessage.NewLighterMessage(0, b.ChainId, &lighterMessage.LighterData{
+				OrderHash:     b.DepositId,
+				Nonce:         b.Nonce.Int,
+				LiquidityPool: common.HexToAddress(b.LiquidityPool),
+				Caller:        common.HexToAddress(b.Caller),
+				ErrChn:        errChn,
+				Source:        0,
+				Destination:   b.ChainId,
+			})
+		}
 	default:
 		JSONError(w, fmt.Errorf("invalid protocol %s", b.Protocol), http.StatusBadRequest)
 		return

api/handlers/signing_test.go

@@ -16,6 +16,7 @@ import (
 	"github.com/sprintertech/sprinter-signing/api/handlers"
 	mock_handlers "github.com/sprintertech/sprinter-signing/api/handlers/mock"
 	across "github.com/sprintertech/sprinter-signing/chains/evm/message"
+	lighter "github.com/sprintertech/sprinter-signing/chains/lighter/message"
 	"github.com/stretchr/testify/suite"
 	"github.com/sygmaprotocol/sygma-core/relayer/message"
 	"go.uber.org/mock/gomock"
@@ -394,6 +395,40 @@ func (s *SigningHandlerTestSuite) Test_HandleSigning_LifiSuccess() {
 	s.Equal(http.StatusAccepted, recorder.Code)
 }
 
+func (s *SigningHandlerTestSuite) Test_HandleSigning_LighterSuccess() {
+	msgChn := make(chan []*message.Message)
+	handler := handlers.NewSigningHandler(msgChn, s.chains)
+
+	input := handlers.SigningBody{
+		DepositId:     "depositID",
+		Protocol:      "lighter",
+		LiquidityPool: "0xbe526bA5d1ad94cC59D7A79d99A59F607d31A657",
+		Caller:        "0xbe526bA5d1ad94cC59D7A79d99A59F607d31A657",
+		Calldata:      "0xbe5",
+		Nonce:         &handlers.BigInt{big.NewInt(1001)},
+		BorrowAmount:  &handlers.BigInt{big.NewInt(1000)},
+	}
+	body, _ := json.Marshal(input)
+
+	req := httptest.NewRequest(http.MethodPost, "/v1/chains/1/signatures", bytes.NewReader(body))
+	req = mux.SetURLVars(req, map[string]string{
+		"chainId": "1",
+	})
+	req.Header.Set("Content-Type", "application/json")
+
+	recorder := httptest.NewRecorder()
+
+	go func() {
+		msg := <-msgChn
+		ad := msg[0].Data.(*lighter.LighterData)
+		ad.ErrChn <- nil
+	}()
+
+	handler.HandleSigning(recorder, req)
+
+	s.Equal(http.StatusAccepted, recorder.Code)
+}
+
 type StatusHandlerTestSuite struct {
 	suite.Suite
 

app/app.go

@@ -32,6 +32,8 @@ import (
 	"github.com/sprintertech/sprinter-signing/chains/evm/calls/events"
 	evmListener "github.com/sprintertech/sprinter-signing/chains/evm/listener"
 	evmMessage "github.com/sprintertech/sprinter-signing/chains/evm/message"
+	"github.com/sprintertech/sprinter-signing/chains/lighter"
+	lighterMessage "github.com/sprintertech/sprinter-signing/chains/lighter/message"
 	"github.com/sprintertech/sprinter-signing/comm"
 	"github.com/sprintertech/sprinter-signing/comm/elector"
 	"github.com/sprintertech/sprinter-signing/comm/p2p"
@@ -43,6 +45,7 @@ import (
 	"github.com/sprintertech/sprinter-signing/price"
 	"github.com/sprintertech/sprinter-signing/protocol/across"
 	"github.com/sprintertech/sprinter-signing/protocol/lifi"
+	lighterAPI "github.com/sprintertech/sprinter-signing/protocol/lighter"
 	"github.com/sprintertech/sprinter-signing/protocol/mayan"
 	"github.com/sprintertech/sprinter-signing/topology"
 	"github.com/sprintertech/sprinter-signing/tss"
@@ -350,6 +353,23 @@ func Run() error {
 		}
 	}
 
+	lighterConfig, err := lighter.NewLighterConfig(*solverConfig)
+	panicOnError(err)
+	lighterAPI := lighterAPI.NewLighterAPI()
+	lighterMessageHandler := lighterMessage.NewLighterMessageHandler(
+		lighterConfig.WithdrawalAddress,
+		lighterConfig.UsdcAddress,
+		lighterAPI,
+		coordinator,
+		host,
+		communication,
+		keyshareStore,
+		sigChn,
+	)
+	go lighterMessageHandler.Listen(ctx)
+	lighterChain := lighter.NewLighterChain(lighterMessageHandler)
+	domains[lighter.LIGHTER_DOMAIN_ID] = lighterChain
+
 	go jobs.StartCommunicationHealthCheckJob(host, configuration.RelayerConfig.MpcConfig.CommHealthCheckInterval, sygmaMetrics)
 
 	r := relayer.NewRelayer(domains, sygmaMetrics)

chains/lighter/chain.go

@@ -0,0 +1,42 @@
+package lighter
+
+import (
+	"context"
+
+	"github.com/sygmaprotocol/sygma-core/relayer/message"
+	"github.com/sygmaprotocol/sygma-core/relayer/proposal"
+)
+
+const (
+	LIGHTER_DOMAIN_ID uint64 = 1513889025
+)
+
+type MessageHandler interface {
+	HandleMessage(m *message.Message) (*proposal.Proposal, error)
+}
+
+type LighterChain struct {
+	messageHandler MessageHandler
+	domainID       uint64
+}
+
+func NewLighterChain(messageHandler MessageHandler) *LighterChain {
+	return &LighterChain{
+		messageHandler: messageHandler,
+		domainID:       LIGHTER_DOMAIN_ID,
+	}
+}
+
+func (c *LighterChain) PollEvents(_ context.Context) {}
+
+func (c *LighterChain) ReceiveMessage(m *message.Message) (*proposal.Proposal, error) {
+	return c.messageHandler.HandleMessage(m)
+}
+
+func (c *LighterChain) Write(_ []*proposal.Proposal) error {
+	return nil
+}
+
+func (c *LighterChain) DomainID() uint64 {
+	return c.domainID
+}