Merge pull request #5 from atakanyenel/sleep

add sleep command
This commit is contained in:
Vladimir Vivien
2019-07-17 12:02:32 -04:00
committed by GitHub
5 changed files with 59 additions and 6 deletions

1
.gitignore vendored
View File

@@ -22,3 +22,4 @@ _testmain.go
*.exe
*.test
*.prof
gosh

View File

@@ -11,7 +11,8 @@ files that implement command plugins.
#### Pre-requisites
* Go 1.8 or above
* Linux (for now)
* Linux
* Mac OSX
Gosh makes it easy to create shell programs. First, download or clone this
repository. For a quick start, run the following:

View File

@@ -9,12 +9,12 @@ import (
)
var (
testPluginsDir = "../plugins"
testPluginsDir = "./plugins"
)
func TestShellNew(t *testing.T) {
shell := New()
if shell.pluginsDir != pluginsDir {
if shell.pluginsDir != testPluginsDir {
t.Error("pluginsDir not set")
}
}
@@ -49,10 +49,10 @@ func TestShellHandle(t *testing.T) {
helloOut := bytes.NewBufferString("")
shell.ctx = context.WithValue(context.TODO(), "gosh.stdout", helloOut)
if err := shell.handle("testhello"); err == nil {
if _, err := shell.handle(shell.ctx, "testhello"); err == nil {
t.Error("this test should have failed with command not found")
}
if err := shell.handle("hello"); err != nil {
if _, err := shell.handle(shell.ctx, "hello"); err != nil {
t.Error(err)
}
printedOut := strings.TrimSpace(helloOut.String())
@@ -62,7 +62,7 @@ func TestShellHandle(t *testing.T) {
byeOut := bytes.NewBufferString("")
shell.ctx = context.WithValue(context.TODO(), "gosh.stdout", byeOut)
if err := shell.handle("goodbye"); err != nil {
if _, err := shell.handle(shell.ctx, "goodbye"); err != nil {
t.Error(err)
}
printedOut = strings.TrimSpace(byeOut.String())

51
plugins/sleepcmd.go Normal file
View File

@@ -0,0 +1,51 @@
package main
import (
"context"
"fmt"
"github.com/vladimirvivien/gosh/api"
"io"
"strconv"
"time"
)
type sleepCmd string
func (s sleepCmd) Name() string {
return string(s)
}
func (s sleepCmd) Usage() string { return "Usage: sleep <duration>" }
func (s sleepCmd) ShortDesc() string { return "sleeps for <duration> seconds" }
func (s sleepCmd) LongDesc() string { return s.ShortDesc() }
func (s sleepCmd) Exec(ctx context.Context, args []string) (context.Context, error) {
if len(args) == 2 {
duration, err := strconv.Atoi(args[1])
if err != nil {
return ctx, err
}
time.Sleep(time.Duration(duration) * time.Second)
return ctx, nil
}
out := ctx.Value("gosh.stdout").(io.Writer)
fmt.Fprintln(out, s.Usage())
return ctx, nil
}
type sleepCmds struct{}
func (s *sleepCmds) Init(ctx context.Context) error {
out := ctx.Value("gosh.stdout").(io.Writer)
fmt.Fprintln(out, "sleep module loaded")
return nil
}
func (s *sleepCmds) Registry() map[string]api.Command {
return map[string]api.Command{
"sleep": sleepCmd("sleep"),
}
}
var Commands sleepCmds