func TestIsTerminal(t *testing.T) {
mpty, mtty, err := pty.Open()
if err != nil {
t.Fatal(err)
}
defer mtty.Close()
defer mpty.Close()
if !ttyutils.IsTerminal(mtty.Fd()) {
t.Error("tty should be reported as a terminal")
}
if !ttyutils.IsTerminal(mpty.Fd()) {
t.Error("pty should be reported as a terminal")
}
null, err := os.Open(os.DevNull)
if err != nil {
t.Fatal(err)
}
defer null.Close()
if ttyutils.IsTerminal(null.Fd()) {
t.Error("/dev/null should not be reported as a terminal")
}
}
func (f *TextFormatter) Format(entry *logrus.Entry) ([]byte, error) {
var serialized []byte
if f.ForceColors || ttyutils.IsTerminal(os.Stdout.Fd()) {
levelText := strings.ToUpper(entry.Data["level"].(string))[0:4]
levelColor := col_bold_blue
switch entry.Data["level"] {
case "warning":
levelColor = col_bold_yellow
case "error", "fatal", "panic":
levelColor = col_bold_red
case "debug":
levelColor = col_magenta
}
timeStr := entry.Data["time"].(string)
when, _ := time.Parse("2006-01-02 15:04:05.999999999 -0700 MST", timeStr)
finalTime := when.Format("2006-01-02 15:04:05.9")
serialized = append(serialized, []byte(fmt.Sprintf("%s[%s] %s\x1b[0m %-45s ", levelColor, finalTime, levelText, entry.Data["msg"]))...)
keys := make([]string, 0)
for k, _ := range entry.Data {
if k != "level" && k != "time" && k != "msg" {
keys = append(keys, k)
}
}
sort.Strings(keys)
first := true
for _, k := range keys {
v := entry.Data[k]
if first {
first = false
} else {
serialized = append(serialized, ' ')
}
serialized = append(serialized, []byte(fmt.Sprintf("\x1b[%dm%s\x1b[0m=%v", levelColor, k, v))...)
}
} else {
serialized = f.AppendKeyValue(serialized, "time", entry.Data["time"].(string))
serialized = f.AppendKeyValue(serialized, "level", entry.Data["level"].(string))
serialized = f.AppendKeyValue(serialized, "msg", entry.Data["msg"].(string))
for key, value := range entry.Data {
if key != "time" && key != "level" && key != "msg" {
serialized = f.AppendKeyValue(serialized, key, value)
}
}
}
return append(serialized, '\n'), nil
}
func (f *TextFormatter) Format(entry *Entry) ([]byte, error) {
var serialized []byte
if f.ForceColors || ttyutils.IsTerminal(os.Stdout.Fd()) {
levelText := strings.ToUpper(entry.Data["level"].(string))[0:4]
levelColor := blue
if entry.Data["level"] == "warning" {
levelColor = yellow
} else if entry.Data["level"] == "error" ||
entry.Data["level"] == "fatal" ||
entry.Data["level"] == "panic" {
levelColor = red
}
serialized = append(serialized, []byte(fmt.Sprintf("\x1b[%dm%s\x1b[0m[%04d] %-45s ", levelColor, levelText, miniTS(), entry.Data["msg"]))...)
keys := make([]string, 0)
for k, _ := range entry.Data {
if k != "level" && k != "time" && k != "msg" {
keys = append(keys, k)
}
}
sort.Strings(keys)
first := true
for _, k := range keys {
v := entry.Data[k]
if first {
first = false
} else {
serialized = append(serialized, ' ')
}
serialized = append(serialized, []byte(fmt.Sprintf("\x1b[%dm%s\x1b[0m=%v", levelColor, k, v))...)
}
} else {
serialized = f.AppendKeyValue(serialized, "time", entry.Data["time"].(string))
serialized = f.AppendKeyValue(serialized, "level", entry.Data["level"].(string))
serialized = f.AppendKeyValue(serialized, "msg", entry.Data["msg"].(string))
for key, value := range entry.Data {
if key != "time" && key != "level" && key != "msg" {
serialized = f.AppendKeyValue(serialized, key, value)
}
}
}
return append(serialized, '\n'), nil
}
func Start(tree *processtree.ProcessTree, done chan bool) chan bool {
quit := make(chan bool)
theChart = &StatusChart{}
theChart.RootSlave = tree.Root
theChart.numberOfSlaves = len(tree.SlavesByName)
theChart.Commands = tree.Commands
theChart.update = make(chan bool, 10)
theChart.directLogger = slog.NewShinyLogger(os.Stdout, os.Stderr)
theChart.terminalSupported = ttyutils.IsTerminal(os.Stdout.Fd())
if theChart.terminalSupported {
ttyStart(tree, done, quit)
} else {
stdoutStart(tree, done, quit)
}
return quit
}
func doRun(color bool) int {
if !color {
slog.DisableColor()
DisableErrorColor()
}
if os.Getenv("RAILS_ENV") != "" {
println("Warning: Specifying a Rails environment via RAILS_ENV has no effect for commands run with zeus.")
}
master, slave, err := pty.Open()
if err != nil {
panic(err)
}
defer master.Close()
if ttyutils.IsTerminal(os.Stdout.Fd()) {
oldState, err := ttyutils.MakeTerminalRaw(os.Stdout.Fd())
if err != nil {
panic(err)
}
defer ttyutils.RestoreTerminalState(os.Stdout.Fd(), oldState)
}
ttyutils.MirrorWinsize(os.Stdout, master)
addr, err := net.ResolveUnixAddr("unixgram", zeusSockName)
if err != nil {
panic("Can't resolve server address")
}
conn, err := net.DialUnix("unix", nil, addr)
if err != nil {
ErrorCantConnectToMaster()
}
usock := unixsocket.NewUsock(conn)
msg := CreateCommandAndArgumentsMessage(os.Args[1], os.Args[2:])
usock.WriteMessage(msg)
usock.WriteFD(int(slave.Fd()))
slave.Close()
msg, err = usock.ReadMessage()
if err != nil {
panic(err)
}
parts := strings.Split(msg, "\000")
commandPid, err := strconv.Atoi(parts[0])
defer func() {
if commandPid > 0 {
// Just in case.
syscall.Kill(commandPid, 9)
}
}()
if err != nil {
panic(err)
}
c := make(chan os.Signal, 1)
signal.Notify(c, syscall.SIGWINCH, syscall.SIGCONT)
go func() {
for sig := range c {
if sig == syscall.SIGCONT {
syscall.Kill(commandPid, syscall.SIGCONT)
} else if sig == syscall.SIGWINCH {
ttyutils.MirrorWinsize(os.Stdout, master)
syscall.Kill(commandPid, syscall.SIGWINCH)
}
}
}()
var exitStatus int = -1
if len(parts) > 2 {
exitStatus, err = strconv.Atoi(parts[0])
if err != nil {
panic(err)
}
}
eof := make(chan bool)
go func() {
for {
buf := make([]byte, 1024)
n, err := master.Read(buf)
if err != nil {
eof <- true
break
}
os.Stdout.Write(buf[:n])
}
}()
go func() {
buf := make([]byte, 8192)
for {
n, err := os.Stdin.Read(buf)
if err != nil {
eof <- true
break
}
for i := 0; i < n; i++ {
switch buf[i] {
case sigInt:
syscall.Kill(commandPid, syscall.SIGINT)
case sigQuit:
syscall.Kill(commandPid, syscall.SIGQUIT)
case sigTstp:
syscall.Kill(commandPid, syscall.SIGTSTP)
syscall.Kill(os.Getpid(), syscall.SIGTSTP)
}
}
master.Write(buf[:n])
}
}()
<-eof
if exitStatus == -1 {
msg, err = usock.ReadMessage()
if err != nil {
panic(err)
}
parts := strings.Split(msg, "\000")
exitStatus, err = strconv.Atoi(parts[0])
if err != nil {
panic(err)
}
}
return exitStatus
}
func Run(args []string, input io.Reader, output *os.File) int {
if os.Getenv("RAILS_ENV") != "" {
println("Warning: Specifying a Rails environment via RAILS_ENV has no effect for commands run with zeus.")
println("As a safety precaution to protect you from nuking your development database,")
println("Zeus will now cowardly refuse to proceed. Please unset RAILS_ENV and try again.")
return 1
}
isTerminal := ttyutils.IsTerminal(output.Fd())
var master, slave *os.File
var err error
if isTerminal {
master, slave, err = pty.Open()
} else {
master, slave, err = unixsocket.Socketpair(syscall.SOCK_STREAM)
}
if err != nil {
slog.ErrorString(err.Error() + "\r")
return 1
}
defer master.Close()
var oldState *ttyutils.Termios
if isTerminal {
oldState, err = ttyutils.MakeTerminalRaw(output.Fd())
if err != nil {
slog.ErrorString(err.Error() + "\r")
return 1
}
defer ttyutils.RestoreTerminalState(output.Fd(), oldState)
}
// should this happen if we're running over a pipe? I think maybe not?
ttyutils.MirrorWinsize(output, master)
addr, err := net.ResolveUnixAddr("unixgram", unixsocket.ZeusSockName())
if err != nil {
slog.ErrorString(err.Error() + "\r")
return 1
}
conn, err := net.DialUnix("unix", nil, addr)
if err != nil {
zerror.ErrorCantConnectToMaster()
return 1
}
usock := unixsocket.New(conn)
msg := messages.CreateCommandAndArgumentsMessage(args, os.Getpid())
usock.WriteMessage(msg)
err = sendCommandLineArguments(usock, args)
if err != nil {
slog.ErrorString(err.Error() + "\r")
return 1
}
usock.WriteFD(int(slave.Fd()))
slave.Close()
msg, err = usock.ReadMessage()
if err != nil {
slog.ErrorString(err.Error() + "\r")
return 1
}
parts := strings.Split(msg, "\000")
commandPid, err := strconv.Atoi(parts[0])
defer func() {
if commandPid > 0 {
// Just in case.
syscall.Kill(commandPid, 9)
}
}()
if err != nil {
slog.ErrorString(err.Error() + "\r")
return 1
}
if isTerminal {
c := make(chan os.Signal, 1)
handledSignals := append(append(terminatingSignals, syscall.SIGWINCH), syscall.SIGCONT)
signal.Notify(c, handledSignals...)
go func() {
for sig := range c {
if sig == syscall.SIGCONT {
syscall.Kill(commandPid, syscall.SIGCONT)
} else if sig == syscall.SIGWINCH {
ttyutils.MirrorWinsize(output, master)
syscall.Kill(commandPid, syscall.SIGWINCH)
} else { // member of terminatingSignals
ttyutils.RestoreTerminalState(output.Fd(), oldState)
print("\r")
syscall.Kill(commandPid, sig.(syscall.Signal))
os.Exit(1)
}
}
}()
}
var exitStatus int = -1
if len(parts) > 2 {
exitStatus, err = strconv.Atoi(parts[0])
if err != nil {
slog.ErrorString(err.Error() + "\r")
return 1
}
}
eof := make(chan bool)
go func() {
for {
buf := make([]byte, 1024)
n, err := master.Read(buf)
if err == nil || (err == io.EOF && n > 0) {
output.Write(buf[:n])
} else {
eof <- true
break
}
}
}()
go func() {
buf := make([]byte, 8192)
for {
n, err := input.Read(buf)
if err != nil {
eof <- true
break
}
if isTerminal {
for i := 0; i < n; i++ {
switch buf[i] {
case sigInt:
syscall.Kill(commandPid, syscall.SIGINT)
case sigQuit:
syscall.Kill(commandPid, syscall.SIGQUIT)
case sigTstp:
syscall.Kill(commandPid, syscall.SIGTSTP)
syscall.Kill(os.Getpid(), syscall.SIGTSTP)
}
}
}
master.Write(buf[:n])
}
}()
<-eof
if exitStatus == -1 {
msg, err = usock.ReadMessage()
if err != nil {
slog.ErrorString(err.Error() + "\r")
return 1
}
parts := strings.Split(msg, "\000")
exitStatus, err = strconv.Atoi(parts[0])
if err != nil {
slog.ErrorString(err.Error() + "\r")
return 1
}
}
return exitStatus
}