package arn import ( "errors" "fmt" "reflect" "strings" "github.com/aerogo/aero" "github.com/aerogo/api" "github.com/aerogo/markdown" "github.com/animenotifier/notify.moe/arn/autocorrect" ) // Force interface implementations var ( _ Postable = (*Post)(nil) _ Likeable = (*Post)(nil) _ LikeEventReceiver = (*Post)(nil) _ PostParent = (*Post)(nil) _ fmt.Stringer = (*Post)(nil) _ api.Newable = (*Post)(nil) _ api.Editable = (*Post)(nil) _ api.Actionable = (*Post)(nil) _ api.Deletable = (*Post)(nil) ) // Actions func init() { API.RegisterActions("Post", []*api.Action{ // Like post LikeAction(), // Unlike post UnlikeAction(), }) } // Authorize returns an error if the given API POST request is not authorized. func (post *Post) Authorize(ctx aero.Context, action string) error { if !ctx.HasSession() { return errors.New("Neither logged in nor in session") } if action == "edit" { user := GetUserFromContext(ctx) if post.CreatedBy != user.ID && user.Role != "admin" { return errors.New("Can't edit the posts of other users") } } return nil } // Create sets the data for a new post with data we received from the API request. func (post *Post) Create(ctx aero.Context) error { data, err := ctx.Request().Body().JSONObject() if err != nil { return err } user := GetUserFromContext(ctx) if user == nil { return errors.New("Not logged in") } post.ID = GenerateID("Post") post.Text, _ = data["text"].(string) post.CreatedBy = user.ID post.ParentID, _ = data["parentId"].(string) post.ParentType, _ = data["parentType"].(string) post.Created = DateTimeUTC() post.Edited = "" // Check parent type if !DB.HasType(post.ParentType) { return errors.New("Invalid parent type: " + post.ParentType) } // Post-process text post.Text = autocorrect.PostText(post.Text) if len(post.Text) < 5 { return errors.New("Text too short: Should be at least 5 characters") } // Tags tags, _ := data["tags"].([]interface{}) post.Tags = make([]string, len(tags)) for i := range post.Tags { post.Tags[i] = tags[i].(string) } // Parent parent := post.Parent() if parent == nil { return errors.New(post.ParentType + " does not exist") } // Is the parent locked? if IsLocked(parent) { return errors.New(post.ParentType + " is locked") } // Don't allow posting when you're not a group member topMostParent := post.TopMostParent() if topMostParent.TypeName() == "Group" { group := topMostParent.(*Group) if !group.HasMember(user.ID) { return errors.New("Only group members can post in groups") } } // Append to posts parent.AddPost(post.ID) // Save the parent thread parent.Save() // Send notification to the author of the parent post go func() { notifyUser := parent.Creator() // Does the parent have a creator? if notifyUser == nil { return } // Don't notify the author himself if notifyUser.ID == post.CreatedBy { return } title := user.Nick + " replied" message := "" switch post.ParentType { case "Post": message = fmt.Sprintf("%s replied to your comment in \"%s\".", user.Nick, parent.(*Post).Parent().TitleByUser(notifyUser)) case "User": title = fmt.Sprintf("%s wrote a comment on your profile.", user.Nick) message = post.Text case "Group": title = fmt.Sprintf(`%s wrote a new post in the group "%s".`, user.Nick, parent.TitleByUser(nil)) message = post.Text default: message = fmt.Sprintf(`%s replied in the %s "%s".`, user.Nick, strings.ToLower(post.ParentType), parent.TitleByUser(notifyUser)) } notification := &PushNotification{ Title: title, Message: message, Icon: "https:" + user.AvatarLink("large"), Link: post.Link(), Type: NotificationTypeForumReply, } // If you're posting to a group, // all members except the author will receive a notification. if post.ParentType == "Group" { group := parent.(*Group) group.SendNotification(notification, user.ID) return } notifyUser.SendNotification(notification) }() // Write log entry logEntry := NewEditLogEntry(user.ID, "create", "Post", post.ID, "", "", "") logEntry.Save() // Create activity activity := NewActivityCreate("Post", post.ID, user.ID) activity.Save() return nil } // Edit saves a log entry for the edit. func (post *Post) Edit(ctx aero.Context, key string, value reflect.Value, newValue reflect.Value) (bool, error) { consumed := false user := GetUserFromContext(ctx) // This should stay a switch statement. // nolint:gocritic switch key { case "ParentID": var newParent PostParent newParentID := newValue.String() newParent, err := GetPost(newParentID) if err != nil { newParent, err = GetThread(newParentID) if err != nil { return false, err } } post.SetParent(newParent) consumed = true } // Write log entry logEntry := NewEditLogEntry(user.ID, "edit", "Post", post.ID, key, fmt.Sprint(value.Interface()), fmt.Sprint(newValue.Interface())) logEntry.Save() return consumed, nil } // OnAppend saves a log entry. func (post *Post) OnAppend(ctx aero.Context, key string, index int, obj interface{}) { onAppend(post, ctx, key, index, obj) } // OnRemove saves a log entry. func (post *Post) OnRemove(ctx aero.Context, key string, index int, obj interface{}) { onRemove(post, ctx, key, index, obj) } // AfterEdit sets the edited date on the post object. func (post *Post) AfterEdit(ctx aero.Context) error { post.Edited = DateTimeUTC() post.html = markdown.Render(post.Text) return nil } // DeleteInContext deletes the post in the given context. func (post *Post) DeleteInContext(ctx aero.Context) error { user := GetUserFromContext(ctx) // Write log entry logEntry := NewEditLogEntry(user.ID, "delete", "Post", post.ID, "", fmt.Sprint(post), "") logEntry.Save() return post.Delete() } // Delete deletes the post from the database. func (post *Post) Delete() error { // Remove child posts first for _, post := range post.Posts() { err := post.Delete() if err != nil { return err } } parent := post.Parent() if parent == nil { return fmt.Errorf("Invalid %s parent ID: %s", post.ParentType, post.ParentID) } // Remove the reference of the post in the thread that contains it if !parent.RemovePost(post.ID) { return fmt.Errorf("This post does not exist in the %s", strings.ToLower(post.ParentType)) } parent.Save() // Remove activities for activity := range StreamActivityCreates() { if activity.ObjectID == post.ID && activity.ObjectType == "Post" { err := activity.Delete() if err != nil { return err } } } DB.Delete("Post", post.ID) return nil } // Save saves the post object in the database. func (post *Post) Save() { DB.Set("Post", post.ID, post) }