mirror of https://github.com/curusarn/resh
Simon Let
3 years ago
parent
dd573cac03
commit
668308a528
@ -0,0 +1,151 @@ |
|||||||
|
package cmd |
||||||
|
|
||||||
|
import ( |
||||||
|
"fmt" |
||||||
|
"os" |
||||||
|
"os/exec" |
||||||
|
"time" |
||||||
|
|
||||||
|
"github.com/curusarn/resh/internal/cfg" |
||||||
|
"github.com/curusarn/resh/internal/check" |
||||||
|
"github.com/curusarn/resh/internal/msg" |
||||||
|
"github.com/curusarn/resh/internal/status" |
||||||
|
"github.com/spf13/cobra" |
||||||
|
"go.uber.org/zap" |
||||||
|
) |
||||||
|
|
||||||
|
func doctorCmdFunc(config cfg.Config) func(*cobra.Command, []string) { |
||||||
|
return func(cmd *cobra.Command, args []string) { |
||||||
|
allOK := true |
||||||
|
if !checkDaemon(config) { |
||||||
|
allOK = false |
||||||
|
printDivider() |
||||||
|
} |
||||||
|
if !checkShellSession() { |
||||||
|
allOK = false |
||||||
|
printDivider() |
||||||
|
} |
||||||
|
if !checkShells() { |
||||||
|
allOK = false |
||||||
|
printDivider() |
||||||
|
} |
||||||
|
|
||||||
|
if allOK { |
||||||
|
out.Info("Everything looks good.") |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
func printDivider() { |
||||||
|
fmt.Printf("\n") |
||||||
|
} |
||||||
|
|
||||||
|
var msgFailedDaemonStart = `Failed to start RESH daemon. |
||||||
|
|
||||||
|
-> Start RESH daemon manually - run: resh-daemon-start |
||||||
|
-> Or restart this terminal window to bring RESH daemon back up |
||||||
|
-> You can check logs: ~/.local/share/resh/log.json (or ~/$XDG_DATA_HOME/resh/log.json) |
||||||
|
-> You can create an issue at: https://github.com/curusarn/resh/issues
|
||||||
|
|
||||||
|
` |
||||||
|
|
||||||
|
func checkDaemon(config cfg.Config) bool { |
||||||
|
ok := true |
||||||
|
resp, err := status.GetDaemonStatus(config.Port) |
||||||
|
if err != nil { |
||||||
|
out.InfoE("RESH Daemon is not running", err) |
||||||
|
out.Info("Attempting to start RESH daemon ...") |
||||||
|
resp, err = startDaemon(config.Port, 5, 200*time.Millisecond) |
||||||
|
if err != nil { |
||||||
|
out.InfoE(msgFailedDaemonStart, err) |
||||||
|
return false |
||||||
|
} |
||||||
|
ok = false |
||||||
|
out.Info("Successfully started daemon.") |
||||||
|
} |
||||||
|
if version != resp.Version { |
||||||
|
out.InfoDaemonVersionMismatch(version, resp.Version) |
||||||
|
return false |
||||||
|
} |
||||||
|
return ok |
||||||
|
} |
||||||
|
|
||||||
|
func startDaemon(port int, maxRetries int, backoff time.Duration) (*msg.StatusResponse, error) { |
||||||
|
err := exec.Command("resh-daemon-start").Run() |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
var resp *msg.StatusResponse |
||||||
|
retry := 0 |
||||||
|
for { |
||||||
|
time.Sleep(backoff) |
||||||
|
resp, err = status.GetDaemonStatus(port) |
||||||
|
if err == nil { |
||||||
|
break |
||||||
|
} |
||||||
|
if retry == maxRetries { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
out.Logger.Error("Failed to get daemon status - retrying", zap.Error(err), zap.Int("retry", retry)) |
||||||
|
retry++ |
||||||
|
continue |
||||||
|
} |
||||||
|
return resp, nil |
||||||
|
} |
||||||
|
|
||||||
|
var msgShellFilesNotLoaded = `RESH shell files were not properly loaded in this terminal. |
||||||
|
|
||||||
|
-> Try restarting this terminal to see if the issue persists |
||||||
|
-> Check your shell rc files (e.g. .zshrc, .bashrc, ...) |
||||||
|
-> You can create an issue at: https://github.com/curusarn/resh/issues
|
||||||
|
|
||||||
|
` |
||||||
|
|
||||||
|
func checkShellSession() bool { |
||||||
|
versionEnv, found := os.LookupEnv("__RESH_VERSION") |
||||||
|
if !found { |
||||||
|
out.Info(msgShellFilesNotLoaded) |
||||||
|
return false |
||||||
|
} |
||||||
|
if version != versionEnv { |
||||||
|
out.InfoTerminalVersionMismatch(version, versionEnv) |
||||||
|
return false |
||||||
|
} |
||||||
|
return true |
||||||
|
} |
||||||
|
|
||||||
|
func checkShells() bool { |
||||||
|
allOK := true |
||||||
|
|
||||||
|
msg, err := check.LoginShell() |
||||||
|
if err != nil { |
||||||
|
out.InfoE("Failed to get login shell", err) |
||||||
|
allOK = false |
||||||
|
} |
||||||
|
if msg != "" { |
||||||
|
out.Info(msg) |
||||||
|
allOK = false |
||||||
|
} |
||||||
|
|
||||||
|
msg, err = check.ZshVersion() |
||||||
|
if err != nil { |
||||||
|
out.InfoE("Failed to check zsh version", err) |
||||||
|
allOK = false |
||||||
|
} |
||||||
|
if msg != "" { |
||||||
|
out.Info(msg) |
||||||
|
allOK = false |
||||||
|
} |
||||||
|
|
||||||
|
msg, err = check.BashVersion() |
||||||
|
if err != nil { |
||||||
|
out.InfoE("Failed to check bash version", err) |
||||||
|
allOK = false |
||||||
|
} |
||||||
|
if msg != "" { |
||||||
|
out.Info(msg) |
||||||
|
allOK = false |
||||||
|
} |
||||||
|
|
||||||
|
return allOK |
||||||
|
} |
||||||
@ -0,0 +1,90 @@ |
|||||||
|
package check |
||||||
|
|
||||||
|
import ( |
||||||
|
"fmt" |
||||||
|
"os" |
||||||
|
"os/exec" |
||||||
|
"strconv" |
||||||
|
"strings" |
||||||
|
) |
||||||
|
|
||||||
|
func LoginShell() (string, error) { |
||||||
|
shellPath, found := os.LookupEnv("SHELL") |
||||||
|
if !found { |
||||||
|
return "", fmt.Errorf("env variable $SHELL is not set") |
||||||
|
} |
||||||
|
parts := strings.Split(shellPath, "/") |
||||||
|
shell := parts[len(parts)-1] |
||||||
|
if shell != "bash" && shell != "zsh" { |
||||||
|
return fmt.Sprintf("Current shell (%s) is unsupported\n", shell), nil |
||||||
|
} |
||||||
|
return "", nil |
||||||
|
} |
||||||
|
|
||||||
|
func msgShellVersion(shell, expectedVer, actualVer string) string { |
||||||
|
return fmt.Sprintf(`Minimal supported %s version is %s. You have %s. |
||||||
|
|
||||||
|
-> Update your %s if you want to use RESH with it. |
||||||
|
`, shell, expectedVer, actualVer, shell) |
||||||
|
} |
||||||
|
|
||||||
|
func BashVersion() (string, error) { |
||||||
|
out, err := exec.Command("bash", "-c", "echo $BASH_VERSION").Output() |
||||||
|
if err != nil { |
||||||
|
return "", fmt.Errorf("command failed: %w", err) |
||||||
|
} |
||||||
|
verStr := strings.TrimSuffix(string(out), "\n") |
||||||
|
ver, err := parseVersion(verStr) |
||||||
|
if err != nil { |
||||||
|
return "", fmt.Errorf("failed to parse version: %w", err) |
||||||
|
} |
||||||
|
|
||||||
|
if ver.Major < 4 || (ver.Major == 4 && ver.Minor < 3) { |
||||||
|
return msgShellVersion("bash", "4.3", verStr), nil |
||||||
|
} |
||||||
|
return "", nil |
||||||
|
} |
||||||
|
|
||||||
|
func ZshVersion() (string, error) { |
||||||
|
out, err := exec.Command("zsh", "-c", "echo $ZSH_VERSION").Output() |
||||||
|
if err != nil { |
||||||
|
return "", fmt.Errorf("command failed: %w", err) |
||||||
|
} |
||||||
|
verStr := strings.TrimSuffix(string(out), "\n") |
||||||
|
ver, err := parseVersion(string(out)) |
||||||
|
if err != nil { |
||||||
|
return "", fmt.Errorf("failed to parse version: %w", err) |
||||||
|
} |
||||||
|
|
||||||
|
if ver.Major < 5 { |
||||||
|
return msgShellVersion("zsh", "5.0", verStr), nil |
||||||
|
} |
||||||
|
return "", nil |
||||||
|
} |
||||||
|
|
||||||
|
type version struct { |
||||||
|
Major int |
||||||
|
Minor int |
||||||
|
Rest string |
||||||
|
} |
||||||
|
|
||||||
|
func parseVersion(str string) (version, error) { |
||||||
|
parts := strings.SplitN(str, ".", 3) |
||||||
|
if len(parts) < 3 { |
||||||
|
return version{}, fmt.Errorf("not enough parts") |
||||||
|
} |
||||||
|
major, err := strconv.Atoi(parts[0]) |
||||||
|
if err != nil { |
||||||
|
return version{}, fmt.Errorf("failed to parse major version: %w", err) |
||||||
|
} |
||||||
|
minor, err := strconv.Atoi(parts[1]) |
||||||
|
if err != nil { |
||||||
|
return version{}, fmt.Errorf("failed to parse minor version: %w", err) |
||||||
|
} |
||||||
|
ver := version{ |
||||||
|
Major: major, |
||||||
|
Minor: minor, |
||||||
|
Rest: parts[2], |
||||||
|
} |
||||||
|
return ver, nil |
||||||
|
} |
||||||
Loading…
Reference in new issue