Added arn to the main repository
This commit is contained in:
288
arn/PostAPI.go
Normal file
288
arn/PostAPI.go
Normal file
@ -0,0 +1,288 @@
|
||||
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)
|
||||
|
||||
// nolint:gocritic (because this should stay a switch statement)
|
||||
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)
|
||||
}
|
||||
Reference in New Issue
Block a user