logos-messaging-go/waku/node_rln.go

//go:build gowaku_rln
// +build gowaku_rln

package waku

import (
	"crypto/ecdsa"
	"errors"
	"github.com/ethereum/go-ethereum/common"
	"github.com/ethereum/go-ethereum/crypto"
	"github.com/status-im/go-rln/rln"
	"github.com/status-im/go-waku/waku/v2/node"
)

var loadedCredentialsFromFile bool = false

func checkForRLN(options Options, nodeOpts *[]node.WakuNodeOption) {
	if options.RLNRelay.Enable {
		if !options.Relay.Enable {
			failOnErr(errors.New("relay not available"), "Could not enable RLN Relay")
		}
		if !options.RLNRelay.Dynamic {
			*nodeOpts = append(*nodeOpts, node.WithStaticRLNRelay(options.RLNRelay.PubsubTopic, options.RLNRelay.ContentTopic, rln.MembershipIndex(options.RLNRelay.MembershipIndex), nil))
		} else {

			var ethPrivKey *ecdsa.PrivateKey
			if options.RLNRelay.ETHPrivateKey != "" {
				k, err := crypto.ToECDSA(common.FromHex(options.RLNRelay.ETHPrivateKey))
				failOnErr(err, "Invalid private key")
				ethPrivKey = k
			}

			loaded, idKey, idCommitment, membershipIndex, err := getMembershipCredentials(options)
			failOnErr(err, "Invalid membership credentials")

			loadedCredentialsFromFile = loaded

			*nodeOpts = append(*nodeOpts, node.WithDynamicRLNRelay(
				options.RLNRelay.PubsubTopic,
				options.RLNRelay.ContentTopic,
				membershipIndex,
				idKey,
				idCommitment,
				nil,
				options.RLNRelay.ETHClientAddress,
				ethPrivKey,
				common.HexToAddress(options.RLNRelay.MembershipContractAddress),
			))
		}
	}
}

func onStartRLN(wakuNode *node.WakuNode, options Options) {
	if options.RLNRelay.Enable && options.RLNRelay.Dynamic && !loadedCredentialsFromFile {
		err := writeRLNMembershipCredentialsToFile(wakuNode.RLNRelay().MembershipKeyPair(), wakuNode.RLNRelay().MembershipIndex(), options.RLNRelay.CredentialsFile, []byte(options.KeyPasswd), options.Overwrite)
		failOnErr(err, "Could not write membership credentials file")
	}
}