github.com/openshift/installer@v1.4.17/pkg/agent/rest.go

package agent

import (
	"context"
	"net"
	"net/url"

	"github.com/go-openapi/strfmt"
	"github.com/pkg/errors"
	"github.com/sirupsen/logrus"

	"github.com/openshift/assisted-service/api/v1beta1"
	"github.com/openshift/assisted-service/client"
	"github.com/openshift/assisted-service/client/events"
	"github.com/openshift/assisted-service/client/installer"
	"github.com/openshift/assisted-service/models"
	"github.com/openshift/installer/pkg/asset/agent/agentconfig"
	"github.com/openshift/installer/pkg/asset/agent/gencrypto"
	"github.com/openshift/installer/pkg/asset/agent/image"
	"github.com/openshift/installer/pkg/asset/agent/manifests"
	"github.com/openshift/installer/pkg/asset/installconfig"
	assetstore "github.com/openshift/installer/pkg/asset/store"
	"github.com/openshift/installer/pkg/types/agent"
)

// NodeZeroRestClient is a struct to interact with the Agent Rest API that is on node zero.
type NodeZeroRestClient struct {
	Client     *client.AssistedInstall
	ctx        context.Context
	config     client.Config
	NodeZeroIP string
	NodeSSHKey []string
}

// NewNodeZeroRestClient Initialize a new rest client to interact with the Agent Rest API on node zero.
func NewNodeZeroRestClient(ctx context.Context, rendezvousIP, sshKey, token string) *NodeZeroRestClient {
	restClient := &NodeZeroRestClient{}

	// Get SSH Keys which can be used to determine if Rest API failures are due to network connectivity issues
	if sshKey != "" {
		restClient.NodeSSHKey = append(restClient.NodeSSHKey, sshKey)
	}

	config := client.Config{}
	config.URL = &url.URL{
		Scheme: "http",
		Host:   net.JoinHostPort(rendezvousIP, "8090"),
		Path:   client.DefaultBasePath,
	}

	config.AuthInfo = gencrypto.UserAuthHeaderWriter(token)

	client := client.New(config)

	restClient.Client = client
	restClient.ctx = ctx
	restClient.config = config
	restClient.NodeZeroIP = rendezvousIP

	return restClient
}

// FindRendezvouIPAndSSHKeyFromAssetStore returns the rendezvousIP and public ssh key.
func FindRendezvouIPAndSSHKeyFromAssetStore(assetDir string) (string, string, error) {
	agentConfigAsset := &agentconfig.AgentConfig{}
	agentManifestsAsset := &manifests.AgentManifests{}
	installConfigAsset := &installconfig.InstallConfig{}
	agentHostsAsset := &agentconfig.AgentHosts{}

	assetStore, err := assetstore.NewStore(assetDir)
	if err != nil {
		return "", "", errors.Wrap(err, "failed to create asset store")
	}

	agentConfig, agentConfigError := assetStore.Load(agentConfigAsset)
	agentManifests, manifestError := assetStore.Load(agentManifestsAsset)
	installConfig, installConfigError := assetStore.Load(installConfigAsset)
	agentHosts, agentHostsError := assetStore.Load(agentHostsAsset)

	if agentConfigError != nil {
		logrus.Debug(errors.Wrapf(agentConfigError, "failed to load %s", agentConfigAsset.Name()))
	}
	if manifestError != nil {
		logrus.Debug(errors.Wrapf(manifestError, "failed to load %s", agentManifestsAsset.Name()))
	}
	if installConfigError != nil {
		logrus.Debug(errors.Wrapf(installConfigError, "failed to load %s", installConfigAsset.Name()))
	}
	if agentHostsError != nil {
		logrus.Debug(errors.Wrapf(agentConfigError, "failed to load %s", agentHostsAsset.Name()))
	}
	if agentConfigError != nil || manifestError != nil || installConfigError != nil || agentHostsError != nil {
		return "", "", errors.New("failed to load AgentConfig, NMStateConfig, InstallConfig, or AgentHosts")
	}

	var rendezvousIP string
	var rendezvousIPError error
	var emptyNMStateConfigs []*v1beta1.NMStateConfig

	if agentConfig != nil && agentManifests != nil {
		rendezvousIP, rendezvousIPError = image.RetrieveRendezvousIP(agentConfig.(*agentconfig.AgentConfig).Config, agentHosts.(*agentconfig.AgentHosts).Hosts, agentManifests.(*manifests.AgentManifests).NMStateConfigs)
	} else if agentConfig == nil && agentManifests != nil {
		rendezvousIP, rendezvousIPError = image.RetrieveRendezvousIP(&agent.Config{}, agentHosts.(*agentconfig.AgentHosts).Hosts, agentManifests.(*manifests.AgentManifests).NMStateConfigs)
	} else if agentConfig != nil && agentManifests == nil {
		rendezvousIP, rendezvousIPError = image.RetrieveRendezvousIP(agentConfig.(*agentconfig.AgentConfig).Config, agentHosts.(*agentconfig.AgentHosts).Hosts, emptyNMStateConfigs)
	} else {
		return "", "", errors.New("both AgentConfig and NMStateConfig are empty")
	}
	if rendezvousIPError != nil {
		return "", "", rendezvousIPError
	}

	var sshKey string
	// Get SSH Keys which can be used to determine if Rest API failures are due to network connectivity issues
	if installConfig != nil {
		sshKey = installConfig.(*installconfig.InstallConfig).Config.SSHKey
	}

	return rendezvousIP, sshKey, nil
}

