network

var DisplayFields = map[string]func(port *host.Port) string{

	"ID": func(port *host.Port) string {
		id := display.FormatSmallID(port.Id)

		if port.State == nil {
			return id
		}

		switch port.State.State {
		case hostdomain.PortOpen:
			return color.HiGreenString(id)
		case hostdomain.PortFiltered:
			return color.HiYellowString(id)
		case hostdomain.PortClosed:
			return color.HiRedString(id)
		}

		return id
	},
	"Num": func(port *host.Port) string {
		return strconv.Itoa(int(port.Number))
	},
	"Proto": func(port *host.Port) string {
		proto := color.HiBlackString(port.Protocol)

		if port.Service == nil {
			return proto
		} else {
			proto = ""
		}

		if port.Service.Protocol != "" {
			proto += port.Service.Protocol
		} else if port.Service.Name != "" {
			proto += port.Service.Name
		}

		return proto
	},
	"Product": func(port *host.Port) string {
		if port.Service == nil {
			return ""
		}
		product := port.Service.Product
		if port.Service.Version == "" {
			return product
		}
		product += color.HiBlackString(fmt.Sprintf(" (v%s)", port.Service.Version))
		return product
	},
	"Version": func(port *host.Port) string {
		serv := port.Service
		version := port.Service.Version
		if serv.HighVersion != "" && serv.LowVersion != "" {
			version += color.HiBlackString(" (%s / %s)", serv.LowVersion, serv.HighVersion)
		}
		return version
	},
	"Info": func(port *host.Port) string {
		if len(port.Scripts) == 0 {
			return port.Service.ExtraInfo
		}

		var scripts string
		for _, script := range port.Scripts {
			name := script.Name
			if script.Output == "" && script.Name == "" {
				continue
			}
			scripts += fmt.Sprintf("(%s) %s\n", name, script.Output)
		}

		return strings.TrimSuffix(scripts, "\n")
	},
	"Extra Info": func(port *host.Port) string {
		if port.Service == nil {
			return ""
		}
		return port.Service.ExtraInfo
	},
	"Fingerprint": func(port *host.Port) string {
		if port.Service == nil {
			return ""
		}
		return port.Service.ServiceFP
	},

	"Method": func(port *host.Port) string {
		if port.Service == nil {
			return ""
		}
		return color.HiBlackString(port.Service.Method)
	},
	"State": func(port *host.Port) string {
		if port.State == nil {
			return "unknown"
		}

		switch port.State.State {
		case hostdomain.PortOpen:
			return color.HiGreenString(hostdomain.PortOpen)
		case hostdomain.PortFiltered:
			return color.HiYellowString(hostdomain.PortFiltered)
		case hostdomain.PortClosed:
			return color.HiRedString(hostdomain.PortClosed)
		}

		return port.State.State
	},
	"Reason": func(port *host.Port) string {
		if port.State == nil {
			return "undefined"
		}
		return port.State.Reason
	},
	"Scripts": func(port *host.Port) string {
		scripts := ""
		for _, script := range port.Scripts {
			scripts += printScript(script, 0)
		}

		return scripts
	},
}