Compare commits

...

5 Commits

Author SHA1 Message Date
a11318072f Write new entries in the same file #3 2019-12-22 12:57:14 +01:00
4f8b167e7d Improve error handling for outputEntries 2019-12-22 11:34:08 +01:00
e6c4095e78 Extract method to output entries 2019-12-22 11:33:24 +01:00
0be444aa29 Clean up error handling of getEntriesForFeed 2019-12-22 11:32:31 +01:00
9dd00d29fd Extract getEntriesForFeed
Create a new function to get the entries for the feed.
2019-12-22 11:25:08 +01:00
3 changed files with 104 additions and 31 deletions

91
main.go
View File

@ -1,11 +1,16 @@
package main
import (
"bytes"
"fmt"
"html/template"
"io"
"io/ioutil"
"log"
"net/http"
"net/url"
"os"
"regexp"
"time"
"github.com/jinzhu/now"
@ -15,36 +20,104 @@ import (
"p83.nl/go/ekster/pkg/jf2"
)
func init() {
log.SetFlags(log.LstdFlags | log.Lshortfile)
}
func main() {
u, err := url.Parse(os.Args[1])
if err != nil {
log.Fatal(err)
}
filename := os.Args[2]
resp, err := http.Get(u.String())
re, err := regexp.Compile(`<span class="next-entry"></span>`)
if err != nil {
log.Fatal(err)
}
defer resp.Body.Close()
if resp.StatusCode != 200 {
log.Fatalf("Status is not 200, but %d", resp.StatusCode)
f, err := openAndCreate(filename)
data, err := ioutil.ReadAll(f)
if err != nil {
log.Fatal(err)
}
md := microformats.Parse(resp.Body, u)
items := jf2.SimplifyMicroformatDataItems(md)
f.Close()
var buf bytes.Buffer
items, err := getEntriesForFeed(u)
if err != nil {
log.Fatal(err)
}
items = filterEntriesAfter(items, now.With(time.Now()).Monday())
err = outputEntries(items, &buf)
if err != nil {
log.Fatal(err)
}
output := re.ReplaceAll(data, buf.Bytes())
err = ioutil.WriteFile(filename, output, 0666)
if err != nil {
log.Fatal(err)
}
}
func openAndCreate(filename string) (io.ReadCloser, error) {
f, err := os.Open(filename)
if err != nil && !os.IsExist(err) {
t, err := template.ParseFiles("templates/full.html")
if err != nil {
return nil, err
}
f, err = os.Create(filename)
if err != nil {
return nil, err
}
err = t.Execute(f, nil)
if err != nil {
return nil, err
}
err = f.Close()
if err != nil {
return nil, err
}
f, err = os.Open(filename)
if err != nil {
return nil, err
}
} else if err != nil {
return nil, err
}
return f, nil
}
func outputEntries(items []microsub.Item, w io.Writer) error {
t, err := template.ParseFiles("templates/weekly.html")
if err != nil {
log.Fatal(err)
}
err = t.Execute(os.Stdout, items)
err = t.Execute(w, items)
if err != nil {
log.Fatal(err)
return err
}
return nil
}
func getEntriesForFeed(u *url.URL) ([]microsub.Item, error) {
resp, err := http.Get(u.String())
if err != nil {
return nil, err
}
defer resp.Body.Close()
if resp.StatusCode != 200 {
return nil, fmt.Errorf("status is not 200, but %d", resp.StatusCode)
}
md := microformats.Parse(resp.Body, u)
items := jf2.SimplifyMicroformatDataItems(md)
return items, nil
}
func filterEntriesAfter(items []microsub.Item, from time.Time) []microsub.Item {

11
templates/full.html Normal file
View File

@ -0,0 +1,11 @@
<html lang="en">
<head>
<title>Weekly summaries</title>
</head>
<body>
<div class="h-feed">
<h1 class="p-name">Weekly</h1>
<span class="next-entry"></span>
</div>
</body>
</html>

View File

@ -1,25 +1,14 @@
<html lang="en">
<head>
<title>Weekly summaries</title>
</head>
<body>
<div class="h-feed">
<h1 class="p-name">Weekly</h1>
<span class="next-entry"></span>
<div class="h-entry">
{{ range . }}
<div class="h-entry">
<h2 class="p-name">{{ .Name }}</h2>
{{ if .Content }}
{{ if .Content.HTML }}
<div class="e-content">{{ .Content.HTML }}</div>
{{ else }}
<div class="p-content">{{ .Content.Text }}</div>
{{ end }}
<h2 class="p-name">{{ .Name }}</h2>
{{ if .Content }}
{{ if .Content.HTML }}
<div class="e-content">{{ .Content.HTML }}</div>
{{ else }}
<div class="p-content">{{ .Content.Text }}</div>
{{ end }}
<time datetime="{{ .Published }}">{{ .Published }}</time>
</div>
{{ end }}
<time datetime="{{ .Published }}">{{ .Published }}</time>
{{ end }}
</div>
</body>
</html>
</div>