package main
import (
"bytes"
"context"
_ "embed"
"encoding/json"
"errors"
"fmt"
"html/template"
"io"
"net/http"
"net/url"
"os"
"path/filepath"
"regexp"
"strconv"
"strings"
"time"
"github.com/dustin/go-humanize"
"github.com/julienschmidt/httprouter"
"github.com/k3a/html2text"
"github.com/rystaf/go-lemmy"
"golang.org/x/text/language"
"golang.org/x/text/message"
)
var pictrs = regexp.MustCompile(`\/pictrs\/image\/([a-z0-9\-]+)\.([a-z]+)$`)
var imgur = regexp.MustCompile(`^https:\/\/(i\.)?imgur.com\/([a-zA-Z0-9]{5,})(\.[a-zA-Z0-9]+)?`)
var isImage = func(u string) bool {
p, err := url.Parse(u)
if err != nil || p.Path == "" {
return false
}
ext := filepath.Ext(p.Path)
if ext == ".jpeg" || ext == ".jpg" || ext == ".png" || ext == ".webp" || ext == ".gif" {
return true
}
return false
}
var funcMap = template.FuncMap{
"host": func(host string) string {
if l := os.Getenv("LEMMY_DOMAIN"); l != "" {
return l
}
return host
},
"localize": func(s string) string {
u, err := url.Parse(s)
if err != nil {
return s
}
return "." + u.Path + "@" + u.Host
},
"printer": func(n any) string {
p := message.NewPrinter(language.English)
return p.Sprintf("%d", n)
},
"likedPerc": func(c lemmy.PostAggregates) string {
return fmt.Sprintf("%.1f", (float64(c.Upvotes)/float64(c.Upvotes+c.Downvotes))*100)
},
"fullname": func(person lemmy.Person) string {
if person.Local {
return person.Name
}
l, err := url.Parse(person.ActorID)
if err != nil {
fmt.Println(err)
return person.Name
}
return person.Name + "@" + l.Host
},
"fullcname": func(c lemmy.Community) string {
if c.Local {
return c.Name
}
l, err := url.Parse(c.ActorID)
if err != nil {
fmt.Println(err)
return c.Name
}
return c.Name + "@" + l.Host
},
"isMod": func(c *lemmy.GetCommunityResponse, username string) bool {
for _, mod := range c.Moderators {
if mod.Moderator.Local && username == mod.Moderator.Name {
return true
}
}
return false
},
"instance": func(actorID string) string {
l, err := url.Parse(actorID)
if err != nil {
fmt.Println(err)
return ""
}
return l.Host
},
"domain": func(p Post) string {
if p.Post.URL.IsValid() {
l, err := url.Parse(p.Post.URL.String())
if err != nil {
return ""
}
return l.Host
}
if p.Post.Local {
return "self." + p.Community.Name
}
l, err := url.Parse(p.Post.ApID)
if err != nil {
return ""
}
return l.Host
},
"membership": func(s lemmy.SubscribedType) string {
switch s {
case lemmy.SubscribedTypeSubscribed:
return "leave"
case lemmy.SubscribedTypeNotSubscribed:
return "join"
case lemmy.SubscribedTypePending:
return "pending"
}
return ""
},
"isImage": isImage,
"isYoutube": func(u string) bool {
re := regexp.MustCompile(`^((?:https?:)?\/\/)?((?:www|m)\.)?((?:youtube(-nocookie)?\.com|youtu.be))(\/(?:[\w\-]+\?v=|embed\/|live\/|v\/)?)([\w\-]+)(\S+)?$`)
return re.MatchString(u)
},
"thumbnail": func(p lemmy.Post) string {
if pictrs.MatchString(p.ThumbnailURL.String()) {
return p.ThumbnailURL.String() + "?format=jpg&thumbnail=96"
} else if pictrs.MatchString(p.URL.String()) {
return p.URL.String() + "?format=jpg&thumbnail=96"
}
if imgur.MatchString(p.URL.String()) {
return imgur.ReplaceAllString(p.URL.String(), "https://i.imgur.com/${2}s.jpg")
}
if p.ThumbnailURL.IsValid() {
return p.ThumbnailURL.String()
}
if isImage(p.URL.String()) {
return "/_/static/photo.png"
}
if p.URL.IsValid() {
return "/_/static/link.png"
}
return "/_/static/text.png"
},
"cbanner": func(c lemmy.Community) string {
bannerURL := ""
if c.Banner.IsValid() {
bannerURL = c.Banner.String()
} else if c.Icon.IsValid() {
bannerURL = c.Icon.String()
}
if pictrs.MatchString(bannerURL) {
return bannerURL + "?format=jpg&thumbnail=300"
}
return bannerURL
},
"shrink": func(u string) string {
if pictrs.MatchString(u) {
return u + "?format=png&thumbnail=60"
}
return u
},
"humanize": humanize.Time,
"markdown": func(host string, body string) template.HTML {
var buf bytes.Buffer
re := regexp.MustCompile(`\s---\s`)
body = re.ReplaceAllString(body, "\n***\n")
// community bangs
body = RegReplace(body, `([^\[])!([a-zA-Z0-9_]+)@([a-zA-Z0-9\-]+(\.[a-zA-Z0-9\-]+)+)`, `$1[!$2@$3](/c/$2@$3)`)
if err := md.Convert([]byte(body), &buf); err != nil {
fmt.Println(err)
return template.HTML(body)
}
body = buf.String()
body = strings.Replace(body, `![]()
$1$2")
return template.HTML(body)
},
"collapsemedia": func(h template.HTML) template.HTML {
body := string(h)
return template.HTML(RegReplace(body, `)
`, `collapsed inline media
