175 lines
4.1 KiB
Go
175 lines
4.1 KiB
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)
|
|
}
|
|
}
|