package server

import (
	"github.com/gorilla/mux"
	"gitlab.com/kbr4/svevijesti/internal/database"
	"net/http"
	"strconv"
	"time"
)

func rootHandler(wr http.ResponseWriter, req *http.Request) {
	title := "Pocetna"
	store, err := database.Connect()
	if err != nil {
		http.Error(wr, err.Error(), http.StatusInternalServerError)
	}

	articles, err := database.ArticlesForDay(store, time.Now())
	if err != nil {
		http.Error(wr, err.Error(), http.StatusInternalServerError)
	}

	data := map[string]interface{}{
		"title":    title,
		"articles": articles,
	}

	err = templates.ExecuteTemplate(wr, "homeHTML", data)
	if err != nil {
		http.Error(wr, err.Error(), http.StatusInternalServerError)
	}
}

func articleHandler(wr http.ResponseWriter, req *http.Request) {
	title := "Pocetna"
	store, err := database.Connect()
	if err != nil {
		http.Error(wr, err.Error(), http.StatusInternalServerError)
	}

	vars := mux.Vars(req)
	articleID, err := strconv.Atoi(vars["id"])
	if err != nil {
		articleID = -1
	}
	articleSlug := vars["slug"]
	article, err := database.ArticleByID(store, articleID, articleSlug)
	if err != nil {
		http.Error(wr, err.Error(), http.StatusNotFound)
	}

	data := map[string]interface{}{
		"title":   title,
		"article": article,
	}

	err = templates.ExecuteTemplate(wr, "articleHTML", data)
	if err != nil {
		http.Error(wr, err.Error(), http.StatusInternalServerError)
	}
}