tilde.town/genfeed.go

122 lines
2.0 KiB
Go

package main
import (
"bytes"
_ "embed"
"encoding/json"
"fmt"
"os"
"os/exec"
"sort"
"text/template"
)
const statsPath = "/usr/local/bin/stats"
//go:embed feed.tmpl.xml
var feedTmpl string
type newsEntry struct {
Title string // Title of entry
Pubdate string // Human readable date
Content string // HTML of entry
}
type User struct {
Username string
Default bool
}
type tildeData struct {
News []newsEntry
Users []User
ActiveUsers []string `json:"active_users"`
}
type ByName []User
func (n ByName) Len() int { return len(n) }
func (n ByName) Swap(i, j int) { n[i], n[j] = n[j], n[i] }
func (n ByName) Less(i, j int) bool { return n[i].Username < n[j].Username }
func _main() error {
data, err := stats()
if err != nil {
return err
}
type tmplData struct {
News []newsEntry
Lights string
}
td := &tmplData{
News: data.News,
Lights: "",
}
sort.Sort(ByName(data.Users))
isActive := func(username string) bool {
for _, u := range data.ActiveUsers {
if u == username {
return true
}
}
return false
}
for _, u := range data.Users {
if isActive(u.Username) {
td.Lights += fmt.Sprintf("<a href=\"/~%s\">*</a>", u.Username)
} else if !u.Default {
td.Lights += fmt.Sprintf("<a href=\"/~%s\">+</a>", u.Username)
} else {
td.Lights += "."
}
}
t, err := template.New("feed").Parse(feedTmpl)
if err != nil {
return fmt.Errorf("failed to parse the feed template: %w", err)
}
out := bytes.Buffer{}
if err = t.Execute(&out, td); err != nil {
return fmt.Errorf("failed to render feed template: %w", err)
}
fmt.Println(out.String())
return nil
}
func stats() (*tildeData, error) {
sout := bytes.Buffer{}
cmd := exec.Command(statsPath)
cmd.Stdout = &sout
err := cmd.Run()
if err != nil {
return nil, err
}
var data tildeData
err = json.Unmarshal(sout.Bytes(), &data)
if err != nil {
return nil, err
}
return &data, nil
}
func main() {
err := _main()
if err != nil {
fmt.Fprintf(os.Stderr, "error: %s\n", err)
os.Exit(1)
}
}