fixed a few things

This commit is contained in:
tsmethurst 2021-05-12 21:47:45 +02:00
parent 884d0ecc8f
commit 3ba528120c
6 changed files with 92 additions and 106 deletions

View File

@ -41,11 +41,11 @@ type Mention struct {
// NameString is for putting in the namestring of the mentioned user
// before the mention is dereferenced. Should be in a form along the lines of:
// @whatever_username@example.org
//
// This will not be put in the database, it's just for convenience.
NameString string `pg:"-"`
// Href is the web URL (not AP uri!) of the user mentioned.
//
// This will not be put in the database, it's just for convenience.
Href string `pg:"-"`
NameString string `pg:"-"`
// MentionedAccountURI is the AP ID (uri) of the user mentioned.
//
// This will not be put in the database, it's just for convenience.
MentionedAccountURI string `pg:"-"`
}

View File

@ -71,12 +71,14 @@ type Status struct {
Text string
/*
NON-DATABASE FIELDS
INTERNAL MODEL NON-DATABASE FIELDS
These are for convenience while passing the status around internally,
but these fields should *never* be put in the db.
*/
// Account that created this status
GTSAccount *Account `pg:"-"`
// Mentions created in this status
GTSMentions []*Mention `pg:"-"`
// Hashtags used in this status
@ -93,6 +95,20 @@ type Status struct {
GTSBoostedStatus *Status `pg:"-"`
// Account of the boosted status
GTSBoostedAccount *Account `pg:"-"`
/*
AP NON-DATABASE FIELDS
These are for convenience while passing the status around internally,
but these fields should *never* be put in the db.
*/
// AP URI of the status being replied to.
// Useful when that status doesn't exist in the database yet and we still need to dereference it.
APReplyToStatusURI string `pg:"-"`
// The AP URI of the owner/creator of the status.
// Useful when that account doesn't exist in the database yet and we still need to dereference it.
APStatusOwnerURI string `pg:"-"`
}
// Visibility represents the visibility granularity of a status.

View File

@ -388,7 +388,6 @@ func extractHashtags(i withTag) ([]*gtsmodel.Tag, error) {
tag, err := extractHashtag(hashtaggable)
if err != nil {
fmt.Println(err)
continue
}
@ -516,13 +515,13 @@ func extractMention(i Mentionable) (*gtsmodel.Mention, error) {
if username == "" || domain == "" {
return nil, errors.New("username or domain was empty")
}
mention.NameString = mentionString
// the href prop should be the URL of a user we know, eg https://example.org/@whatever_user
// the href prop should be the AP URI of a user we know, eg https://example.org/users/whatever_user
hrefProp := i.GetActivityStreamsHref()
if hrefProp == nil || !hrefProp.IsIRI() {
return nil, errors.New("no href prop")
}
mention.Href = hrefProp.GetIRI().String()
mention.MentionedAccountURI = hrefProp.GetIRI().String()
return mention, nil
}

View File

@ -161,134 +161,108 @@ func (c *converter) ASRepresentationToAccount(accountable Accountable) (*gtsmode
func (c *converter) ASStatusToStatus(statusable Statusable) (*gtsmodel.Status, error) {
status := &gtsmodel.Status{}
// uri at which this status is reachable
uriProp := statusable.GetJSONLDId()
if uriProp == nil || !uriProp.IsIRI() {
return nil, errors.New("no id property found, or id was not an iri")
}
status.URI = uriProp.GetIRI().String()
statusURL, err := extractURL(statusable)
if err == nil {
// web url for viewing this status
if statusURL, err := extractURL(statusable); err == nil {
status.URL = statusURL.String()
}
// the html-formatted content of this status
if content, err := extractContent(statusable); err == nil {
status.Content = content
}
attachments, err := extractAttachments(statusable); if err == nil {
// attachments to dereference and fetch later on (we don't do that here)
if attachments, err := extractAttachments(statusable); err == nil {
status.GTSMediaAttachments = attachments
}
hashtags, err := extractHashtags(statusable)
if err == nil {
// hashtags to dereference later on
if hashtags, err := extractHashtags(statusable); err == nil {
status.GTSTags = hashtags
}
emojis, err := extractEmojis(statusable)
if err == nil {
// emojis to dereference and fetch later on
if emojis, err := extractEmojis(statusable); err == nil {
status.GTSEmojis = emojis
}
mentions, err := extractMentions(statusable)
if err == nil {
// mentions to dereference later on
if mentions, err := extractMentions(statusable); err == nil {
status.GTSMentions = mentions
}
cw, err := extractSummary(statusable)
if err == nil && cw != "" {
// cw string for this status
if cw, err := extractSummary(statusable); err == nil {
status.ContentWarning = cw
}
inReplyToURI, err := extractInReplyToURI(statusable)
if err == nil {
inReplyToStatus := &gtsmodel.Status{}
if err := c.db.GetWhere("uri", inReplyToURI.String(), inReplyToStatus); err == nil {
status.InReplyToID = inReplyToStatus.ID
}
}
// when was this status created?
published, err := extractPublished(statusable)
if err == nil {
status.CreatedAt = published
}
// which account posted this status?
// if we don't know the account yet we can dereference it later
attributedTo, err := extractAttributedTo(statusable)
if err != nil {
return nil, errors.New("attributedTo was empty")
}
status.APStatusOwnerURI = attributedTo.String()
// if we don't know the account yet we can dereference it later
statusOwner := &gtsmodel.Status{}
statusOwner := &gtsmodel.Account{}
if err := c.db.GetWhere("uri", attributedTo.String(), statusOwner); err == nil {
status.AccountID = statusOwner.ID
status.GTSAccount = statusOwner
}
// check if there's a post that this is a reply to
inReplyToURI, err := extractInReplyToURI(statusable)
if err == nil {
// something is set so we can at least set this field on the
// status and dereference using this later if we need to
status.APReplyToStatusURI = inReplyToURI.String()
// now we can check if we have the replied-to status in our db already
inReplyToStatus := &gtsmodel.Status{}
if err := c.db.GetWhere("uri", inReplyToURI.String(), inReplyToStatus); err == nil {
// we have the status in our database already
// so we can set these fields here and then...
status.InReplyToID = inReplyToStatus.ID
status.InReplyToAccountID = inReplyToStatus.AccountID
status.GTSReplyToStatus = inReplyToStatus
// ... check if we've seen the account already
inReplyToAccount := &gtsmodel.Account{}
if err := c.db.GetByID(inReplyToStatus.AccountID, inReplyToAccount); err == nil {
status.GTSReplyToAccount = inReplyToAccount
}
}
}
// visibility entry for this status
// TODO: if it's just got followers in TO and it's not CC'ed to public, it's followers only
// TODO: if it's CC'ed to public, it's public or unlocked
// TODO: if it's a DM then it's addressed to SPECIFIC ACCOUNTS and not followers or public
// TODO: mentioned SPECIFIC ACCOUNTS also get added to CC'es if it's not a direct message
// advanced visibility for this status
// TODO: a lot of work to be done here -- a new type needs to be created for this in go-fed/activity using ASTOOL
// sensitive
// TODO: this is a bool
// language
// we might be able to extract this from the contentMap field
status.ActivityStreamsType = gtsmodel.ActivityStreamsObject(statusable.GetTypeName())
return status, nil
}
// // id of the status in the database
// ID string `pg:"type:uuid,default:gen_random_uuid(),pk,notnull"`
// // uri at which this status is reachable
// URI string `pg:",unique"`
// // web url for viewing this status
// URL string `pg:",unique"`
// // the html-formatted content of this status
// Content string
// // Database IDs of any media attachments associated with this status
// Attachments []string `pg:",array"`
// // Database IDs of any tags used in this status
// Tags []string `pg:",array"`
// // Database IDs of any accounts mentioned in this status
// Mentions []string `pg:",array"`
// // Database IDs of any emojis used in this status
// Emojis []string `pg:",array"`
// // when was this status created?
// CreatedAt time.Time `pg:"type:timestamp,notnull,default:now()"`
// // when was this status updated?
// UpdatedAt time.Time `pg:"type:timestamp,notnull,default:now()"`
// // is this status from a local account?
// Local bool
// // which account posted this status?
// AccountID string
// // id of the status this status is a reply to
// InReplyToID string
// // id of the account that this status replies to
// InReplyToAccountID string
// // id of the status this status is a boost of
// BoostOfID string
// // cw string for this status
// ContentWarning string
// // visibility entry for this status
// Visibility Visibility `pg:",notnull"`
// // mark the status as sensitive?
// Sensitive bool
// // what language is this status written in?
// Language string
// // Which application was used to create this status?
// CreatedWithApplicationID string
// // advanced visibility for this status
// VisibilityAdvanced *VisibilityAdvanced
// // What is the activitystreams type of this status? See: https://www.w3.org/TR/activitystreams-vocabulary/#object-types
// // Will probably almost always be Note but who knows!.
// ActivityStreamsType ActivityStreamsObject
// // Original text of the status without formatting
// Text string
// // Mentions created in this status
// GTSMentions []*Mention `pg:"-"`
// // Hashtags used in this status
// GTSTags []*Tag `pg:"-"`
// // Emojis used in this status
// GTSEmojis []*Emoji `pg:"-"`
// // MediaAttachments used in this status
// GTSMediaAttachments []*MediaAttachment `pg:"-"`
// // Status being replied to
// GTSReplyToStatus *Status `pg:"-"`
// // Account being replied to
// GTSReplyToAccount *Account `pg:"-"`
// // Status being boosted
// GTSBoostedStatus *Status `pg:"-"`
// // Account of the boosted status
// GTSBoostedAccount *Account `pg:"-"`

View File

@ -35,8 +35,10 @@ const (
)
var (
mentionNameRegexString = `@([a-zA-Z0-9_]+)(?:@([a-zA-Z0-9_\-\.]+)?)`
mentionNameRegex = regexp.MustCompile(fmt.Sprintf(`^%s$`, mentionNameRegexString))
mentionNameRegexString = `^@([a-zA-Z0-9_]+)(?:@([a-zA-Z0-9_\-\.]+)?)$`
// mention name regex captures the username and domain part from a mention string
// such as @whatever_user@example.org, returning whatever_user and example.org (without the @ symbols)
mentionNameRegex = regexp.MustCompile(mentionNameRegexString)
// mention regex can be played around with here: https://regex101.com/r/qwM9D3/1
mentionFinderRegexString = `(?: |^|\W)(@[a-zA-Z0-9_]+(?:@[a-zA-Z0-9_\-\.]+)?)(?: |\n)`

View File

@ -62,23 +62,18 @@ func DeriveEmojisFromStatus(status string) []string {
return lower(unique(emojis))
}
// ExtractMentionParts extracts the username @test_user and the domain @example.org
// ExtractMentionParts extracts the username test_user and the domain example.org
// from a mention string like @test_user@example.org.
//
// If no domain is provided, it will return just the username part.
//
// If nothing is matched, it will return an error.
func ExtractMentionParts(mention string) (username, domain string, err error) {
matches := mentionNameRegex.FindStringSubmatch(mention)
if matches == nil {
if matches == nil || len(matches) != 3 {
err = fmt.Errorf("could't match mention %s", mention)
return
}
fmt.Println(matches)
username = matches[1]
if len(matches) == 2 {
domain = matches[2]
}
domain = matches[2]
return
}