hosts_pool_list.go

// Copyright 2018 Bull S.A.S. Atos Technologies - Bull, Rue Jean Jaures, B.P.68, 78340, Les Clayes-sous-Bois, France.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package hostspool

import (
	"encoding/json"
	"fmt"
	"github.com/pkg/errors"
	"io/ioutil"
	"net/http"

	"github.com/fatih/color"
	"github.com/spf13/cobra"

	"github.com/ystia/yorc/v4/commands/httputil"
	"github.com/ystia/yorc/v4/helper/tabutil"
	"github.com/ystia/yorc/v4/rest"
)

func init() {
	var location string
	var filters []string
	hpListCmd := &cobra.Command{
		Use:   "list -l <locationName>",
		Short: "List hosts pool for a specified location",
		Long:  `Lists hosts of a hosts pool for a specified location managed by this Yorc cluster.`,
		RunE: func(cmd *cobra.Command, args []string) error {
			client, err := httputil.GetClient(clientConfig)
			if err != nil {
				httputil.ErrExit(err)
			}
			return listHostsPool(client, args, location, filters)
		},
	}
	hpListCmd.Flags().StringVarP(&location, "location", "l", "", "Need to provide the specified hosts pool location name")
	hpListCmd.Flags().StringSliceVarP(&filters, "filter", "f", nil, "Filter hosts based on their labels. May be specified several time, filters are joined by a logical 'and'. See the documentation for the filters grammar.")
	hostsPoolCmd.AddCommand(hpListCmd)
}

func listHostsPool(client httputil.HTTPClient, args []string, location string, filters []string) error {
	if location == "" {
		return errors.Errorf("Expecting a hosts pool location name")
	}
	colorize := !noColor
	request, err := client.NewRequest("GET", "/hosts_pool/"+location, nil)
	if err != nil {
		return err
	}
	q := request.URL.Query()
	for i := range filters {
		q.Add("filter", (filters[i]))
	}
	request.URL.RawQuery = q.Encode()
	request.Header.Add("Accept", "application/json")
	response, err := client.Do(request)
	if err != nil {
		return err
	}
	defer response.Body.Close()
	httputil.HandleHTTPStatusCode(response, "", "hosts pool", http.StatusOK)
	var hostsColl rest.HostsCollection
	body, err := ioutil.ReadAll(response.Body)
	if err != nil {
		return err
	}
	err = json.Unmarshal(body, &hostsColl)
	if err != nil {
		return err
	}

	hostsTable := tabutil.NewTable()
	hostsTable.AddHeaders("Name", "Connection", "Status", "Allocations", "Message", "Labels")
	for _, hostLink := range hostsColl.Hosts {
		if hostLink.Rel == rest.LinkRelHost {
			var host rest.Host
			err = httputil.GetJSONEntityFromAtomGetRequest(client, hostLink, &host)
			if err != nil {
				return err
			}
			// If no label was defined, define an empty map
			// to still consider there is a column to display
			if host.Labels == nil {
				host.Labels = map[string]string{}
			}
			addRow(hostsTable, colorize, hostList, &host, true)
		}
	}
	if colorize {
		defer color.Unset()
	}
	fmt.Printf("Host pools for location %q:\n", location)
	fmt.Println(hostsTable.Render())
	return nil
}