120 lines
2.4 KiB
Go
120 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"strings"
|
|
"text/template"
|
|
|
|
"code.ewintr.nl/emdb/storage"
|
|
"code.ewintr.nl/go-kit/slugify"
|
|
)
|
|
|
|
const (
|
|
pageTemplate = `+++
|
|
title = "{{ .Title }}"
|
|
date = {{ .Date }}
|
|
draft = false
|
|
extra.movie.year = {{ .Year }}
|
|
extra.movie.directors = "{{ .Directors }}"
|
|
extra.movie.en_title = "{{ .EnTitle }}"
|
|
extra.movie.rating = {{ .Rating }}
|
|
+++
|
|
|
|
{{ .Comment }}<!-- more -->`
|
|
)
|
|
|
|
func main() {
|
|
dbHost := os.Getenv("EMDB_DB_HOST")
|
|
dbName := os.Getenv("EMDB_DB_NAME")
|
|
dbUser := os.Getenv("EMDB_DB_USER")
|
|
dbPassword := os.Getenv("EMDB_DB_PASSWORD")
|
|
pgConnStr := fmt.Sprintf("host=%s user=%s password=%s dbname=%s sslmode=disable", dbHost, dbUser, dbPassword, dbName)
|
|
dbPostgres, err := storage.NewPostgres(pgConnStr)
|
|
if err != nil {
|
|
fmt.Printf("could not create new postgres repo: %s", err.Error())
|
|
os.Exit(1)
|
|
}
|
|
movieRepo := storage.NewMovieRepository(dbPostgres)
|
|
movies, err := movieRepo.FindAll()
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
tpl, err := template.New("page").Parse(pageTemplate)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
path := "public"
|
|
Empty(path)
|
|
|
|
for _, m := range movies {
|
|
filename := fmt.Sprintf("%s.md", slugify.Slugify(m.EnglishTitle))
|
|
|
|
watchedOn := strings.Split(m.WatchedOn, "-")
|
|
watchedOnYear := watchedOn[0]
|
|
|
|
if err := os.MkdirAll(fmt.Sprintf("%s/%s", path, watchedOnYear), os.ModePerm); err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
}
|
|
filePath := fmt.Sprintf("%s/%s/%s", path, watchedOnYear, filename)
|
|
f, err := os.Create(filePath)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
enTitle := m.EnglishTitle
|
|
if enTitle == m.Title {
|
|
enTitle = ""
|
|
}
|
|
|
|
data := struct {
|
|
Title string
|
|
Date string
|
|
Year int
|
|
Directors string
|
|
EnTitle string
|
|
Rating string
|
|
Comment string
|
|
}{
|
|
Title: m.Title,
|
|
Date: m.WatchedOn,
|
|
Year: m.Year,
|
|
Directors: strings.Join(m.Directors, ", "),
|
|
EnTitle: enTitle,
|
|
Rating: fmt.Sprintf("%d", m.Rating),
|
|
Comment: m.Comment,
|
|
}
|
|
if err := tpl.Execute(f, data); err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
}
|
|
if err := f.Close(); err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
}
|
|
|
|
func Empty(dir string) error {
|
|
files, err := ioutil.ReadDir(dir)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, file := range files {
|
|
err = os.RemoveAll(dir + "/" + file.Name())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|