feat: add event management and provider loading to bus (#606)

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
This commit is contained in:
Itxaka
2025-07-22 10:20:55 +02:00
committed by GitHub
parent ffdc1ef892
commit fe4095c5b9
3 changed files with 121 additions and 3 deletions

View File

@@ -83,14 +83,16 @@ var AllEvents = []pluggable.EventType{
EventRecoveryStop,
EventAvailableReleases,
EventVersionImage,
InitProviderInstall,
InitProviderConfigure,
}
// IsEventDefined checks wether an event is defined in the bus.
// It accepts strings or EventType, returns a boolean indicating that
// the event was defined among the events emitted by the bus.
func IsEventDefined(i interface{}) bool {
func IsEventDefined(i interface{}, events ...pluggable.EventType) bool {
checkEvent := func(e pluggable.EventType) bool {
for _, ee := range AllEvents {
for _, ee := range append(AllEvents, events...) {
if ee == e {
return true
}

116
bus/manager.go Normal file
View File

@@ -0,0 +1,116 @@
package bus
import (
"os"
"github.com/kairos-io/kairos-sdk/types"
"github.com/mudler/go-pluggable"
)
const (
DefaultProviderPrefix = "agent-provider"
DefaultLogName = "kairos-bus"
)
var DefaultProviderPaths = []string{"/system/providers", "/usr/local/system/providers"}
func NewBus(withEvents ...pluggable.EventType) *Bus {
if len(withEvents) == 0 {
withEvents = AllEvents
}
return &Bus{
Manager: pluggable.NewManager(withEvents),
}
}
type Bus struct {
*pluggable.Manager
registered bool
logger *types.KairosLogger // Fully override the logger
logLevel string // Log level for the logger, defaults to "info" unless BUS_DEBUG is set to "true". This only valid if logger is not set.
logName string // Name of the logger, defaults to "bus". This only valid if logger is not set.
providerPrefix string // Prefix for provider plugins, defaults to "agent-provider". This is used to autoload providers.
providerPaths []string // Paths to search for provider plugins, defaults to system and current working directory.
}
func (b *Bus) LoadProviders() *pluggable.Manager {
return b.Autoload(b.providerPrefix, b.providerPaths...).Register()
}
func (b *Bus) Initialize(o ...Options) {
if b.registered {
return
}
for _, opt := range o {
opt(b)
}
// If no provider prefix is set, use the default "agent-provider"
if b.providerPrefix == "" {
b.providerPrefix = DefaultProviderPrefix
}
// If no provider paths are set, use the default system paths and current working directory
if b.providerPaths == nil {
wd, _ := os.Getwd()
b.providerPaths = append(DefaultProviderPaths, wd)
}
// If no logger is set, create a new one with the default log level and name
if b.logger == nil {
if b.logLevel == "" {
b.logLevel = "info"
}
if os.Getenv("BUS_DEBUG") == "true" {
b.logLevel = "debug"
}
if b.logName == "" {
b.logName = DefaultLogName
}
l := types.NewKairosLogger(b.logName, b.logLevel, false)
b.logger = &l
}
b.LoadProviders()
b.registered = true
b.logger.Logger.Debug().Interface("bus", b).Msg("Bus initialized with options")
}
type Options func(d *Bus)
// WithLogger allows to set a custom logger for the bus. If set, it will override the default logger.
func WithLogger(logger *types.KairosLogger) Options {
return func(d *Bus) {
d.logger = logger
}
}
// WithLoggerLevel allows to set the log level for the bus logger. If set, it will override the default log level.
func WithLoggerLevel(level string) Options {
return func(d *Bus) {
d.logLevel = level
}
}
// WithLoggerName allows to set the name of the logger for the bus. If set, it will override the default logger name.
func WithLoggerName(name string) Options {
return func(d *Bus) {
d.logName = name
}
}
// WithProviderPrefix allows to set the prefix for provider plugins. If set, it will override the default prefix.
func WithProviderPrefix(prefix string) Options {
return func(d *Bus) {
d.providerPrefix = prefix
}
}
// WithProviderPaths allows to set the paths to search for provider plugins. If set, it will override the default paths.
func WithProviderPaths(paths ...string) Options {
return func(d *Bus) {
d.providerPaths = paths
}
}

2
go.mod
View File

@@ -1,6 +1,6 @@
module github.com/kairos-io/kairos-sdk
go 1.24.2
go 1.24.5
require (
github.com/anatol/luks.go v0.0.0-20250316021219-8cd744c3576f