Add support for reblog

This commit is contained in:
Romain de Laage 2021-05-02 19:30:47 +02:00
parent 8494120df0
commit 1c5ead27b0
Signed by: rdelaage
GPG Key ID: 534845FADDF0C329
4 changed files with 143 additions and 24 deletions

18
main.go
View File

@ -22,13 +22,19 @@ import (
"time" "time"
) )
type OriginalBlog struct {
Id string `json:"id"`
Author Account `json:"account"`
}
type Blog struct { type Blog struct {
Id string `json:"id"` Id string `json:"id"`
Content string `json:"content"` Content string `json:"content"`
Date string `json:"created_at"` Date string `json:"created_at"`
Author Account `json:"account"` Author Account `json:"account"`
Tags []Tag `json:"tags"` Tags []Tag `json:"tags"`
Mentions []Mention `json:"mentions"` Mentions []Mention `json:"mentions"`
Reblog *OriginalBlog `json:"reblog"`
} }
type Config struct { type Config struct {

View File

@ -25,6 +25,41 @@ import (
"fmt" "fmt"
) )
func getBlogAndReblog(baseURL, account string) ([]Blog, error) {
if baseURL == "" || account == "" {
log.Println("baseURL or account is empty")
return nil, fmt.Errorf("BaseURL or account is empty")
}
resp, err := http.Get(baseURL + "/api/v1/accounts/" + account + "/statuses?exclude_reblogs=false&exclude_replies=true")
if err != nil {
log.Println("Mastodon API request: %s", err)
return nil, fmt.Errorf("Failed to request Mastodon API")
}
defer resp.Body.Close()
if resp.StatusCode != 200 {
log.Println("Mastodon API response: %s", resp.Status)
return nil, fmt.Errorf("Mastodon instance failed")
}
body, err := ioutil.ReadAll(resp.Body)
if err != nil {
log.Println("Mastodon response body: %s", err)
return nil, fmt.Errorf("Failed to read Mastodon response body")
}
var blogs []Blog
err = json.Unmarshal(body, &blogs)
if err != nil {
log.Println("Mastodon response %s", err)
return nil, fmt.Errorf("Failed to parse Mastodon response")
}
return blogs, nil
}
func getBlog(baseURL, account string) ([]Blog, error) { func getBlog(baseURL, account string) ([]Blog, error) {
if baseURL == "" || account == "" { if baseURL == "" || account == "" {
log.Println("baseURL or account is empty") log.Println("baseURL or account is empty")

101
server.go
View File

@ -125,22 +125,39 @@ func handleConn(conn *tls.Conn, baseURL, title, home_message string, rateLimit i
// profile // profile
if strings.HasPrefix(path, "/profile/") { if strings.HasPrefix(path, "/profile/") {
// skip prefix if strings.HasSuffix(path, "/reblog") {
path = path[9:] // skip prefix and suffix
_, err = strconv.ParseUint(path, 10, 64) path = path[9:len(path)-len("/reblog")]
if err != nil { _, err = strconv.ParseUint(path, 10, 64)
log.Println("invalid request: %s", err)
_, err = fmt.Fprintf(conn, "59 Can't parse request\r\n")
if err != nil { if err != nil {
log.Println("send error: %s", err) log.Println("invalid request: %s", err)
_, err = fmt.Fprintf(conn, "59 Can't parse request\r\n")
if err != nil {
log.Println("send error: %s", err)
return
}
return return
} }
return
log.Println("Received request for account with reblog " + path)
printProfileWithReblog(conn, baseURL, path)
} else {
// skip prefix
path = path[9:]
_, err = strconv.ParseUint(path, 10, 64)
if err != nil {
log.Println("invalid request: %s", err)
_, err = fmt.Fprintf(conn, "59 Can't parse request\r\n")
if err != nil {
log.Println("send error: %s", err)
return
}
return
}
log.Println("Received request for account " + path)
printProfile(conn, baseURL, path)
} }
log.Println("Received request for account " + path)
printProfile(conn, baseURL, path)
} /* thread */ else if strings.HasPrefix(path, "/thread/") { } /* thread */ else if strings.HasPrefix(path, "/thread/") {
// skip prefix // skip prefix
path = path[8:] path = path[8:]
@ -257,7 +274,49 @@ func printProfile(conn *tls.Conn, baseURL, profileID string) {
} }
} }
_, err = fmt.Fprintf(conn, "\n=> %s Go to %s account", account.Url, account.Name) _, err = fmt.Fprintf(conn, "\n=> /profile/%s/reblog This profile with reblog\n=> %s Go to %s account", account.Id, account.Url, account.Name)
if err != nil {
log.Println("add link: %s", err)
return
}
}
func printProfileWithReblog(conn *tls.Conn, baseURL, profileID string) {
account, err := getAccount(baseURL, profileID)
if err != nil {
_, err = fmt.Fprintf(conn, "40 Remote mastodon instance failed\r\n")
if err != nil {
log.Println("handleConn: %s", err)
return
}
return
}
blogs, err := getBlogAndReblog(baseURL, profileID)
if err != nil {
_, err = fmt.Fprintf(conn, "40 Remote mastodon instance failed\r\n")
if err != nil {
log.Println("handleConn: %s", err)
return
}
return
}
_, err = fmt.Fprintf(conn, "20 text/gemini\r\n# Toots for %s account\n", account.Name)
if err != nil {
log.Println("handleConn: %s", err)
return
}
for _, blog := range blogs {
_, err = fmt.Fprintf(conn, "\n%s\n=> /thread/%s View the thread\n", formatBlog(blog), blog.Id)
if err != nil {
log.Println("read blogs: %s", err)
return
}
}
_, err = fmt.Fprintf(conn, "\n=> /profile/%s This profile without reblog\n=> %s Go to %s account", account.Id, account.Url, account.Name)
if err != nil { if err != nil {
log.Println("add link: %s", err) log.Println("add link: %s", err)
return return
@ -283,10 +342,18 @@ func printToot(conn *tls.Conn, baseURL, tootID string) {
} }
// Print toot // Print toot
_, err = fmt.Fprintf(conn, "# Toot\n\n%s\n=> /thread/%s View the thread\n=> /profile/%s More toots from %s\n", formatBlog(toot), toot.Id, toot.Author.Id, toot.Author.Name) if toot.Reblog == nil {
if err != nil { _, err = fmt.Fprintf(conn, "# Toot\n\n%s\n=> /thread/%s View the thread\n=> /profile/%s More toots from %s\n", formatBlog(toot), toot.Id, toot.Author.Id, toot.Author.Name)
log.Println("handleConn: %s", err) if err != nil {
return log.Println("handleConn: %s", err)
return
}
} else {
_, err = fmt.Fprintf(conn, "# Toot\n\n%s\n=> /toot/%s Original toot\n=> /thread/%s View the thread\n=> /profile/%s More toots from %s\n=> /profile/%s More toots from %s\n", formatBlog(toot), toot.Reblog.Id, toot.Id, toot.Author.Id, toot.Author.Name, toot.Reblog.Author.Id, toot.Reblog.Author.Name)
if err != nil {
log.Println("handleConn: %s", err)
return
}
} }
// print mentions // print mentions

13
util.go
View File

@ -104,7 +104,18 @@ func formatBlog(toot Blog) string {
author = toot.Author.DisplayName author = toot.Author.DisplayName
} }
return "### Written by " + author + " on " + toot.Date[0:10] + " at " + toot.Date[11:16] + "\n" + content + "\n=> /toot/" + toot.Id + " More informations about this toot" if toot.Reblog == nil {
return "### Written by " + author + " on " + toot.Date[0:10] + " at " + toot.Date[11:16] + "\n" + content + "\n=> /toot/" + toot.Id + " More informations about this toot"
} else {
var originalAuthor string
if toot.Reblog.Author.DisplayName == "" {
originalAuthor = toot.Reblog.Author.Name
} else {
originalAuthor = toot.Reblog.Author.DisplayName
}
return "### Shared by " + author + " on " + toot.Date[0:10] + " at " + toot.Date[11:16] + " (original by " + originalAuthor + ")\n" + content + "\n=> /toot/" + toot.Id + " More informations about this toot"
}
} }
func rateIsOk(tab map[string]Rate, remoteIP string, limit int) bool { func rateIsOk(tab map[string]Rate, remoteIP string, limit int) bool {