mirror of
https://github.com/weaveworks/scope.git
synced 2026-03-04 02:30:45 +00:00
41193b428e
It is not a singleton anymore. Instead it is an object with a registry backend. The default registry backend is provided, which is equivalent to what used to be before. Custom backend can be provided for testing purposes. The registry also supports batch operations to remove and add handlers as an atomic step.
59 lines
1.2 KiB
Go
59 lines
1.2 KiB
Go
package host
|
|
|
|
import (
|
|
"os/exec"
|
|
|
|
log "github.com/Sirupsen/logrus"
|
|
"github.com/kr/pty"
|
|
|
|
"github.com/weaveworks/scope/common/xfer"
|
|
"github.com/weaveworks/scope/probe/controls"
|
|
)
|
|
|
|
// Control IDs used by the host integration.
|
|
const (
|
|
ExecHost = "host_exec"
|
|
)
|
|
|
|
func (r *Reporter) registerControls() {
|
|
r.handlerRegistry.Register(ExecHost, r.execHost)
|
|
}
|
|
|
|
func (r *Reporter) deregisterControls() {
|
|
r.handlerRegistry.Rm(ExecHost)
|
|
}
|
|
|
|
func (r *Reporter) execHost(req xfer.Request) xfer.Response {
|
|
cmd := exec.Command(r.hostShellCmd[0], r.hostShellCmd[1:]...)
|
|
cmd.Env = []string{"TERM=xterm"}
|
|
ptyPipe, err := pty.Start(cmd)
|
|
if err != nil {
|
|
return xfer.ResponseError(err)
|
|
}
|
|
|
|
id, pipe, err := controls.NewPipeFromEnds(nil, ptyPipe, r.pipes, req.AppID)
|
|
if err != nil {
|
|
return xfer.ResponseError(err)
|
|
}
|
|
pipe.OnClose(func() {
|
|
if err := cmd.Process.Kill(); err != nil {
|
|
log.Errorf("Error stopping host shell: %v", err)
|
|
}
|
|
if err := ptyPipe.Close(); err != nil {
|
|
log.Errorf("Error closing host shell's pty: %v", err)
|
|
}
|
|
log.Info("Host shell closed.")
|
|
})
|
|
go func() {
|
|
if err := cmd.Wait(); err != nil {
|
|
log.Errorf("Error waiting on host shell: %v", err)
|
|
}
|
|
pipe.Close()
|
|
}()
|
|
|
|
return xfer.Response{
|
|
Pipe: id,
|
|
RawTTY: true,
|
|
}
|
|
}
|