Weather and Category

internal/server/articles.go

@@ -25,11 +25,25 @@ func rootHandler(wr http.ResponseWriter, req *http.Request) {
 
 	dayBefore := "/dan/" + time.Now().Add(-24*time.Hour).Format("2006-01-02")
 
+	cities := []string{"Sarajevo", "Banja Luka", "Zenica", "Tuzla", "Mostar"}
+
+	var weatherInfo []WeatherData
+	for _, city := range cities {
+		data, err := getWeather(city)
+		if err != nil {
+			fmt.Printf("Error fetching weather for %s: %v\n", city, err)
+			continue
+		}
+		weatherInfo = append(weatherInfo, data)
+	}
+
 	data := map[string]interface{}{
-		"title":    title,
-		"articles": articles,
-		"previous": dayBefore,
-		"next":     "/",
+		"title":       title,
+		"articles":    articles,
+		"previous":    dayBefore,
+		"next":        "/",
+		"weatherInfo": weatherInfo,
+		"categories":  CategoryMenu,
 	}
 
 	err = templates.ExecuteTemplate(wr, "homeHTML", data)
@@ -63,11 +77,24 @@ func dailyArticlesHandler(wr http.ResponseWriter, req *http.Request) {
 		http.Error(wr, err.Error(), http.StatusInternalServerError)
 	}
 
+	cities := []string{"Sarajevo", "Banja Luka", "Zenica", "Tuzla", "Mostar"}
+
+	var weatherInfo []WeatherData
+	for _, city := range cities {
+		data, err := getWeather(city)
+		if err != nil {
+			fmt.Printf("Error fetching weather for %s: %v\n", city, err)
+			continue
+		}
+		weatherInfo = append(weatherInfo, data)
+	}
+
 	data := map[string]interface{}{
-		"title":    title,
-		"articles": articles,
-		"previous": dayBefore,
-		"next":     dayAfter,
+		"title":       title,
+		"articles":    articles,
+		"previous":    dayBefore,
+		"next":        dayAfter,
+		"weatherInfo": weatherInfo,
 	}
 
 	err = templates.ExecuteTemplate(wr, "homeHTML", data)
@@ -98,10 +125,11 @@ func articleHandler(wr http.ResponseWriter, req *http.Request) {
 
 	title := article.Title
 	data := map[string]interface{}{
-		"title":    title,
-		"article":  article,
-		"previous": previous,
-		"next":     next,
+		"title":      title,
+		"article":    article,
+		"previous":   previous,
+		"next":       next,
+		"categories": CategoryMenu,
 	}
 
 	err = templates.ExecuteTemplate(wr, "articleHTML", data)

internal/server/category.go

@@ -4,31 +4,65 @@ import (
 	"net/http"
 	"time"
 
+	"github.com/gorilla/mux"
 	"gitlab.com/kbr4/svevijesti/internal/database"
 	"gitlab.com/kbr4/svevijesti/internal/model"
 )
 
-func categoryHandler(wr http.ResponseWriter, r *http.Request) {
+var CategoryMenu = []string{
+	"Politika",
+	"Biznis",
+	"Sport",
+	"Magazin",
+	"Nauka i tehnologija",
+	"Ostalo",
+}
+
+func handleCategory(wr http.ResponseWriter, r *http.Request) {
+	vars := mux.Vars(r)
+	category := vars["category"]
 
 	store, err := database.Connect()
-
 	if err != nil {
 		http.Error(wr, err.Error(), http.StatusInternalServerError)
+		return
+	}
+	defer store.Close()
+
+	currentDate, err := time.Parse("2006-01-02", category)
+	if err != nil {
+		currentDate = time.Now()
 	}
 
-	articles, err := database.ArticleByCategory(store, time.Now())
+	articles, err := database.ArticlesForDay(store, currentDate)
+	if err != nil {
+		http.Error(wr, err.Error(), http.StatusInternalServerError)
+		return
+	}
 
 	articlesByCategory := make(map[string][]model.DisplayArticle)
-
 	for _, article := range articles {
 		articlesByCategory[article.Category] = append(articlesByCategory[article.Category], article)
 	}
 
-	for category, articles := range articlesByCategory {
-		data := map[string]interface{}{"Category": category, "Articles": articles}
-		err := templates.ExecuteTemplate(wr, "categoryHTML", data)
-		if err != nil {
-			panic(err)
-		}
+	var categories []string
+	for cat := range articlesByCategory {
+		categories = append(categories, cat)
+	}
+
+	prevDay := currentDate.AddDate(0, 0, -1)
+
+	data := map[string]interface{}{
+		"title":           category,
+		"currentCategory": category,
+		"articles":        articlesByCategory[category],
+		"categories":      CategoryMenu,
+		"previous":        prevDay.Format("2006-01-02"),
+		"next":            "/",
+	}
+	err = templates.ExecuteTemplate(wr, "categoryHTML", data)
+	if err != nil {
+		http.Error(wr, err.Error(), http.StatusInternalServerError)
+		return
 	}
 }

internal/server/server.go

@@ -44,6 +44,6 @@ func CreateRoutes() *mux.Router {
 	r.HandleFunc("/{id:[0-9]+}/{slug}", articleHandler)
 	r.HandleFunc("/", rootHandler)
 	r.HandleFunc("/weather", WeatherHandler)
-	r.HandleFunc("/{category}", categoryHandler)
+	r.HandleFunc("/{category}", handleCategory)
 	return r
 }

internal/server/weather.go

@@ -5,9 +5,27 @@ import (
 	"fmt"
 	"io/ioutil"
 	"net/http"
+	"os"
+
+	"github.com/joho/godotenv"
 )
 
-const apiKey = "abb35e21bdcbad6d1b00141a2b25cf5a"
+var apiKey string
+
+func init() {
+	err := godotenv.Load()
+	if err != nil {
+		fmt.Println("Error loading .env file:", err)
+		os.Exit(1)
+	}
+
+	apiKey = os.Getenv("API_KEY")
+
+	if apiKey == "" {
+		fmt.Println("API_KEY environment variable not set.")
+		os.Exit(1)
+	}
+}
 
 type WeatherData struct {
 	Coord struct {
@@ -74,24 +92,13 @@ func WeatherHandler(w http.ResponseWriter, r *http.Request) {
 	data := map[string]interface{}{
 		"title":       title,
 		"weatherInfo": weatherInfo,
+		"categories":  CategoryMenu,
 	}
 
-	err := templates.ExecuteTemplate(w, "fullweatherHTML", data)
+	err := templates.ExecuteTemplate(w, "weatherHTML", data)
 	if err != nil {
 		fmt.Println("Error executing template:", err)
 		http.Error(w, "Internal Server Error", http.StatusInternalServerError)
 	}
-	for _, info := range weatherInfo {
-		widgetData := map[string]interface{}{
-			"Temperature": info.Main.Temp,
-			"City":        info.Name,
-		}
-		err := templates.ExecuteTemplate(w, "weatherwidgetHTML", widgetData)
-		if err != nil {
-			fmt.Println("Error executing template:", err)
-			http.Error(w, "Internal Server Error", http.StatusInternalServerError)
-			return
-		}
-	}
 
 }