1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
|
package cli
import (
"fmt"
"os"
"path/filepath"
"runtime"
"strconv"
"github.com/roadrunner-server/errors"
"github.com/roadrunner-server/roadrunner/v2/internal/cli/reset"
"github.com/roadrunner-server/roadrunner/v2/internal/cli/serve"
"github.com/roadrunner-server/roadrunner/v2/internal/cli/stop"
"github.com/roadrunner-server/roadrunner/v2/internal/cli/workers"
dbg "github.com/roadrunner-server/roadrunner/v2/internal/debug"
"github.com/roadrunner-server/roadrunner/v2/internal/meta"
"github.com/joho/godotenv"
"github.com/spf13/cobra"
)
const (
// env var name: path to the .env file
envDotenv string = "DOTENV_PATH"
pidFileName string = ".pid"
)
// NewCommand creates root command.
func NewCommand(cmdName string) *cobra.Command { //nolint:funlen,gocognit
// path to the .rr.yaml
cfgFile := toPtr("")
// pidfile path
pidFile := toPtr(false)
// override config values
override := &[]string{}
// do not print startup message
silent := toPtr(false)
// working directory
var workDir string
// path to the .env file
var dotenv string
// debug mode
var debug bool
cmd := &cobra.Command{
Use: cmdName,
Short: "High-performance PHP application server, load-balancer and process manager",
SilenceErrors: true,
SilenceUsage: true,
Version: fmt.Sprintf("%s (build time: %s, %s)", meta.Version(), meta.BuildTime(), runtime.Version()),
PersistentPreRunE: func(*cobra.Command, []string) error {
// cfgFile could be defined by user or default `.rr.yaml`
// this check added just to be safe
if cfgFile == nil || *cfgFile == "" {
return errors.Str("no configuration file provided")
}
// if user set the wd, change the current wd
if workDir != "" {
if err := os.Chdir(workDir); err != nil {
return err
}
}
// try to get the absolute path to the configuration
if absPath, err := filepath.Abs(*cfgFile); err == nil {
*cfgFile = absPath // switch config path to the absolute
// if workDir is empty - force working absPath related to config file
if workDir == "" {
if err = os.Chdir(filepath.Dir(absPath)); err != nil {
return err
}
}
}
if v, ok := os.LookupEnv(envDotenv); ok { // read path to the dotenv file from environment variable
dotenv = v
}
if dotenv != "" {
err := godotenv.Load(dotenv)
if err != nil {
return err
}
}
if debug {
srv := dbg.NewServer()
go func() { _ = srv.Start(":6061") }() // TODO implement graceful server stopping
}
// user wanted to write a .pid file
if *pidFile {
f, err := os.Create(pidFileName)
if err != nil {
return err
}
defer func() {
_ = f.Close()
}()
_, err = f.WriteString(strconv.Itoa(os.Getpid()))
if err != nil {
return err
}
}
return nil
},
}
f := cmd.PersistentFlags()
f.BoolVarP(pidFile, "pid", "p", false, "create a .pid file")
f.StringVarP(cfgFile, "config", "c", ".rr.yaml", "config file")
f.StringVarP(&workDir, "WorkDir", "w", "", "working directory")
f.StringVarP(&dotenv, "dotenv", "", "", fmt.Sprintf("dotenv file [$%s]", envDotenv))
f.BoolVarP(&debug, "debug", "d", false, "debug mode")
f.BoolVarP(silent, "silent", "s", false, "print startup message")
f.StringArrayVarP(override, "override", "o", nil, "override config value (dot.notation=value)")
cmd.AddCommand(
workers.NewCommand(cfgFile, override),
reset.NewCommand(cfgFile, override, silent),
serve.NewCommand(override, cfgFile, silent),
stop.NewCommand(silent),
)
return cmd
}
func toPtr[T any](val T) *T {
return &val
}
|
