prysm-pulse/runtime/tos/tos.go
terence 5a66807989
Update to V5 (#13622)
* First take at updating everything to v5

* Patch gRPC gateway to use prysm v5

Fix patch

* Update go ssz

---------

Co-authored-by: Preston Van Loon <pvanloon@offchainlabs.com>
2024-02-15 05:46:47 +00:00

80 lines
2.4 KiB
Go

package tos
import (
"errors"
"path/filepath"
"strings"
"github.com/logrusorgru/aurora"
"github.com/prysmaticlabs/prysm/v5/cmd"
"github.com/prysmaticlabs/prysm/v5/io/file"
"github.com/prysmaticlabs/prysm/v5/io/prompt"
"github.com/sirupsen/logrus"
"github.com/urfave/cli/v2"
)
const (
acceptTosFilename = "tosaccepted"
acceptTosPromptText = `
Prysm Terms of Use
By downloading, accessing or using the Prysm implementation (“Prysm”), you (referenced herein
as “you” or the “user”) certify that you have read and agreed to the terms and conditions below.
TERMS AND CONDITIONS: https://github.com/prysmaticlabs/prysm/blob/develop/TERMS_OF_SERVICE.md
Type "accept" to accept this terms and conditions [accept/decline]:`
acceptTosPromptErrText = `could not scan text input, if you are trying to run in non-interactive environment, you
can use the --accept-terms-of-use flag after reading the terms and conditions here:
https://github.com/prysmaticlabs/prysm/blob/develop/TERMS_OF_SERVICE.md`
)
var (
au = aurora.NewAurora(true)
log = logrus.WithField("prefix", "tos")
)
// VerifyTosAcceptedOrPrompt checks if Tos was accepted before or asks to accept.
func VerifyTosAcceptedOrPrompt(ctx *cli.Context) error {
tosFilePath := filepath.Join(ctx.String(cmd.DataDirFlag.Name), acceptTosFilename)
if file.Exists(tosFilePath) {
return nil
}
if ctx.Bool(cmd.AcceptTosFlag.Name) {
saveTosAccepted(ctx)
return nil
}
input, err := prompt.DefaultPrompt(au.Bold(acceptTosPromptText).String(), "decline")
if err != nil {
return errors.New(acceptTosPromptErrText)
}
if !strings.EqualFold(input, "accept") {
return errors.New("you have to accept Terms and Conditions in order to continue")
}
saveTosAccepted(ctx)
return nil
}
// saveTosAccepted creates a file when Tos accepted.
func saveTosAccepted(ctx *cli.Context) {
dataDir := ctx.String(cmd.DataDirFlag.Name)
dataDirExists, err := file.HasDir(dataDir)
if err != nil {
log.WithError(err).Warnf("error checking directory: %s", dataDir)
}
if !dataDirExists {
if err := file.MkdirAll(dataDir); err != nil {
log.WithError(err).Warnf("error creating directory: %s", dataDir)
}
}
if err := file.WriteFile(filepath.Join(dataDir, acceptTosFilename), []byte("")); err != nil {
log.WithError(err).Warnf("error writing %s to file: %s", cmd.AcceptTosFlag.Name,
filepath.Join(dataDir, acceptTosFilename))
}
}