func runRun(args *docopt.Args, client *controller.Client) error {
runDetached := args.Bool["--detached"]
runRelease := args.String["-r"]
if runRelease == "" {
release, err := client.GetAppRelease(mustApp())
if err == controller.ErrNotFound {
return errors.New("No app release, specify a release with -release")
}
if err != nil {
return err
}
runRelease = release.ID
}
req := &ct.NewJob{
Cmd: append([]string{args.String["<command>"]}, args.All["<argument>"].([]string)...),
TTY: term.IsTerminal(os.Stdin) && term.IsTerminal(os.Stdout) && !runDetached,
ReleaseID: runRelease,
}
if args.String["-e"] != "" {
req.Entrypoint = []string{args.String["-e"]}
}
if req.TTY {
cols, err := term.Cols()
if err != nil {
return err
}
lines, err := term.Lines()
if err != nil {
return err
}
req.Columns = cols
req.Lines = lines
req.Env = map[string]string{
"COLUMNS": strconv.Itoa(cols),
"LINES": strconv.Itoa(lines),
"TERM": os.Getenv("TERM"),
}
}
if runDetached {
job, err := client.RunJobDetached(mustApp(), req)
if err != nil {
return err
}
log.Println(job.ID)
return nil
}
rwc, err := client.RunJobAttached(mustApp(), req)
if err != nil {
return err
}
defer rwc.Close()
attachClient := cluster.NewAttachClient(rwc)
if req.TTY {
if err := term.MakeRaw(os.Stdin); err != nil {
return err
}
defer term.Restore(os.Stdin)
go func() {
ch := make(chan os.Signal)
signal.Notify(ch, SIGWINCH)
<-ch
height, err := term.Lines()
if err != nil {
return
}
width, err := term.Cols()
if err != nil {
return
}
attachClient.ResizeTTY(uint16(height), uint16(width))
attachClient.Signal(int(SIGWINCH))
}()
}
go func() {
ch := make(chan os.Signal)
signal.Notify(ch, syscall.SIGINT, syscall.SIGTERM)
sig := <-ch
attachClient.Signal(int(sig.(syscall.Signal)))
time.Sleep(10 * time.Second)
attachClient.Signal(int(syscall.SIGKILL))
}()
go func() {
io.Copy(attachClient, os.Stdin)
attachClient.CloseWrite()
}()
exitStatus, err := attachClient.Receive(os.Stdout, os.Stderr)
if err != nil {
return err
}
if req.TTY {
term.Restore(os.Stdin)
}
os.Exit(exitStatus)
panic("unreached")
}
func runRun(cmd *Command, args []string, client *controller.Client) error {
if len(args) == 0 {
cmd.printUsage(true)
}
if runRelease == "" {
release, err := client.GetAppRelease(mustApp())
if err == controller.ErrNotFound {
return errors.New("No app release, specify a release with -release")
}
if err != nil {
return err
}
runRelease = release.ID
}
req := &ct.NewJob{
Cmd: args,
TTY: term.IsTerminal(os.Stdin) && term.IsTerminal(os.Stdout) && !runDetached,
ReleaseID: runRelease,
}
if req.TTY {
cols, err := term.Cols()
if err != nil {
return err
}
lines, err := term.Lines()
if err != nil {
return err
}
req.Columns = cols
req.Lines = lines
req.Env = map[string]string{
"COLUMNS": strconv.Itoa(cols),
"LINES": strconv.Itoa(lines),
"TERM": os.Getenv("TERM"),
}
}
if runDetached {
job, err := client.RunJobDetached(mustApp(), req)
if err != nil {
return err
}
log.Println(job.ID)
return nil
}
rwc, err := client.RunJobAttached(mustApp(), req)
if err != nil {
return err
}
defer rwc.Close()
if req.TTY {
if err := term.MakeRaw(os.Stdin); err != nil {
return err
}
defer term.Restore(os.Stdin)
}
go func() {
io.Copy(rwc, os.Stdin)
rwc.CloseWrite()
}()
if req.TTY {
_, err = io.Copy(os.Stdout, rwc)
} else {
err = demultiplex.Copy(os.Stdout, os.Stderr, rwc)
}
// TODO: get exit code and use it
return err
}