main.go

package main

import (
	"encoding/json"
	"fmt"
	"log"
	"net/http"
	"os"
	"os/signal"
	"runtime"

	gotdotenv "github.com/joho/godotenv"

	"github.com/javokhirbek1999/tez-spider/handlers"
)

type Response struct {
	Data       []handlers.Song
	Total      int64
	StatusCode int
}

func init() {
	runtime.GOMAXPROCS(runtime.NumCPU())
}

func enableCORS(w *http.ResponseWriter) {
	(*w).Header().Set("Access-Control-Allow-Origin", "*")
}

func main() {

	if err := gotdotenv.Load(".env"); err != nil {
		log.Fatalln("Couldn't load environment variables")
		return
	}
	port := os.Getenv("PORT")

	if port == "" {
		port = "4000"
	}

	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		enableCORS(&w)
		fmt.Fprintf(w, "Welcome")
	})

	http.HandleFunc("/search", func(w http.ResponseWriter, r *http.Request) {

		enableCORS(&w)

		query := r.URL.Query().Get("query")

		songs := handlers.Crawler(query)
		encoder := json.NewEncoder(w)

		totalSongs := len(songs)

		if totalSongs == 0 {
			encoder.Encode(Response{
				Data:       nil,
				Total:      0,
				StatusCode: http.StatusBadRequest,
			})
		} else {
			encoder.Encode(Response{
				Data:       songs,
				Total:      int64(totalSongs),
				StatusCode: http.StatusOK,
			})
		}
	})

	go func() {

		err := http.ListenAndServe(":"+port, nil)

		if err != nil {
			log.Fatalf("Server Error: %v", err)
			return
		}
	}()

	sig := make(chan os.Signal)

	signal.Notify(sig, os.Interrupt)
	signal.Notify(sig, os.Kill)

	signalMessage := <-sig

	log.Fatalf("Server stopped: %v\n", signalMessage)

}