package main

import (
	"fmt"
	"io"
)

func (msg *MsgName) iterTop(w io.Writer) {
	var BASE string = msg.Name

	fmt.Fprintln(w, "type "+BASE+"Iterator struct {")
	fmt.Fprintln(w, "	sync.RWMutex")
	fmt.Fprintln(w, "")
	fmt.Fprintln(w, "	things []*"+BASE)
	fmt.Fprintln(w, "	index int")
	fmt.Fprintln(w, "}")
	fmt.Fprintln(w, "")
	fmt.Fprintln(w, "// New"+BASE+"Iterator initializes a new iterator.")
	fmt.Fprintln(w, "func New"+BASE+"Iterator(things []*"+BASE+") *"+BASE+"Iterator {")
	fmt.Fprintln(w, "	return &"+BASE+"Iterator{things: things}")
	fmt.Fprintln(w, "}")
	fmt.Fprintln(w, "")
	fmt.Fprintln(w, "// Scan moves to the next element and returns false if there are no more things.")
	fmt.Fprintln(w, "// Use Scan() in a loop, similar to a while loop")
	fmt.Fprintln(w, "//")
	fmt.Fprintln(w, "//	for iterator.Scan() ")
	fmt.Fprintln(w, "//		d := iterator.Next(")
	fmt.Fprintln(w, "//		fmt.Println(\"found UUID:\", d.Uuid")
	fmt.Fprintln(w, "//	}")
	fmt.Fprintln(w, "func (it *"+BASE+"Iterator) Scan() bool {")
	fmt.Fprintln(w, "	if it.index >= len(it.things) {")
	fmt.Fprintln(w, "		return false")
	fmt.Fprintln(w, "	}")
	fmt.Fprintln(w, "	it.index++")
	fmt.Fprintln(w, "	return true")
	fmt.Fprintln(w, "}")
	fmt.Fprintln(w, "")
}

func (msg *MsgName) iterNext(w io.Writer) {
	var BASE string = msg.Name

	fmt.Fprintln(w, "// Next() returns the next thing in the array")
	fmt.Fprintln(w, "func (it *"+BASE+"Iterator) Next() *"+BASE+" {")
	fmt.Fprintln(w, "	if it.things[it.index-1] == nil {")
	fmt.Fprintln(w, "		for i, d := range it.things {")
	fmt.Fprintln(w, "			fmt.Println(\"i =\", i, d)")
	fmt.Fprintln(w, "		}")
	fmt.Fprintln(w, "	}")
	fmt.Fprintln(w, "	return it.things[it.index-1]")
	fmt.Fprintln(w, "}")
	fmt.Fprintln(w, "")
}

func (pf *File) selectAllFunc(w io.Writer) {
	var BASES string = pf.Bases.Name
	var BASE string = pf.Base.Name
	var LOCK string = pf.Bases.Lockname

	fmt.Fprintln(w, "func (all *"+BASES+") All() *"+BASE+"Iterator {")
	fmt.Fprintln(w, "	"+BASE+"Pointers := all.selectAll"+BASE+"()")
	fmt.Fprintln(w, "")
	fmt.Fprintln(w, "	iterator := New"+BASE+"Iterator("+BASE+"Pointers)")
	fmt.Fprintln(w, "	return iterator")
	fmt.Fprintln(w, "}")
	fmt.Fprintln(w, "")
	fmt.Fprintln(w, "func (all *"+BASES+") Len() int {")
	fmt.Fprintln(w, "	"+LOCK+".RLock()")
	fmt.Fprintln(w, "	defer "+LOCK+".RUnlock()")
	fmt.Fprintln(w, "")
	fmt.Fprintln(w, "	return len(all."+BASES+")")
	fmt.Fprintln(w, "}")
	fmt.Fprintln(w, "")
}

func (pf *File) sortByFunc(w io.Writer) {
	var BASES string = pf.Bases.Name
	var BASE string = pf.Base.Name

	for _, SORT := range pf.Base.Sort {
		fmt.Fprintln(w, "func (all *"+BASES+") SortBy"+SORT+"() *"+BASE+"Iterator {")
		fmt.Fprintln(w, "	things := all.selectAll"+BASE+"()")
		fmt.Fprintln(w, "")
		fmt.Fprintln(w, "	sort.Sort("+BASE+""+SORT+"(things))")
		fmt.Fprintln(w, "")
		fmt.Fprintln(w, "	iterator := New"+BASE+"Iterator(things)")
		fmt.Fprintln(w, "	return iterator")
		fmt.Fprintln(w, "}")
		fmt.Fprintln(w, "")

		fmt.Fprintln(w, "type "+BASE+""+SORT+" []*"+BASE+"")
		fmt.Fprintln(w, "")
		fmt.Fprintln(w, "func (a "+BASE+""+SORT+") Len() int           { return len(a) }")
		fmt.Fprintln(w, "func (a "+BASE+""+SORT+") Less(i, j int) bool { return a[i]."+SORT+" < a[j]."+SORT+" }")
		fmt.Fprintln(w, "func (a "+BASE+""+SORT+") Swap(i, j int)      { a[i], a[j] = a[j], a[i] }")
		fmt.Fprintln(w, "")
	}
}

