package vm

import (
	"errors"
	"fmt"
	"strconv"

	"github.com/mediocregopher/ginger/lang"
	"llvm.org/llvm/bindings/go/llvm"
)

type op interface {
	inType() valType
	outType() valType
	build(*Module) (llvm.Value, error)
}

type valType struct {
	term lang.Term
	llvm llvm.Type
}

func (vt valType) isInt() bool {
	return lang.Equal(Int, vt.term)
}

func (vt valType) eq(vt2 valType) bool {
	return lang.Equal(vt.term, vt2.term) && vt.llvm == vt2.llvm
}

// primitive valTypes
var (
	valTypeVoid = valType{term: lang.Tuple{}, llvm: llvm.VoidType()}
	valTypeInt  = valType{term: Int, llvm: llvm.Int64Type()}
)

////////////////////////////////////////////////////////////////////////////////

// most types don't have an input, so we use this as a shortcut
type voidIn struct{}

func (voidIn) inType() valType {
	return valTypeVoid
}

////////////////////////////////////////////////////////////////////////////////

type intOp struct {
	voidIn
	c lang.Const
}

func (io intOp) outType() valType {
	return valTypeInt
}

func (io intOp) build(mod *Module) (llvm.Value, error) {
	ci, err := strconv.ParseInt(string(io.c), 10, 64)
	if err != nil {
		return llvm.Value{}, err
	}
	return llvm.ConstInt(llvm.Int64Type(), uint64(ci), false), nil
}

////////////////////////////////////////////////////////////////////////////////

type tupOp struct {
	voidIn
	els []op
}

func (to tupOp) outType() valType {
	termTypes := make(lang.Tuple, len(to.els))
	llvmTypes := make([]llvm.Type, len(to.els))
	for i := range to.els {
		elValType := to.els[i].outType()
		termTypes[i] = elValType.term
		llvmTypes[i] = elValType.llvm
	}
	vt := valType{term: lang.Tuple{Tuple, termTypes}}
	if len(llvmTypes) == 0 {
		vt.llvm = llvm.VoidType()
	} else {
		vt.llvm = llvm.StructType(llvmTypes, false)
	}
	return vt
}

func (to tupOp) build(mod *Module) (llvm.Value, error) {
	str := llvm.Undef(to.outType().llvm)
	var val llvm.Value
	var err error
	for i := range to.els {
		if val, err = to.els[i].build(mod); err != nil {
			return llvm.Value{}, err
		}
		str = mod.b.CreateInsertValue(str, val, i, "")
	}
	return str, err
}

////////////////////////////////////////////////////////////////////////////////

type tupElOp struct {
	voidIn
	tup op
	i   int
}

func (teo tupElOp) outType() valType {
	tupType := teo.tup.outType()
	return valType{
		llvm: tupType.llvm.StructElementTypes()[teo.i],
		term: tupType.term.(lang.Tuple)[1].(lang.Tuple)[1],
	}
}

func (teo tupElOp) build(mod *Module) (llvm.Value, error) {
	if to, ok := teo.tup.(tupOp); ok {
		return to.els[teo.i].build(mod)
	}

	tv, err := teo.tup.build(mod)
	if err != nil {
		return llvm.Value{}, err
	}
	return mod.b.CreateExtractValue(tv, teo.i, ""), nil
}

////////////////////////////////////////////////////////////////////////////////

type varOp struct {
	op
	v     llvm.Value
	built bool
}

func (vo *varOp) build(mod *Module) (llvm.Value, error) {
	if !vo.built {
		var err error
		if vo.v, err = vo.op.build(mod); err != nil {
			return llvm.Value{}, err
		}
		vo.built = true
	}
	return vo.v, nil
}

type varCtx map[string]*varOp

func (c varCtx) assign(name string, vo *varOp) error {
	if _, ok := c[name]; ok {
		return fmt.Errorf("var %q already assigned", name)
	}
	c[name] = vo
	return nil
}

