Something went wrong. Try again.
This repository has no description
Something went wrong. Try again.
31 kB · 1127 lines
Go
at master
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128package state
import ( "context" "fmt" "net/http" "slices" "strings" "time"
comatproto "github.com/bluesky-social/indigo/api/atproto" "github.com/bluesky-social/indigo/atproto/identity" "github.com/bluesky-social/indigo/atproto/syntax" lexutil "github.com/bluesky-social/indigo/lex/util" "github.com/go-chi/chi/v5" "github.com/gorilla/feeds" "tangled.org/core/api/tangled" "tangled.org/core/appview/cache" "tangled.org/core/appview/db" "tangled.org/core/appview/middleware" "tangled.org/core/appview/models" "tangled.org/core/appview/pages" "tangled.org/core/appview/pagination" "tangled.org/core/appview/searchquery" "tangled.org/core/orm" "tangled.org/core/xrpc")
func (s *State) Profile(w http.ResponseWriter, r *http.Request) { tabVal := r.URL.Query().Get("tab") switch tabVal { case "repos": middleware. Paginate(http.HandlerFunc(s.reposPage)). ServeHTTP(w, r) case "followers": s.followersPage(w, r) case "following": s.followingPage(w, r) case "starred": middleware. Paginate(http.HandlerFunc(s.starredPage)). ServeHTTP(w, r) case "strings": s.stringsPage(w, r) default: s.profileOverview(w, r) }}
func (s *State) profile(r *http.Request) (*pages.ProfileCard, error) { didOrHandle := chi.URLParam(r, "user") if didOrHandle == "" { return nil, fmt.Errorf("empty DID or handle") }
ident, ok := r.Context().Value("resolvedId").(identity.Identity) if !ok { return nil, fmt.Errorf("failed to resolve ID") } did := ident.DID.String()
profile, err := db.GetProfile(s.db, did) if err != nil { return nil, fmt.Errorf("failed to get profile: %w", err) }
hasProfile := profile != nil if !hasProfile { profile = &models.Profile{Did: did} }
repoCount, err := db.CountRepos(s.db, orm.FilterEq("did", did)) if err != nil { return nil, fmt.Errorf("failed to get repo count: %w", err) }
stringCount, err := db.CountStrings(s.db, orm.FilterEq("did", did)) if err != nil { return nil, fmt.Errorf("failed to get string count: %w", err) }
starredCount, err := db.CountStars(s.db, orm.FilterEq("did", did)) if err != nil { return nil, fmt.Errorf("failed to get starred repo count: %w", err) }
followStats, err := db.GetFollowerFollowingCount(s.db, did) if err != nil { return nil, fmt.Errorf("failed to get follower stats: %w", err) }
loggedInUser := s.oauth.GetMultiAccountUser(r) followStatus := models.IsNotFollowing var loggedInDid string if loggedInUser != nil { followStatus = db.GetFollowStatus(s.db, loggedInUser.Did, did) loggedInDid = loggedInUser.Did }
showPunchcard := s.shouldShowPunchcard(did, loggedInDid)
var punchcard *models.Punchcard if showPunchcard { now := time.Now() startOfYear := time.Date(now.Year(), 1, 1, 0, 0, 0, 0, time.UTC) punchcard, err = db.MakePunchcard( s.db, orm.FilterEq("did", did), orm.FilterGte("date", startOfYear.Format(time.DateOnly)), orm.FilterLte("date", now.Format(time.DateOnly)), ) if err != nil { return nil, fmt.Errorf("failed to get punchcard for %s: %w", did, err) } }
return &pages.ProfileCard{ UserDid: did, HasProfile: hasProfile, Profile: profile, FollowStatus: followStatus, Stats: pages.ProfileStats{ RepoCount: repoCount, StringCount: stringCount, StarredCount: starredCount, FollowersCount: followStats.Followers, FollowingCount: followStats.Following, }, Punchcard: punchcard, }, nil}
func (s *State) profileOverview(w http.ResponseWriter, r *http.Request) { l := s.logger.With("handler", "profileHomePage")
profile, err := s.profile(r) if err != nil { l.Error("failed to build profile card", "err", err) s.pages.Error500(w) return } l = l.With("profileDid", profile.UserDid)
repos, err := db.GetRepos( s.db, orm.FilterEq("did", profile.UserDid), ) if err != nil { l.Error("failed to fetch repos", "err", err) }
// filter out ones that are pinned pinnedRepos := []models.Repo{} for i, r := range repos { if profile.Profile.MatchesPinnedRepo(r) { pinnedRepos = append(pinnedRepos, r) } else if profile.Profile.IsPinnedReposEmpty() && i < 4 { pinnedRepos = append(pinnedRepos, r) } }
collaboratingRepos, err := db.CollaboratingIn(s.db, profile.UserDid) if err != nil { l.Error("failed to fetch collaborating repos", "err", err) }
pinnedCollaboratingRepos := []models.Repo{} for _, r := range collaboratingRepos { if profile.Profile.MatchesPinnedRepo(r) { pinnedCollaboratingRepos = append(pinnedCollaboratingRepos, r) } }
timeline, err := db.MakeProfileTimeline(s.db, profile.UserDid) if err != nil { l.Error("failed to create timeline", "err", err) }
s.pages.ProfileOverview(w, pages.ProfileOverviewParams{ LoggedInUser: s.oauth.GetMultiAccountUser(r), Card: profile, Repos: pinnedRepos, CollaboratingRepos: pinnedCollaboratingRepos, ProfileTimeline: timeline, })}
func (s *State) shouldShowPunchcard(targetDid, requesterDid string) bool { l := s.logger.With("helper", "shouldShowPunchcard")
targetPunchcardPreferences, err := db.GetPunchcardPreference(s.db, targetDid) if err != nil { l.Error("failed to get target users punchcard preferences", "err", err) return true }
requesterPunchcardPreferences, err := db.GetPunchcardPreference(s.db, requesterDid) if err != nil { l.Error("failed to get requester users punchcard preferences", "err", err) return true }
showPunchcard := true
// looking at their own profile if targetDid == requesterDid { if targetPunchcardPreferences.HideMine { return false } return true }
if targetPunchcardPreferences.HideMine || requesterPunchcardPreferences.HideOthers { showPunchcard = false } return showPunchcard}
func (s *State) reposPage(w http.ResponseWriter, r *http.Request) { l := s.logger.With("handler", "reposPage")
profile, err := s.profile(r) if err != nil { l.Error("failed to build profile card", "err", err) s.pages.Error500(w) return } l = l.With("profileDid", profile.UserDid)
params := r.URL.Query() page := pagination.FromContext(r.Context())
query := searchquery.Parse(params.Get("q"))
var language string if lang := query.Get("language"); lang != nil { language = *lang }
tf := searchquery.ExtractTextFilters(query)
searchOpts := models.RepoSearchOptions{ Keywords: tf.Keywords, Phrases: tf.Phrases, NegatedKeywords: tf.NegatedKeywords, NegatedPhrases: tf.NegatedPhrases, Did: profile.UserDid, Language: language, Page: page, }
var repos []models.Repo var totalRepos int64
if searchOpts.HasSearchFilters() { res, err := s.indexer.Repos.Search(r.Context(), searchOpts) if err != nil { l.Error("failed to search repos", "err", err) s.pages.Error500(w) return }
if len(res.Hits) > 0 { repos, err = db.GetRepos(s.db, orm.FilterIn("id", res.Hits)) if err != nil { l.Error("failed to get repos by IDs", "err", err) s.pages.Error500(w) return }
// sort repos to match search result order (by relevance) repoMap := make(map[int64]models.Repo, len(repos)) for _, repo := range repos { repoMap[repo.Id] = repo } repos = make([]models.Repo, 0, len(res.Hits)) for _, id := range res.Hits { if repo, ok := repoMap[id]; ok { repos = append(repos, repo) } } } totalRepos = int64(res.Total) } else { repos, err = db.GetReposPaginated( s.db, page, orm.FilterEq("did", profile.UserDid), ) if err != nil { l.Error("failed to get repos", "err", err) s.pages.Error500(w) return }
totalRepos, err = db.CountRepos( s.db, orm.FilterEq("did", profile.UserDid), ) if err != nil { l.Error("failed to count repos", "err", err) s.pages.Error500(w) return } }
err = s.pages.ProfileRepos(w, pages.ProfileReposParams{ LoggedInUser: s.oauth.GetMultiAccountUser(r), Repos: repos, Card: profile, Page: page, RepoCount: int(totalRepos), FilterQuery: query.String(), }) if err != nil { l.Error("failed to render page", "err", err) }}
func (s *State) starredPage(w http.ResponseWriter, r *http.Request) { l := s.logger.With("handler", "starredPage")
page := pagination.FromContext(r.Context()) l = l.With("page", page)
profile, err := s.profile(r) if err != nil { l.Error("failed to build profile card", "err", err) s.pages.Error500(w) return } l = l.With("profileDid", profile.UserDid)
stars, err := db.GetRepoStars(s.db, page, orm.FilterEq("did", profile.UserDid)) if err != nil { l.Error("failed to get stars", "err", err) s.pages.Error500(w) return } var repos []models.Repo for _, s := range stars { repos = append(repos, *s.Repo) }
err = s.pages.ProfileStarred(w, pages.ProfileStarredParams{ LoggedInUser: s.oauth.GetMultiAccountUser(r), Repos: repos, Total: int(profile.Stats.StarredCount), Card: profile, Page: page, }) if err != nil { l.Error("failed to render", "err", err) }}
func (s *State) stringsPage(w http.ResponseWriter, r *http.Request) { l := s.logger.With("handler", "stringsPage")
profile, err := s.profile(r) if err != nil { l.Error("failed to build profile card", "err", err) s.pages.Error500(w) return } l = l.With("profileDid", profile.UserDid)
strings, err := db.GetStrings(s.db, 0, orm.FilterEq("did", profile.UserDid)) if err != nil { l.Error("failed to get strings", "err", err) s.pages.Error500(w) return }
err = s.pages.ProfileStrings(w, pages.ProfileStringsParams{ LoggedInUser: s.oauth.GetMultiAccountUser(r), Strings: strings, Card: profile, })}
type FollowsPageParams struct { Follows []pages.FollowCard Card *pages.ProfileCard}
func (s *State) followPage( r *http.Request, fetchFollows func(db.Execer, string) ([]models.Follow, error), extractDid func(models.Follow) string,) (*FollowsPageParams, error) { l := s.logger.With("handler", "reposPage")
profile, err := s.profile(r) if err != nil { return nil, err } l = l.With("profileDid", profile.UserDid)
loggedInUser := s.oauth.GetMultiAccountUser(r) params := FollowsPageParams{ Card: profile, }
follows, err := fetchFollows(s.db, profile.UserDid) if err != nil { l.Error("failed to fetch follows", "err", err) return ¶ms, err }
if len(follows) == 0 { return ¶ms, nil }
followDids := make([]string, 0, len(follows)) for _, follow := range follows { followDids = append(followDids, extractDid(follow)) }
profiles, err := db.GetProfiles(s.db, orm.FilterIn("did", followDids)) if err != nil { l.Error("failed to get profiles", "followDids", followDids, "err", err) return ¶ms, err }
followStatsMap, err := db.GetFollowerFollowingCounts(s.db, followDids) if err != nil { l.Error("getting follow counts", "followDids", followDids, "err", err) }
loggedInUserFollowing := make(map[string]struct{}) if loggedInUser != nil { following, err := db.GetFollowing(s.db, loggedInUser.Did) if err != nil { l.Error("failed to get follow list", "err", err, "loggedInUser", loggedInUser.Did) return ¶ms, err } loggedInUserFollowing = make(map[string]struct{}, len(following)) for _, follow := range following { loggedInUserFollowing[follow.SubjectDid] = struct{}{} } }
followCards := make([]pages.FollowCard, len(follows)) for i, did := range followDids { followStats := followStatsMap[did] followStatus := models.IsNotFollowing if _, exists := loggedInUserFollowing[did]; exists { followStatus = models.IsFollowing } else if loggedInUser != nil && loggedInUser.Did == did { followStatus = models.IsSelf }
var profile *models.Profile if p, exists := profiles[did]; exists { profile = p } else { profile = &models.Profile{} profile.Did = did } followCards[i] = pages.FollowCard{ LoggedInUser: loggedInUser, UserDid: did, FollowStatus: followStatus, FollowersCount: followStats.Followers, FollowingCount: followStats.Following, Profile: profile, } }
params.Follows = followCards
return ¶ms, nil}
func (s *State) followersPage(w http.ResponseWriter, r *http.Request) { followPage, err := s.followPage(r, db.GetFollowers, func(f models.Follow) string { return f.UserDid }) if err != nil { s.pages.Notice(w, "all-followers", "Failed to load followers") return }
s.pages.ProfileFollowers(w, pages.ProfileFollowersParams{ LoggedInUser: s.oauth.GetMultiAccountUser(r), Followers: followPage.Follows, Card: followPage.Card, })}
func (s *State) followingPage(w http.ResponseWriter, r *http.Request) { followPage, err := s.followPage(r, db.GetFollowing, func(f models.Follow) string { return f.SubjectDid }) if err != nil { s.pages.Notice(w, "all-following", "Failed to load following") return }
s.pages.ProfileFollowing(w, pages.ProfileFollowingParams{ LoggedInUser: s.oauth.GetMultiAccountUser(r), Following: followPage.Follows, Card: followPage.Card, })}
func (s *State) AtomFeedPage(w http.ResponseWriter, r *http.Request) { ident, ok := r.Context().Value("resolvedId").(identity.Identity) if !ok { s.pages.Error404(w) return }
feed, err := s.getProfileFeed(r.Context(), &ident) if err != nil { s.pages.Error500(w) return }
if feed == nil { return }
atom, err := feed.ToAtom() if err != nil { s.pages.Error500(w) return }
w.Header().Set("content-type", "application/atom+xml") w.Write([]byte(atom))}
func (s *State) getProfileFeed(ctx context.Context, id *identity.Identity) (*feeds.Feed, error) { timeline, err := db.MakeProfileTimeline(s.db, id.DID.String()) if err != nil { return nil, err }
author := &feeds.Author{ Name: fmt.Sprintf("@%s", id.Handle), }
feed := feeds.Feed{ Title: fmt.Sprintf("%s's timeline", author.Name), Link: &feeds.Link{Href: fmt.Sprintf("%s/@%s", s.config.Core.BaseUrl(), id.Handle), Type: "text/html", Rel: "alternate"}, Items: make([]*feeds.Item, 0), Updated: time.UnixMilli(0), Author: author, }
for _, byMonth := range timeline.ByMonth { if err := s.addPullRequestItems(ctx, &feed, byMonth.PullEvents.Items, author); err != nil { return nil, err } if err := s.addIssueItems(ctx, &feed, byMonth.IssueEvents.Items, author); err != nil { return nil, err } if err := s.addRepoItems(ctx, &feed, byMonth.RepoEvents, author); err != nil { return nil, err } }
slices.SortFunc(feed.Items, func(a *feeds.Item, b *feeds.Item) int { return int(b.Created.UnixMilli()) - int(a.Created.UnixMilli()) })
if len(feed.Items) > 0 { feed.Updated = feed.Items[0].Created }
return &feed, nil}
func (s *State) addPullRequestItems(ctx context.Context, feed *feeds.Feed, pulls []*models.Pull, author *feeds.Author) error { for _, pull := range pulls { owner, err := s.idResolver.ResolveIdent(ctx, pull.Repo.Did) if err != nil { return err }
// Add pull request creation item feed.Items = append(feed.Items, s.createPullRequestItem(pull, owner, author)) } return nil}
func (s *State) addIssueItems(ctx context.Context, feed *feeds.Feed, issues []*models.Issue, author *feeds.Author) error { for _, issue := range issues { owner, err := s.idResolver.ResolveIdent(ctx, issue.Repo.Did) if err != nil { return err }
feed.Items = append(feed.Items, s.createIssueItem(issue, owner, author)) } return nil}
func (s *State) addRepoItems(ctx context.Context, feed *feeds.Feed, repos []models.RepoEvent, author *feeds.Author) error { for _, repo := range repos { item, err := s.createRepoItem(ctx, repo, author) if err != nil { return err } feed.Items = append(feed.Items, item) } return nil}
func (s *State) createPullRequestItem(pull *models.Pull, owner *identity.Identity, author *feeds.Author) *feeds.Item { return &feeds.Item{ Title: fmt.Sprintf("%s created pull request '%s' in @%s/%s", author.Name, pull.Title, owner.Handle, pull.Repo.Name), Link: &feeds.Link{Href: fmt.Sprintf("%s/@%s/%s/pulls/%d", s.config.Core.BaseUrl(), owner.Handle, pull.Repo.Name, pull.PullId), Type: "text/html", Rel: "alternate"}, Created: pull.Created, Author: author, }}
func (s *State) createIssueItem(issue *models.Issue, owner *identity.Identity, author *feeds.Author) *feeds.Item { return &feeds.Item{ Title: fmt.Sprintf("%s created issue '%s' in @%s/%s", author.Name, issue.Title, owner.Handle, issue.Repo.Name), Link: &feeds.Link{Href: fmt.Sprintf("%s/@%s/%s/issues/%d", s.config.Core.BaseUrl(), owner.Handle, issue.Repo.Name, issue.IssueId), Type: "text/html", Rel: "alternate"}, Created: issue.Created, Author: author, }}
func (s *State) createRepoItem(ctx context.Context, repo models.RepoEvent, author *feeds.Author) (*feeds.Item, error) { var title string if repo.Source != nil { sourceOwner, err := s.idResolver.ResolveIdent(ctx, repo.Source.Did) if err != nil { return nil, err } title = fmt.Sprintf("%s forked repository @%s/%s to '%s'", author.Name, sourceOwner.Handle, repo.Source.Name, repo.Repo.Name) } else { title = fmt.Sprintf("%s created repository '%s'", author.Name, repo.Repo.Name) }
return &feeds.Item{ Title: title, Link: &feeds.Link{Href: fmt.Sprintf("%s/@%s/%s", s.config.Core.BaseUrl(), author.Name[1:], repo.Repo.Name), Type: "text/html", Rel: "alternate"}, // Remove @ prefix Created: repo.Repo.Created, Author: author, }, nil}
func (s *State) UpdateProfileBio(w http.ResponseWriter, r *http.Request) { l := s.logger.With("handler", "UpdateProfileBio") user := s.oauth.GetMultiAccountUser(r)
err := r.ParseForm() if err != nil { l.Error("invalid profile update form", "err", err) s.pages.Notice(w, "update-profile", "Invalid form.") return }
profile, err := db.GetProfile(s.db, user.Did) if err != nil { l.Error("getting profile data", "did", user.Did, "err", err) } if profile == nil { profile = &models.Profile{Did: user.Did} }
profile.Description = r.FormValue("description") profile.IncludeBluesky = r.FormValue("includeBluesky") == "on" profile.Location = r.FormValue("location") profile.Pronouns = r.FormValue("pronouns") rawPreferredHandle := strings.TrimSpace(r.FormValue("preferredHandle")) if rawPreferredHandle != "" { h, err := syntax.ParseHandle(rawPreferredHandle) if err != nil { s.pages.Notice(w, "update-profile", "Invalid handle format.") return }
ident, err := s.idResolver.ResolveIdent(r.Context(), user.Did) if err != nil || !slices.Contains(ident.AlsoKnownAs, "at://"+rawPreferredHandle) { s.pages.Notice(w, "update-profile", "Handle not found in your DID document.") return } profile.PreferredHandle = h } else { profile.PreferredHandle = "" }
var links [5]string for i := range 5 { iLink := r.FormValue(fmt.Sprintf("link%d", i)) links[i] = iLink } profile.Links = links
// Parse stats (exactly 2) stat0 := r.FormValue("stat0") stat1 := r.FormValue("stat1")
profile.Stats[0].Kind = models.ParseVanityStatKind(stat0) profile.Stats[1].Kind = models.ParseVanityStatKind(stat1)
if err := db.ValidateProfile(s.db, profile); err != nil { l.Error("invalid profile", "err", err) s.pages.Notice(w, "update-profile", err.Error()) return }
s.updateProfile(profile, w, r)}
func (s *State) UpdateProfilePins(w http.ResponseWriter, r *http.Request) { l := s.logger.With("handler", "UpdateProfilePins") user := s.oauth.GetMultiAccountUser(r)
err := r.ParseForm() if err != nil { l.Error("invalid profile update form", "err", err) s.pages.Notice(w, "update-profile", "Invalid form.") return }
profile, err := db.GetProfile(s.db, user.Did) if err != nil { l.Error("getting profile data", "did", user.Did, "err", err) } if profile == nil { profile = &models.Profile{Did: user.Did} }
i := 0 var pinnedRepos [6]string for key, values := range r.Form { if i >= 6 { l.Warn("too many pinned repos") s.pages.Notice(w, "update-profile", "Only 6 repositories can be pinned at a time.") return } if strings.HasPrefix(key, "pinnedRepo") && len(values) > 0 && values[0] != "" && i < 6 { pinnedRepos[i] = values[0] i++ } } profile.PinnedRepos = pinnedRepos
s.updateProfile(profile, w, r)}
func (s *State) updateProfile(profile *models.Profile, w http.ResponseWriter, r *http.Request) { l := s.logger.With("handler", "updateProfile") user := s.oauth.GetMultiAccountUser(r)
client, err := s.oauth.AuthorizedClient(r) if err != nil { l.Error("failed to get authorized client", "err", err) s.pages.Notice(w, "update-profile", "Failed to update profile, try again later.") return }
var pinnedRepoStrings []string for _, r := range profile.PinnedRepos { if r != "" { pinnedRepoStrings = append(pinnedRepoStrings, r) } }
var vanityStats []string for _, v := range profile.Stats { vanityStats = append(vanityStats, string(v.Kind)) }
ex, _ := comatproto.RepoGetRecord(r.Context(), client, "", tangled.ActorProfileNSID, user.Did, "self") var cid *string var existingAvatar *lexutil.LexBlob if ex != nil { cid = ex.Cid if rec, ok := ex.Value.Val.(*tangled.ActorProfile); ok { existingAvatar = rec.Avatar } }
_, err = comatproto.RepoPutRecord(r.Context(), client, &comatproto.RepoPutRecord_Input{ Collection: tangled.ActorProfileNSID, Repo: user.Did, Rkey: "self", Record: &lexutil.LexiconTypeDecoder{ Val: &tangled.ActorProfile{ Avatar: existingAvatar, Bluesky: profile.IncludeBluesky, Description: &profile.Description, Links: profile.Links[:], Location: &profile.Location, PinnedRepositories: pinnedRepoStrings, Stats: vanityStats[:], Pronouns: &profile.Pronouns, PreferredHandle: (*string)(&profile.PreferredHandle), }}, SwapRecord: cid, }) if err != nil { l.Error("failed to update profile on PDS", "err", err) s.pages.Notice(w, "update-profile", "Failed to update PDS, try again later.") return }
tx, err := s.db.BeginTx(r.Context(), nil) if err != nil { l.Error("failed to start transaction", "err", err) s.pages.Notice(w, "update-profile", "Failed to update profile, try again later.") return }
if err := db.UpsertProfile(tx, profile); err != nil { l.Error("failed to update profile in DB", "err", err) s.pages.Notice(w, "update-profile", "Failed to update profile, try again later.") return }
if s.rdb != nil { ctx := r.Context() pipe := s.rdb.Pipeline() didKey := fmt.Sprintf(cache.PreferredHandleByDid, profile.Did) if profile.PreferredHandle != "" { pipe.Set(ctx, didKey, string(profile.PreferredHandle), cache.PreferredHandleTTL) pipe.Set(ctx, fmt.Sprintf(cache.PreferredHandleByHandle, string(profile.PreferredHandle)), profile.Did, cache.PreferredHandleTTL) } else { pipe.Del(ctx, didKey) } if _, execErr := pipe.Exec(ctx); execErr != nil { l.Warn("failed to update preferred handle cache", "err", execErr) } }
s.notifier.UpdateProfile(r.Context(), profile)
s.pages.HxRedirect(w, "/"+user.Did)}
func (s *State) EditBioFragment(w http.ResponseWriter, r *http.Request) { l := s.logger.With("handler", "EditBioFragment") user := s.oauth.GetMultiAccountUser(r)
profile, err := db.GetProfile(s.db, user.Did) if err != nil { l.Error("getting profile data", "did", user.Did, "err", err) } if profile == nil { profile = &models.Profile{Did: user.Did} }
var alsoKnownAs []string ident, err := s.idResolver.ResolveIdent(r.Context(), user.Did) if err == nil { alsoKnownAs = ident.AlsoKnownAs }
s.pages.EditBioFragment(w, pages.EditBioParams{ LoggedInUser: user, Profile: profile, AlsoKnownAs: alsoKnownAs, })}
func (s *State) EditPinsFragment(w http.ResponseWriter, r *http.Request) { l := s.logger.With("handler", "EditPinsFragment") user := s.oauth.GetMultiAccountUser(r)
profile, err := db.GetProfile(s.db, user.Did) if err != nil { l.Error("getting profile data", "did", user.Did, "err", err) } if profile == nil { profile = &models.Profile{Did: user.Did} }
repos, err := db.GetRepos(s.db, orm.FilterEq("did", user.Did)) if err != nil { l.Error("getting repos", "did", user.Did, "err", err) }
collaboratingRepos, err := db.CollaboratingIn(s.db, user.Did) if err != nil { l.Error("getting collaborating repos", "did", user.Did, "err", err) }
allRepos := []pages.PinnedRepo{}
for _, r := range repos { allRepos = append(allRepos, pages.PinnedRepo{ IsPinned: profile.MatchesPinnedRepo(r), Repo: r, }) } for _, r := range collaboratingRepos { allRepos = append(allRepos, pages.PinnedRepo{ IsPinned: profile.MatchesPinnedRepo(r), Repo: r, }) }
s.pages.EditPinsFragment(w, pages.EditPinsParams{ LoggedInUser: user, Profile: profile, AllRepos: allRepos, })}
func (s *State) UploadProfileAvatar(w http.ResponseWriter, r *http.Request) { l := s.logger.With("handler", "UploadProfileAvatar") user := s.oauth.GetMultiAccountUser(r) l = l.With("did", user.Did)
// Parse multipart form (10MB max) if err := r.ParseMultipartForm(10 << 20); err != nil { l.Error("failed to parse form", "err", err) s.pages.Notice(w, "avatar-error", "Failed to parse form") return }
file, header, err := r.FormFile("avatar") if err != nil { l.Error("failed to read avatar file", "err", err) s.pages.Notice(w, "avatar-error", "Failed to read avatar file") return } defer file.Close()
if header.Size > 5000000 { l.Warn("avatar file too large", "size", header.Size) s.pages.Notice(w, "avatar-error", "Avatar file too large (max 5MB)") return }
contentType := header.Header.Get("Content-Type") if contentType != "image/png" && contentType != "image/jpeg" { l.Warn("invalid image type", "contentType", contentType) s.pages.Notice(w, "avatar-error", "Invalid image type (only PNG and JPEG allowed)") return }
client, err := s.oauth.AuthorizedClient(r) if err != nil { l.Error("failed to get PDS client", "err", err) s.pages.Notice(w, "avatar-error", "Failed to connect to your PDS") return }
uploadBlobResp, err := xrpc.RepoUploadBlob(r.Context(), client, file, header.Header.Get("Content-Type")) if err != nil { l.Error("failed to upload avatar blob", "err", err) s.pages.Notice(w, "avatar-error", "Failed to upload avatar to your PDS") return }
l.Info("uploaded avatar blob", "cid", uploadBlobResp.Blob.Ref.String())
// get current profile record from PDS to get its CID for swap getRecordResp, err := comatproto.RepoGetRecord(r.Context(), client, "", tangled.ActorProfileNSID, user.Did, "self") if err != nil { l.Error("failed to get current profile record", "err", err) s.pages.Notice(w, "avatar-error", "Failed to get current profile from your PDS") return }
var profileRecord *tangled.ActorProfile if getRecordResp.Value != nil { if val, ok := getRecordResp.Value.Val.(*tangled.ActorProfile); ok { profileRecord = val } else { l.Warn("profile record type assertion failed, creating new record") profileRecord = &tangled.ActorProfile{} } } else { l.Warn("no existing profile record, creating new record") profileRecord = &tangled.ActorProfile{} }
profileRecord.Avatar = uploadBlobResp.Blob
_, err = comatproto.RepoPutRecord(r.Context(), client, &comatproto.RepoPutRecord_Input{ Collection: tangled.ActorProfileNSID, Repo: user.Did, Rkey: "self", Record: &lexutil.LexiconTypeDecoder{Val: profileRecord}, SwapRecord: getRecordResp.Cid, })
if err != nil { l.Error("failed to update profile record", "err", err) s.pages.Notice(w, "avatar-error", "Failed to update profile on your PDS") return }
l.Info("successfully updated profile with avatar")
profile, err := db.GetProfile(s.db, user.Did) if err != nil { l.Warn("getting profile data from DB", "err", err) } if profile == nil { profile = &models.Profile{Did: user.Did} } profile.Avatar = uploadBlobResp.Blob.Ref.String()
tx, err := s.db.BeginTx(r.Context(), nil) if err != nil { l.Error("failed to start transaction", "err", err) s.pages.HxRefresh(w) w.WriteHeader(http.StatusOK) return }
err = db.UpsertProfile(tx, profile) if err != nil { l.Error("failed to update profile in DB", "err", err) s.pages.HxRefresh(w) w.WriteHeader(http.StatusOK) return }
s.pages.HxRedirect(w, r.Header.Get("Referer"))}
func (s *State) RemoveProfileAvatar(w http.ResponseWriter, r *http.Request) { l := s.logger.With("handler", "RemoveProfileAvatar") user := s.oauth.GetMultiAccountUser(r) l = l.With("did", user.Did)
client, err := s.oauth.AuthorizedClient(r) if err != nil { l.Error("failed to get PDS client", "err", err) s.pages.Notice(w, "avatar-error", "Failed to connect to your PDS") return }
getRecordResp, err := comatproto.RepoGetRecord(r.Context(), client, "", tangled.ActorProfileNSID, user.Did, "self") if err != nil { l.Error("failed to get current profile record", "err", err) s.pages.Notice(w, "avatar-error", "Failed to get current profile from your PDS") return }
var profileRecord *tangled.ActorProfile if getRecordResp.Value != nil { if val, ok := getRecordResp.Value.Val.(*tangled.ActorProfile); ok { profileRecord = val } else { l.Warn("profile record type assertion failed") profileRecord = &tangled.ActorProfile{} } } else { l.Warn("no existing profile record") profileRecord = &tangled.ActorProfile{} }
profileRecord.Avatar = nil
_, err = comatproto.RepoPutRecord(r.Context(), client, &comatproto.RepoPutRecord_Input{ Collection: tangled.ActorProfileNSID, Repo: user.Did, Rkey: "self", Record: &lexutil.LexiconTypeDecoder{Val: profileRecord}, SwapRecord: getRecordResp.Cid, })
if err != nil { l.Error("failed to update profile record", "err", err) s.pages.Notice(w, "avatar-error", "Failed to remove avatar from your PDS") return }
l.Info("successfully removed avatar from PDS")
profile, err := db.GetProfile(s.db, user.Did) if err != nil { l.Warn("getting profile data from DB", "err", err) } if profile == nil { profile = &models.Profile{Did: user.Did} } profile.Avatar = ""
tx, err := s.db.BeginTx(r.Context(), nil) if err != nil { l.Error("failed to start transaction", "err", err) s.pages.HxRefresh(w) w.WriteHeader(http.StatusOK) return }
err = db.UpsertProfile(tx, profile) if err != nil { l.Error("failed to update profile in DB", "err", err) s.pages.HxRefresh(w) w.WriteHeader(http.StatusOK) return }
s.pages.HxRedirect(w, r.Header.Get("Referer"))}
func (s *State) UpdateProfilePunchcardSetting(w http.ResponseWriter, r *http.Request) { l := s.logger.With("handler", "UpdateProfilePunchcardSetting") err := r.ParseForm() if err != nil { l.Error("invalid profile update form", "err", err) return } user := s.oauth.GetMultiAccountUser(r)
hideOthers := false hideMine := false
if r.Form.Get("hideMine") == "on" { hideMine = true } if r.Form.Get("hideOthers") == "on" { hideOthers = true }
err = db.UpsertPunchcardPreference(s.db, user.Did, hideMine, hideOthers) if err != nil { l.Error("failed to update punchcard preferences", "err", err) return }
s.pages.HxRefresh(w)}