115 lines
2.2 KiB
Go
115 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
e "codeberg.org/pebbe/errors"
|
|
|
|
u "git.web.rug.nl/p209327/nlnieuws/internal/util"
|
|
|
|
"encoding/xml"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"net/url"
|
|
"os"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
type Rss struct {
|
|
XMLName xml.Name `xml:"rss"`
|
|
Items []ItemT `xml:"channel>item"`
|
|
}
|
|
|
|
type ItemT struct {
|
|
PubDate string `xml:"pubDate"`
|
|
UnixTime int64 `xml:"unixTime"`
|
|
Guid string `xml:"guid"`
|
|
Link string `xml:"link"`
|
|
Data []byte `xml:",innerxml"`
|
|
}
|
|
|
|
var (
|
|
p = e.PanicErr
|
|
agent = "AhrefsBot/7.0"
|
|
)
|
|
|
|
func exists(filename string) bool {
|
|
_, err := os.Stat(filename)
|
|
return err == nil
|
|
}
|
|
|
|
func main() {
|
|
defer func() {
|
|
if e.Panicked {
|
|
_ = recover()
|
|
os.Exit(1)
|
|
}
|
|
}()
|
|
|
|
myLock := "/net/corpora/nlnieuws/Sargasso/lock"
|
|
u.MkLock(myLock)
|
|
defer func() {
|
|
_ = os.Remove(myLock)
|
|
}()
|
|
|
|
req, err := http.NewRequest("GET", "https://sargasso.nl/feed/", nil)
|
|
p(err)
|
|
req.Header.Set("User-Agent", agent)
|
|
|
|
client := &http.Client{}
|
|
resp, err := client.Do(req)
|
|
p(err)
|
|
body, err := io.ReadAll(resp.Body)
|
|
p(err)
|
|
p(resp.Body.Close())
|
|
|
|
var rss Rss
|
|
p(xml.Unmarshal(body, &rss))
|
|
|
|
if len(rss.Items) == 0 {
|
|
p(fmt.Errorf("len(rss.Items) == 0"))
|
|
}
|
|
|
|
for _, item := range rss.Items {
|
|
t, err := time.Parse(time.RFC1123Z, item.PubDate)
|
|
if err != nil {
|
|
t, err = time.Parse(time.RFC1123, item.PubDate)
|
|
}
|
|
p(err)
|
|
year, week := t.ISOWeek()
|
|
dirname := fmt.Sprintf("/net/corpora/nlnieuws/Sargasso/%d/w%02d", year, week)
|
|
if exists(dirname + "/lock") {
|
|
continue
|
|
}
|
|
basename := strings.TrimPrefix(item.Guid, "https://sargasso.nl/?")
|
|
if i := strings.LastIndex(basename, "p="); i >= 0 {
|
|
basename = basename[i+2:]
|
|
}
|
|
filename := dirname + "/" + url.PathEscape(basename)
|
|
|
|
p(os.MkdirAll(dirname, 0777))
|
|
func() {
|
|
var ok bool
|
|
defer func() {
|
|
if e.Panicked {
|
|
fmt.Fprintln(os.Stderr, "----", filename)
|
|
fmt.Fprintln(os.Stderr, "----", item.Link)
|
|
}
|
|
if !ok {
|
|
_ = os.Remove(filename + ".xml")
|
|
}
|
|
}()
|
|
fp, err := os.Create(filename + ".xml")
|
|
p(err)
|
|
p(fp.WriteString("<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n<item>\n"))
|
|
p(fmt.Fprintf(fp, "<unixTime>%d</unixTime>", t.Unix()))
|
|
p(fp.Write(item.Data))
|
|
p(fp.WriteString("</item>\n"))
|
|
p(fp.Close())
|
|
p(os.Chtimes(filename+".xml", t, t))
|
|
ok = true
|
|
}()
|
|
}
|
|
|
|
}
|