warewulf/internal/pkg/api/profile/list.go

package apiprofile

import (
	"encoding/json"
	"fmt"
	"sort"

	"github.com/warewulf/warewulf/internal/pkg/api/routes/wwapiv1"
	"github.com/warewulf/warewulf/internal/pkg/node"
	"github.com/warewulf/warewulf/internal/pkg/wwlog"
	"gopkg.in/yaml.v2"
)

/*
Returns the formatted list of profiles as string
*/
func ProfileList(ShowOpt *wwapiv1.GetProfileList) (profileList wwapiv1.ProfileList, err error) {
	profileList.Output = []string{}
	nodeDB, err := node.New()
	if err != nil {
		return
	}
	profiles, err := nodeDB.FindAllProfiles()
	if err != nil {
		return
	}
	profiles = node.FilterByName(profiles, ShowOpt.Profiles)
	sort.Slice(profiles, func(i, j int) bool {
		return profiles[i].Id() < profiles[j].Id()
	})
	if ShowOpt.ShowAll {
		for _, p := range profiles {
			profileList.Output = append(profileList.Output,
				fmt.Sprintf("%s:=:%s:=:%s", "PROFILE", "FIELD", "VALUE"))
			fields := p.GetFields(ShowOpt.ShowAll)
			for _, f := range fields {
				profileList.Output = append(profileList.Output,
					fmt.Sprintf("%s:=:%s:=:%s", p.Id.Print(), f.Field, f.Value))
			}
		}
	} else if ShowOpt.ShowYaml {
		profileMap := make(map[string]node.NodeInfo)
		for _, profile := range profiles {
			profileMap[profile.Id.Get()] = profile
		}

		buf, _ := yaml.Marshal(profileMap)
		profileList.Output = append(profileList.Output, string(buf))
	} else if ShowOpt.ShowJson {
		profileMap := make(map[string]node.NodeInfo)
		for _, profile := range profiles {
			profileMap[profile.Id.Get()] = profile
		}

		buf, _ := json.MarshalIndent(profileMap, "", "  ")
		profileList.Output = append(profileList.Output, string(buf))
	} else {
		profileList.Output = append(profileList.Output,
			fmt.Sprintf("%s:=:%s", "PROFILE NAME", "COMMENT/DESCRIPTION"))

		for _, profile := range profiles {
			profileList.Output = append(profileList.Output,
				fmt.Sprintf("%s:=:%s", profile.Id.Print(), profile.Comment.Print()))
		}
	}
	return
}