mirror of
https://github.com/kairos-io/kairos-sdk.git
synced 2025-04-27 19:15:23 +00:00
541728fa22
* feat: add support for extra plugins in cluster provider Signed-off-by: Nianyu Shen <xiaoyu9964@gmail.com> * add cluster reset event Signed-off-by: Nianyu Shen <xiaoyu9964@gmail.com> --------- Signed-off-by: Nianyu Shen <xiaoyu9964@gmail.com>
97 lines
2.7 KiB
Go
97 lines
2.7 KiB
Go
package clusterplugin
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/kairos-io/kairos-sdk/bus"
|
|
"github.com/mudler/go-pluggable"
|
|
yip "github.com/mudler/yip/pkg/schema"
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
const clusterProviderCloudConfigFile = "/usr/local/cloud-config/cluster.kairos.yaml"
|
|
|
|
const EventClusterReset pluggable.EventType = "cluster.reset"
|
|
|
|
// ClusterProvider returns a yip configuration that configures a Kubernetes engine. The yip config may use any elemental
|
|
// stages after initramfs.
|
|
type ClusterProvider func(cluster Cluster) yip.YipConfig
|
|
|
|
// ClusterPlugin creates a cluster plugin from a `ClusterProvider`. It calls the cluster provider at the appropriate events
|
|
// and ensures it configuration is written where it will be executed.
|
|
type ClusterPlugin struct {
|
|
Provider ClusterProvider
|
|
}
|
|
|
|
func (p ClusterPlugin) onBoot(event *pluggable.Event) pluggable.EventResponse {
|
|
var payload bus.EventPayload
|
|
var config Config
|
|
var response pluggable.EventResponse
|
|
|
|
// parse the boot payload
|
|
if err := json.Unmarshal([]byte(event.Data), &payload); err != nil {
|
|
response.Error = fmt.Sprintf("failed to parse boot event: %s", err.Error())
|
|
return response
|
|
}
|
|
|
|
// parse config from boot payload
|
|
if err := yaml.Unmarshal([]byte(payload.Config), &config); err != nil {
|
|
response.Error = fmt.Sprintf("failed to parse config from boot event: %s", err.Error())
|
|
return response
|
|
}
|
|
|
|
if config.Cluster == nil {
|
|
return response
|
|
}
|
|
|
|
// request the cloud configuration of the provider
|
|
cc := p.Provider(*config.Cluster)
|
|
|
|
configFilePath := clusterProviderCloudConfigFile
|
|
|
|
if len(config.Cluster.ClusterConfigPath) != 0 {
|
|
configFilePath = config.Cluster.ClusterConfigPath
|
|
}
|
|
|
|
// open our cloud configuration file for writing
|
|
f, err := filesystem.OpenFile(configFilePath, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0600)
|
|
if err != nil {
|
|
response.Error = fmt.Sprintf("failed to parse boot event: %s", err.Error())
|
|
return response
|
|
}
|
|
|
|
defer f.Close()
|
|
|
|
// write the cloud configuration header
|
|
_, err = f.WriteString("#cloud-config\n")
|
|
if err != nil {
|
|
response.Error = fmt.Sprintf("failed to parse boot event: %s", err.Error())
|
|
return response
|
|
}
|
|
|
|
// encode the provider's configuration
|
|
err = yaml.NewEncoder(f).Encode(cc)
|
|
if err != nil {
|
|
response.Error = fmt.Sprintf("failed to parse boot event: %s", err.Error())
|
|
return response
|
|
}
|
|
|
|
return response
|
|
}
|
|
|
|
func (p ClusterPlugin) Run(extraPlugins ...pluggable.FactoryPlugin) error {
|
|
plugins := []pluggable.FactoryPlugin{
|
|
{
|
|
EventType: bus.EventBoot,
|
|
PluginHandler: p.onBoot,
|
|
},
|
|
}
|
|
plugins = append(plugins, extraPlugins...)
|
|
|
|
f := pluggable.NewPluginFactory(plugins...)
|
|
|
|
return f.Run(pluggable.EventType(os.Args[1]), os.Stdin, os.Stdout)
|
|
}
|