24129c4852
- ensure all scraper functions have a `api.XYZ` version and a package-level convenience function - isolate `the_api` to top-level convenience functions, in preparation for removal - move a bunch of scraper functions around to be nearby their related functions - new ErrLoginRequired - remove obsolete APIv1 stuff (Feed, TweetDetail) - rename scraper function GetUserFeedGraphqlFor => GetUserFeed - fix go.mod Go version incorrectly claiming it's compatible with Go 1.16 (should be Go 1.17)
95 lines
3.3 KiB
Go
95 lines
3.3 KiB
Go
package scraper
|
|
|
|
import (
|
|
"fmt"
|
|
"net/url"
|
|
"path"
|
|
)
|
|
|
|
type DMChatRoomID string
|
|
|
|
// A participant in a chat room.
|
|
//
|
|
// Most settings will only be fetched for the logged-in user (other chat members will just be "false" for
|
|
// everything). The "IsChatSettingsValid" flag indicates whether this is the case.
|
|
type DMChatParticipant struct {
|
|
DMChatRoomID DMChatRoomID `db:"chat_room_id"`
|
|
UserID UserID `db:"user_id"`
|
|
LastReadEventID DMMessageID `db:"last_read_event_id"`
|
|
|
|
IsChatSettingsValid bool `db:"is_chat_settings_valid"`
|
|
IsNotificationsDisabled bool `db:"is_notifications_disabled"`
|
|
IsMentionNotificationsDisabled bool `db:"is_mention_notifications_disabled"`
|
|
IsReadOnly bool `db:"is_read_only"`
|
|
IsTrusted bool `db:"is_trusted"`
|
|
IsMuted bool `db:"is_muted"`
|
|
Status string `db:"status"`
|
|
}
|
|
|
|
// A chat room. Stores a map of chat participants and a reference to the most recent message,
|
|
// for preview purposes.
|
|
type DMChatRoom struct {
|
|
ID DMChatRoomID `db:"id"`
|
|
Type string `db:"type"`
|
|
LastMessagedAt Timestamp `db:"last_messaged_at"` // Used for ordering the chats in the UI
|
|
IsNSFW bool `db:"is_nsfw"`
|
|
|
|
// GROUP_DM rooms
|
|
CreatedAt Timestamp `db:"created_at"`
|
|
CreatedByUserID UserID `db:"created_by_user_id"`
|
|
Name string `db:"name"`
|
|
AvatarImageRemoteURL string `db:"avatar_image_remote_url"`
|
|
AvatarImageLocalPath string `db:"avatar_image_local_path"`
|
|
|
|
LastMessageID DMMessageID `db:"last_message_id"` // Not stored, but used to generate preview
|
|
Participants map[UserID]DMChatParticipant
|
|
}
|
|
|
|
func (r DMChatRoom) GetParticipantIDs() []UserID {
|
|
ret := []UserID{}
|
|
for user_id := range r.Participants {
|
|
ret = append(ret, user_id)
|
|
}
|
|
return ret
|
|
}
|
|
|
|
func ParseAPIDMChatRoom(api_room APIDMConversation, current_user_id UserID) DMChatRoom {
|
|
ret := DMChatRoom{}
|
|
ret.ID = DMChatRoomID(api_room.ConversationID)
|
|
ret.Type = api_room.Type
|
|
ret.LastMessagedAt = TimestampFromUnixMilli(int64(api_room.SortTimestamp))
|
|
ret.IsNSFW = api_room.NSFW
|
|
|
|
if ret.Type == "GROUP_DM" {
|
|
ret.CreatedAt = TimestampFromUnixMilli(int64(api_room.CreateTime))
|
|
ret.CreatedByUserID = UserID(api_room.CreatedByUserID)
|
|
ret.Name = api_room.Name
|
|
ret.AvatarImageRemoteURL = api_room.AvatarImage
|
|
tmp_url, err := url.Parse(ret.AvatarImageRemoteURL)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
ret.AvatarImageLocalPath = fmt.Sprintf("%s_avatar_%s.%s", ret.ID, path.Base(tmp_url.Path), tmp_url.Query().Get("format"))
|
|
}
|
|
|
|
ret.Participants = make(map[UserID]DMChatParticipant)
|
|
for _, api_participant := range api_room.Participants {
|
|
participant := DMChatParticipant{}
|
|
participant.UserID = UserID(api_participant.UserID)
|
|
participant.DMChatRoomID = ret.ID
|
|
participant.LastReadEventID = DMMessageID(api_participant.LastReadEventID)
|
|
|
|
// Process chat settings if this is the logged-in user
|
|
if participant.UserID == current_user_id {
|
|
participant.IsNotificationsDisabled = api_room.NotificationsDisabled
|
|
participant.IsReadOnly = api_room.ReadOnly
|
|
participant.IsTrusted = api_room.Trusted
|
|
participant.IsMuted = api_room.Muted
|
|
participant.Status = api_room.Status
|
|
participant.IsChatSettingsValid = true
|
|
}
|
|
ret.Participants[participant.UserID] = participant
|
|
}
|
|
return ret
|
|
}
|