func (c varCtx) get(name string) (*varOp, error) {
	if o, ok := c[name]; ok {
		return o, nil
	}
	return nil, fmt.Errorf("var %q referenced before assignment", name)
}

////////////////////////////////////////////////////////////////////////////////

type addOp struct {
	voidIn
	a, b op
}

func (ao addOp) outType() valType {
	return ao.a.outType()
}

func (ao addOp) build(mod *Module) (llvm.Value, error) {
	av, err := ao.a.build(mod)
	if err != nil {
		return llvm.Value{}, err
	}
	bv, err := ao.b.build(mod)
	if err != nil {
		return llvm.Value{}, err
	}
	return mod.b.CreateAdd(av, bv, ""), nil
}

////////////////////////////////////////////////////////////////////////////////

func termToOp(ctx varCtx, t lang.Term) (op, error) {
	aPat := func(a lang.Atom) lang.Tuple {
		return lang.Tuple{lang.AAtom, a}
	}
	cPat := func(t lang.Term) lang.Tuple {
		return lang.Tuple{lang.AConst, t}
	}
	tPat := func(el ...lang.Term) lang.Tuple {
		return lang.Tuple{Tuple, lang.Tuple(el)}
	}

	if !lang.Match(tPat(aPat(lang.AUnder), lang.TDblUnder), t) {
		return nil, fmt.Errorf("term %v does not look like a vm command", t)
	}
	k := t.(lang.Tuple)[0].(lang.Atom)
	v := t.(lang.Tuple)[1]

	// for when v is a Tuple argument, convenience function for casting
	vAsTup := func(n int) ([]op, error) {
		vop, err := termToOp(ctx, v)
		if err != nil {
			return nil, err
		}
		ops := make([]op, n)
		for i := range ops {
			ops[i] = tupElOp{tup: vop, i: i}
		}

		return ops, nil
	}

	switch k {
	case Int:
		if !lang.Match(cPat(lang.AUnder), v) {
			return nil, errors.New("int requires constant arg")
		}
		return intOp{c: v.(lang.Const)}, nil
	case Tuple:
		if !lang.Match(lang.Tuple{Tuple, lang.AUnder}, v) {
			return nil, errors.New("tup requires tuple arg")
		}
		tup := v.(lang.Tuple)
		tc := tupOp{els: make([]op, len(tup))}
		var err error
		for i := range tup {
			if tc.els[i], err = termToOp(ctx, tup[i]); err != nil {
				return nil, err
			}
		}
		return tc, nil
	case Var:
		if !lang.Match(aPat(lang.AUnder), v) {
			return nil, errors.New("var requires atom arg")
		}
		name := v.(lang.Atom).String()
		return ctx.get(name)

	case Assign:
		if !lang.Match(tPat(tPat(aPat(Var), aPat(lang.AUnder)), lang.TDblUnder), v) {
			return nil, errors.New("assign requires 2-tuple arg, the first being a var")
		}
		tup := v.(lang.Tuple)
		name := tup[0].(lang.Tuple)[1].String()
		o, err := termToOp(ctx, tup[1])
		if err != nil {
			return nil, err
		}

		vo := &varOp{op: o}
		if err := ctx.assign(name, vo); err != nil {
			return nil, err
		}
		return vo, nil

	// Add is special in some way, I think it's a function not a compiler op,
	// not sure yet though
	case Add:
		els, err := vAsTup(2)
		if err != nil {
			return nil, err
		} else if !els[0].outType().eq(valTypeInt) {
			return nil, errors.New("add args must be numbers of the same type")
		} else if !els[1].outType().eq(valTypeInt) {
			return nil, errors.New("add args must be numbers of the same type")
		}
		return addOp{a: els[0], b: els[1]}, nil
	default:
		return nil, fmt.Errorf("op %v unknown, or its args are malformed", t)
	}
}