package sandbox

import (
	"context"
	"fmt"
	"time"

	"github.com/yaoapp/yao/monitor"
)

func init() {
	monitor.Register(&sandboxWatcher{})
}

type sandboxWatcher struct{}

func (w *sandboxWatcher) Name() string            { return "sandbox" }
func (w *sandboxWatcher) Interval() time.Duration { return 30 * time.Second }

func (w *sandboxWatcher) Check(ctx context.Context) []monitor.Alert {
	if mgr == nil {
		return nil
	}

	var alerts []monitor.Alert
	mgr.boxes.Range(func(_, v any) bool {
		b := v.(*Box)

		status := b.inspectStatus(ctx)
		old, _ := b.status.Swap(status).(string)
		if old != "" && old != status {
			alerts = append(alerts, monitor.Alert{
				Level:   monitor.Info,
				Target:  "box:" + b.id,
				Message: fmt.Sprintf("status %s → %s", old, status),
			})
		}

		if status != "running" {
			return true
		}

		idle := time.Since(b.idleSince())
		timeout := b.idleTimeout()
		if timeout <= 0 || idle <= timeout {
			return true
		}

		switch b.policy {
		case Session:
			alerts = append(alerts, monitor.Alert{
				Level:   monitor.Warn,
				Target:  "box:" + b.id,
				Message: fmt.Sprintf("session idle expired (idle=%s, timeout=%s), removing", idle.Round(time.Second), timeout),
				Action: func(ctx context.Context) {
					mgr.Remove(ctx, b.id)
				},
			})

		case LongRunning:
			alerts = append(alerts, monitor.Alert{
				Level:   monitor.Warn,
				Target:  "box:" + b.id,
				Message: fmt.Sprintf("longrunning idle expired (idle=%s, timeout=%s), stopping", idle.Round(time.Second), timeout),
				Action: func(ctx context.Context) {
					b.Stop(ctx)
				},
			})
		}

		if b.policy == LongRunning {
			if lifetime := b.maxLifetime(); lifetime > 0 && time.Since(b.createdAt) > lifetime {
				alerts = append(alerts, monitor.Alert{
					Level:   monitor.Warn,
					Target:  "box:" + b.id,
					Message: fmt.Sprintf("lifetime expired (%s), removing", lifetime),
					Action: func(ctx context.Context) {
						mgr.Remove(ctx, b.id)
					},
				})
			}
		}

		return true
	})
	return alerts
}