mirror of
https://github.com/rystaf/mlmym.git
synced 2024-11-05 06:43:21 +00:00
1552 lines
44 KiB
Go
1552 lines
44 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
_ "embed"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"html/template"
|
|
"io"
|
|
"net/http"
|
|
"net/url"
|
|
"os"
|
|
"regexp"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"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 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
|
|
},
|
|
"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": func(url string) bool {
|
|
ext := url[len(url)-4:]
|
|
if ext == "jpeg" || ext == ".jpg" || ext == ".png" || ext == "webp" || ext == ".gif" {
|
|
return true
|
|
}
|
|
return false
|
|
},
|
|
"thumbnail": func(p lemmy.Post) string {
|
|
if p.ThumbnailURL.IsValid() {
|
|
return p.ThumbnailURL.String() + "?format=jpg&thumbnail=96"
|
|
}
|
|
re := regexp.MustCompile(`\/pictrs\/image\/([a-z0-9\-]+)\.([a-z]+)$`)
|
|
if re.MatchString(p.URL.String()) {
|
|
return p.URL.String() + "?format=jpg&thumbnail=96"
|
|
}
|
|
re = regexp.MustCompile(`^https:\/\/(i\.)?imgur.com\/([a-zA-Z0-9]{5,})(\.[a-zA-Z0-9]+)?`)
|
|
if re.MatchString(p.URL.String()) {
|
|
return re.ReplaceAllString(p.URL.String(), "https://i.imgur.com/${2}s.jpg")
|
|
}
|
|
if p.URL.IsValid() {
|
|
return "/_/static/link.png"
|
|
}
|
|
return "/_/static/text.png"
|
|
},
|
|
"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, `<img `, `<img loading="lazy" `, -1)
|
|
body = RegReplace(body, `href="(https:\/\/[a-zA-Z0-9\.\-]+\/(c|u|comment|post)\/[^#\?]*?)"`, `href="/`+host+`/link?url=$1"`)
|
|
body = RegReplace(body, `::: ?spoiler (.*?)\n([\S\s]*?):::`, "<details><summary>$1</summary>$2</details>")
|
|
return template.HTML(body)
|
|
},
|
|
"rmmarkdown": func(body string) string {
|
|
var buf bytes.Buffer
|
|
if err := md.Convert([]byte(body), &buf); err != nil {
|
|
fmt.Println(err)
|
|
return body
|
|
}
|
|
text := html2text.HTML2TextWithOptions(buf.String(), html2text.WithLinksInnerText())
|
|
re := regexp.MustCompile(`\<(https?:\/\/|mailto)(?:www\.)?[-a-zA-Z0-9@:%._\+~#=]{1,256}\.[a-zA-Z0-9()]{1,6}\b(?:[-a-zA-Z0-9()@:%_\+.~#?&\/=]*)\>`)
|
|
return re.ReplaceAllString(text, "")
|
|
},
|
|
"contains": strings.Contains,
|
|
"sub": func(a int64, b int) int {
|
|
return int(a) - b
|
|
},
|
|
"add": func(a int64, b int) int {
|
|
return int(a) + b
|
|
},
|
|
}
|
|
|
|
func LemmyLinkRewrite(input string, host string, lemmy_domain string) (body string) {
|
|
body = input
|
|
// localize community and user links
|
|
body = RegReplace(body, `href="https:\/\/([a-zA-Z0-9\.\-]+)\/((c|u|comment|post)\/[^#\?]*?)"`, `href="/$2@$1"`)
|
|
// remove extra instance tag
|
|
body = RegReplace(body, `href="(https:\/)?(\/[a-zA-Z0-9\.\-]+)?\/((c|u)\/[a-zA-Z0-9]+@[a-zA-Z0-9\.\-]+)@([a-zA-Z0-9\.\-]+)"`, `href="/$3"`)
|
|
if lemmy_domain == "" {
|
|
// add domain to relative links
|
|
body = RegReplace(body, `href="\/((c|u|post|comment)\/(.*?)")`, `href="/`+host+`/$1`)
|
|
// convert links to relative
|
|
body = RegReplace(body, `href="https:\/\/([a-zA-Z0-9\.\-]+\/((c|u|post|comment)\/[a-zA-Z0-9]+"))`, `href="/$1`)
|
|
} else {
|
|
// convert local links to relative
|
|
body = RegReplace(body, `href="https:\/\/`+lemmy_domain+`\/(c\/[a-zA-Z0-9]+"|(c|u|post|comment)\/(.*?)")`, `href="/$1`)
|
|
body = RegReplace(body, `href="(.*)@`+lemmy_domain+`"`, `href="$1"`)
|
|
}
|
|
|
|
re := regexp.MustCompile(`href="\/?([a-zA-Z0-9\.\-]*)\/(c|u|post|comment)\/(.*?)@(.*?)"`)
|
|
// assume "old." subdomain is mlmym and remove
|
|
matches := re.FindAllStringSubmatch(body, -1)
|
|
for _, match := range matches {
|
|
if match[4][0:4] == "old." {
|
|
s := 1
|
|
if match[1] == "" {
|
|
s += 1
|
|
}
|
|
body = strings.Replace(body, match[0], `href="/`+strings.Join(match[s:4], "/")+"@"+match[4][4:]+`"`, -1)
|
|
}
|
|
}
|
|
// remove redundant instance tag
|
|
matches = re.FindAllStringSubmatch(body, -1)
|
|
for _, match := range matches {
|
|
if match[1] == match[4] {
|
|
body = strings.Replace(body, match[0], `href="/`+strings.Join(match[1:4], "/")+`"`, -1)
|
|
}
|
|
}
|
|
return body
|
|
}
|
|
|
|
func RegReplace(input string, match string, replace string) string {
|
|
re := regexp.MustCompile(match)
|
|
return re.ReplaceAllString(input, replace)
|
|
}
|
|
|
|
func getenv(key, fallback string) string {
|
|
value := os.Getenv(key)
|
|
if len(value) == 0 {
|
|
return fallback
|
|
}
|
|
return value
|
|
}
|
|
|
|
func Initialize(Host string, r *http.Request) (State, error) {
|
|
state := State{
|
|
Host: Host,
|
|
Page: 1,
|
|
Status: http.StatusOK,
|
|
Version: version,
|
|
}
|
|
if watch != nil {
|
|
state.Watch = *watch
|
|
}
|
|
lemmyDomain := os.Getenv("LEMMY_DOMAIN")
|
|
if lemmyDomain != "" {
|
|
state.Host = "."
|
|
Host = lemmyDomain
|
|
}
|
|
remoteAddr := r.RemoteAddr
|
|
if r.Header.Get("CF-Connecting-IP") != "" {
|
|
remoteAddr = r.Header.Get("CF-Connecting-IP")
|
|
}
|
|
client := http.Client{Transport: NewAddHeaderTransport(remoteAddr)}
|
|
c, err := lemmy.NewWithClient("https://"+Host, &client)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
state.Status = http.StatusInternalServerError
|
|
return state, err
|
|
}
|
|
state.HTTPClient = &client
|
|
state.Client = c
|
|
token := getCookie(r, "jwt")
|
|
user := getCookie(r, "user")
|
|
parts := strings.Split(user, ":")
|
|
if len(parts) == 2 && token != "" {
|
|
if id, err := strconv.Atoi(parts[1]); err == nil {
|
|
state.Client.Token = token
|
|
sess := Session{
|
|
UserName: parts[0],
|
|
UserID: id,
|
|
}
|
|
state.Session = &sess
|
|
}
|
|
}
|
|
state.Listing = getCookie(r, "DefaultListingType")
|
|
state.Sort = getCookie(r, "DefaultSortType")
|
|
state.CommentSort = getCookie(r, "DefaultCommentSortType")
|
|
if dark := getCookie(r, "Dark"); dark != "" {
|
|
state.Dark = dark != "0"
|
|
} else {
|
|
state.Dark = os.Getenv("DARK") != ""
|
|
}
|
|
state.ShowNSFW = getCookie(r, "ShowNSFW") != ""
|
|
state.HideInstanceNames = getCookie(r, "HideInstanceNames") != ""
|
|
if hide := getCookie(r, "HideThumbnails"); hide != "" {
|
|
state.HideThumbnails = hide != "0"
|
|
} else {
|
|
state.HideThumbnails = os.Getenv("HIDE_THUMBNAILS") != ""
|
|
}
|
|
state.ParseQuery(r.URL.RawQuery)
|
|
if state.Sort == "" {
|
|
state.Sort = getenv("SORT", "Hot")
|
|
}
|
|
if state.CommentSort == "" {
|
|
state.CommentSort = getenv("COMMENT_SORT", "Hot")
|
|
}
|
|
if state.Listing == "" || state.Session == nil && state.Listing == "Subscribed" {
|
|
state.Listing = getenv("LISTING", "All")
|
|
}
|
|
if linksInNewWindow := getCookie(r, "LinksInNewWindow"); linksInNewWindow != "" {
|
|
state.LinksInNewWindow = linksInNewWindow != "0"
|
|
} else {
|
|
state.LinksInNewWindow = os.Getenv("LINKS_IN_NEW_WINDOW") != ""
|
|
}
|
|
return state, nil
|
|
}
|
|
func GetTemplate(name string) (*template.Template, error) {
|
|
if *watch {
|
|
t := template.New(name).Funcs(funcMap)
|
|
glob, err := t.ParseGlob("templates/*")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return glob, nil
|
|
}
|
|
t, ok := templates[name]
|
|
if !ok {
|
|
return nil, errors.New("template not found: " + name)
|
|
}
|
|
return t, nil
|
|
}
|
|
func Render(w http.ResponseWriter, templateName string, state State) {
|
|
tmpl, err := GetTemplate(templateName)
|
|
if err != nil {
|
|
w.Write([]byte("500 - Server Error"))
|
|
fmt.Println(err)
|
|
return
|
|
}
|
|
if len(state.TopCommunities) == 0 {
|
|
state.GetCommunities()
|
|
}
|
|
if state.Session != nil {
|
|
state.GetUnreadCount()
|
|
}
|
|
if state.Status != http.StatusOK {
|
|
w.WriteHeader(state.Status)
|
|
}
|
|
header := w.Header()
|
|
header.Set("Content-Security-Policy", "script-src 'self'")
|
|
err = tmpl.Execute(w, state)
|
|
if err != nil {
|
|
fmt.Println("execute fail", err)
|
|
w.Write([]byte("500 - Server Error"))
|
|
return
|
|
}
|
|
}
|
|
func GetRoot(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
|
|
data := make(map[string]any)
|
|
data["Title"] = r.Host
|
|
if dark := getCookie(r, "Dark"); dark != "" {
|
|
data["Dark"] = dark != "0"
|
|
} else {
|
|
data["Dark"] = os.Getenv("DARK") != ""
|
|
}
|
|
tmpl, err := GetTemplate("root.html")
|
|
if err != nil {
|
|
fmt.Println("execute fail", err)
|
|
w.Write([]byte("500 - Server Error"))
|
|
return
|
|
}
|
|
tmpl.Execute(w, data)
|
|
}
|
|
|
|
type NodeSoftware struct {
|
|
Name string `json:"name"`
|
|
Version string `json:"version"`
|
|
}
|
|
type NodeInfo struct {
|
|
Software NodeSoftware `json:"software"`
|
|
}
|
|
|
|
func IsLemmy(domain string, remoteAddr string) bool {
|
|
client := http.Client{Transport: NewAddHeaderTransport(remoteAddr)}
|
|
var nodeInfo NodeInfo
|
|
res, err := client.Get("https://" + domain + "/nodeinfo/2.0.json")
|
|
if err != nil {
|
|
return false
|
|
}
|
|
err = json.NewDecoder(res.Body).Decode(&nodeInfo)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
if nodeInfo.Software.Name == "lemmy" {
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func PostRoot(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
|
|
data := make(map[string]any)
|
|
tmpl, err := GetTemplate("root.html")
|
|
if err != nil {
|
|
fmt.Println("execute fail", err)
|
|
w.Write([]byte("500 - Server Error"))
|
|
return
|
|
}
|
|
input := r.FormValue("destination")
|
|
re := regexp.MustCompile(`^(([a-zA-Z]{1})|([a-zA-Z]{1}[a-zA-Z]{1})|([a-zA-Z]{1}[0-9]{1})|([0-9]{1}[a-zA-Z]{1})|([a-zA-Z0-9][a-zA-Z0-9-_]{1,61}[a-zA-Z0-9]))\.([a-zA-Z]{2,6}|[a-zA-Z0-9-]{2,30}\.[a-zA-Z
|
|
]{2,3})`)
|
|
if re.MatchString(input) {
|
|
input = "https://" + input
|
|
}
|
|
dest, err := url.Parse(input)
|
|
if dest.Host != "" {
|
|
state, _ := Initialize(dest.Host, r)
|
|
if err := state.LemmyError(dest.Host); err != nil {
|
|
data["Error"] = err
|
|
} else {
|
|
redirectUrl := "/" + dest.Host + dest.Path
|
|
if dest.RawQuery != "" {
|
|
redirectUrl = redirectUrl + "?" + dest.RawQuery
|
|
}
|
|
http.Redirect(w, r, redirectUrl, 302)
|
|
return
|
|
}
|
|
} else {
|
|
data["Error"] = "Invalid destination"
|
|
}
|
|
tmpl.Execute(w, data)
|
|
}
|
|
func GetIcon(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
|
|
if ps.ByName("host") == "favicon.ico" {
|
|
w.WriteHeader(http.StatusNotFound)
|
|
w.Write([]byte("404 - Not Found"))
|
|
}
|
|
state, err := Initialize(ps.ByName("host"), r)
|
|
state.Client.Token = ""
|
|
resp, err := state.Client.Site(context.Background())
|
|
if err != nil {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
w.Write([]byte("500 - Server Error"))
|
|
return
|
|
}
|
|
if !resp.SiteView.Site.Icon.IsValid() {
|
|
w.WriteHeader(http.StatusNotFound)
|
|
w.Write([]byte("404 - Not Found"))
|
|
return
|
|
}
|
|
iresp, err := state.HTTPClient.Get(resp.SiteView.Site.Icon.String())
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
w.Write([]byte("500 - Server Error"))
|
|
return
|
|
}
|
|
defer iresp.Body.Close()
|
|
w.Header().Set("Content-Type", "image/jpeg")
|
|
w.Header().Set("Cache-Control", "max-age=2592000")
|
|
io.Copy(w, iresp.Body)
|
|
return
|
|
|
|
}
|
|
func GetFrontpage(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
|
|
state, err := Initialize(ps.ByName("host"), r)
|
|
if err != nil {
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
m, _ := url.ParseQuery(r.URL.RawQuery)
|
|
if len(m["edit"]) > 0 {
|
|
state.Op = "edit_community"
|
|
}
|
|
if ps.ByName("community") == "" || state.Op == "edit_community" {
|
|
state.GetSite()
|
|
}
|
|
state.GetCommunity(ps.ByName("community"))
|
|
if state.Op == "" {
|
|
state.GetPosts()
|
|
}
|
|
if state.XHR {
|
|
Render(w, "xhr.html", state)
|
|
} else {
|
|
Render(w, "frontpage.html", state)
|
|
}
|
|
}
|
|
|
|
func GetCommunities(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
|
|
r.URL.Path = "/search"
|
|
if ps.ByName("host") != "" {
|
|
r.URL.Path = "/" + ps.ByName("host") + "/search"
|
|
}
|
|
r.URL.RawQuery = "searchtype=Communities&sort=TopMonth"
|
|
http.Redirect(w, r, r.URL.String(), 301)
|
|
}
|
|
|
|
func ResolveId(r *http.Request, class string, id string, host string) string {
|
|
remoteAddr := r.RemoteAddr
|
|
if r.Header.Get("CF-Connecting-IP") != "" {
|
|
remoteAddr = r.Header.Get("CF-Connecting-IP")
|
|
}
|
|
client := http.Client{Transport: NewAddHeaderTransport(remoteAddr)}
|
|
c, err := lemmy.NewWithClient("https://"+host, &client)
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
idn, _ := strconv.ParseInt(id, 10, 64)
|
|
if class == "post" {
|
|
resp, err := c.Post(context.Background(), lemmy.GetPost{
|
|
ID: lemmy.NewOptional(idn),
|
|
})
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
return resp.PostView.Post.ApID
|
|
}
|
|
resp, err := c.Comment(context.Background(), lemmy.GetComment{
|
|
ID: idn,
|
|
})
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
return resp.CommentView.Comment.ApID
|
|
}
|
|
|
|
func GetPost(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
|
|
state, err := Initialize(ps.ByName("host"), r)
|
|
if err != nil {
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
// redirect /post/remote_id@instance to /post/local_id
|
|
if path := strings.Split(ps.ByName("postid"), "@"); len(path) > 1 {
|
|
apid := ResolveId(r, "post", path[0], path[1])
|
|
if apid != "" {
|
|
resp, err := state.Client.ResolveObject(context.Background(), lemmy.ResolveObject{
|
|
Q: apid,
|
|
})
|
|
if err != nil {
|
|
dest := apid
|
|
if os.Getenv("LEMMY_DOMAIN") == "" {
|
|
dest = RegReplace(dest, `https:\/\/([a-zA-Z0-9\.\-]+\/post\/\d+)`, `/$1`)
|
|
}
|
|
http.Redirect(w, r, dest, 302)
|
|
return
|
|
}
|
|
post, _ := resp.Post.Value()
|
|
if post.Post.ID > 0 {
|
|
dest := RegReplace(r.URL.String(), `(([a-zA-Z0-9\.\-]+)?/post/)([a-zA-Z0-9\-\.@]+)`, `$1`)
|
|
dest += strconv.FormatInt(post.Post.ID, 10)
|
|
http.Redirect(w, r, dest, 302)
|
|
return
|
|
} else {
|
|
http.Redirect(w, r, apid, 302)
|
|
return
|
|
}
|
|
}
|
|
}
|
|
m, _ := url.ParseQuery(r.URL.RawQuery)
|
|
if len(m["edit"]) > 0 {
|
|
state.Op = "edit_post"
|
|
state.GetSite()
|
|
}
|
|
if len(m["content"]) > 0 {
|
|
state.Content = m["content"][0]
|
|
}
|
|
postid, _ := strconv.ParseInt(ps.ByName("postid"), 10, 64)
|
|
state.GetPost(postid)
|
|
if ps.ByName("op") == "block" {
|
|
state.Op = "block"
|
|
Render(w, "block.html", state)
|
|
return
|
|
}
|
|
state.GetComments()
|
|
Render(w, "index.html", state)
|
|
}
|
|
func GetComment(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
|
|
state, err := Initialize(ps.ByName("host"), r)
|
|
if err != nil {
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
if path := strings.Split(ps.ByName("commentid"), "@"); len(path) > 1 {
|
|
apid := ResolveId(r, "comment", path[0], path[1])
|
|
if apid != "" {
|
|
resp, err := state.Client.ResolveObject(context.Background(), lemmy.ResolveObject{
|
|
Q: apid,
|
|
})
|
|
if err != nil {
|
|
dest := apid
|
|
if os.Getenv("LEMMY_DOMAIN") == "" {
|
|
dest = RegReplace(dest, `https:\/\/([a-zA-Z0-9\.\-]+\/comment\/\d+)`, `/$1`)
|
|
}
|
|
http.Redirect(w, r, dest, 302)
|
|
return
|
|
}
|
|
comment, _ := resp.Comment.Value()
|
|
if comment.Comment.ID > 0 {
|
|
dest := RegReplace(r.URL.String(), `(([a-zA-Z0-9\.\-]+)?/comment/)([a-zA-Z0-9\-\.@]+)`, `$1`)
|
|
dest += strconv.FormatInt(comment.Comment.ID, 10)
|
|
http.Redirect(w, r, dest, 302)
|
|
return
|
|
} else {
|
|
http.Redirect(w, r, apid, 302)
|
|
return
|
|
}
|
|
}
|
|
}
|
|
m, _ := url.ParseQuery(r.URL.RawQuery)
|
|
if len(m["reply"]) > 0 {
|
|
state.Op = "reply"
|
|
}
|
|
if len(m["edit"]) > 0 {
|
|
state.Op = "edit"
|
|
}
|
|
if r.Method == "POST" && len(m["content"]) > 0 {
|
|
state.Content = m["content"][0]
|
|
}
|
|
if len(m["source"]) > 0 {
|
|
state.Op = "source"
|
|
}
|
|
if len(m["context"]) > 0 {
|
|
ctx, _ := strconv.Atoi(m["context"][0])
|
|
state.Context = ctx
|
|
}
|
|
commentid, _ := strconv.ParseInt(ps.ByName("commentid"), 10, 64)
|
|
state.GetComment(commentid)
|
|
if state.XHR && len(m["content"]) > 0 {
|
|
Render(w, "create_comment.html", state)
|
|
return
|
|
}
|
|
state.GetPost(state.PostID)
|
|
Render(w, "index.html", state)
|
|
}
|
|
func GetUser(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
|
|
state, err := Initialize(ps.ByName("host"), r)
|
|
if err != nil {
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
state.GetUser(ps.ByName("username"))
|
|
if state.Site == nil {
|
|
state.GetSite()
|
|
}
|
|
Render(w, "index.html", state)
|
|
}
|
|
func GetMessageForm(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
|
|
state, err := Initialize(ps.ByName("host"), r)
|
|
if err != nil {
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
state.Op = "send_message"
|
|
state.GetUser(ps.ByName("username"))
|
|
Render(w, "index.html", state)
|
|
}
|
|
func SendMessage(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
|
|
state, err := Initialize(ps.ByName("host"), r)
|
|
if err != nil {
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
userid, _ := strconv.ParseInt(r.FormValue("userid"), 10, 64)
|
|
_, err = state.Client.CreatePrivateMessage(context.Background(), lemmy.CreatePrivateMessage{
|
|
Content: r.FormValue("content"),
|
|
RecipientID: userid,
|
|
})
|
|
if err != nil {
|
|
state.Error = err
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
r.URL.Path = "/" + state.Host + "/inbox"
|
|
http.Redirect(w, r, r.URL.String(), 301)
|
|
}
|
|
func GetCreatePost(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
|
|
state, err := Initialize(ps.ByName("host"), r)
|
|
if err != nil {
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
m, _ := url.ParseQuery(r.URL.RawQuery)
|
|
|
|
if len(m["url"]) > 0 {
|
|
state.SubmitURL = m["url"][0]
|
|
}
|
|
if len(m["title"]) > 0 {
|
|
state.SubmitTitle = m["title"][0]
|
|
}
|
|
if len(m["body"]) > 0 {
|
|
state.SubmitBody = m["body"][0]
|
|
}
|
|
state.GetSite()
|
|
state.GetCommunity("")
|
|
state.Op = "create_post"
|
|
Render(w, "index.html", state)
|
|
}
|
|
func GetCreateCommunity(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
|
|
state, err := Initialize(ps.ByName("host"), r)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
state.GetSite()
|
|
state.Op = "create_community"
|
|
if ps.ByName("community") != "" {
|
|
state.GetCommunity(ps.ByName("community"))
|
|
state.Op = "edit_community"
|
|
}
|
|
Render(w, "index.html", state)
|
|
}
|
|
|
|
func Inbox(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
|
|
state, err := Initialize(ps.ByName("host"), r)
|
|
if err != nil {
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
state.GetMessages()
|
|
Render(w, "index.html", state)
|
|
state.MarkAllAsRead()
|
|
}
|
|
func getCookie(r *http.Request, name string) string {
|
|
cookie, err := r.Cookie(name)
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
return cookie.Value
|
|
}
|
|
func setCookie(w http.ResponseWriter, host string, name string, value string) {
|
|
if host == "." {
|
|
host = ""
|
|
}
|
|
cookie := http.Cookie{
|
|
Name: name,
|
|
Value: value,
|
|
MaxAge: 86400 * 30,
|
|
HttpOnly: true,
|
|
SameSite: http.SameSiteNoneMode,
|
|
Secure: true,
|
|
Path: "/" + host,
|
|
}
|
|
http.SetCookie(w, &cookie)
|
|
}
|
|
func deleteCookie(w http.ResponseWriter, host string, name string) {
|
|
if host == "." {
|
|
host = ""
|
|
}
|
|
cookie := http.Cookie{
|
|
Name: name,
|
|
Path: "/" + host,
|
|
MaxAge: -1,
|
|
}
|
|
http.SetCookie(w, &cookie)
|
|
}
|
|
func Settings(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
|
|
state, err := Initialize(ps.ByName("host"), r)
|
|
if err != nil {
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
state.GetSite()
|
|
switch r.Method {
|
|
case "POST":
|
|
for _, name := range []string{"DefaultSortType", "DefaultListingType", "DefaultCommentSortType"} {
|
|
deleteCookie(w, state.Host, name)
|
|
setCookie(w, "", name, r.FormValue(name))
|
|
}
|
|
if r.FormValue("darkmode") != "" {
|
|
setCookie(w, "", "Dark", "1")
|
|
state.Dark = true
|
|
} else {
|
|
setCookie(w, "", "Dark", "0")
|
|
state.Dark = false
|
|
}
|
|
if r.FormValue("shownsfw") != "" {
|
|
setCookie(w, "", "ShowNSFW", "1")
|
|
state.ShowNSFW = true
|
|
} else {
|
|
deleteCookie(w, state.Host, "ShowNSFW")
|
|
deleteCookie(w, "", "ShowNSFW")
|
|
state.ShowNSFW = false
|
|
}
|
|
if r.FormValue("hideInstanceNames") != "" {
|
|
setCookie(w, "", "HideInstanceNames", "1")
|
|
state.HideInstanceNames = true
|
|
} else {
|
|
deleteCookie(w, "", "HideInstanceNames")
|
|
state.HideInstanceNames = false
|
|
}
|
|
if r.FormValue("hideThumbnails") != "" {
|
|
setCookie(w, "", "HideThumbnails", "1")
|
|
state.HideInstanceNames = true
|
|
} else {
|
|
setCookie(w, "", "HideThumbnails", "0")
|
|
state.HideInstanceNames = false
|
|
}
|
|
if r.FormValue("linksInNewWindow") != "" {
|
|
setCookie(w, "", "LinksInNewWindow", "1")
|
|
state.LinksInNewWindow = true
|
|
} else {
|
|
setCookie(w, "", "LinksInNewWindow", "0")
|
|
state.LinksInNewWindow = false
|
|
}
|
|
state.Listing = r.FormValue("DefaultListingType")
|
|
state.Sort = r.FormValue("DefaultSortType")
|
|
state.CommentSort = r.FormValue("DefaultCommentSortType")
|
|
// TODO save user settings
|
|
case "GET":
|
|
if state.Session != nil {
|
|
// TODO fetch user settings
|
|
}
|
|
}
|
|
Render(w, "settings.html", state)
|
|
}
|
|
|
|
func SignUpOrLogin(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
|
|
state, err := Initialize(ps.ByName("host"), r)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
var token string
|
|
var username string
|
|
switch r.FormValue("submit") {
|
|
case "log in":
|
|
login := lemmy.Login{
|
|
UsernameOrEmail: r.FormValue("username"),
|
|
Password: r.FormValue("password"),
|
|
}
|
|
if r.FormValue("totp") != "" {
|
|
login.TOTP2FAToken = lemmy.NewOptional(r.FormValue("totp"))
|
|
}
|
|
resp, err := state.Client.Login(context.Background(), login)
|
|
if err != nil {
|
|
if strings.Contains(fmt.Sprintf("%v", err), "missing_totp_token") {
|
|
state.Op = "2fa"
|
|
}
|
|
fmt.Println(err)
|
|
state.Error = err
|
|
state.GetSite()
|
|
state.GetCaptcha()
|
|
Render(w, "login.html", state)
|
|
return
|
|
}
|
|
if resp.JWT.IsValid() {
|
|
token = resp.JWT.String()
|
|
username = r.FormValue("username")
|
|
deleteCookie(w, state.Host, "ShowNSFW")
|
|
}
|
|
case "sign up":
|
|
register := lemmy.Register{
|
|
Username: r.FormValue("username"),
|
|
Password: r.FormValue("password"),
|
|
PasswordVerify: r.FormValue("passwordverify"),
|
|
ShowNSFW: r.FormValue("nsfw") != "",
|
|
}
|
|
if r.FormValue("email") != "" {
|
|
register.Email = lemmy.NewOptional(r.FormValue("email"))
|
|
}
|
|
if r.FormValue("answer") != "" {
|
|
register.Answer = lemmy.NewOptional(r.FormValue("answer"))
|
|
}
|
|
if r.FormValue("captchauuid") != "" {
|
|
register.CaptchaUUID = lemmy.NewOptional(r.FormValue("captchauuid"))
|
|
}
|
|
if r.FormValue("captchaanswer") != "" {
|
|
register.CaptchaAnswer = lemmy.NewOptional(r.FormValue("captchaanswer"))
|
|
}
|
|
resp, err := state.Client.Register(context.Background(), register)
|
|
if err != nil {
|
|
state.Error = err
|
|
state.GetSite()
|
|
state.GetCaptcha()
|
|
Render(w, "login.html", state)
|
|
return
|
|
}
|
|
if resp.JWT.IsValid() {
|
|
username = r.FormValue("username")
|
|
token = resp.JWT.String()
|
|
} else {
|
|
var alert string
|
|
if resp.RegistrationCreated {
|
|
alert = "Registration application submitted. "
|
|
}
|
|
if resp.VerifyEmailSent {
|
|
alert = alert + "Email verification sent. "
|
|
}
|
|
q := r.URL.Query()
|
|
q.Add("alert", alert)
|
|
r.URL.RawQuery = q.Encode()
|
|
http.Redirect(w, r, r.URL.String(), 301)
|
|
return
|
|
}
|
|
}
|
|
if token != "" {
|
|
state.GetUser(username)
|
|
if state.User == nil {
|
|
return
|
|
}
|
|
setCookie(w, state.Host, "jwt", token)
|
|
userid := strconv.FormatInt(state.User.PersonView.Person.ID, 10)
|
|
setCookie(w, state.Host, "user", state.User.PersonView.Person.Name+":"+userid)
|
|
setCookie(w, state.Host, "jwt", token)
|
|
r.URL.Path = "/" + state.Host
|
|
http.Redirect(w, r, r.URL.String(), 301)
|
|
return
|
|
}
|
|
}
|
|
func GetLogin(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
|
|
state, err := Initialize(ps.ByName("host"), r)
|
|
if err != nil {
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
state.GetSite()
|
|
if state.Site != nil && state.Site.SiteView.LocalSite.CaptchaEnabled {
|
|
state.GetCaptcha()
|
|
}
|
|
m, _ := url.ParseQuery(r.URL.RawQuery)
|
|
if len(m["alert"]) > 0 {
|
|
state.Alert = m["alert"][0]
|
|
}
|
|
Render(w, "login.html", state)
|
|
}
|
|
func Search(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
|
|
state, err := Initialize(ps.ByName("host"), r)
|
|
if err != nil {
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
if state.CommunityName != "" {
|
|
state.GetCommunity(state.CommunityName)
|
|
}
|
|
if state.UserName != "" {
|
|
state.GetUser(state.UserName)
|
|
} else if state.Community == nil {
|
|
state.GetSite()
|
|
}
|
|
m, _ := url.ParseQuery(r.URL.RawQuery)
|
|
state.SearchType = "Posts"
|
|
if len(m["searchtype"]) > 0 {
|
|
switch m["searchtype"][0] {
|
|
case "Comments":
|
|
state.SearchType = "Comments"
|
|
case "Communities":
|
|
state.SearchType = "Communities"
|
|
}
|
|
}
|
|
state.Search(state.SearchType)
|
|
Render(w, "index.html", state)
|
|
}
|
|
|
|
type PictrsFile struct {
|
|
Filename string `json:"file"`
|
|
DeleteToken string `json:"delete_token"`
|
|
}
|
|
|
|
type PictrsResponse struct {
|
|
Message string `json:"msg"`
|
|
Files []PictrsFile `json:"files"`
|
|
}
|
|
|
|
func UserOp(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
|
|
state, err := Initialize(ps.ByName("host"), r)
|
|
if err != nil {
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
switch r.FormValue("op") {
|
|
case "leave":
|
|
communityid, _ := strconv.ParseInt(r.FormValue("communityid"), 10, 64)
|
|
state.Client.FollowCommunity(context.Background(), lemmy.FollowCommunity{
|
|
CommunityID: communityid,
|
|
Follow: false,
|
|
})
|
|
case "join":
|
|
communityid, _ := strconv.ParseInt(r.FormValue("communityid"), 10, 64)
|
|
state.Client.FollowCommunity(context.Background(), lemmy.FollowCommunity{
|
|
CommunityID: communityid,
|
|
Follow: true,
|
|
})
|
|
case "block":
|
|
communityid, _ := strconv.ParseInt(r.FormValue("communityid"), 10, 64)
|
|
state.Client.BlockCommunity(context.Background(), lemmy.BlockCommunity{
|
|
CommunityID: communityid,
|
|
Block: true,
|
|
})
|
|
case "unblock":
|
|
communityid, _ := strconv.ParseInt(r.FormValue("communityid"), 10, 64)
|
|
state.Client.BlockCommunity(context.Background(), lemmy.BlockCommunity{
|
|
CommunityID: communityid,
|
|
Block: false,
|
|
})
|
|
case "block_user":
|
|
personId, _ := strconv.ParseInt(r.FormValue("user_id"), 10, 64)
|
|
if personId == 0 {
|
|
state.GetUser(ps.ByName("username"))
|
|
personId = state.User.PersonView.Person.ID
|
|
}
|
|
state.Client.BlockPerson(context.Background(), lemmy.BlockPerson{
|
|
PersonID: personId,
|
|
Block: r.FormValue("submit") == "block",
|
|
})
|
|
if r.FormValue("xhr") == "1" {
|
|
w.Write([]byte{})
|
|
return
|
|
}
|
|
case "logout":
|
|
deleteCookie(w, state.Host, "jwt")
|
|
deleteCookie(w, state.Host, "user")
|
|
case "login":
|
|
login := lemmy.Login{
|
|
UsernameOrEmail: r.FormValue("username"),
|
|
Password: r.FormValue("password"),
|
|
}
|
|
if r.FormValue("totp") != "" {
|
|
login.TOTP2FAToken = lemmy.NewOptional(r.FormValue("totp"))
|
|
}
|
|
resp, err := state.Client.Login(context.Background(), login)
|
|
if err != nil {
|
|
if strings.Contains(fmt.Sprintf("%v", err), "missing_totp_token") {
|
|
state.Op = "2fa"
|
|
}
|
|
state.GetSite()
|
|
if state.Site != nil && state.Site.SiteView.LocalSite.CaptchaEnabled {
|
|
state.GetCaptcha()
|
|
}
|
|
state.Status = http.StatusUnauthorized
|
|
state.Error = err
|
|
Render(w, "login.html", state)
|
|
return
|
|
} else if resp.JWT.IsValid() {
|
|
state.GetUser(r.FormValue("username"))
|
|
if state.User != nil {
|
|
setCookie(w, state.Host, "jwt", resp.JWT.String())
|
|
userid := strconv.FormatInt(state.User.PersonView.Person.ID, 10)
|
|
setCookie(w, state.Host, "user", state.User.PersonView.Person.Name+":"+userid)
|
|
}
|
|
}
|
|
case "create_community":
|
|
state.GetSite()
|
|
community := lemmy.CreateCommunity{
|
|
Name: r.FormValue("name"),
|
|
Title: r.FormValue("title"),
|
|
}
|
|
if r.FormValue("description") != "" {
|
|
community.Description = lemmy.NewOptional(r.FormValue("description"))
|
|
}
|
|
if file, handler, err := r.FormFile("icon"); err == nil {
|
|
pres, err := state.UploadImage(file, handler)
|
|
if err != nil {
|
|
state.Error = err
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
community.Icon = lemmy.NewOptional("https://" + state.Host + "/pictrs/image/" + pres.Files[0].Filename)
|
|
}
|
|
if file, handler, err := r.FormFile("banner"); err == nil {
|
|
pres, err := state.UploadImage(file, handler)
|
|
if err != nil {
|
|
state.Error = err
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
community.Banner = lemmy.NewOptional("https://" + state.Host + "/pictrs/image/" + pres.Files[0].Filename)
|
|
}
|
|
resp, err := state.Client.CreateCommunity(context.Background(), community)
|
|
if err == nil {
|
|
r.URL.Path = "/" + state.Host + "/c/" + resp.CommunityView.Community.Name
|
|
} else {
|
|
fmt.Println(err)
|
|
}
|
|
case "edit_community":
|
|
state.CommunityName = ps.ByName("community")
|
|
state.GetCommunity("")
|
|
if state.Community == nil {
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
state.GetSite()
|
|
community := lemmy.EditCommunity{
|
|
CommunityID: state.Community.CommunityView.Community.ID,
|
|
}
|
|
if r.FormValue("title") != "" {
|
|
community.Title = lemmy.NewOptional(r.FormValue("title"))
|
|
}
|
|
if r.FormValue("description") != "" {
|
|
community.Description = lemmy.NewOptional(r.FormValue("description"))
|
|
}
|
|
if file, handler, err := r.FormFile("icon"); err == nil {
|
|
pres, err := state.UploadImage(file, handler)
|
|
if err != nil {
|
|
state.Error = err
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
community.Icon = lemmy.NewOptional("https://" + state.Host + "/pictrs/image/" + pres.Files[0].Filename)
|
|
}
|
|
if file, handler, err := r.FormFile("banner"); err == nil {
|
|
pres, err := state.UploadImage(file, handler)
|
|
if err != nil {
|
|
state.Error = err
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
community.Banner = lemmy.NewOptional("https://" + state.Host + "/pictrs/image/" + pres.Files[0].Filename)
|
|
}
|
|
resp, err := state.Client.EditCommunity(context.Background(), community)
|
|
if err == nil {
|
|
r.URL.Path = "/" + state.Host + "/c/" + resp.CommunityView.Community.Name
|
|
} else {
|
|
fmt.Println(err)
|
|
}
|
|
case "create_post":
|
|
if state.CommunityName == "" {
|
|
state.CommunityName = r.FormValue("communityname")
|
|
}
|
|
state.GetCommunity(state.CommunityName)
|
|
state.GetSite()
|
|
if state.Community == nil {
|
|
state.Status = http.StatusBadRequest
|
|
state.Op = "create_post"
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
post := lemmy.CreatePost{
|
|
Name: r.FormValue("name"),
|
|
CommunityID: state.Community.CommunityView.Community.ID,
|
|
}
|
|
if r.FormValue("url") != "" {
|
|
post.URL = lemmy.NewOptional(r.FormValue("url"))
|
|
}
|
|
file, handler, err := r.FormFile("file")
|
|
if err == nil {
|
|
pres, err := state.UploadImage(file, handler)
|
|
if err != nil {
|
|
state.Error = err
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
post.URL = lemmy.NewOptional("https://" + state.Host + "/pictrs/image/" + pres.Files[0].Filename)
|
|
}
|
|
if r.FormValue("body") != "" {
|
|
post.Body = lemmy.NewOptional(r.FormValue("body"))
|
|
}
|
|
if r.FormValue("language") != "" {
|
|
languageid, _ := strconv.ParseInt(r.FormValue("language"), 10, 64)
|
|
post.LanguageID = lemmy.NewOptional(languageid)
|
|
}
|
|
resp, err := state.Client.CreatePost(context.Background(), post)
|
|
if err == nil {
|
|
postid := strconv.FormatInt(resp.PostView.Post.ID, 10)
|
|
r.URL.Path = "/" + state.Host + "/post/" + postid
|
|
} else {
|
|
fmt.Println(err)
|
|
}
|
|
case "edit_post":
|
|
r.ParseMultipartForm(10 << 20)
|
|
state.GetSite()
|
|
postid, _ := strconv.ParseInt(ps.ByName("postid"), 10, 64)
|
|
post := lemmy.EditPost{
|
|
PostID: postid,
|
|
Body: lemmy.NewOptional(r.FormValue("body")),
|
|
Name: lemmy.NewOptional(r.FormValue("name")),
|
|
URL: lemmy.NewOptional(r.FormValue("url")),
|
|
}
|
|
if r.FormValue("url") == "" {
|
|
post.URL = lemmy.Optional[string]{}
|
|
}
|
|
if r.FormValue("language") != "" {
|
|
languageid, _ := strconv.ParseInt(r.FormValue("language"), 10, 64)
|
|
post.LanguageID = lemmy.NewOptional(languageid)
|
|
}
|
|
file, handler, err := r.FormFile("file")
|
|
if err == nil {
|
|
pres, err := state.UploadImage(file, handler)
|
|
if err != nil {
|
|
state.Error = err
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
post.URL = lemmy.NewOptional("https://" + state.Host + "/pictrs/image/" + pres.Files[0].Filename)
|
|
}
|
|
|
|
resp, err := state.Client.EditPost(context.Background(), post)
|
|
if err == nil {
|
|
postid := strconv.FormatInt(resp.PostView.Post.ID, 10)
|
|
r.URL.Path = "/" + state.Host + "/post/" + postid
|
|
r.URL.RawQuery = ""
|
|
} else {
|
|
state.Status = http.StatusBadRequest
|
|
state.Error = err
|
|
fmt.Println(err)
|
|
}
|
|
case "save_post":
|
|
postid, _ := strconv.ParseInt(r.FormValue("postid"), 10, 64)
|
|
_, err := state.Client.SavePost(context.Background(), lemmy.SavePost{
|
|
PostID: postid,
|
|
Save: r.FormValue("submit") == "save",
|
|
})
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
if r.FormValue("xhr") != "" {
|
|
state.GetPost(postid)
|
|
state.PostID = 0
|
|
state.Op = "save_post"
|
|
state.XHR = true
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
case "save_comment":
|
|
commentid, _ := strconv.ParseInt(r.FormValue("commentid"), 10, 64)
|
|
_, err := state.Client.SaveComment(context.Background(), lemmy.SaveComment{
|
|
CommentID: commentid,
|
|
Save: r.FormValue("submit") == "save",
|
|
})
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
if r.FormValue("xhr") != "" {
|
|
state.XHR = true
|
|
state.GetComment(commentid)
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
case "delete_post":
|
|
postid, _ := strconv.ParseInt(r.FormValue("postid"), 10, 64)
|
|
post := lemmy.DeletePost{
|
|
PostID: postid,
|
|
Deleted: true,
|
|
}
|
|
if r.FormValue("undo") != "" {
|
|
post.Deleted = false
|
|
}
|
|
resp, err := state.Client.DeletePost(context.Background(), post)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
} else {
|
|
r.URL.Path = "/" + state.Host + "/c/" + resp.PostView.Community.Name
|
|
r.URL.RawQuery = ""
|
|
}
|
|
case "block_post":
|
|
postid, _ := strconv.ParseInt(r.FormValue("postid"), 10, 64)
|
|
state.GetPost(postid)
|
|
if r.FormValue("blockcommunity") != "" && len(state.Posts) > 0 {
|
|
state.Client.BlockCommunity(context.Background(), lemmy.BlockCommunity{
|
|
CommunityID: state.Posts[0].Post.CommunityID,
|
|
Block: true,
|
|
})
|
|
}
|
|
if r.FormValue("blockuser") != "" && len(state.Posts) > 0 {
|
|
state.Client.BlockPerson(context.Background(), lemmy.BlockPerson{
|
|
PersonID: state.Posts[0].Post.CreatorID,
|
|
Block: true,
|
|
})
|
|
}
|
|
case "read_post":
|
|
postid, _ := strconv.ParseInt(r.FormValue("postid"), 10, 64)
|
|
post := lemmy.MarkPostAsRead{
|
|
PostID: lemmy.NewOptional(postid),
|
|
Read: true,
|
|
}
|
|
if r.FormValue("submit") == "mark unread" {
|
|
post.Read = false
|
|
}
|
|
_, err := state.Client.MarkPostAsRead(context.Background(), post)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
} else if r.FormValue("xhr") != "" {
|
|
w.Write([]byte{})
|
|
return
|
|
}
|
|
case "vote_post":
|
|
var score int64
|
|
score = 1
|
|
if r.FormValue("vote") != "▲" {
|
|
score = -1
|
|
}
|
|
if r.FormValue("undo") == strconv.Itoa(int(score)) {
|
|
score = 0
|
|
}
|
|
postid, _ := strconv.ParseInt(r.FormValue("postid"), 10, 64)
|
|
post := lemmy.CreatePostLike{
|
|
PostID: postid,
|
|
Score: score,
|
|
}
|
|
state.Client.LikePost(context.Background(), post)
|
|
if r.FormValue("xhr") != "" {
|
|
state.GetPost(postid)
|
|
state.PostID = 0
|
|
state.Op = "vote_post"
|
|
state.XHR = true
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
case "vote_comment":
|
|
var score int64
|
|
score = 1
|
|
if r.FormValue("submit") != "▲" {
|
|
score = -1
|
|
}
|
|
if r.FormValue("undo") == strconv.Itoa(int(score)) {
|
|
score = 0
|
|
}
|
|
commentid, _ := strconv.ParseInt(r.FormValue("commentid"), 10, 64)
|
|
post := lemmy.CreateCommentLike{
|
|
CommentID: commentid,
|
|
Score: score,
|
|
}
|
|
state.Client.LikeComment(context.Background(), post)
|
|
if r.FormValue("xhr") != "" {
|
|
state.XHR = true
|
|
state.GetComment(commentid)
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
case "create_comment":
|
|
if ps.ByName("postid") != "" {
|
|
postid, _ := strconv.ParseInt(ps.ByName("postid"), 10, 64)
|
|
state.PostID = postid
|
|
}
|
|
if r.FormValue("parentid") != "" && r.FormValue("parentid") != "0" {
|
|
parentid, _ := strconv.ParseInt(r.FormValue("parentid"), 10, 64)
|
|
state.GetComment(parentid)
|
|
}
|
|
content := r.FormValue("content")
|
|
file, handler, err := r.FormFile("file")
|
|
if err == nil {
|
|
pres, err := state.UploadImage(file, handler)
|
|
if err != nil {
|
|
state.Error = err
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
content += ("![](https://" + state.Host + "/pictrs/image/" + pres.Files[0].Filename + ")")
|
|
}
|
|
if r.FormValue("submit") == "save" {
|
|
createComment := lemmy.CreateComment{
|
|
Content: content,
|
|
PostID: state.PostID,
|
|
}
|
|
if state.CommentID > 0 {
|
|
createComment.ParentID = lemmy.NewOptional(state.CommentID)
|
|
}
|
|
resp, err := state.Client.CreateComment(context.Background(), createComment)
|
|
if err == nil {
|
|
if r.FormValue("xhr") != "" {
|
|
state.XHR = true
|
|
state.Comments = nil
|
|
state.GetComment(resp.CommentView.Comment.ID)
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
postid := strconv.FormatInt(state.PostID, 10)
|
|
commentid := strconv.FormatInt(resp.CommentView.Comment.ID, 10)
|
|
r.URL.Path = "/" + state.Host + "/post/" + postid
|
|
r.URL.Fragment = "c" + commentid
|
|
} else {
|
|
fmt.Println(err)
|
|
}
|
|
} else if r.FormValue("submit") == "preview" {
|
|
q := r.URL.Query()
|
|
q.Set("content", content)
|
|
q.Set("reply", "")
|
|
if r.FormValue("xhr") != "" {
|
|
q.Set("xhr", "1")
|
|
}
|
|
r.URL.RawQuery = q.Encode()
|
|
if ps.ByName("postid") != "" {
|
|
GetPost(w, r, ps)
|
|
return
|
|
}
|
|
if ps.ByName("commentid") != "" {
|
|
GetComment(w, r, ps)
|
|
return
|
|
}
|
|
} else if r.FormValue("xhr") != "" {
|
|
w.Write([]byte{})
|
|
return
|
|
}
|
|
if r.FormValue("submit") == "cancel" {
|
|
r.URL.RawQuery = ""
|
|
}
|
|
case "edit_comment":
|
|
commentid, _ := strconv.ParseInt(r.FormValue("commentid"), 10, 64)
|
|
q := r.URL.Query()
|
|
content := r.FormValue("content")
|
|
file, handler, err := r.FormFile("file")
|
|
if err == nil {
|
|
pres, err := state.UploadImage(file, handler)
|
|
if err != nil {
|
|
state.Error = err
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
content += ("![](https://" + state.Host + "/pictrs/image/" + pres.Files[0].Filename + ")")
|
|
}
|
|
|
|
if r.FormValue("submit") == "save" {
|
|
resp, err := state.Client.EditComment(context.Background(), lemmy.EditComment{
|
|
CommentID: commentid,
|
|
Content: lemmy.NewOptional(content),
|
|
})
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
} else {
|
|
commentid := strconv.FormatInt(resp.CommentView.Comment.ID, 10)
|
|
r.URL.Fragment = "c" + commentid
|
|
r.URL.RawQuery = ""
|
|
}
|
|
} else if r.FormValue("submit") == "preview" {
|
|
q.Set("content", content)
|
|
q.Set("edit", "")
|
|
if r.FormValue("xhr") != "" {
|
|
q.Set("xhr", "1")
|
|
}
|
|
r.URL.RawQuery = q.Encode()
|
|
if ps.ByName("commentid") != "" {
|
|
GetComment(w, r, ps)
|
|
return
|
|
}
|
|
} else if r.FormValue("submit") == "cancel" {
|
|
if ps.ByName("commentid") != "" {
|
|
if r.FormValue("xhr") != "" {
|
|
q.Set("xhr", "1")
|
|
}
|
|
r.URL.RawQuery = q.Encode()
|
|
GetComment(w, r, ps)
|
|
return
|
|
}
|
|
} else if r.FormValue("xhr") != "" {
|
|
w.Write([]byte{})
|
|
return
|
|
}
|
|
if r.FormValue("xhr") != "" {
|
|
state.XHR = true
|
|
state.GetComment(commentid)
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
if r.FormValue("submit") == "cancel" {
|
|
r.URL.RawQuery = ""
|
|
}
|
|
case "delete_comment":
|
|
commentid, _ := strconv.ParseInt(r.FormValue("commentid"), 10, 64)
|
|
resp, err := state.Client.DeleteComment(context.Background(), lemmy.DeleteComment{
|
|
CommentID: commentid,
|
|
Deleted: true,
|
|
})
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
} else {
|
|
if r.FormValue("xhr") != "" {
|
|
state.XHR = true
|
|
state.Comments = append(state.Comments, Comment{P: resp.CommentView, State: &state})
|
|
state.CommentID = commentid
|
|
Render(w, "index.html", state)
|
|
return
|
|
}
|
|
commentid := strconv.FormatInt(resp.CommentView.Comment.ID, 10)
|
|
r.URL.Fragment = "c" + commentid
|
|
r.URL.RawQuery = ""
|
|
}
|
|
case "shownsfw":
|
|
if r.FormValue("submit") == "continue" {
|
|
setCookie(w, "", "ShowNSFW", "1")
|
|
} else {
|
|
r.URL.Path = "/" + state.Host
|
|
}
|
|
}
|
|
http.Redirect(w, r, r.URL.String(), 301)
|
|
}
|
|
func GetLink(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
|
|
var dest *url.URL
|
|
m, _ := url.ParseQuery(r.URL.RawQuery)
|
|
if len(m["url"]) > 0 {
|
|
dest, _ = url.Parse(m["url"][0])
|
|
}
|
|
if dest.Host == r.Host || !IsLemmy(dest.Host, RemoteAddr(r)) {
|
|
http.Redirect(w, r, dest.String(), 302)
|
|
return
|
|
}
|
|
if host := ps.ByName("host"); host != "" {
|
|
redirect := "/" + host + dest.Path
|
|
if host != dest.Host && !strings.Contains(redirect, "@") {
|
|
redirect += ("@" + dest.Host)
|
|
}
|
|
http.Redirect(w, r, redirect, 302)
|
|
return
|
|
}
|
|
if host := os.Getenv("LEMMY_DOMAIN"); host != "" {
|
|
redirect := dest.Path
|
|
if host != dest.Host && !strings.Contains(redirect, "@") {
|
|
redirect += ("@" + dest.Host)
|
|
}
|
|
http.Redirect(w, r, redirect, 302)
|
|
return
|
|
}
|
|
}
|
|
func GetRouter() *httprouter.Router {
|
|
host := os.Getenv("LEMMY_DOMAIN")
|
|
router := httprouter.New()
|
|
if host == "" {
|
|
router.ServeFiles("/:host/static/*filepath", http.Dir("public"))
|
|
router.GET("/", GetRoot)
|
|
router.POST("/", PostRoot)
|
|
router.GET("/:host/", middleware(GetFrontpage))
|
|
router.GET("/:host/search", middleware(Search))
|
|
router.POST("/:host/search", middleware(UserOp))
|
|
router.GET("/:host/inbox", middleware(Inbox))
|
|
router.POST("/:host/inbox", middleware(UserOp))
|
|
router.GET("/:host/login", middleware(GetLogin))
|
|
router.POST("/:host/login", middleware(SignUpOrLogin))
|
|
router.GET("/:host/settings", middleware(Settings))
|
|
router.POST("/:host/settings", middleware(Settings))
|
|
router.POST("/:host/", middleware(UserOp))
|
|
router.GET("/:host/icon.jpg", middleware(GetIcon))
|
|
router.GET("/:host/c/:community", middleware(GetFrontpage))
|
|
router.POST("/:host/c/:community", middleware(UserOp))
|
|
router.GET("/:host/c/:community/search", middleware(Search))
|
|
router.GET("/:host/c/:community/edit", middleware(GetCreateCommunity))
|
|
router.GET("/:host/post/:postid", middleware(GetPost))
|
|
router.POST("/:host/post/:postid", middleware(UserOp))
|
|
router.GET("/:host/post/:postid/:op", middleware(GetPost))
|
|
router.GET("/:host/comment/:commentid", middleware(GetComment))
|
|
router.GET("/:host/comment/:commentid/:op", middleware(GetComment))
|
|
router.POST("/:host/comment/:commentid", middleware(UserOp))
|
|
router.GET("/:host/u/:username", middleware(GetUser))
|
|
router.GET("/:host/u/:username/message", middleware(GetMessageForm))
|
|
router.POST("/:host/u/:username/message", middleware(SendMessage))
|
|
router.POST("/:host/u/:username", middleware(UserOp))
|
|
router.GET("/:host/u/:username/search", middleware(Search))
|
|
router.GET("/:host/create_post", middleware(GetCreatePost))
|
|
router.POST("/:host/create_post", middleware(UserOp))
|
|
router.GET("/:host/create_community", middleware(GetCreateCommunity))
|
|
router.POST("/:host/create_community", middleware(UserOp))
|
|
router.GET("/:host/communities", middleware(GetCommunities))
|
|
router.GET("/:host/link", middleware(GetLink))
|
|
} else {
|
|
router.ServeFiles("/_/static/*filepath", http.Dir("public"))
|
|
router.GET("/", middleware(GetFrontpage))
|
|
router.GET("/search", middleware(Search))
|
|
router.POST("/search", middleware(UserOp))
|
|
router.GET("/inbox", middleware(Inbox))
|
|
router.POST("/inbox", middleware(UserOp))
|
|
router.GET("/login", middleware(GetLogin))
|
|
router.POST("/login", middleware(SignUpOrLogin))
|
|
router.GET("/settings", middleware(Settings))
|
|
router.POST("/settings", middleware(Settings))
|
|
router.POST("/", middleware(UserOp))
|
|
router.GET("/icon.jpg", middleware(GetIcon))
|
|
router.GET("/c/:community", middleware(GetFrontpage))
|
|
router.POST("/c/:community", middleware(UserOp))
|
|
router.GET("/c/:community/search", middleware(Search))
|
|
router.GET("/c/:community/edit", middleware(GetCreateCommunity))
|
|
router.GET("/post/:postid", middleware(GetPost))
|
|
router.POST("/post/:postid", middleware(UserOp))
|
|
router.GET("/comment/:commentid", middleware(GetComment))
|
|
router.GET("/comment/:commentid/:op", middleware(GetComment))
|
|
router.POST("/comment/:commentid", middleware(UserOp))
|
|
router.GET("/u/:username", middleware(GetUser))
|
|
router.GET("/u/:username/message", middleware(GetMessageForm))
|
|
router.POST("/u/:username/message", middleware(SendMessage))
|
|
router.POST("/u/:username", middleware(UserOp))
|
|
router.GET("/u/:username/search", middleware(Search))
|
|
router.GET("/create_post", middleware(GetCreatePost))
|
|
router.POST("/create_post", middleware(UserOp))
|
|
router.GET("/create_community", middleware(GetCreateCommunity))
|
|
router.POST("/create_community", middleware(UserOp))
|
|
router.GET("/communities", middleware(GetCommunities))
|
|
router.GET("/link", middleware(GetLink))
|
|
}
|
|
return router
|
|
}
|