// FindAuthTokenFromAssetStore returns the auth token from asset store.
func FindAuthTokenFromAssetStore(assetDir string) (string, error) {
	authConfigAsset := &gencrypto.AuthConfig{}

	assetStore, err := assetstore.NewStore(assetDir)
	if err != nil {
		return "", errors.Wrap(err, "failed to create asset store")
	}

	authConfig, authConfigError := assetStore.Load(authConfigAsset)

	if authConfigError != nil {
		logrus.Debug(errors.Wrapf(authConfigError, "failed to load %s", authConfigAsset.Name()))
		return "", errors.New("failed to load AuthConfig")
	}

	var authToken string
	if authConfig != nil {
		authToken = authConfig.(*gencrypto.AuthConfig).AgentAuthToken
	}

	return authToken, nil
}

// IsRestAPILive Determine if the Agent Rest API on node zero has initialized
func (rest *NodeZeroRestClient) IsRestAPILive() bool {
	// GET /v2/infraenvs
	listInfraEnvsParams := installer.NewListInfraEnvsParams()
	_, err := rest.Client.Installer.ListInfraEnvs(rest.ctx, listInfraEnvsParams)
	return err == nil
}

// GetRestAPIServiceBaseURL Return the url of the Agent Rest API on node zero
func (rest *NodeZeroRestClient) GetRestAPIServiceBaseURL() *url.URL {
	return rest.config.URL
}

// GetInfraEnvEvents Return the event list for the provided infraEnvID from the Agent Rest API
func (rest *NodeZeroRestClient) GetInfraEnvEvents(infraEnvID *strfmt.UUID) (models.EventList, error) {
	listEventsParams := &events.V2ListEventsParams{InfraEnvID: infraEnvID}
	clusterEventsResult, err := rest.Client.Events.V2ListEvents(rest.ctx, listEventsParams)
	if err != nil {
		return nil, err
	}
	return clusterEventsResult.Payload, nil
}

// getClusterID Return the cluster ID assigned by the Agent Rest API
func (rest *NodeZeroRestClient) getClusterID() (*strfmt.UUID, error) {
	// GET /v2/clusters and return first result
	listClusterParams := installer.NewV2ListClustersParams()
	clusterResult, err := rest.Client.Installer.V2ListClusters(rest.ctx, listClusterParams)
	if err != nil {
		return nil, err
	}
	clusterList := clusterResult.Payload
	if len(clusterList) == 1 {
		clusterID := clusterList[0].ID
		return clusterID, nil
	} else if len(clusterList) == 0 {
		logrus.Debug("cluster is not registered in rest API")
		return nil, nil
	} else {
		logrus.Infof("found too many clusters. number of clusters found: %d", len(clusterList))
		return nil, nil
	}
}

// getClusterID Return the infraEnv ID associated with the cluster in the Agent Rest API
func (rest *NodeZeroRestClient) getClusterInfraEnvID() (*strfmt.UUID, error) {
	// GET /v2/infraenvs and return first result
	listInfraEnvParams := installer.NewListInfraEnvsParams()
	infraEnvResult, err := rest.Client.Installer.ListInfraEnvs(rest.ctx, listInfraEnvParams)
	if err != nil {
		return nil, err
	}
	infraEnvList := infraEnvResult.Payload
	if len(infraEnvList) == 1 {
		clusterInfraEnvID := infraEnvList[0].ID
		return clusterInfraEnvID, nil
	} else if len(infraEnvList) == 0 {
		logrus.Debug("infraenv is not registered in rest API")
		return nil, nil
	} else {
		logrus.Infof("found too many infraenvs. number of infraenvs found: %d", len(infraEnvList))
		return nil, nil
	}
}