moved go files to src

src/orchestrator.go

@@ -0,0 +1,95 @@
+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...)
+		}
+	}
+
+	UpdateTapology(q, client, &events)
+
+	if len(events) > 0 {
+		err := q.UpsertEvents(context.Background(), events)
+		if err != nil {
+			fmt.Println("Error updating events in database:", err)
+		}
+	}
+}
+
+func UpdateAllTapology(q *event.Queries, client ClientScraper) error {
+	events_, err := q.ListEvents(context.Background())
+	if err != nil {
+		return err
+	}
+
+	events := []*event.Event{}
+	for _, e := range events_ {
+		events = append(events, &e)
+	}
+
+	UpdateTapology(q, client, &events)
+
+	err = q.UpsertEvents(context.Background(), events)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func UpdateTapology(q *event.Queries, client ClientScraper, events *[]*event.Event) {
+	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 {
+				fmt.Println("Getting tapology link for", ff.Name)
+				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)
+	}
+}