func (pf *File) iterSelect(w io.Writer) {
	var BASES string = pf.Bases.Name
	var BASE string = pf.Base.Name
	var LOCK string = pf.Bases.Lockname

	fmt.Fprintln(w, "// safely returns a slice of pointers to the "+BASE+" protobufs")
	fmt.Fprintln(w, "func (all *"+BASES+") selectAll"+BASE+"() []*"+BASE+" {")
	fmt.Fprintln(w, "	"+LOCK+".RLock()")
	fmt.Fprintln(w, "	defer "+LOCK+".RUnlock()")
	fmt.Fprintln(w, "")
	fmt.Fprintln(w, "	// Create a new slice to hold pointers to each "+BASE+"")
	fmt.Fprintln(w, "	var tmp []*"+BASE+"")
	fmt.Fprintln(w, "	tmp = make([]*"+BASE+", len(all."+BASES+"))")
	fmt.Fprintln(w, "	for i, p := range all."+BASES+" {")
	fmt.Fprintln(w, "		tmp[i] = p // Copy pointers for safe iteration")
	fmt.Fprintln(w, "	}")
	fmt.Fprintln(w, "")
	fmt.Fprintln(w, "	return tmp")
	fmt.Fprintln(w, "}")
}

func (pf *File) appendUniqueOld(w io.Writer) {
	var MSG string = pf.Bases.Name
	var BASE string = pf.Base.Name
	var LOCK string = pf.Bases.Lockname

	if argv.Mutex {
		// use the mutex lock from the modified protoc.pb.go file
		LOCK = "all.Lock"
	} else {
		LOCK = pf.Bases.Lockname
	}

	// append -- no check at all
	fmt.Fprintln(w, "// just a simple Append() with no checking (but still uses the mutex lock)")
	fmt.Fprintln(w, "func (all *"+MSG+") Append(newP *"+BASE+") bool {")
	fmt.Fprintln(w, "	"+LOCK+".RLock()")
	fmt.Fprintln(w, "	defer "+LOCK+".RUnlock()")
	fmt.Fprintln(w, "")
	fmt.Fprintln(w, "	all."+MSG+" = append(all."+MSG+", newP)")
	fmt.Fprintln(w, "	return true")
	fmt.Fprintln(w, "}")
	fmt.Fprintln(w, "")

	// append for single keys
	for _, KEY := range pf.Base.Unique {
		fmt.Fprintln(w, "// enforces "+BASE+" is unique")
		fmt.Fprintln(w, "func (all *"+MSG+") AppendUnique"+KEY+"(newP *"+BASE+") bool {")
		fmt.Fprintln(w, "	"+LOCK+".RLock()")
		fmt.Fprintln(w, "	defer "+LOCK+".RUnlock()")
		fmt.Fprintln(w, "")
		fmt.Fprintln(w, "	for _, p := range all."+MSG+" {")
		fmt.Fprintln(w, "		if p."+KEY+" == newP."+KEY+" {")
		fmt.Fprintln(w, "			return false")
		fmt.Fprintln(w, "		}")
		fmt.Fprintln(w, "	}")
		fmt.Fprintln(w, "")
		fmt.Fprintln(w, "	all."+MSG+" = append(all."+MSG+", newP)")
		fmt.Fprintln(w, "	return true")
		fmt.Fprintln(w, "}")
		fmt.Fprintln(w, "")
	}

	// append check for every key
	if len(pf.Base.Unique) == 0 {
		// there are no keys defined
		return
	}
	fmt.Fprintln(w, "// enforces "+BASE+" is unique")
	fmt.Fprintln(w, "func (all *"+MSG+") AppendUnique(newP *"+BASE+") bool {")
	fmt.Fprintln(w, "	"+LOCK+".RLock()")
	fmt.Fprintln(w, "	defer "+LOCK+".RUnlock()")
	fmt.Fprintln(w, "")
	fmt.Fprintln(w, "	for _, p := range all."+MSG+" {")
	for _, KEY := range pf.Base.Unique {
		fmt.Fprintln(w, "		if p."+KEY+" == newP."+KEY+" {")
		fmt.Fprintln(w, "			return false")
		fmt.Fprintln(w, "		}")
	}
	fmt.Fprintln(w, "	}")
	fmt.Fprintln(w, "")
	fmt.Fprintln(w, "	all."+MSG+" = append(all."+MSG+", newP)")
	fmt.Fprintln(w, "	return true")
	fmt.Fprintln(w, "}")
	fmt.Fprintln(w, "")
}

