-
-
Notifications
You must be signed in to change notification settings - Fork 424
Expand file tree
/
Copy pathexample_test.go
More file actions
101 lines (91 loc) · 2.63 KB
/
Copy pathexample_test.go
File metadata and controls
101 lines (91 loc) · 2.63 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
// Copyright (c) 2018, Daniel Martí <mvdan@mvdan.cc>
// See LICENSE for licensing information
package interp_test
import (
"context"
"fmt"
"io"
"os"
"strings"
"mvdan.cc/sh/v3/expand"
"mvdan.cc/sh/v3/interp"
"mvdan.cc/sh/v3/syntax"
)
func Example() {
src := `
foo=abc
for i in 1 2 3; do
foo+=$i
done
let bar=(2 + 3)
echo $foo $bar
echo $GLOBAL
`
file, _ := syntax.NewParser().Parse(strings.NewReader(src), "")
runner, _ := interp.New(
// Use [interp.Interactive] to enable interactive shell defaults like expanding aliases.
interp.Env(expand.ListEnviron("GLOBAL=global_value")),
interp.StdIO(nil, os.Stdout, os.Stdout),
)
runner.Run(context.TODO(), file)
// Output:
// abc123 5
// global_value
}
func ExampleExecHandlers() {
src := "echo foo; join ! foo bar baz; missing-program bar"
file, _ := syntax.NewParser().Parse(strings.NewReader(src), "")
execJoin := func(next interp.ExecHandlerFunc) interp.ExecHandlerFunc {
return func(ctx context.Context, args []string) error {
hc := interp.HandlerCtx(ctx)
if args[0] == "join" {
fmt.Fprintln(hc.Stdout, strings.Join(args[2:], args[1]))
return nil
}
return next(ctx, args)
}
}
execNotInstalled := func(next interp.ExecHandlerFunc) interp.ExecHandlerFunc {
return func(ctx context.Context, args []string) error {
hc := interp.HandlerCtx(ctx)
if _, err := interp.LookPathDir(hc.Dir, hc.Env, args[0]); err != nil {
fmt.Printf("%s is not installed\n", args[0])
return interp.ExitStatus(1)
}
return next(ctx, args)
}
}
runner, _ := interp.New(
interp.StdIO(nil, os.Stdout, os.Stdout),
interp.ExecHandlers(execJoin, execNotInstalled),
)
runner.Run(context.TODO(), file)
// Output:
// foo
// foo!bar!baz
// missing-program is not installed
}
type nopWriterCloser struct {
*strings.Reader
}
func (nopWriterCloser) Write([]byte) (int, error) { return 0, io.EOF }
func (nopWriterCloser) Close() error { return nil }
func ExampleOpenHandler() {
src := "echo $(</etc/hostname); echo bar >/dev/null"
file, _ := syntax.NewParser().Parse(strings.NewReader(src), "")
open := func(ctx context.Context, path string, flag int, perm os.FileMode) (io.ReadWriteCloser, error) {
// Hard-code the contents of the hostname file for all platforms.
if path == "/etc/hostname" {
return nopWriterCloser{strings.NewReader("mymachine")}, nil
}
// DefaultOpenHandler already redirects /dev/null to NUL on Windows.
return interp.DefaultOpenHandler()(ctx, path, flag, perm)
}
runner, _ := interp.New(
interp.StdIO(nil, os.Stdout, os.Stdout),
interp.OpenHandler(open),
)
runner.Run(context.TODO(), file)
// Output:
// mymachine
}