111 lines
2.8 KiB
Go
111 lines
2.8 KiB
Go
package webserver
|
|
|
|
import (
|
|
"errors"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"gitlab.com/offline-twitter/twitter_offline_engine/pkg/persistence"
|
|
"gitlab.com/offline-twitter/twitter_offline_engine/pkg/scraper"
|
|
)
|
|
|
|
type UserProfileData struct {
|
|
persistence.Feed
|
|
scraper.UserID
|
|
FeedType string
|
|
}
|
|
|
|
func (t UserProfileData) Tweet(id scraper.TweetID) scraper.Tweet {
|
|
return t.Tweets[id]
|
|
}
|
|
func (t UserProfileData) User(id scraper.UserID) scraper.User {
|
|
return t.Users[id]
|
|
}
|
|
func (t UserProfileData) Retweet(id scraper.TweetID) scraper.Retweet {
|
|
return t.Retweets[id]
|
|
}
|
|
func (t UserProfileData) Space(id scraper.SpaceID) scraper.Space {
|
|
return t.Spaces[id]
|
|
}
|
|
func (t UserProfileData) FocusedTweetID() scraper.TweetID {
|
|
return scraper.TweetID(0)
|
|
}
|
|
|
|
func (app *Application) UserFeed(w http.ResponseWriter, r *http.Request) {
|
|
app.traceLog.Printf("'UserFeed' handler (path: %q)", r.URL.Path)
|
|
|
|
parts := strings.Split(strings.Trim(r.URL.Path, "/"), "/")
|
|
|
|
user, err := app.Profile.GetUserByHandle(scraper.UserHandle(parts[0]))
|
|
if err != nil {
|
|
app.error_404(w)
|
|
return
|
|
}
|
|
|
|
if len(parts) > 1 && parts[len(parts)-1] == "scrape" {
|
|
if app.IsScrapingDisabled {
|
|
http.Error(w, "Scraping is disabled (are you logged in?)", 401)
|
|
return
|
|
}
|
|
|
|
if len(parts) == 2 { // Already checked the last part is "scrape"
|
|
// Run scraper
|
|
trove, err := scraper.GetUserFeedGraphqlFor(user.ID, 50) // TODO: parameterizable
|
|
if err != nil {
|
|
app.ErrorLog.Print(err)
|
|
// TOOD: show error in UI
|
|
}
|
|
app.Profile.SaveTweetTrove(trove)
|
|
} else if len(parts) == 3 && parts[1] == "likes" {
|
|
trove, err := scraper.GetUserLikes(user.ID, 50) // TODO: parameterizable
|
|
if err != nil {
|
|
app.ErrorLog.Print(err)
|
|
// TOOD: show error in UI
|
|
}
|
|
app.Profile.SaveTweetTrove(trove)
|
|
}
|
|
}
|
|
|
|
var c persistence.Cursor
|
|
if len(parts) > 1 && parts[1] == "likes" {
|
|
c = persistence.NewUserFeedLikesCursor(user.Handle)
|
|
} else {
|
|
c = persistence.NewUserFeedCursor(user.Handle)
|
|
}
|
|
if len(parts) > 1 && parts[1] == "without_replies" {
|
|
c.FilterReplies = persistence.EXCLUDE
|
|
}
|
|
if len(parts) > 1 && parts[1] == "media" {
|
|
c.FilterMedia = persistence.REQUIRE
|
|
}
|
|
err = parse_cursor_value(&c, r)
|
|
if err != nil {
|
|
app.error_400_with_message(w, "invalid cursor (must be a number)")
|
|
return
|
|
}
|
|
|
|
feed, err := app.Profile.NextPage(c, app.ActiveUser.ID)
|
|
if err != nil {
|
|
if errors.Is(err, persistence.ErrEndOfFeed) {
|
|
// TODO
|
|
} else {
|
|
panic(err)
|
|
}
|
|
}
|
|
feed.Users[user.ID] = user
|
|
|
|
data := UserProfileData{Feed: feed, UserID: user.ID}
|
|
if len(parts) == 2 {
|
|
data.FeedType = parts[1]
|
|
} else {
|
|
data.FeedType = ""
|
|
}
|
|
|
|
if r.Header.Get("HX-Request") == "true" && c.CursorPosition == persistence.CURSOR_MIDDLE {
|
|
// It's a Show More request
|
|
app.buffered_render_tweet_htmx(w, "timeline", data)
|
|
} else {
|
|
app.buffered_render_tweet_page(w, "tpl/user_feed.tpl", data)
|
|
}
|
|
}
|