package node

import (
	"regexp"
	"strings"

	"github.com/hpcng/warewulf/internal/pkg/util"
)

/**********
 *
 * Filters
 *
 *********/

func FilterByName(set []NodeInfo, searchList []string) []NodeInfo {
	var ret []NodeInfo
	unique := make(map[string]NodeInfo)

	if len(searchList) > 0 {
		for _, search := range searchList {
			for _, entry := range set {
				b, _ := regexp.MatchString("^"+search+"$", entry.Id.Get())
				if b {
					unique[entry.Id.Get()] = entry
				}
			}
		}
		for _, n := range unique {
			ret = append(ret, n)
		}
	} else {
		ret = set
	}

	return ret
}

/**********
 *
 * Sets
 *
 *********/

func (ent *Entry) Set(val string) {
	if val == "" {
		return
	}

	if val == "UNDEF" || val == "DELETE" || val == "UNSET" || val == "--" {
		ent.value = ""
	} else {
		ent.value = val
	}
}

func (ent *Entry) SetB(val bool) {
	if val {
		ent.value = "true"
	}
}

func (ent *Entry) SetAlt(val string, from string) {
	if val == "" {
		return
	}

	ent.altvalue = val
	ent.from = from

}

func (ent *Entry) SetAltB(val bool, from string) {
	if val {
		ent.altvalue = "true"
		ent.from = from
	}
}

func (ent *Entry) SetDefault(val string) {
	if val == "" {
		return
	}

	ent.def = val

}

/**********
*
* Gets
*
*********/

func (ent *Entry) Get() string {
	if ent.value != "" {
		return ent.value
	}
	if ent.altvalue != "" {
		return ent.altvalue
	}
	if ent.def != "" {
		return ent.def
	}
	return ""
}

func (ent *Entry) GetB() bool {
	if ent.value == "false" || ent.value == "no" || ent.value == "" {
		if ent.altvalue == "false" || ent.altvalue == "no" || ent.altvalue == "" {
			return false
		}
		return false
	}
	return true
}

func (ent *Entry) GetReal() string {
	return ent.value
}

/*
Returns a string slice, which is the combination of value, altvalue and def.
The elemtent in the slice are uniq.
*/
func (ent *Entry) GetSlice() []string {
	var retval []string
	if ent.value != "" {
		retval = util.SliceAppendUniq(retval, strings.Split(ent.value, ","))
	}
	if ent.altvalue != "" {
		retval = util.SliceAppendUniq(retval, strings.Split(ent.altvalue, ","))
	}
	if ent.def != "" && len(retval) == 0 {
		retval = util.SliceAppendUniq(retval, strings.Split(ent.def, ","))

	}
	return retval
}

/**********
 *
 * Misc
 *
 *********/

func (ent *Entry) Print() string {
	if ent.value != "" {
		return ent.value
	}
	if ent.altvalue != "" {
		return ent.altvalue
	}
	if ent.def != "" {
		return "(" + ent.def + ")"
	}
	return "--"
}

func (ent *Entry) PrintComb() string {
	if ent.value != "" && ent.def != "" {
		return "[" + ent.value + "," + ent.def + "]"
	}
	return ent.PrintComb()
}

func (ent *Entry) PrintB() bool {
	return ent.GetB()
}

func (ent *Entry) Source() string {
	if ent.value != "" && ent.altvalue != "" {
		return "SUPERSEDED"
		//return fmt.Sprintf("[%s]", ent.from)
	} else if ent.from == "" {
		return "--"
	}
	return ent.from
}

func (ent *Entry) Defined() bool {
	if ent.value != "" {
		return true
	}
	if ent.altvalue != "" {
		return true
	}
	if ent.def != "" {
		return true
	}
	return false
}