135 lines
3.0 KiB
Go
135 lines
3.0 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
|
|
artmdl "go-common/app/interface/openplatform/article/model"
|
|
"go-common/library/ecode"
|
|
"go-common/library/log"
|
|
)
|
|
|
|
func (s Service) checkDraftAuthor(c context.Context, aid, mid int64) (d *artmdl.Draft, err error) {
|
|
if d, err = s.dao.ArtDraft(c, mid, aid); err != nil {
|
|
return
|
|
}
|
|
if d == nil {
|
|
err = ecode.NothingFound
|
|
return
|
|
}
|
|
if d.Author.Mid != mid {
|
|
err = ecode.ArtCreationMIDErr
|
|
}
|
|
return
|
|
}
|
|
|
|
// ArtDraft get article draft by id.
|
|
func (s *Service) ArtDraft(c context.Context, aid, mid int64) (res *artmdl.Draft, err error) {
|
|
if err = s.checkPrivilege(c, mid); err != nil {
|
|
return
|
|
}
|
|
if res, err = s.checkDraftAuthor(c, aid, mid); err != nil {
|
|
return
|
|
}
|
|
if res.ListID > 0 {
|
|
res.List, _ = s.dao.List(c, res.ListID)
|
|
}
|
|
return
|
|
}
|
|
|
|
// AddArtDraft add article draft .
|
|
func (s *Service) AddArtDraft(c context.Context, a *artmdl.Draft) (id int64, err error) {
|
|
log.Infov(c, log.KV("AddArtDraft", a))
|
|
if err = s.checkPrivilege(c, a.Author.Mid); err != nil {
|
|
return
|
|
}
|
|
a.Content = xssFilter(a.Content)
|
|
if err = s.preDraftCheck(c, a); err != nil {
|
|
return
|
|
}
|
|
if _, err = s.checkList(c, a.Author.Mid, a.ListID); err != nil {
|
|
return
|
|
}
|
|
// if a.ListID > 0 {
|
|
// var exist bool
|
|
// for _, cid := range s.novelCIDs() {
|
|
// if cid == a.Category.ID {
|
|
// exist = true
|
|
// break
|
|
// }
|
|
// }
|
|
// if !exist {
|
|
// a.ListID = 0
|
|
// }
|
|
// }
|
|
if a.ID > 0 {
|
|
if _, err = s.checkDraftAuthor(c, a.ID, a.Author.Mid); err != nil {
|
|
return
|
|
}
|
|
_, err = s.dao.AddArtDraft(c, a)
|
|
id = a.ID
|
|
log.Info("update draft success mid(%d) aid(%d)", a.Author.Mid, a.ID)
|
|
return
|
|
}
|
|
var total int
|
|
if total, err = s.dao.CountUpperDraft(c, a.Author.Mid); err != nil {
|
|
return
|
|
}
|
|
if total > s.c.Article.UpperDraftLimit {
|
|
err = ecode.ArtCreationDraftFull
|
|
return
|
|
}
|
|
id, err = s.dao.AddArtDraft(c, a)
|
|
return
|
|
}
|
|
|
|
// DelArtDraft deletes article draft.
|
|
func (s *Service) DelArtDraft(c context.Context, aid, mid int64) (err error) {
|
|
if err = s.checkPrivilege(c, mid); err != nil {
|
|
return
|
|
}
|
|
if _, err = s.checkDraftAuthor(c, aid, mid); err != nil {
|
|
return
|
|
}
|
|
err = s.dao.DelArtDraft(c, mid, aid)
|
|
return
|
|
}
|
|
|
|
// UpperDrafts batch get draft by mid.
|
|
func (s *Service) UpperDrafts(c context.Context, mid int64, pn, ps int) (res *artmdl.Drafts, err error) {
|
|
if err = s.checkPrivilege(c, mid); err != nil {
|
|
return
|
|
}
|
|
var (
|
|
total int
|
|
ds []*artmdl.Draft
|
|
start = (pn - 1) * ps
|
|
page = &artmdl.ArtPage{Pn: pn, Ps: ps}
|
|
)
|
|
res = &artmdl.Drafts{Page: page}
|
|
if total, err = s.dao.CountUpperDraft(c, mid); err != nil {
|
|
return
|
|
} else if total == 0 {
|
|
return
|
|
}
|
|
page.Total = total
|
|
if ds, err = s.dao.UpperDrafts(c, mid, start, ps); err != nil {
|
|
return
|
|
}
|
|
for _, v := range ds {
|
|
// 用户没选择分区，返回空分区信息
|
|
if v.Category.ID == 0 {
|
|
continue
|
|
}
|
|
var pid int64
|
|
if pid, err = s.CategoryToRoot(v.Category.ID); err != nil {
|
|
log.Error("s.CategoryToRoot(%d) error(%+v)", v.Category.ID, err)
|
|
err = nil
|
|
continue
|
|
}
|
|
v.Category = s.categoriesMap[pid]
|
|
v.List, _ = s.dao.List(c, v.ListID)
|
|
}
|
|
res.Drafts = ds
|
|
return
|
|
}
|