68 lines
1.9 KiB
Go
68 lines
1.9 KiB
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 || ShowOpt.ShowFullAll {
|
|
for _, p := range profiles {
|
|
profileList.Output = append(profileList.Output,
|
|
fmt.Sprintf("%s:=:%s:=:%s", "PROFILE", "FIELD", "VALUE"))
|
|
fields := p.GetFields(ShowOpt.ShowFullAll)
|
|
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
|
|
}
|