package main

import (
	"fmt"

	"github.com/mediocregopher/ginger/expr"

	"llvm.org/llvm/bindings/go/llvm"
)

func main() {
	//ee, err := expr.Parse(os.Stdin)
	//if err != nil {
	//	panic(err)
	//}
	//for _, e := range ee {
	//	fmt.Println(e)
	//}

	llvm.LinkInMCJIT()
	llvm.InitializeNativeTarget()
	llvm.InitializeNativeAsmPrinter()

	// setup our context, builder, and module
	ctx := expr.NewCtx()
	bctx := expr.BuildCtx{
		C: &ctx,
		B: llvm.NewBuilder(),
		M: llvm.NewModule("my_module"),
	}

	// do the work in the function
	add := expr.Macro("add")
	bind := expr.Macro("bind")
	idA := expr.Identifier("A")
	idB := expr.Identifier("B")
	idC := expr.Identifier("C")
	stmts := []expr.Statement{
		expr.NewStatement(bind, idA, expr.NewStatement(add, expr.Int(1), expr.Int(2))),
		expr.NewStatement(bind, idB, expr.Int(3)),
		expr.NewStatement(bind, idC, expr.NewTuple(idA, idB)),
		expr.NewStatement(add, expr.NewStatement(add, idC), expr.NewStatement(add, idC)),
	}

	//block := expr.Block([]expr.Expr{stmt})
	//fn := block.LLVMVal(expr.RootCtx, lctx)

	// create main and call our function
	mainFn := llvm.AddFunction(bctx.M, "main", llvm.FunctionType(llvm.Int64Type(), []llvm.Type{}, false))
	mainBlock := llvm.AddBasicBlock(mainFn, "entry")
	bctx.B.SetInsertPoint(mainBlock, mainBlock.FirstInstruction())
	v := bctx.Build(stmts...)
	bctx.B.CreateRet(v)

	//ret := lctx.B.CreateCall(fn, []llvm.Value{}, "")
	//lctx.B.CreateRet(ret)

	// verify it's all good
	if err := llvm.VerifyModule(bctx.M, llvm.ReturnStatusAction); err != nil {
		panic(err)
	}
	fmt.Println("# verified")

	// Dump the IR
	fmt.Println("# dumping IR")
	bctx.M.Dump()
	fmt.Println("# done dumping IR")

	// create our exe engine
	fmt.Println("# creating new execution engine")
	engine, err := llvm.NewExecutionEngine(bctx.M)
	if err != nil {
		panic(err)
	}

	// run the function!
	fmt.Println("# running the function main")
	funcResult := engine.RunFunction(bctx.M.NamedFunction("main"), []llvm.GenericValue{})
	fmt.Printf("%d\n", funcResult.Int(false))
}