|
3 | 3 | package main |
4 | 4 |
|
5 | 5 | import ( |
6 | | - "flag" |
7 | 6 | "fmt" |
8 | | - "strings" |
9 | | -) |
| 7 | + "io" |
| 8 | + "os" |
10 | 9 |
|
11 | | -var ( |
12 | | - version string |
| 10 | + "foxygo.at/evy/pkg/evaluator" |
| 11 | + "foxygo.at/evy/pkg/lexer" |
| 12 | + "foxygo.at/evy/pkg/parser" |
| 13 | + "github.com/alecthomas/kong" |
13 | 14 | ) |
14 | 15 |
|
15 | | -func main() { |
16 | | - versionFlag := flag.Bool("version", false, "Print version information") |
17 | | - flag.Parse() |
18 | | - if *versionFlag { |
19 | | - fmt.Println("Version", version) |
20 | | - return |
| 16 | +var version string = "v0.0.0" |
| 17 | + |
| 18 | +const description = ` |
| 19 | +evy is a tool for managing evy source code. |
| 20 | +` |
| 21 | + |
| 22 | +type config struct { |
| 23 | + Version kong.VersionFlag `short:"V" help:"Print version information"` |
| 24 | + Run cmdRun `cmd:"" help:"Run evy program"` |
| 25 | + Tokenize cmdTokenize `cmd:"" help:"Tokenize evy program"` |
| 26 | + Parse cmdParse `cmd:"" help:"Parse evy program"` |
| 27 | +} |
| 28 | + |
| 29 | +type cmdRun struct { |
| 30 | + Source string `arg:"" help:"Source file. Default stdin" default:"-"` |
| 31 | +} |
| 32 | +type cmdTokenize struct { |
| 33 | + Source string `arg:"" help:"Source file. Default stdin" default:"-"` |
| 34 | +} |
| 35 | +type cmdParse struct { |
| 36 | + Source string `arg:"" help:"Source file. Default stdin" default:"-"` |
| 37 | +} |
| 38 | + |
| 39 | +func (c *cmdRun) Run() error { |
| 40 | + b, err := fileBytes(c.Source) |
| 41 | + if err != nil { |
| 42 | + return err |
| 43 | + } |
| 44 | + print := func(s string) { fmt.Print(s) } |
| 45 | + evaluator.Run(string(b), print) |
| 46 | + return nil |
| 47 | +} |
| 48 | + |
| 49 | +func (c *cmdTokenize) Run() error { |
| 50 | + b, err := fileBytes(c.Source) |
| 51 | + if err != nil { |
| 52 | + return err |
21 | 53 | } |
| 54 | + result := lexer.Run(string(b)) |
| 55 | + fmt.Println(result) |
| 56 | + return nil |
| 57 | +} |
22 | 58 |
|
23 | | - fmt.Println(evaluate("some program")) |
| 59 | +func (c *cmdParse) Run() error { |
| 60 | + b, err := fileBytes(c.Source) |
| 61 | + if err != nil { |
| 62 | + return err |
| 63 | + } |
| 64 | + result := parser.Run(string(b)) |
| 65 | + fmt.Println(result) |
| 66 | + return nil |
24 | 67 | } |
25 | 68 |
|
26 | | -func evaluate(program string) string { |
27 | | - return strings.ToUpper(truncate(program, 20)) |
| 69 | +func main() { |
| 70 | + kctx := kong.Parse(&config{}, |
| 71 | + kong.Description(description), |
| 72 | + kong.Vars{"version": version}, |
| 73 | + ) |
| 74 | + kctx.FatalIfErrorf(kctx.Run()) |
28 | 75 | } |
29 | 76 |
|
30 | | -func truncate(s string, max int) string { |
31 | | - r := []rune(s) |
32 | | - if len(r) <= max { |
33 | | - return s |
| 77 | +func fileBytes(filename string) ([]byte, error) { |
| 78 | + if filename == "-" { |
| 79 | + return io.ReadAll(os.Stdin) |
34 | 80 | } |
35 | | - return string(r[:max]) |
| 81 | + return os.ReadFile(filename) |
36 | 82 | } |
0 commit comments