1 files changed, 184 insertions, 0 deletions
diff --git a/weed/worker/tasks/ui_base.go b/weed/worker/tasks/ui_base.go
new file mode 100644
index 000000000..ac22c20c4
--- /dev/null
+++ b/weed/worker/tasks/ui_base.go
@@ -0,0 +1,184 @@
+package tasks
+
+import (
+	"reflect"
+
+	"github.com/seaweedfs/seaweedfs/weed/glog"
+	"github.com/seaweedfs/seaweedfs/weed/pb/worker_pb"
+	"github.com/seaweedfs/seaweedfs/weed/worker/types"
+)
+
+// BaseUIProvider provides common UIProvider functionality for all tasks
+type BaseUIProvider struct {
+	taskType            types.TaskType
+	displayName         string
+	description         string
+	icon                string
+	schemaFunc          func() *TaskConfigSchema
+	configFunc          func() types.TaskConfig
+	applyTaskPolicyFunc func(policy *worker_pb.TaskPolicy) error
+	applyTaskConfigFunc func(config types.TaskConfig) error
+}
+
+// NewBaseUIProvider creates a new base UI provider
+func NewBaseUIProvider(
+	taskType types.TaskType,
+	displayName string,
+	description string,
+	icon string,
+	schemaFunc func() *TaskConfigSchema,
+	configFunc func() types.TaskConfig,
+	applyTaskPolicyFunc func(policy *worker_pb.TaskPolicy) error,
+	applyTaskConfigFunc func(config types.TaskConfig) error,
+) *BaseUIProvider {
+	return &BaseUIProvider{
+		taskType:            taskType,
+		displayName:         displayName,
+		description:         description,
+		icon:                icon,
+		schemaFunc:          schemaFunc,
+		configFunc:          configFunc,
+		applyTaskPolicyFunc: applyTaskPolicyFunc,
+		applyTaskConfigFunc: applyTaskConfigFunc,
+	}
+}
+
+// GetTaskType returns the task type
+func (ui *BaseUIProvider) GetTaskType() types.TaskType {
+	return ui.taskType
+}
+
+// GetDisplayName returns the human-readable name
+func (ui *BaseUIProvider) GetDisplayName() string {
+	return ui.displayName
+}
+
+// GetDescription returns a description of what this task does
+func (ui *BaseUIProvider) GetDescription() string {
+	return ui.description
+}
+
+// GetIcon returns the icon CSS class for this task type
+func (ui *BaseUIProvider) GetIcon() string {
+	return ui.icon
+}
+
+// GetCurrentConfig returns the current configuration as TaskConfig
+func (ui *BaseUIProvider) GetCurrentConfig() types.TaskConfig {
+	return ui.configFunc()
+}
+
+// ApplyTaskPolicy applies protobuf TaskPolicy configuration
+func (ui *BaseUIProvider) ApplyTaskPolicy(policy *worker_pb.TaskPolicy) error {
+	return ui.applyTaskPolicyFunc(policy)
+}
+
+// ApplyTaskConfig applies TaskConfig interface configuration
+func (ui *BaseUIProvider) ApplyTaskConfig(config types.TaskConfig) error {
+	return ui.applyTaskConfigFunc(config)
+}
+
+// CommonConfigGetter provides a common pattern for getting current configuration
+type CommonConfigGetter[T any] struct {
+	defaultConfig T
+	detectorFunc  func() T
+	schedulerFunc func() T
+}
+
+// NewCommonConfigGetter creates a new common config getter
+func NewCommonConfigGetter[T any](
+	defaultConfig T,
+	detectorFunc func() T,
+	schedulerFunc func() T,
+) *CommonConfigGetter[T] {
+	return &CommonConfigGetter[T]{
+		defaultConfig: defaultConfig,
+		detectorFunc:  detectorFunc,
+		schedulerFunc: schedulerFunc,
+	}
+}
+
+// GetConfig returns the merged configuration
+func (cg *CommonConfigGetter[T]) GetConfig() T {
+	config := cg.defaultConfig
+
+	// Apply detector values if available
+	if cg.detectorFunc != nil {
+		detectorConfig := cg.detectorFunc()
+		mergeConfigs(&config, detectorConfig)
+	}
+
+	// Apply scheduler values if available
+	if cg.schedulerFunc != nil {
+		schedulerConfig := cg.schedulerFunc()
+		mergeConfigs(&config, schedulerConfig)
+	}
+
+	return config
+}
+
+// mergeConfigs merges non-zero values from source into dest
+func mergeConfigs[T any](dest *T, source T) {
+	destValue := reflect.ValueOf(dest).Elem()
+	sourceValue := reflect.ValueOf(source)
+
+	if destValue.Kind() != reflect.Struct || sourceValue.Kind() != reflect.Struct {
+		return
+	}
+
+	for i := 0; i < destValue.NumField(); i++ {
+		destField := destValue.Field(i)
+		sourceField := sourceValue.Field(i)
+
+		if !destField.CanSet() {
+			continue
+		}
+
+		// Only copy non-zero values
+		if !sourceField.IsZero() {
+			if destField.Type() == sourceField.Type() {
+				destField.Set(sourceField)
+			}
+		}
+	}
+}
+
+// RegisterUIFunc provides a common registration function signature
+type RegisterUIFunc[D, S any] func(uiRegistry *types.UIRegistry, detector D, scheduler S)
+
+// CommonRegisterUI provides a common registration implementation
+func CommonRegisterUI[D, S any](
+	taskType types.TaskType,
+	displayName string,
+	uiRegistry *types.UIRegistry,
+	detector D,
+	scheduler S,
+	schemaFunc func() *TaskConfigSchema,
+	configFunc func() types.TaskConfig,
+	applyTaskPolicyFunc func(policy *worker_pb.TaskPolicy) error,
+	applyTaskConfigFunc func(config types.TaskConfig) error,
+) {
+	// Get metadata from schema
+	schema := schemaFunc()
+	description := "Task configuration"
+	icon := "fas fa-cog"
+
+	if schema != nil {
+		description = schema.Description
+		icon = schema.Icon
+	}
+
+	uiProvider := NewBaseUIProvider(
+		taskType,
+		displayName,
+		description,
+		icon,
+		schemaFunc,
+		configFunc,
+		applyTaskPolicyFunc,
+		applyTaskConfigFunc,
+	)
+
+	uiRegistry.RegisterUI(uiProvider)
+	glog.V(1).Infof("✅ Registered %s task UI provider", taskType)
+}