gosora/common/profile_reply.go

package common

import (
	"database/sql"
	"html"
	"strconv"
	"time"

	qgen "git.tuxpa.in/a/gosora/query_gen"
)

var profileReplyStmts ProfileReplyStmts

type ProfileReply struct {
    ID           int
    ParentID     int
    Content      string
    CreatedBy    int
    Group        int
    CreatedAt    time.Time
    LastEdit     int
    LastEditBy   int
    ContentLines int
    IP           string
}

type ProfileReplyStmts struct {
    edit   *sql.Stmt
    delete *sql.Stmt
}

func init() {
    DbInits.Add(func(acc *qgen.Accumulator) error {
        ur := "users_replies"
        profileReplyStmts = ProfileReplyStmts{
            edit:   acc.Update(ur).Set("content=?,parsed_content=?").Where("rid=?").Prepare(),
            delete: acc.Delete(ur).Where("rid=?").Prepare(),
        }
        return acc.FirstError()
    })
}

// Mostly for tests, so we don't wind up with out-of-date profile reply initialisation logic there
func BlankProfileReply(id int) *ProfileReply {
    return &ProfileReply{ID: id}
}

// TODO: Write tests for this
func (r *ProfileReply) Delete() error {
    _, err := profileReplyStmts.delete.Exec(r.ID)
    if err != nil {
        return err
    }
    // TODO: Better coupling between the two paramsextra queries
    aids, err := Activity.AidsByParamsExtra("reply", r.ParentID, "user", strconv.Itoa(r.ID))
    if err != nil {
        return err
    }
    for _, aid := range aids {
        DismissAlert(r.ParentID, aid)
    }
    err = Activity.DeleteByParamsExtra("reply", r.ParentID, "user", strconv.Itoa(r.ID))
    return err
}

func (r *ProfileReply) SetBody(content string) error {
    content = PreparseMessage(html.UnescapeString(content))
    _, err := profileReplyStmts.edit.Exec(content, ParseMessage(content, 0, "", nil, nil), r.ID)
    return err
}

// TODO: We can get this from the topic store instead of a query which will always miss the cache...
func (r *ProfileReply) Creator() (*User, error) {
    return Users.Get(r.CreatedBy)
}