69 lines
1.4 KiB
Go
69 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"gitlab.com/kbr4/svevijesti/internal/database"
|
|
"html/template"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"path/filepath"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
var tPath = "./web/tpl/"
|
|
var dPath = "./web/data/"
|
|
|
|
var templateDirs = []string{"./web/tpl", "./web/data"}
|
|
var templates *template.Template
|
|
|
|
func getTemplates() (templates *template.Template, err error) {
|
|
var allFiles []string
|
|
for _, dir := range templateDirs {
|
|
files2, _ := ioutil.ReadDir(dir)
|
|
for _, file := range files2 {
|
|
filename := file.Name()
|
|
if strings.HasSuffix(filename, ".html") {
|
|
filePath := filepath.Join(dir, filename)
|
|
fmt.Println("Template found: ", filePath)
|
|
allFiles = append(allFiles, filePath)
|
|
}
|
|
}
|
|
}
|
|
|
|
templates, err = template.New("").ParseFiles(allFiles...)
|
|
return
|
|
}
|
|
|
|
func init() {
|
|
templates, _ = getTemplates()
|
|
}
|
|
|
|
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 main() {
|
|
http.HandleFunc("/", rootHandler)
|
|
http.ListenAndServe(":8080", nil)
|
|
}
|