func (pf *File) replaceFunc(w io.Writer) {
	var MSG string = pf.Bases.Name
	var BASE string = pf.Base.Name
	var LOCK string = pf.Bases.Lockname

	for _, KEY := range pf.Base.Unique {
		fmt.Fprintln(w, "// enforces "+KEY+" is unique")
		fmt.Fprintln(w, "func (all *"+MSG+") Replace"+KEY+"(newP *"+BASE+") bool { // todo: make unique name here")
		fmt.Fprintln(w, "	"+LOCK+".RLock()")
		fmt.Fprintln(w, "	defer "+LOCK+".RUnlock()")
		fmt.Fprintln(w, "")
		fmt.Fprintln(w, "	for _, p := range all."+MSG+" {")
		fmt.Fprintln(w, "		if p."+KEY+" == newP."+KEY+" {")
		fmt.Fprintln(w, "			return false")
		fmt.Fprintln(w, "		}")
		fmt.Fprintln(w, "	}")
		fmt.Fprintln(w, "")
		fmt.Fprintln(w, "	all."+MSG+" = append(all."+MSG+", newP)")
		fmt.Fprintln(w, "	return true")
		fmt.Fprintln(w, "}")
		fmt.Fprintln(w, "")
	}
}

func (pf *File) deleteFunc(w io.Writer) {
	var MSG string = pf.Bases.Name
	var LOCK string = pf.Bases.Lockname

	for _, KEY := range pf.Base.Unique {
		fmt.Fprintln(w, "func (all *"+MSG+") DeleteBy"+KEY+"(s string) bool {")
		fmt.Fprintln(w, "	"+LOCK+".RLock()")
		fmt.Fprintln(w, "	defer "+LOCK+".RUnlock()")
		fmt.Fprintln(w, "")
		fmt.Fprintln(w, "	for i, _ := range all."+MSG+" {")
		fmt.Fprintln(w, "		if all."+MSG+"[i]."+KEY+" == s {")
		fmt.Fprintln(w, "			all."+MSG+"[i] = all."+MSG+"[len(all."+MSG+")-1]")
		fmt.Fprintln(w, "			all."+MSG+" = all."+MSG+"[:len(all."+MSG+")-1]")
		fmt.Fprintln(w, "			return true")
		fmt.Fprintln(w, "		}")
		fmt.Fprintln(w, "	}")
		fmt.Fprintln(w, "	return false")
		fmt.Fprintln(w, "}")
		fmt.Fprintln(w, "")
	}
}

// this tries to return the deleted one but is wrong/gives warning if mutex lock is in struct
func (pf *File) deleteWithCopyFunc(w io.Writer) {
	var MSG string = pf.Bases.Name
	var BASE string = pf.Base.Name
	var LOCK string = pf.Bases.Lockname

	for _, KEY := range pf.Base.Unique {
		fmt.Fprintln(w, "func (all *"+MSG+") DeleteBy"+KEY+"(s string) *"+BASE+" {")
		fmt.Fprintln(w, "	"+LOCK+".RLock()")
		fmt.Fprintln(w, "	defer "+LOCK+".RUnlock()")
		fmt.Fprintln(w, "")
		fmt.Fprintln(w, "	var newr "+BASE)
		fmt.Fprintln(w, "")
		fmt.Fprintln(w, "	for i, _ := range all."+MSG+" {")
		fmt.Fprintln(w, "		if all."+MSG+"[i]."+KEY+" == s {")
		fmt.Fprintln(w, "			newr = *all."+MSG+"[i]")
		fmt.Fprintln(w, "			all."+MSG+"[i] = all."+MSG+"[len(all."+MSG+")-1]")
		fmt.Fprintln(w, "			all."+MSG+" = all."+MSG+"[:len(all."+MSG+")-1]")
		fmt.Fprintln(w, "			return &newr")
		fmt.Fprintln(w, "		}")
		fmt.Fprintln(w, "	}")
		fmt.Fprintln(w, "	return nil")
		fmt.Fprintln(w, "}")
		fmt.Fprintln(w, "")
	}
}

func (pf *File) findFunc(w io.Writer) {
	var MSG string = pf.Bases.Name
	var BASE string = pf.Base.Name
	var LOCK string = pf.Bases.Lockname

	for _, KEY := range pf.Base.Unique {
		fmt.Fprintln(w, "// find a dependancy by the go path")
		fmt.Fprintln(w, "func (all *"+MSG+") FindBy"+KEY+"(s string) *"+BASE+" {")
		fmt.Fprintln(w, "	if all == nil {")
		fmt.Fprintln(w, "		return nil")
		fmt.Fprintln(w, "	}")
		fmt.Fprintln(w, "")
		fmt.Fprintln(w, "	"+LOCK+".RLock()")
		fmt.Fprintln(w, "	defer "+LOCK+".RUnlock()")
		fmt.Fprintln(w, "")
		fmt.Fprintln(w, "	for i, _ := range all."+MSG+" {")
		fmt.Fprintln(w, "		if all."+MSG+"[i]."+KEY+" == s {")
		fmt.Fprintln(w, "			return all."+MSG+"[i]")
		fmt.Fprintln(w, "		}")
		fmt.Fprintln(w, "	}")
		fmt.Fprintln(w, "	return nil")
		fmt.Fprintln(w, "}")
		fmt.Fprintln(w, "")
	}
}