fix: adding fraud service to handler

celestiaorg · Oct 5, 2022 · b7ba257 · b7ba257
1 parent 5535eb9
commit b7ba257
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 4 deletions.
diff --git a/api/rpc/handler.go b/api/rpc/handler.go
@@ -3,9 +3,9 @@ package rpc
 import (
 	logging "github.com/ipfs/go-log/v2"
 
-	"github.com/celestiaorg/celestia-node/nodebuilder/daser"
-
 	"github.com/celestiaorg/celestia-node/das"
+	"github.com/celestiaorg/celestia-node/nodebuilder/daser"
+	"github.com/celestiaorg/celestia-node/nodebuilder/fraud"
 	"github.com/celestiaorg/celestia-node/nodebuilder/header"
 	"github.com/celestiaorg/celestia-node/nodebuilder/share"
 	"github.com/celestiaorg/celestia-node/nodebuilder/state"
@@ -17,19 +17,22 @@ type Handler struct {
 	state  state.Module
 	share  share.Module
 	header header.Module
+	fraud  fraud.Module
 	das    daser.Module
 }
 
 func NewHandler(
 	state state.Module,
 	share share.Module,
 	header header.Module,
+	fraud fraud.Module,
 	das *das.DASer,
 ) *Handler {
 	return &Handler{
 		state:  state,
 		share:  share,
 		header: header,
+		fraud:  fraud,
 		das:    das,
 	}
 }
@@ -38,5 +41,6 @@ func (h *Handler) RegisterEndpoints(rpc *Server) {
 	rpc.handler.Register("Handler", h.state)
 	rpc.handler.Register("Handler", h.share)
 	rpc.handler.Register("Handler", h.header)
+	rpc.handler.Register("Handler", h.fraud)
 	rpc.handler.Register("Handler", h.das)
 }
diff --git a/nodebuilder/rpc/module.go b/nodebuilder/rpc/module.go
@@ -3,6 +3,8 @@ package rpc
 import (
 	"context"
 
+	fraudServ "github.com/celestiaorg/celestia-node/nodebuilder/fraud"
+
 	"github.com/celestiaorg/celestia-node/api/rpc"
 
 	"go.uber.org/fx"
@@ -45,10 +47,11 @@ func ConstructModule(tp node.Type, cfg *Config) fx.Option {
 			fx.Invoke(func(
 				state stateServ.Module,
 				share shareServ.Module,
+				fraud fraudServ.Module,
 				header headerServ.Module,
 				rpcSrv *rpc.Server,
 			) {
-				Handler(state, share, header, rpcSrv, nil)
+				Handler(state, share, header, fraud, rpcSrv, nil)
 			}),
 		)
 	default:

diff --git a/nodebuilder/rpc/rpc.go b/nodebuilder/rpc/rpc.go
@@ -3,6 +3,7 @@ package rpc
 import (
 	"github.com/celestiaorg/celestia-node/api/rpc"
 	"github.com/celestiaorg/celestia-node/das"
+	"github.com/celestiaorg/celestia-node/nodebuilder/fraud"
 	"github.com/celestiaorg/celestia-node/nodebuilder/header"
 	"github.com/celestiaorg/celestia-node/nodebuilder/share"
 	"github.com/celestiaorg/celestia-node/nodebuilder/state"
@@ -13,10 +14,11 @@ func Handler(
 	state state.Module,
 	share share.Module,
 	header header.Module,
+	fraud fraud.Module,
 	serv *rpc.Server,
 	daser *das.DASer,
 ) {
-	handler := rpc.NewHandler(state, share, header, daser)
+	handler := rpc.NewHandler(state, share, header, fraud, daser)
 	handler.RegisterEndpoints(serv)
 }