package main import ( "io" "os" "bufio" "main/walk" "main/json" ) type ProgramState struct { value, xreg, yreg, zreg []walk.Value in walk.StredReader out walk.StredWriter program []Command pc int } type config struct { quiet bool program string in io.Reader out io.Writer } func run(config config) { tokens := Lex(config.program) program := Parse(tokens) stdin := bufio.NewReader(config.in) stdout := bufio.NewWriter(config.out) state := ProgramState { in: json.NewJSONReader(stdin), out: json.NewJSONWriter(stdout), program: program, } for { walkItem, err := state.in.Read() if err != nil { break } state.value = []walk.Value{walkItem.Value} state.pc = 0 for state.pc < len(state.program) { state.program[state.pc].exec(&state) } if !config.quiet { for _, value := range state.value { err := state.out.Write(value) if err != nil { panic("Error while outputting") } } } } state.in.AssertDone() state.out.AssertDone() } func main() { config := config { quiet: false, in: os.Stdin, out: os.Stdout, } hasInput := false for i := 1; i < len(os.Args); i += 1 { switch os.Args[i] { case "-n": config.quiet = true continue } if i < len(os.Args) - 1 { panic("Unexpected arguments after program") } config.program = os.Args[i] hasInput = true } if !hasInput { panic("Missing program") } run(config) }