mirror of https://github.com/sunface/rust-course
sunface
4 years ago
parent
3a1c406da4
commit
453d8c7718
@ -1,141 +0,0 @@
|
|||||||
package posts
|
|
||||||
|
|
||||||
import (
|
|
||||||
"database/sql"
|
|
||||||
"net/http"
|
|
||||||
"sort"
|
|
||||||
"strings"
|
|
||||||
"time"
|
|
||||||
"unicode/utf8"
|
|
||||||
|
|
||||||
"github.com/asaskevich/govalidator"
|
|
||||||
"github.com/gin-gonic/gin"
|
|
||||||
"github.com/imdotdev/im.dev/server/internal/session"
|
|
||||||
"github.com/imdotdev/im.dev/server/pkg/config"
|
|
||||||
"github.com/imdotdev/im.dev/server/pkg/db"
|
|
||||||
"github.com/imdotdev/im.dev/server/pkg/e"
|
|
||||||
"github.com/imdotdev/im.dev/server/pkg/models"
|
|
||||||
"github.com/imdotdev/im.dev/server/pkg/utils"
|
|
||||||
)
|
|
||||||
|
|
||||||
func UserPosts(uid int64) (models.Posts, *e.Error) {
|
|
||||||
ars := make(models.Posts, 0)
|
|
||||||
rows, err := db.Conn.Query("select id,title,url,cover,brief,created,updated from posts where creator=?", uid)
|
|
||||||
if err != nil {
|
|
||||||
if err == sql.ErrNoRows {
|
|
||||||
return ars, e.New(http.StatusNotFound, e.NotFound)
|
|
||||||
}
|
|
||||||
logger.Warn("get user posts error", "error", err)
|
|
||||||
return ars, e.New(http.StatusInternalServerError, e.Internal)
|
|
||||||
}
|
|
||||||
|
|
||||||
creator := &models.UserSimple{ID: uid}
|
|
||||||
creator.Query()
|
|
||||||
for rows.Next() {
|
|
||||||
ar := &models.Post{}
|
|
||||||
err := rows.Scan(&ar.ID, &ar.Title, &ar.URL, &ar.Cover, &ar.Brief, &ar.Created, &ar.Updated)
|
|
||||||
if err != nil {
|
|
||||||
logger.Warn("scan post error", "error", err)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
ar.Creator = creator
|
|
||||||
ars = append(ars, ar)
|
|
||||||
}
|
|
||||||
|
|
||||||
sort.Sort(ars)
|
|
||||||
return ars, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func SubmitPost(c *gin.Context) *e.Error {
|
|
||||||
user := session.CurrentUser(c)
|
|
||||||
if !user.Role.IsEditor() {
|
|
||||||
return e.New(http.StatusForbidden, e.NoEditorPermission)
|
|
||||||
}
|
|
||||||
|
|
||||||
ar := &models.Post{}
|
|
||||||
err := c.Bind(&ar)
|
|
||||||
if err != nil {
|
|
||||||
return e.New(http.StatusBadRequest, e.ParamInvalid)
|
|
||||||
}
|
|
||||||
|
|
||||||
if strings.TrimSpace(ar.Title) == "" || utf8.RuneCountInString(ar.Brief) > config.Data.Posts.BriefMaxLen {
|
|
||||||
return e.New(http.StatusBadRequest, e.ParamInvalid)
|
|
||||||
}
|
|
||||||
|
|
||||||
if strings.TrimSpace(ar.URL) != "" && !govalidator.IsURL(ar.URL) {
|
|
||||||
return e.New(http.StatusBadRequest, e.ParamInvalid)
|
|
||||||
}
|
|
||||||
|
|
||||||
if strings.TrimSpace(ar.Cover) != "" && !govalidator.IsURL(ar.Cover) {
|
|
||||||
return e.New(http.StatusBadRequest, e.ParamInvalid)
|
|
||||||
}
|
|
||||||
|
|
||||||
now := time.Now()
|
|
||||||
|
|
||||||
md := utils.Compress(ar.Md)
|
|
||||||
if ar.ID == 0 {
|
|
||||||
//create
|
|
||||||
_, err = db.Conn.Exec("INSERT INTO posts (creator, title, md, url, cover, brief, created, updated) VALUES(?,?,?,?,?,?,?,?)",
|
|
||||||
user.ID, ar.Title, md, ar.URL, ar.Cover, ar.Brief, now, now)
|
|
||||||
if err != nil {
|
|
||||||
logger.Warn("submit post error", "error", err)
|
|
||||||
return e.New(http.StatusInternalServerError, e.Internal)
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
_, err = db.Conn.Exec("UPDATE posts SET title=?, md=?, url=?, cover=?, brief=?, updated=? WHERE id=?",
|
|
||||||
ar.Title, md, ar.URL, ar.Cover, ar.Brief, now, ar.ID)
|
|
||||||
if err != nil {
|
|
||||||
logger.Warn("upate post error", "error", err)
|
|
||||||
return e.New(http.StatusInternalServerError, e.Internal)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func DeletePost(id int64) *e.Error {
|
|
||||||
_, err := db.Conn.Exec("DELETE FROM posts WHERE id=?", id)
|
|
||||||
if err != nil {
|
|
||||||
logger.Warn("delete post error", "error", err)
|
|
||||||
return e.New(http.StatusInternalServerError, e.Internal)
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func GetPost(id int64) (*models.Post, *e.Error) {
|
|
||||||
ar := &models.Post{}
|
|
||||||
var rawmd []byte
|
|
||||||
err := db.Conn.QueryRow("select id,title,md,url,cover,brief,creator,created,updated from posts where id=?", id).Scan(
|
|
||||||
&ar.ID, &ar.Title, &rawmd, &ar.URL, &ar.Cover, &ar.Brief, &ar.CreatorID, &ar.Created, &ar.Updated,
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
if err == sql.ErrNoRows {
|
|
||||||
return nil, e.New(http.StatusNotFound, e.NotFound)
|
|
||||||
}
|
|
||||||
logger.Warn("get post error", "error", err)
|
|
||||||
return nil, e.New(http.StatusInternalServerError, e.Internal)
|
|
||||||
}
|
|
||||||
|
|
||||||
md, _ := utils.Uncompress(rawmd)
|
|
||||||
ar.Md = string(md)
|
|
||||||
ar.Creator = &models.UserSimple{ID: ar.CreatorID}
|
|
||||||
err = ar.Creator.Query()
|
|
||||||
|
|
||||||
return ar, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func GetPostCreator(id int64) (int64, *e.Error) {
|
|
||||||
var uid int64
|
|
||||||
err := db.Conn.QueryRow("SELECT creator FROM posts WHERE id=?", id).Scan(&uid)
|
|
||||||
if err != nil {
|
|
||||||
if err == sql.ErrNoRows {
|
|
||||||
return 0, e.New(http.StatusNotFound, e.NotFound)
|
|
||||||
}
|
|
||||||
logger.Warn("get post creator error", "error", err)
|
|
||||||
return 0, e.New(http.StatusInternalServerError, e.Internal)
|
|
||||||
}
|
|
||||||
|
|
||||||
return uid, nil
|
|
||||||
}
|
|
||||||
@ -0,0 +1,23 @@
|
|||||||
|
package utils
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/base64"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/gosimple/slug"
|
||||||
|
)
|
||||||
|
|
||||||
|
func Slugify(raw string) string {
|
||||||
|
s := slug.Make(strings.ToLower(raw))
|
||||||
|
if s == "" {
|
||||||
|
// If the raw name is only characters outside of the
|
||||||
|
// sluggable characters, the slug creation will return an
|
||||||
|
// empty string which will mess up URLs. This failsafe picks
|
||||||
|
// that up and creates the slug as a base64 identifier instead.
|
||||||
|
s = base64.RawURLEncoding.EncodeToString([]byte(raw))
|
||||||
|
if slug.MaxLength != 0 && len(s) > slug.MaxLength {
|
||||||
|
s = s[:slug.MaxLength]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return s
|
||||||
|
}
|
||||||
@ -1,6 +1,7 @@
|
|||||||
export enum Role {
|
export enum Role {
|
||||||
NORMAL = "Normal",
|
NORMAL = "Normal",
|
||||||
EDITOR = "Editor",
|
EDITOR = "Editor",
|
||||||
|
ROLE_CREATOR = "Creator",
|
||||||
ADMIN = "Admin",
|
ADMIN = "Admin",
|
||||||
SUPER_ADMIN = "SuperAdmin"
|
SUPER_ADMIN = "SuperAdmin"
|
||||||
}
|
}
|
||||||
Loading…
Reference in new issue