1 files changed, 361 insertions, 0 deletions
diff --git a/weed/worker/tasks/balance/ui.go b/weed/worker/tasks/balance/ui.go
new file mode 100644
index 000000000..88f7bb4a9
--- /dev/null
+++ b/weed/worker/tasks/balance/ui.go
@@ -0,0 +1,361 @@
+package balance
+
+import (
+	"fmt"
+	"html/template"
+	"strconv"
+	"time"
+
+	"github.com/seaweedfs/seaweedfs/weed/glog"
+	"github.com/seaweedfs/seaweedfs/weed/worker/types"
+)
+
+// UIProvider provides the UI for balance task configuration
+type UIProvider struct {
+	detector  *BalanceDetector
+	scheduler *BalanceScheduler
+}
+
+// NewUIProvider creates a new balance UI provider
+func NewUIProvider(detector *BalanceDetector, scheduler *BalanceScheduler) *UIProvider {
+	return &UIProvider{
+		detector:  detector,
+		scheduler: scheduler,
+	}
+}
+
+// GetTaskType returns the task type
+func (ui *UIProvider) GetTaskType() types.TaskType {
+	return types.TaskTypeBalance
+}
+
+// GetDisplayName returns the human-readable name
+func (ui *UIProvider) GetDisplayName() string {
+	return "Volume Balance"
+}
+
+// GetDescription returns a description of what this task does
+func (ui *UIProvider) GetDescription() string {
+	return "Redistributes volumes across volume servers to optimize storage utilization and performance"
+}
+
+// GetIcon returns the icon CSS class for this task type
+func (ui *UIProvider) GetIcon() string {
+	return "fas fa-balance-scale text-secondary"
+}
+
+// BalanceConfig represents the balance configuration
+type BalanceConfig struct {
+	Enabled             bool    `json:"enabled"`
+	ImbalanceThreshold  float64 `json:"imbalance_threshold"`
+	ScanIntervalSeconds int     `json:"scan_interval_seconds"`
+	MaxConcurrent       int     `json:"max_concurrent"`
+	MinServerCount      int     `json:"min_server_count"`
+	MoveDuringOffHours  bool    `json:"move_during_off_hours"`
+	OffHoursStart       string  `json:"off_hours_start"`
+	OffHoursEnd         string  `json:"off_hours_end"`
+	MinIntervalSeconds  int     `json:"min_interval_seconds"`
+}
+
+// Helper functions for duration conversion
+func secondsToDuration(seconds int) time.Duration {
+	return time.Duration(seconds) * time.Second
+}
+
+func durationToSeconds(d time.Duration) int {
+	return int(d.Seconds())
+}
+
+// formatDurationForUser formats seconds as a user-friendly duration string
+func formatDurationForUser(seconds int) string {
+	d := secondsToDuration(seconds)
+	if d < time.Minute {
+		return fmt.Sprintf("%ds", seconds)
+	}
+	if d < time.Hour {
+		return fmt.Sprintf("%.0fm", d.Minutes())
+	}
+	if d < 24*time.Hour {
+		return fmt.Sprintf("%.1fh", d.Hours())
+	}
+	return fmt.Sprintf("%.1fd", d.Hours()/24)
+}
+
+// RenderConfigForm renders the configuration form HTML
+func (ui *UIProvider) RenderConfigForm(currentConfig interface{}) (template.HTML, error) {
+	config := ui.getCurrentBalanceConfig()
+
+	// Build form using the FormBuilder helper
+	form := types.NewFormBuilder()
+
+	// Detection Settings
+	form.AddCheckboxField(
+		"enabled",
+		"Enable Balance Tasks",
+		"Whether balance tasks should be automatically created",
+		config.Enabled,
+	)
+
+	form.AddNumberField(
+		"imbalance_threshold",
+		"Imbalance Threshold (%)",
+		"Trigger balance when storage imbalance exceeds this percentage (0.0-1.0)",
+		config.ImbalanceThreshold,
+		true,
+	)
+
+	form.AddDurationField("scan_interval", "Scan Interval", "How often to scan for imbalanced volumes", secondsToDuration(config.ScanIntervalSeconds), true)
+
+	// Scheduling Settings
+	form.AddNumberField(
+		"max_concurrent",
+		"Max Concurrent Tasks",
+		"Maximum number of balance tasks that can run simultaneously",
+		float64(config.MaxConcurrent),
+		true,
+	)
+
+	form.AddNumberField(
+		"min_server_count",
+		"Minimum Server Count",
+		"Only balance when at least this many servers are available",
+		float64(config.MinServerCount),
+		true,
+	)
+
+	// Timing Settings
+	form.AddCheckboxField(
+		"move_during_off_hours",
+		"Restrict to Off-Hours",
+		"Only perform balance operations during off-peak hours",
+		config.MoveDuringOffHours,
+	)
+
+	form.AddTextField(
+		"off_hours_start",
+		"Off-Hours Start Time",
+		"Start time for off-hours window (e.g., 23:00)",
+		config.OffHoursStart,
+		false,
+	)
+
+	form.AddTextField(
+		"off_hours_end",
+		"Off-Hours End Time",
+		"End time for off-hours window (e.g., 06:00)",
+		config.OffHoursEnd,
+		false,
+	)
+
+	// Timing constraints
+	form.AddDurationField("min_interval", "Min Interval", "Minimum time between balance operations", secondsToDuration(config.MinIntervalSeconds), true)
+
+	// Generate organized form sections using Bootstrap components
+	html := `
+<div class="row">
+	<div class="col-12">
+		<div class="card mb-4">
+			<div class="card-header">
+				<h5 class="mb-0">
+					<i class="fas fa-balance-scale me-2"></i>
+					Balance Configuration
+				</h5>
+			</div>
+			<div class="card-body">
+` + string(form.Build()) + `
+			</div>
+		</div>
+	</div>
+</div>
+
+<div class="row">
+	<div class="col-12">
+		<div class="card mb-3">
+			<div class="card-header">
+				<h5 class="mb-0">
+					<i class="fas fa-exclamation-triangle me-2"></i>
+					Performance Considerations
+				</h5>
+			</div>
+			<div class="card-body">
+				<div class="alert alert-warning" role="alert">
+					<h6 class="alert-heading">Important Considerations:</h6>
+					<p class="mb-2"><strong>Performance:</strong> Volume balancing involves data movement and can impact cluster performance.</p>
+					<p class="mb-2"><strong>Recommendation:</strong> Enable off-hours restriction to minimize impact on production workloads.</p>
+					<p class="mb-0"><strong>Safety:</strong> Requires at least ` + fmt.Sprintf("%d", config.MinServerCount) + ` servers to ensure data safety during moves.</p>
+				</div>
+			</div>
+		</div>
+	</div>
+</div>`
+
+	return template.HTML(html), nil
+}
+
+// ParseConfigForm parses form data into configuration
+func (ui *UIProvider) ParseConfigForm(formData map[string][]string) (interface{}, error) {
+	config := &BalanceConfig{}
+
+	// Parse enabled
+	config.Enabled = len(formData["enabled"]) > 0
+
+	// Parse imbalance threshold
+	if values, ok := formData["imbalance_threshold"]; ok && len(values) > 0 {
+		threshold, err := strconv.ParseFloat(values[0], 64)
+		if err != nil {
+			return nil, fmt.Errorf("invalid imbalance threshold: %v", err)
+		}
+		if threshold < 0 || threshold > 1 {
+			return nil, fmt.Errorf("imbalance threshold must be between 0.0 and 1.0")
+		}
+		config.ImbalanceThreshold = threshold
+	}
+
+	// Parse scan interval
+	if values, ok := formData["scan_interval"]; ok && len(values) > 0 {
+		duration, err := time.ParseDuration(values[0])
+		if err != nil {
+			return nil, fmt.Errorf("invalid scan interval: %v", err)
+		}
+		config.ScanIntervalSeconds = int(duration.Seconds())
+	}
+
+	// Parse max concurrent
+	if values, ok := formData["max_concurrent"]; ok && len(values) > 0 {
+		maxConcurrent, err := strconv.Atoi(values[0])
+		if err != nil {
+			return nil, fmt.Errorf("invalid max concurrent: %v", err)
+		}
+		if maxConcurrent < 1 {
+			return nil, fmt.Errorf("max concurrent must be at least 1")
+		}
+		config.MaxConcurrent = maxConcurrent
+	}
+
+	// Parse min server count
+	if values, ok := formData["min_server_count"]; ok && len(values) > 0 {
+		minServerCount, err := strconv.Atoi(values[0])
+		if err != nil {
+			return nil, fmt.Errorf("invalid min server count: %v", err)
+		}
+		if minServerCount < 2 {
+			return nil, fmt.Errorf("min server count must be at least 2")
+		}
+		config.MinServerCount = minServerCount
+	}
+
+	// Parse off-hours settings
+	config.MoveDuringOffHours = len(formData["move_during_off_hours"]) > 0
+
+	if values, ok := formData["off_hours_start"]; ok && len(values) > 0 {
+		config.OffHoursStart = values[0]
+	}
+
+	if values, ok := formData["off_hours_end"]; ok && len(values) > 0 {
+		config.OffHoursEnd = values[0]
+	}
+
+	// Parse min interval
+	if values, ok := formData["min_interval"]; ok && len(values) > 0 {
+		duration, err := time.ParseDuration(values[0])
+		if err != nil {
+			return nil, fmt.Errorf("invalid min interval: %v", err)
+		}
+		config.MinIntervalSeconds = int(duration.Seconds())
+	}
+
+	return config, nil
+}
+
+// GetCurrentConfig returns the current configuration
+func (ui *UIProvider) GetCurrentConfig() interface{} {
+	return ui.getCurrentBalanceConfig()
+}
+
+// ApplyConfig applies the new configuration
+func (ui *UIProvider) ApplyConfig(config interface{}) error {
+	balanceConfig, ok := config.(*BalanceConfig)
+	if !ok {
+		return fmt.Errorf("invalid config type, expected *BalanceConfig")
+	}
+
+	// Apply to detector
+	if ui.detector != nil {
+		ui.detector.SetEnabled(balanceConfig.Enabled)
+		ui.detector.SetThreshold(balanceConfig.ImbalanceThreshold)
+		ui.detector.SetMinCheckInterval(secondsToDuration(balanceConfig.ScanIntervalSeconds))
+	}
+
+	// Apply to scheduler
+	if ui.scheduler != nil {
+		ui.scheduler.SetEnabled(balanceConfig.Enabled)
+		ui.scheduler.SetMaxConcurrent(balanceConfig.MaxConcurrent)
+		ui.scheduler.SetMinServerCount(balanceConfig.MinServerCount)
+		ui.scheduler.SetMoveDuringOffHours(balanceConfig.MoveDuringOffHours)
+		ui.scheduler.SetOffHoursStart(balanceConfig.OffHoursStart)
+		ui.scheduler.SetOffHoursEnd(balanceConfig.OffHoursEnd)
+	}
+
+	glog.V(1).Infof("Applied balance configuration: enabled=%v, threshold=%.1f%%, max_concurrent=%d, min_servers=%d, off_hours=%v",
+		balanceConfig.Enabled, balanceConfig.ImbalanceThreshold*100, balanceConfig.MaxConcurrent,
+		balanceConfig.MinServerCount, balanceConfig.MoveDuringOffHours)
+
+	return nil
+}
+
+// getCurrentBalanceConfig gets the current configuration from detector and scheduler
+func (ui *UIProvider) getCurrentBalanceConfig() *BalanceConfig {
+	config := &BalanceConfig{
+		// Default values (fallback if detectors/schedulers are nil)
+		Enabled:             true,
+		ImbalanceThreshold:  0.1, // 10% imbalance
+		ScanIntervalSeconds: durationToSeconds(4 * time.Hour),
+		MaxConcurrent:       1,
+		MinServerCount:      3,
+		MoveDuringOffHours:  true,
+		OffHoursStart:       "23:00",
+		OffHoursEnd:         "06:00",
+		MinIntervalSeconds:  durationToSeconds(1 * time.Hour),
+	}
+
+	// Get current values from detector
+	if ui.detector != nil {
+		config.Enabled = ui.detector.IsEnabled()
+		config.ImbalanceThreshold = ui.detector.GetThreshold()
+		config.ScanIntervalSeconds = int(ui.detector.ScanInterval().Seconds())
+	}
+
+	// Get current values from scheduler
+	if ui.scheduler != nil {
+		config.MaxConcurrent = ui.scheduler.GetMaxConcurrent()
+		config.MinServerCount = ui.scheduler.GetMinServerCount()
+		config.MoveDuringOffHours = ui.scheduler.GetMoveDuringOffHours()
+		config.OffHoursStart = ui.scheduler.GetOffHoursStart()
+		config.OffHoursEnd = ui.scheduler.GetOffHoursEnd()
+	}
+
+	return config
+}
+
+// RegisterUI registers the balance UI provider with the UI registry
+func RegisterUI(uiRegistry *types.UIRegistry, detector *BalanceDetector, scheduler *BalanceScheduler) {
+	uiProvider := NewUIProvider(detector, scheduler)
+	uiRegistry.RegisterUI(uiProvider)
+
+	glog.V(1).Infof("✅ Registered balance task UI provider")
+}
+
+// DefaultBalanceConfig returns default balance configuration
+func DefaultBalanceConfig() *BalanceConfig {
+	return &BalanceConfig{
+		Enabled:             false,
+		ImbalanceThreshold:  0.3,
+		ScanIntervalSeconds: durationToSeconds(4 * time.Hour),
+		MaxConcurrent:       1,
+		MinServerCount:      3,
+		MoveDuringOffHours:  false,
+		OffHoursStart:       "22:00",
+		OffHoursEnd:         "06:00",
+		MinIntervalSeconds:  durationToSeconds(1 * time.Hour),
+	}
+}