package main

import (
	"context"
	"fmt"
	"log/slog"
	"os"
	"os/signal"
	
	"github.com/Telenor-NMS-SE/ottomato/worker"
)

type ExampleDevice struct {
	Hostname string
}

func (d *ExampleDevice) Init(ctx context.Context) error {
	slog.Info("i have been initated", "device", d.Hostname)
	return nil
}

func (d *ExampleDevice) Ping(ctx context.Context) error {
	slog.Info("i have been pinged", "device", d.Hostname)
	return nil
}

func (d *ExampleDevice) RunTask(ctx context.Context, target string, task *worker.Task) (worker.Result, error) {
	slog.Info("i have received a task", "device", d.Hostname, "command", task.Command)
	return worker.Result{}, nil
}

func (d *ExampleDevice) Stop() error {
	slog.Info("i have been told to stop", "device", d.Hostname)
	return nil
}

func (d *ExampleDevice) Name() string {
	return d.Hostname
}

func (d *ExampleDevice) Info() map[string]any {
	return map[string]any{}
}

func main() {
	ctx, cancel := signal.NotifyContext(context.Background(), os.Interrupt)
	defer cancel()
	
	w, err := worker.New(ctx)
	if err != nil {
		slog.Error("failed to create new worker", "error", err)
	}
	defer func() {
		if err := w.Stop(); err != nil {
			slog.Error("failed to stop worker", "error", err)
		}
	}()
	
	for i := range 10 {
		name := fmt.Sprintf("device-%d", i)
		
		slog.Info("adding workload", "name", name)
		
		if err := w.AddWorkload(ctx, &ExampleDevice{Hostname: name}); err != nil {
			slog.Error("failed to add workload", "error", err)
		}
	}
	
	for _, wl := range w.Workloads() {
		res, err := w.RunTask(ctx, wl, &worker.Task{Command: "hello"})
		if err != nil {
			slog.Error("failed to run task", "workload", wl, "error", err)
		}
		slog.Info("got response", "workload", wl, "response", res)
	    
		if err := w.DeleteWorkload(wl); err != nil {
			slog.Error("failed to delete workload", "workload", wl, "error", err)
		}
	}
	
	<-ctx.Done()
}