Initial commit

[vaultmon/.git] / vault.go

package main

import (
	"encoding/json"
	"fmt"
	"io/ioutil"
	"net/http"

	log "github.com/sirupsen/logrus"
)

type Vault struct {
	URL              string
	ConnectionStatus string
	ClusterStatus    string
}

func (v *Vault) CheckHealth() (bool, error) {
	healthURL := v.URL + "/v1/sys/health"
	cnx := NewConnection(healthURL)
	log.Debugf("Vault Connection on %s", healthURL)
	resp, err := cnx.Head()
	if err != nil {
		return false, err
	}
	defer resp.Body.Close()

	log.Debugf("Vault CheckHealth response: %s", resp.Status)

	return resp.StatusCode == http.StatusOK, nil
}

func (v *Vault) CheckCluster() (bool, error) {
	ClusterURL := v.URL + "/v1/sys/leader"
	cnx := NewConnection(ClusterURL)
	log.Debugf("Vault Connection on %s", ClusterURL)
	resp, err := cnx.Get()
	if err != nil {
		return false, err
	}
	defer resp.Body.Close()

	log.Debugf("Vault CheckCluster response: %s", resp.Status)

	buf, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		return false, err
	}

	var result struct {
		LeaderAddress        string `json:"leader_address"`
		IsSelf               bool   `json:"is_self"`
		HAEnabled            bool   `json:"ha_enabled"`
		LeaderClusterAddress string `json:"leader_cluster_address"`
	}
	err = json.Unmarshal(buf, &result)
	if err != nil {
		return false, err
	}

	log.Debugf("Vault CheckCluster JSON result: %+v", result)

	return result.HAEnabled == true, nil
}

func (v *Vault) Test() (bool, string) {
	// test health
	result, err := v.CheckHealth()
	if err != nil {
		log.Errorf("Vault CheckHealth error: %s", err)
		v.ConnectionStatus = "KO"
	}
	v.ConnectionStatus = "OK"
	log.Debugf("Vault CheckHealth result: %t", result)

	// test cluster
	result, err = v.CheckCluster()
	if err != nil {
		log.Errorf("Vault CheckCluster error: %s", err)
		v.ClusterStatus = "KO"
	}
	v.ClusterStatus = "OK"
	log.Debugf("Vault CheckCluster result: %t", result)

	return result, fmt.Sprintf("> %s `%s` status *%s*. Cluster status *%s*.", v.Name(), v.URL, v.ConnectionStatus, v.ClusterStatus)
}

func (v *Vault) Name() string {
	return "Vault"
}

func NewVault(url string) CheckProvider {
	return &Vault{URL: url}
}