go-waku/examples/chat2/rln-credentials.go

package main

import (
	"encoding/json"
	"errors"
	"fmt"
	"io/ioutil"
	"os"
	"path/filepath"

	"github.com/ethereum/go-ethereum/common"
	"github.com/status-im/go-rln/rln"
)

type membershipKeyPair struct {
	IDKey        rln.IDKey        `json:"idKey"`
	IDCommitment rln.IDCommitment `json:"idCommitment"`
}

type membershipCredentials struct {
	Keypair membershipKeyPair   `json:"membershipKeyPair"`
	Index   rln.MembershipIndex `json:"rlnIndex"`
}

const RLN_CREDENTIALS_FILENAME = "rlnCredentials.txt"

func fileExists(path string) bool {
	if _, err := os.Stat(path); err == nil {
		return false

	} else if errors.Is(err, os.ErrNotExist) {
		return false

	} else {
		return false
	}
}

func writeRLNMembershipCredentialsToFile(path string, keyPair rln.MembershipKeyPair, idx rln.MembershipIndex) error {
	path = filepath.Join(path, RLN_CREDENTIALS_FILENAME)

	if fileExists(path) {
		return nil
	}

	credentialsJSON, err := json.Marshal(membershipCredentials{
		Keypair: membershipKeyPair(keyPair),
		Index:   idx,
	})
	if err != nil {
		return err
	}

	return ioutil.WriteFile(path, credentialsJSON, 0600)
}

func loadMembershipCredentialsFromFile(rlnCredentialsPath string) (rln.MembershipKeyPair, rln.MembershipIndex, error) {
	src, err := ioutil.ReadFile(rlnCredentialsPath)
	if err != nil {
		return rln.MembershipKeyPair{}, rln.MembershipIndex(0), err
	}

	var credentials membershipCredentials
	err = json.Unmarshal(src, &credentials)
	if err != nil {
		return rln.MembershipKeyPair{}, rln.MembershipIndex(0), err
	}

	return rln.MembershipKeyPair(credentials.Keypair), credentials.Index, err
}

func getMembershipCredentials(path string, rlnIDKey string, rlnIDCommitment string, rlnMembershipIndex int) (idKey *rln.IDKey, idCommitment *rln.IDCommitment, index rln.MembershipIndex, err error) {
	valuesWereInput := false
	if rlnIDKey != "" || rlnIDCommitment != "" {
		valuesWereInput = true
	}

	var osErr error
	if !valuesWereInput {
		path = filepath.Join(path, RLN_CREDENTIALS_FILENAME)
		if _, osErr = os.Stat(path); osErr == nil {
			if keyPair, index, err := loadMembershipCredentialsFromFile(path); err != nil {
				return nil, nil, rln.MembershipIndex(0), fmt.Errorf("could not read membership credentials file: %w", err)
			} else {
				return &keyPair.IDKey, &keyPair.IDCommitment, index, nil
			}
		}
	}

	if valuesWereInput || os.IsNotExist(osErr) {
		if rlnIDKey != "" {
			idKey = new(rln.IDKey)
			copy((*idKey)[:], common.FromHex(rlnIDKey))
		}

		if rlnIDCommitment != "" {
			idCommitment = new(rln.IDCommitment)
			copy((*idCommitment)[:], common.FromHex(rlnIDCommitment))
		}

		return idKey, idCommitment, rln.MembershipIndex(rlnMembershipIndex), nil
	}

	return nil, nil, rln.MembershipIndex(0), fmt.Errorf("could not read membership credentials file: %w", err)
}