main.go

package main

import (
	"database/sql"
	"log"
	"net/http"
	"os"

	"github.com/go-chi/chi"
	"github.com/go-chi/cors"
	"github.com/joho/godotenv"
	_ "github.com/lib/pq"
	"github.com/lordmoma/blog-aggregator/internal/database"
)

type apiConfig struct {
	DB *database.Queries
}

func main() {
	err := godotenv.Load()
	if err != nil {
		log.Fatal("Error loading .env file")
	}

	port := os.Getenv("PORT")
	if port == "" {
		log.Fatal("PORT is not set")
	}

	dbURL := os.Getenv("DATABASE_URL")
	if dbURL == "" {
		log.Fatal("DATABASE_URL is not set")
	}

	db, err := sql.Open("postgres", dbURL)
	if err != nil {
		log.Fatal(err)
	}
	dbQueries := database.New(db)

	apiCfg := &apiConfig{
		DB: dbQueries,
	}

	r := chi.NewRouter()

	r.Use(cors.Handler(cors.Options{
		AllowedOrigins:   []string{"https://*", "http://*"},
		AllowedMethods:   []string{"GET", "POST", "PUT", "DELETE", "OPTIONS"},
		AllowedHeaders:   []string{"*"},
		ExposedHeaders:   []string{"Link"},
		AllowCredentials: false,
		MaxAge:           300,
	}))

	r2 := chi.NewRouter()
	r2.Get("/readiness", readinessHandler)
	r2.Get("/err", errHandler)

	// user routes
	r2.Post("/users", apiCfg.createUserHandler)
	r2.Get("/users", apiCfg.middlewareAuth(apiCfg.getUserHandler))

	// feed routes
	r2.Post("/feeds", apiCfg.middlewareAuth(apiCfg.createFeedHandler))
	r2.Get("/feeds", apiCfg.getFeedHandler)

	// feed_follows routes
	r2.Get("/feed_follows", apiCfg.middlewareAuth(apiCfg.getFeedFollowHandler))
	r2.Post("/feed_follows", apiCfg.middlewareAuth(apiCfg.createFeedFollowHandler))
	r2.Delete("/feed_follows/{feedFollowID}", apiCfg.middlewareAuth(apiCfg.deleteFeedFollowHandler))

	// posts routes
	r2.Get("/posts", apiCfg.middlewareAuth(apiCfg.getPostsHandler))

	r.Mount("/v1", r2)

	srv := &http.Server{
		Addr:    ":" + port,
		Handler: r,
	}

	go fetchFeedsWorker(apiCfg.DB, 5)

	log.Printf("Serving files on port: %s\n", port)
	if err := srv.ListenAndServe(); err != http.ErrServerClosed {
		log.Fatalf("HTTP server ListenAndServe: %v", err)
	}
}