mirror of
https://github.com/gravitational/teleport.git
synced 2026-11-01 05:27:24 +08:00
Lisa Kim
fe1acfc6d3
* Fix yaml formatting with yaml lists * Snuck in adding db access checker result with webapi create db endpoint
106 lines
3.1 KiB
Go
106 lines
3.1 KiB
Go
/*
|
|
Copyright 2015-2022 Gravitational, Inc.
|
|
|
|
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 scripts
|
|
|
|
import (
|
|
_ "embed"
|
|
"fmt"
|
|
"net/http"
|
|
"sort"
|
|
"strings"
|
|
"text/template"
|
|
|
|
"github.com/gravitational/trace"
|
|
"gopkg.in/yaml.v3"
|
|
|
|
"github.com/gravitational/teleport/api/types"
|
|
"github.com/gravitational/teleport/api/utils"
|
|
)
|
|
|
|
// SetScriptHeaders sets response headers to plain text.
|
|
func SetScriptHeaders(h http.Header) {
|
|
h.Set("Content-Type", "text/plain")
|
|
}
|
|
|
|
// ErrorBashScript is used to display friendly error message when
|
|
// there is an error prepping the actual script.
|
|
var ErrorBashScript = []byte(`
|
|
#!/bin/sh
|
|
echo -e "An error has occurred. \nThe token may be expired or invalid. \nPlease check log for further details."
|
|
exit 1
|
|
`)
|
|
|
|
// InstallNodeBashScript is the script that will run on user's machine
|
|
// to install teleport and join a teleport cluster.
|
|
//
|
|
//go:embed node-join/install.sh
|
|
var installNodeBashScript string
|
|
|
|
var InstallNodeBashScript = template.Must(template.New("nodejoin").Parse(installNodeBashScript))
|
|
|
|
// MarshalLabelsYAML returns a list of strings, each one containing a
|
|
// label key and list of value's pair.
|
|
// This is used to create yaml sections within the join scripts.
|
|
//
|
|
// The arg `extraListIndent` allows adding `extra` indent space on
|
|
// top of the default space already used, for the default yaml listing
|
|
// format (the listing values with the dashes). If `extraListIndent`
|
|
// is zero, it's equivalent to using default space only (which is 4 spaces).
|
|
func MarshalLabelsYAML(resourceMatcherLabels types.Labels, extraListIndent int) ([]string, error) {
|
|
if len(resourceMatcherLabels) == 0 {
|
|
return []string{"{}"}, nil
|
|
}
|
|
|
|
ret := []string{}
|
|
|
|
// Consistently iterate over fields
|
|
labelKeys := make([]string, 0, len(resourceMatcherLabels))
|
|
for k := range resourceMatcherLabels {
|
|
labelKeys = append(labelKeys, k)
|
|
}
|
|
|
|
sort.Strings(labelKeys)
|
|
|
|
for _, labelName := range labelKeys {
|
|
labelValues := resourceMatcherLabels[labelName]
|
|
bs, err := yaml.Marshal(map[string]utils.Strings{labelName: labelValues})
|
|
if err != nil {
|
|
return nil, trace.Wrap(err)
|
|
}
|
|
|
|
labelStr := strings.TrimSpace(string(bs))
|
|
if len(labelValues) > 1 && extraListIndent > 0 {
|
|
labelStr = addExtraListIndentToYAMLLabelStr(labelStr, extraListIndent)
|
|
}
|
|
|
|
ret = append(ret, labelStr)
|
|
}
|
|
|
|
return ret, nil
|
|
}
|
|
|
|
func addExtraListIndentToYAMLLabelStr(labelStr string, indent int) string {
|
|
words := strings.Split(labelStr, "\n")
|
|
// Skip the first word, since that is the label key.
|
|
// Add extra spaces defined by `yamlListIndent` arg.
|
|
for i := 1; i < len(words); i++ {
|
|
words[i] = fmt.Sprintf("%s%s", strings.Repeat(" ", indent), words[i])
|
|
}
|
|
|
|
return strings.Join(words, "\n")
|
|
}
|