posh-providers/gravitational/teleport/config.go

package teleport

import (
	"fmt"
	"os"
	"strings"
)

type (
	Config struct {
		Path       string              `json:"path" yaml:"path"`
		Labels     map[string]string   `json:"labels" yaml:"labels"`
		Hostname   string              `json:"hostname" yaml:"hostname"`
		Kubernetes Kubernetes          `json:"kubernetes" yaml:"kubernetes"`
		Apps       map[string][]string `json:"apps" yaml:"apps"`
		Database   Database            `json:"database" yaml:"database"`
	}
	Kubernetes struct {
		Aliases map[string]string `json:"aliases" yaml:"aliases"`
	}
	Database struct {
		User string `json:"user" yaml:"user"`
	}
)

func (c Config) Query() string {
	if len(c.Labels) == 0 {
		return ""
	}
	var ret []string
	for k, v := range c.Labels {
		ret = append(ret, fmt.Sprintf("labels[\"%s\"] == \"%s\"", k, v))
	}
	return strings.Join(ret, " && ")
}

func (c Database) EnvUser() string {
	if value := os.Getenv("TELEPORT_DATABASE_USER"); value != "" {
		return value
	}
	return c.User
}

func (c Kubernetes) Alias(name string) string {
	if c.Aliases == nil {
		return name
	}
	if value, ok := c.Aliases[name]; ok {
		return value
	}
	return name
}

func (c Kubernetes) Name(alias string) string {
	for name, value := range c.Aliases {
		if value == alias {
			return name
		}
	}
	return alias
}