package publisher

import (
	"context"
	"log"

	"lmika.dev/lmika/weiro/models"
)

type Queue struct {
	publisher *Publisher
	pending   chan models.Site
}

func NewQueue(publisher *Publisher) *Queue {
	return &Queue{
		publisher: publisher,
		pending:   make(chan models.Site, 1),
	}
}

func (q *Queue) Queue(site models.Site) bool {
	select {
	case q.pending <- site:
		return true
	default:
		return false
	}
}

func (q *Queue) Start(ctx context.Context) {
	go func() {
		for {
			select {
			case site := <-q.pending:
				if err := q.publisher.Publish(ctx, site); err != nil {
					log.Printf("error publishing site: %v", err)
				}
			case <-ctx.Done():
				return
			}
		}
	}()
}