graphql-engine/cli/util/server.go

package util

import (
	"crypto/tls"
	"fmt"
	"net/http"
	"net/url"
	"path"

	"github.com/parnurzeal/gorequest"
	"github.com/sirupsen/logrus"
)

// ServerState is the state of Hasura stored on the server.
type ServerState struct {
	UUID     string
	CLIState map[string]interface{}
}

type hdbVersion struct {
	UUID     string                 `json:"hasura_uuid"`
	CLIState map[string]interface{} `json:"cli_state"`
}

// GetServerState queries a server for the state.
func GetServerState(endpoint string, adminSecret string, config *tls.Config, hasMetadataV3 bool, log *logrus.Logger) *ServerState {
	state := &ServerState{
		UUID: "00000000-0000-0000-0000-000000000000",
	}

	if hasMetadataV3 {
		payload := `
	{
    "type": "get_catalog_state",
    "args": {}
	}
`
		req := gorequest.New()
		if config != nil {
			req.TLSClientConfig(config)
		}
		req.Post(endpoint).Send(payload)
		req.Set("X-Hasura-Admin-Secret", adminSecret)

		var r struct {
			ID string `json:"id"`
		}
		_, _, errs := req.EndStruct(&r)
		if len(errs) != 0 {
			log.Debugf("server state: errors: %v", errs)
			return state
		}

		state.UUID = r.ID
	} else {
		state := &ServerState{
			UUID: "00000000-0000-0000-0000-000000000000",
		}
		payload := `{
		"type": "select",
		"args": {
			"table": {
				"schema": "hdb_catalog",
				"name": "hdb_version"
			},
			"columns": [
				"hasura_uuid",
				"cli_state"
			]
		}
	}`

		req := gorequest.New()
		if config != nil {
			req.TLSClientConfig(config)
		}
		req.Post(endpoint).Send(payload)
		req.Set("X-Hasura-Admin-Secret", adminSecret)

		var r []hdbVersion
		_, _, errs := req.EndStruct(&r)
		if len(errs) != 0 {
			log.Debugf("server state: errors: %v", errs)
			return state
		}

		if len(r) != 1 {
			log.Debugf("invalid response: %v", r)
			return state
		}

		state.UUID = r[0].UUID
		state.CLIState = r[0].CLIState
	}
	return state

}

func GetServerStatus(endpoint string) (err error) {
	uri, err := url.Parse(endpoint)
	if err != nil {
		return fmt.Errorf("error while parsing the endpoint :%w", err)
	}
	uri.Path = path.Join(uri.Path, "healthz")
	resp, err := http.Get(uri.String())
	if err != nil {
		return fmt.Errorf("making http request failed: %s", err.Error())
	}
	if resp.StatusCode != http.StatusOK {
		return fmt.Errorf("request failed: url: %s status code: %v status: %s", uri.String(), resp.StatusCode, resp.Status)
	}
	return nil
}