57 lines
1.3 KiB
Go
57 lines
1.3 KiB
Go
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)
|
|
}
|
|
}
|