1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
package main
import (
"database/sql"
"log"
"net/http"
"net/url"
"os"
"path/filepath"
"code.laria.me/laria.me/article"
"code.laria.me/laria.me/config"
"code.laria.me/laria.me/environment"
)
func allArticlesFromDir(dir string) ([]article.Article, error) {
f, err := os.Open(dir)
if err != nil {
return nil, err
}
defer f.Close()
infos, err := f.Readdir(-1)
if err != nil {
return nil, err
}
articles := make([]article.Article, 0)
for _, info := range infos {
if info.IsDir() {
continue
}
fullname := filepath.Join(dir, info.Name())
a, err := article.LoadArticle(fullname)
if err != nil {
return nil, err
}
articles = append(articles, a)
}
return articles, nil
}
func updateArticles(conf *config.Config, db *sql.DB) {
articles := []article.Article{}
for _, dir := range conf.ArticleDirs {
dirArticles, err := allArticlesFromDir(dir)
if err != nil {
log.Fatalf("allArticlesFromDir(%s): %s", dir, err)
}
articles = append(articles, dirArticles...)
}
slugs := make([]string, 0, len(articles))
for _, article := range articles {
if _, err := article.SaveToDb(db); err != nil {
log.Fatalf("SaveToDb: %s", err)
}
slugs = append(slugs, article.Slug)
}
if err := article.DeleteArticlesFromDbExcept(db, slugs); err != nil {
log.Fatalf("DeleteArticlesFromDbExcept: %s", err)
}
}
func cmdUpdate(progname string, env *environment.Env, args []string) {
conf, err := env.Config()
if err != nil {
log.Fatalf("env.Config() failed: %s", err)
}
db, err := env.DB()
if err != nil {
log.Fatalf("env.Config() failed: %s", err)
}
updateArticles(conf, db)
resp, err := http.PostForm(conf.UpdateUrl, url.Values{"secret": {conf.Secret}})
if err != nil {
log.Fatalf("triggering server update failed: %s", err)
}
if resp.StatusCode != 200 {
log.Fatalf("server update unexpectedly responded with %d %s", resp.StatusCode, resp.Status)
}
}
|
