status-go/services/wakuv2ext/service.go

package wakuv2ext

import (
	"github.com/syndtr/goleveldb/leveldb"

	gethrpc "github.com/ethereum/go-ethereum/rpc"
	"github.com/status-im/status-go/eth-node/types"
	"github.com/status-im/status-go/params"
	"github.com/status-im/status-go/rpc"
	"github.com/status-im/status-go/services/ext"
)

type Service struct {
	*ext.Service
	w types.Waku
}

func New(config params.NodeConfig, n types.Node, rpcClient *rpc.Client, handler ext.EnvelopeEventsHandler, ldb *leveldb.DB) *Service {
	w, err := n.GetWakuV2(nil)
	if err != nil {
		panic(err)
	}
	delay := ext.DefaultRequestsDelay
	if config.ShhextConfig.RequestsDelay != 0 {
		delay = config.ShhextConfig.RequestsDelay
	}
	requestsRegistry := ext.NewRequestsRegistry(delay)
	mailMonitor := ext.NewMailRequestMonitor(w, handler, requestsRegistry)
	return &Service{
		Service: ext.New(config, n, rpcClient, ldb, mailMonitor, w),
		w:       w,
	}
}

func (s *Service) PublicWakuAPI() types.PublicWakuAPI {
	return s.w.PublicWakuAPI()
}

// APIs returns a list of new APIs.
func (s *Service) APIs() []gethrpc.API {
	apis := []gethrpc.API{
		{
			Namespace: "wakuext",
			Version:   "1.0",
			Service:   NewPublicAPI(s),
			Public:    false,
		},
	}
	return apis
}