func TestEvalAddition(t *testing.T) {
var tests = []struct {
expr string
res int
}{
{"(+ 2)", 2},
{"(+ 1 2)", 3},
{"(+ 1 2 3)", 6},
{"(+ 1 2 -3)", 0},
{"(+ 1 (+ 2 3))", 6},
{"(+ (+ 1 2) (+ 3 4))", 10},
}
for x, test := range tests {
res := eval.EvalExpr(test.expr)
i, ok := res.(int)
if !ok || i != test.res {
t.Log(x, "- Expected:", test.res)
t.Fatal(x, "- Got:", i)
}
}
}
func main() {
t := flag.Bool("t", false, "Transpile")
flag.Parse()
if flag.NArg() >= 1 {
if *t == true {
fmt.Println("transpiling")
out, err := os.Create("out.c")
if err != nil {
fmt.Println("create:", err)
os.Exit(1)
}
data, err := ioutil.ReadFile(flag.Arg(0))
if err != nil {
fmt.Println(err)
}
transpileCode(out, string(data)) //flag.Arg(0))
return
}
f, err := os.Open(flag.Arg(0))
if err != nil {
fmt.Println(err)
os.Exit(1)
}
defer f.Close()
var fi os.FileInfo
fi, err = f.Stat()
if !fi.IsDir() {
data, err := ioutil.ReadFile(flag.Arg(0))
if err != nil {
fmt.Println(err)
} else {
eval.EvalFile(flag.Arg(0), string(stripCR(data)))
}
} else {
fset := token.NewFileSet()
names, err := f.Readdirnames(0)
if err != nil {
fmt.Println(err)
os.Exit(1)
}
for _, name := range names {
data, err := ioutil.ReadFile(filepath.Join(flag.Arg(0), name))
if err != nil {
fmt.Println(err)
os.Exit(1)
}
fset.AddFile(name, string(stripCR(data)))
}
eval.EvalPackage(flag.Arg(0), fset)
}
} else {
fmt.Println("Welcome to Calc REPL", version)
fmt.Println()
fmt.Println("Type in expression(s) to evaluate on one or more lines.")
fmt.Println("Press enter on an empty line to execute the expression(s).")
fmt.Println("Type 'q' (without quotes) on an empty line to exit.")
in := bufio.NewReader(os.Stdin)
for {
fmt.Print(">>>")
var expr string
stop := false
for !stop {
b, _ := in.ReadBytes('\n')
b = stripCR(b)
if len(b) <= 2 {
switch b[0] {
case 'q':
fmt.Println("QUIT!")
os.Exit(0)
case '\n':
stop = true
}
}
expr += string(b)
}
res := eval.EvalExpr(expr)
if res != nil {
fmt.Println(res)
}
}
}
}