package worker

import (
	"context"
	"time"

	"git.sr.ht/~gabrielgio/midr/db"
	"git.sr.ht/~gabrielgio/midr/yt"
	work "git.sr.ht/~sircmpwn/dowork"
)

const (
	statusNotQueued = "NOTQUEUED"
	statusQueued    = "QUEUED"
	statusStarted   = "RUNNING"

	commandStart   = "START"
	commandEnqueue = "ENQUEUE"
	commandDequeue = "DEQUEUE"
)

type command struct {
	action string
	index  uint
}

type Worker struct {
	jobs map[uint]string
	c    chan command
}

type Job struct {
	Id     uint
	Status string
}

func (w *Worker) CanEnqueue(index uint) bool {
	v, found := w.jobs[index]
	return !found || v == statusNotQueued
}

func (w *Worker) SpawnWorker(index uint, link string, output string) {

	if !w.CanEnqueue(index) {
		return
	}

	w.c <- command{action: commandEnqueue, index: index}
	task := work.NewTask(func(ctx context.Context) error {
		w.c <- command{action: commandStart, index: index}
		yt.RunYtDlpProcess(link, output)
		return nil
	}).After(func(ctx context.Context, task *work.Task) {
		w.c <- command{action: commandDequeue, index: index}
	})

	work.Enqueue(task)
}

func (w *Worker) startReader() {
	for true {
		command := <-w.c

		if command.action == commandEnqueue {
			w.jobs[command.index] = statusQueued
		} else if command.action == commandStart {
			w.jobs[command.index] = statusStarted
		} else if command.action == commandDequeue {
			w.jobs[command.index] = statusNotQueued
		} else {
			panic(1)
		}
	}
}

func (w *Worker) startScheduler(model db.EntryModel) {
	for true {
		entries := model.All()
		for _, e := range entries {
			w.SpawnWorker(e.ID, e.Link, e.OutputFolder)
		}
		time.Sleep(30 * time.Minute)
	}
}

func (w *Worker) StartWorker(model db.EntryModel) {
	w.c = make(chan command, 10)
	w.jobs = make(map[uint]string)
	go w.startReader()
	go w.startScheduler(model)
}

func (w *Worker) GetJobs() []Job {
	jobs := []Job{}

	for k, v := range w.jobs {
		jobs = append(jobs, Job{Id: k, Status: v})
	}

	return jobs
}