123 lines
4.7 KiB
Go
123 lines
4.7 KiB
Go
package cmd
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"runtime"
|
|
"time"
|
|
|
|
"github.com/warrenronsiek/ctask/internal/session"
|
|
"github.com/warrenronsiek/ctask/internal/shell"
|
|
)
|
|
|
|
// isTTYCheck is the test seam for terminal detection. Tests override this
|
|
// package-level variable to control the TTY refusal path without depending
|
|
// on the real process's stdin/stdout state. Production callers go through
|
|
// defaultIsTTYCheck.
|
|
var isTTYCheck = defaultIsTTYCheck
|
|
|
|
func defaultIsTTYCheck() bool {
|
|
return shell.IsTTY(os.Stdin) && shell.IsTTY(os.Stdout)
|
|
}
|
|
|
|
// preflightPersistentEntry validates the host environment supports
|
|
// tmux-based persistent mode and returns the validated tmux binary path
|
|
// for callers to pass through `session.LaunchOpts.TmuxPath`.
|
|
//
|
|
// Order of checks:
|
|
//
|
|
// 1. Native Windows refusal — tmux is not supported; recommend WSL.
|
|
// 2. Nested tmux refusal — `$TMUX` is set in the parent process.
|
|
// 3. Non-TTY refusal — stdin or stdout is not a terminal.
|
|
// 4. shell.LookupTmux — handles ErrTmuxNotFound and ErrTmuxTooOld
|
|
// with platform-aware install hints.
|
|
//
|
|
// commandName ("new", "resume", "last", "open", "attach") is rendered into
|
|
// the bypass hint so each command tells the user the right form.
|
|
func preflightPersistentEntry(commandName string) (string, error) {
|
|
bypass := " ctask " + commandName + " <workspace> --direct"
|
|
|
|
if runtime.GOOS == "windows" && os.Getenv("WSL_DISTRO_NAME") == "" {
|
|
return "", fmt.Errorf(
|
|
"ctask persistent mode requires tmux, which is not supported on native Windows.\n\n"+
|
|
"Recommended:\n Run ctask from WSL and install tmux there:\n sudo apt install tmux\n\n"+
|
|
"Or bypass persistent mode:\n%s", bypass)
|
|
}
|
|
if os.Getenv("TMUX") != "" {
|
|
return "", fmt.Errorf(
|
|
"ctask persistent mode cannot attach while already inside tmux.\n\n"+
|
|
"Run ctask from outside tmux, or bypass persistent mode:\n%s", bypass)
|
|
}
|
|
if !isTTYCheck() {
|
|
return "", fmt.Errorf(
|
|
"ctask persistent mode requires an interactive terminal.\n\n"+
|
|
"Over SSH, use:\n ssh -t <host> ctask %s <workspace>\n\n"+
|
|
"Or bypass persistent mode:\n%s", commandName, bypass)
|
|
}
|
|
tmuxPath, ver, err := shell.LookupTmux()
|
|
if err != nil {
|
|
if errors.Is(err, shell.ErrTmuxNotFound) {
|
|
return "", fmt.Errorf(
|
|
"ctask is configured for persistent sessions, but tmux is not installed.\n\n"+
|
|
"Install tmux:\n"+
|
|
" Debian/Ubuntu/WSL: sudo apt install tmux\n"+
|
|
" macOS: brew install tmux\n"+
|
|
" Arch: sudo pacman -S tmux\n"+
|
|
" Fedora: sudo dnf install tmux\n\n"+
|
|
"Or bypass persistent mode for this command:\n%s\n\n"+
|
|
"To disable persistent mode:\n unset CTASK_SESSION_MODE", bypass)
|
|
}
|
|
if errors.Is(err, shell.ErrTmuxTooOld) {
|
|
raw := ver.Raw
|
|
if raw == "" {
|
|
raw = "unknown version"
|
|
}
|
|
return "", fmt.Errorf(
|
|
"ctask persistent mode requires tmux 3.0 or newer (found: %s).\n\n"+
|
|
"Update tmux:\n"+
|
|
" Debian/Ubuntu/WSL: sudo apt install tmux (Debian 10+ ships 2.8; consider backports or a newer release)\n"+
|
|
" macOS: brew upgrade tmux\n"+
|
|
" Arch: sudo pacman -Syu tmux\n"+
|
|
" Fedora: sudo dnf upgrade tmux\n\n"+
|
|
"Or bypass persistent mode for this command:\n%s", raw, bypass)
|
|
}
|
|
return "", err
|
|
}
|
|
return tmuxPath, nil
|
|
}
|
|
|
|
// confirmFreshRemoteAdoption prompts the user to confirm adoption of a
|
|
// workspace whose lease is fresh but from a different host. Refuses on
|
|
// non-TTY environments — silent overwrite of a fresh remote lease is
|
|
// never appropriate.
|
|
func confirmFreshRemoteAdoption(wsPath string) error {
|
|
l, _ := session.ReadLease(session.LeasePath(wsPath))
|
|
hostStr := "unknown"
|
|
heartbeatAgo := "unknown"
|
|
if l != nil {
|
|
hostStr = l.Hostname
|
|
heartbeatAgo = session.FormatAgo(time.Since(l.LastHeartbeatAt))
|
|
}
|
|
if !isTTYCheck() {
|
|
return fmt.Errorf(
|
|
"ctask refused to adopt persistent session: lease is fresh from another host (%s, last heartbeat %s ago); rerun in an interactive terminal to confirm",
|
|
hostStr, heartbeatAgo)
|
|
}
|
|
fmt.Fprintf(os.Stderr,
|
|
"[ctask] tmux session exists, but a fresh lease from another host is present:\n"+
|
|
"[ctask] hostname: %s\n"+
|
|
"[ctask] last heartbeat: %s ago\n"+
|
|
"[ctask]\n"+
|
|
"[ctask] This may indicate another machine is actively using this workspace\n"+
|
|
"[ctask] (e.g., shared filesystem). Adopting will overwrite the remote lease\n"+
|
|
"[ctask] and may create conflicting workspace activity.\n"+
|
|
"[ctask]\n"+
|
|
"Adopt anyway? [y/N] ",
|
|
hostStr, heartbeatAgo)
|
|
if !session.ConfirmYN(os.Stdin, os.Stderr, "", false) {
|
|
return fmt.Errorf("[ctask] adoption refused; wait for the remote session to end or unset CTASK_SESSION_MODE")
|
|
}
|
|
return nil
|
|
}
|