httpref.go

package httpref

import (
	"strings"
)

// Reference is a single reference item
type Reference struct {
	Name        string
	IsTitle     bool
	Summary     string
	Description string
}

// References is a collection of Reference objects
type References []Reference

// ByName finds all of the Reference with a matching Name field
func (r References) ByName(code string) References {
	results := References{}

	wildcard := strings.HasSuffix(code, "*")
	code = strings.ReplaceAll(code, "*", "")

	for _, v := range r {
		if !wildcard && v.Name == code {
			return References{v}
		}

		if strings.HasPrefix(v.Name, code) {
			results = append(results, v)
		}
	}

	return results
}

// InRange attempts to find a numeric in a range in the Name field
func (r References) InRange(code string) References {
	for _, v := range r {
		if strings.Contains(v.Name, "-") {
			parts := strings.Split(v.Name, "-")

			start, end := parts[0], parts[1]
			if code >= start && code <= end {
				return References{v}
			}
		}
	}

	return References{}
}

// Titles gives all of the Reference objects marked as a title
func (r References) Titles() References {
	results := References{}

	for _, v := range r {
		if v.IsTitle {
			results = append(results, v)
		}
	}

	return results
}

// Search looks for references that contain the search term in their fields
func (r References) Search(term string) References {
	results := References{}

	for _, v := range r {
		if strings.Contains(strings.ToLower(v.Name), term) ||
			strings.Contains(strings.ToLower(v.Summary), term) ||
			strings.Contains(strings.ToLower(v.Description), term) {
			results = append(results, v)
		}
	}

	return results
}