mirror of
https://github.com/yudai/gotty.git
synced 2026-01-01 23:28:49 +01:00
2424e48e17
Adjusted dependencies to use our own repository mirrors. Modified the Makefile for proper build output and target directory.
48 lines
1.3 KiB
Go
48 lines
1.3 KiB
Go
package localcommand
|
|
|
|
import (
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/kernel-punk/gotty/server"
|
|
)
|
|
|
|
type Options struct {
|
|
CloseSignal int `hcl:"close_signal" flagName:"close-signal" flagSName:"" flagDescribe:"Signal sent to the command process when gotty close it (default: SIGHUP)" default:"1"`
|
|
CloseTimeout int `hcl:"close_timeout" flagName:"close-timeout" flagSName:"" flagDescribe:"Time in seconds to force kill process after client is disconnected (default: -1)" default:"-1"`
|
|
}
|
|
|
|
type Factory struct {
|
|
command string
|
|
argv []string
|
|
options *Options
|
|
opts []Option
|
|
}
|
|
|
|
func NewFactory(command string, argv []string, options *Options) (*Factory, error) {
|
|
opts := []Option{WithCloseSignal(syscall.Signal(options.CloseSignal))}
|
|
if options.CloseTimeout >= 0 {
|
|
opts = append(opts, WithCloseTimeout(time.Duration(options.CloseTimeout)*time.Second))
|
|
}
|
|
|
|
return &Factory{
|
|
command: command,
|
|
argv: argv,
|
|
options: options,
|
|
opts: opts,
|
|
}, nil
|
|
}
|
|
|
|
func (factory *Factory) Name() string {
|
|
return "local command"
|
|
}
|
|
|
|
func (factory *Factory) New(params map[string][]string) (server.Slave, error) {
|
|
argv := make([]string, len(factory.argv))
|
|
copy(argv, factory.argv)
|
|
if params["arg"] != nil && len(params["arg"]) > 0 {
|
|
argv = append(argv, params["arg"]...)
|
|
}
|
|
|
|
return New(factory.command, argv, factory.opts...)
|
|
}
|