status-im/status-go

package main

import (
    "log"
    "os"
    "time"

    "go.uber.org/zap"

    "github.com/status-im/status-go/api"
    "github.com/status-im/status-go/protocol"

    "github.com/urfave/cli/v2"
)

const LightFlag = "light"
const InteractiveFlag = "interactive"
const CountFlag = "count"
const NameFlag = "name"
const AddFlag = "add"
const PortFlag = "port"
const APIModulesFlag = "api-modules"
const TelemetryServerURLFlag = "telemetry-server-url"
const KeyUIDFlag = "key-uid"
const DebugLevel = "debug"
const MessageFailureFlag = "fail"

const RetrieveInterval = 300 * time.Millisecond
const SendInterval = 1 * time.Second
const WaitingInterval = 5 * time.Second

var CommonFlags = []cli.Flag{
    &cli.BoolFlag{
        Name:    LightFlag,
        Aliases: []string{"l"},
        Usage:   "Enable light mode",
    },
    &cli.StringFlag{
        Name:    APIModulesFlag,
        Aliases: []string{"m"},
        Value:   "waku,wakuext,wakuv2,permissions,eth",
        Usage:   "API modules to enable",
    },
    &cli.StringFlag{
        Name:    TelemetryServerURLFlag,
        Aliases: []string{"t"},
        Usage:   "Telemetry server URL",
    },
    &cli.BoolFlag{
        Name:    DebugLevel,
        Aliases: []string{"d"},
        Usage:   "Enable CLI's debug level logging",
        Value:   false,
    },
}

var SimulateFlags = append([]cli.Flag{
    &cli.BoolFlag{
        Name:    InteractiveFlag,
        Aliases: []string{"i"},
        Usage:   "Use interactive mode to input the messages",
    },
    &cli.IntFlag{
        Name:    CountFlag,
        Aliases: []string{"c"},
        Value:   1,
        Usage:   "How many messages to sent from each user",
    },
    &cli.BoolFlag{
        Name:    MessageFailureFlag,
        Aliases: []string{"f"},
        Usage:   "Causes messages to fail about 25% of the time",
        Value:   false,
    },
}, CommonFlags...)

var ServeFlags = append([]cli.Flag{
    &cli.StringFlag{
        Name:    NameFlag,
        Aliases: []string{"n"},
        Value:   "Alice",
        Usage:   "Name of the user",
    },
    &cli.StringFlag{
        Name:    AddFlag,
        Aliases: []string{"a"},
        Usage:   "Add a friend with the public key",
    },
    &cli.IntFlag{
        Name:    PortFlag,
        Aliases: []string{"p"},
        Value:   8545,
        Usage:   "HTTP Server port to listen on",
    },
    &cli.BoolFlag{
        Name:    InteractiveFlag,
        Aliases: []string{"i"},
        Usage:   "Use interactive mode to input the messages",
    },
}, CommonFlags...)

type StatusCLI struct {
    name      string
    messenger *protocol.Messenger
    backend   *api.GethStatusBackend
    logger    *zap.SugaredLogger
}

func main() {
    app := &cli.App{
        Commands: []*cli.Command{
            {
                Name:  "simulate",
                Usage: "Simulate the process of sending direct messages",
                Flags: SimulateFlags,
                Action: func(cCtx *cli.Context) error {
                    return simulate(cCtx)
                },
            },
            {
                Name:    "serve",
                Aliases: []string{"s"},
                Usage:   "Start a server to send and receive messages",
                Flags:   ServeFlags,
                Action: func(cCtx *cli.Context) error {
                    return serve(cCtx)
                },
            },
            {
                Name:    "serve-account",
                Aliases: []string{"sl"},
                Usage:   "Start a server with the lastest input name's account\n\n  E.g.: if last time you created an account with name 'Alice',\n  you can start the server with 'Alice' account by running 'servelast -n Alice'",
                Flags: []cli.Flag{
                    &cli.StringFlag{
                        Name:     NameFlag,
                        Aliases:  []string{"n"},
                        Usage:    "Name of the existing user",
                        Required: true,
                    },
                    &cli.StringFlag{
                        Name:     KeyUIDFlag,
                        Aliases:  []string{"kid"},
                        Usage:    "Key ID of the existing user (find them under '<data-dir>/keystore' on in logs when using the 'serve' command)",
                        Required: true,
                    },
                    &cli.BoolFlag{
                        Name:    InteractiveFlag,
                        Aliases: []string{"i"},
                        Usage:   "Use interactive mode to input the messages",
                        Value:   false,
                    },
                    &cli.StringFlag{
                        Name:    AddFlag,
                        Aliases: []string{"a"},
                        Usage:   "Add a friend with the public key",
                    },
                    &cli.BoolFlag{
                        Name:    DebugLevel,
                        Aliases: []string{"d"},
                        Usage:   "Enable CLI's debug level logging",
                        Value:   false,
                    },
                },
                Action: func(cCtx *cli.Context) error {
                    return serve(cCtx)
                },
            },
        },
    }

    if err := app.Run(os.Args); err != nil {
        log.Fatal(err)
    }
}