package main

import (
	"encoding/json"
	"fmt"
	"io"
	"log"
	"net/http"
	"time"
)

var server = New(10 * time.Second)

func wsHandler(responseWriter http.ResponseWriter, request *http.Request) {
	conn, err := server.Upgrader.Upgrade(responseWriter, request, nil)
	if err != nil {
		fmt.Println("Error during connection upgrade:", err)
		return
	}
	server.HandleNew(conn)
}

type DownloadReq struct {
	Link string `json:"link"`
}

func handleDownload(responseWriter http.ResponseWriter, request *http.Request) {
	body, err := io.ReadAll(request.Body)
	if err != nil {
		log.Printf("Error reading request body: %v", err)
		http.Error(responseWriter, "Error reading request body", http.StatusBadRequest)
		return
	}
	defer request.Body.Close()

	req := DownloadReq{}
	err = json.Unmarshal(body, &req)
	if err != nil {
		log.Printf("Error parsing JSON: %v", err)
		http.Error(responseWriter, "Error parsing JSON", http.StatusBadRequest)
		return
	}
	server.Broadcast <- req.Link
}

func main() {
	log.SetFlags(log.Lmicroseconds)

	http.HandleFunc("/ws", wsHandler)
	http.HandleFunc("/download", handleDownload)
	log.Println("Server starting on :8080")
	err := http.ListenAndServe(":8080", nil)
	if err != nil {
		log.Println("Error starting server:", err)
	}
}