package main

import (
	"context"
	"fmt"
	"mmaschedule-go/event"
	"time"
)

type Scraper func(ClientScraper) (*[]*event.Event, error)

func ScrapeEvents(q *event.Queries, client ClientScraper) {
	events := []*event.Event{}
	scrapers := []Scraper{
		ScrapeONE,
		ScrapeUFC,
	}

	for _, scraper := range scrapers {
		e, err := scraper(client)
		if err != nil {
			fmt.Println("Error scraping events: ", err)
		} else {
			events = append(events, *e...)
		}
	}

	err := SetTapologyCSRF(client)
	if err != nil {
		fmt.Println("Error settings tapology CSRF: ", err)
	}

	for _, e := range events {
		fights := e.UnmarshalFights()
		for _, f := range fights {
			fighters := []*event.Fighter{
				f.FighterA,
				f.FighterB,
			}
			for _, ff := range fighters {
				tapology, err := q.GetTapology(context.Background(), ff.Name)
				if err != nil {
					fmt.Println("Error getting tapology from database: ", err)
					link, err := GetTapologyLink(client, ff.Name)
					if err != nil {
						fmt.Println("Error getting tapology from site: ", err)
					} else if len(link) > 0 {
						err := q.CreateTapology(context.Background(), event.CreateTapologyParams{Name: ff.Name, Url: link})
						if err != nil {
							fmt.Println("Error creating tapology in database: ", err)
						}
						ff.Link = link
					}
					time.Sleep(time.Duration(5_000_000_000))
				} else {
					ff.Link = tapology.Url
				}
			}
		}
		e.MarshalFights(fights)
	}

	if len(events) > 0 {
		err := q.UpsertEvents(context.Background(), events)
		if err != nil {
			fmt.Println("Error updating events in database: ", err)
		}
	}
}