mirror of
https://github.com/danielgormly/drone-plugin-kube.git
synced 2026-06-04 18:23:48 +08:00
danielgormly
105 lines
2.7 KiB
Go
105 lines
2.7 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
"regexp"
|
|
"strings"
|
|
|
|
"github.com/aymerick/raymond"
|
|
appv1 "k8s.io/api/apps/v1"
|
|
corev1 "k8s.io/api/core/v1"
|
|
"k8s.io/client-go/kubernetes/scheme"
|
|
)
|
|
|
|
type (
|
|
// KubeConfig -- Contains connection settings for Kube client
|
|
KubeConfig struct {
|
|
Ca string
|
|
Server string
|
|
Token string
|
|
Namespace string
|
|
InsecureSkipTLSVerify bool
|
|
}
|
|
// Plugin -- Contains config for plugin
|
|
Plugin struct {
|
|
Template string
|
|
KubeConfig KubeConfig
|
|
ConfigMapFile string // Optional
|
|
}
|
|
)
|
|
|
|
// Exec -- Runs plugin
|
|
func (p Plugin) Exec() error {
|
|
if p.KubeConfig.Server == "" {
|
|
return errors.New("PLUGIN_SERVER is not defined")
|
|
}
|
|
if p.KubeConfig.Token == "" {
|
|
return errors.New("PLUGIN_TOKEN is not defined")
|
|
}
|
|
if p.KubeConfig.Ca == "" {
|
|
return errors.New("PLUGIN_CA is not defined")
|
|
}
|
|
if p.KubeConfig.Namespace == "" {
|
|
p.KubeConfig.Namespace = "default"
|
|
}
|
|
if p.Template == "" {
|
|
return errors.New("PLUGIN_TEMPLATE, or template must be defined")
|
|
}
|
|
// Make map of environment variables set by Drone
|
|
ctx := make(map[string]string)
|
|
pluginEnv := os.Environ()
|
|
for _, value := range pluginEnv {
|
|
re := regexp.MustCompile(`^PLUGIN_(.*)=(.*)`)
|
|
if re.MatchString(value) {
|
|
matches := re.FindStringSubmatch(value)
|
|
key := strings.ToLower(matches[1])
|
|
ctx[key] = matches[2]
|
|
}
|
|
}
|
|
// Grab template from filesystem
|
|
raw, err := ioutil.ReadFile(p.Template)
|
|
if err != nil {
|
|
log.Print("⛔️ Error reading template file:")
|
|
return err
|
|
}
|
|
// Parse template
|
|
templateYaml, err := raymond.Render(string(raw), ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
// Connect to Kubernetes
|
|
clientset, err := p.CreateKubeClient()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
// Decode
|
|
kubernetesObject, _, err := scheme.Codecs.UniversalDeserializer().Decode([]byte(templateYaml), nil, nil)
|
|
if err != nil {
|
|
log.Print("⛔️ Error decoding template into valid Kubernetes object:")
|
|
return err
|
|
}
|
|
switch o := kubernetesObject.(type) {
|
|
case *appv1.Deployment:
|
|
log.Print("📦 Resource type: Deployment")
|
|
err = CreateOrUpdateDeployment(clientset, p.KubeConfig.Namespace, o)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
// Watch for successful update
|
|
log.Print("📦 Watching deployment until no unavailable replicas.")
|
|
state, watchErr := waitUntilDeploymentSettled(clientset, p.KubeConfig.Namespace, o.ObjectMeta.Name, 120)
|
|
log.Printf("%s", state)
|
|
return watchErr
|
|
case *corev1.ConfigMap:
|
|
log.Print("📦 Resource type: ConfigMap")
|
|
err = ApplyConfigMapFromFile(clientset, p.KubeConfig.Namespace, o, p.ConfigMapFile)
|
|
default:
|
|
err = errors.New("⛔️ This plugin doesn't support that resource type")
|
|
return err
|
|
}
|
|
return err
|
|
}
|