mirror of https://github.com/sunface/rust-course
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
107 lines
2.4 KiB
107 lines
2.4 KiB
package internal
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/imdotdev/im.dev/server/internal/api"
|
|
"github.com/imdotdev/im.dev/server/internal/session"
|
|
"github.com/imdotdev/im.dev/server/internal/storage"
|
|
"github.com/imdotdev/im.dev/server/pkg/common"
|
|
"github.com/imdotdev/im.dev/server/pkg/config"
|
|
"github.com/imdotdev/im.dev/server/pkg/errcode"
|
|
"github.com/imdotdev/im.dev/server/pkg/log"
|
|
)
|
|
|
|
type Server struct {
|
|
}
|
|
|
|
// New ...
|
|
func New() *Server {
|
|
return &Server{}
|
|
}
|
|
|
|
var logger = log.RootLogger.New("logger", "server")
|
|
|
|
// Start ...1=
|
|
func (s *Server) Start() error {
|
|
err := storage.Init()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if config.Data.Common.IsProd {
|
|
gin.SetMode((gin.ReleaseMode))
|
|
} else {
|
|
gin.SetMode(gin.DebugMode)
|
|
}
|
|
|
|
go func() {
|
|
router := gin.New()
|
|
router.Use(Cors())
|
|
|
|
r := router.Group("/api")
|
|
{
|
|
r.POST("/login", session.Login)
|
|
r.POST("/logout", session.Logout)
|
|
r.GET("/uiconfig", GetUIConfig)
|
|
}
|
|
|
|
// login apis
|
|
lr := r.Group("", IsLogin())
|
|
{
|
|
editorR := lr.Group("/editor")
|
|
{
|
|
editorR.GET("/articles", api.GetEditorArticles)
|
|
editorR.POST("/article", api.PostEditorArticle)
|
|
editorR.DELETE("/article/:id", api.DeleteEditorArticle)
|
|
}
|
|
}
|
|
err := router.Run(config.Data.Server.Addr)
|
|
if err != nil {
|
|
logger.Crit("start backend server error", "error", err)
|
|
panic(err)
|
|
}
|
|
}()
|
|
return nil
|
|
}
|
|
|
|
// Close ...
|
|
func (s *Server) Close() error {
|
|
return nil
|
|
}
|
|
|
|
// Cors is a gin middleware for cross domain.
|
|
func Cors() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
method := c.Request.Method
|
|
|
|
c.Header("Access-Control-Allow-Origin", "*")
|
|
c.Header("Access-Control-Allow-Headers", "Content-Type,AccessToken,X-CSRF-Token, Authorization,X-Token,*")
|
|
c.Header("Access-Control-Allow-Methods", "POST, GET, PUT, DELETE, OPTIONS")
|
|
c.Header("Access-Control-Expose-Headers", "Content-Length, Access-Control-Allow-Origin, Access-Control-Allow-Headers, Content-Type")
|
|
c.Header("Access-Control-Allow-Credentials", "true")
|
|
|
|
//放行所有OPTIONS方法
|
|
if method == "OPTIONS" {
|
|
c.AbortWithStatus(http.StatusNoContent)
|
|
}
|
|
|
|
// 处理请求
|
|
c.Next()
|
|
}
|
|
}
|
|
|
|
// Auth is a gin middleware for user auth
|
|
func IsLogin() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
user := session.CurrentUser(c)
|
|
if user == nil {
|
|
c.JSON(http.StatusUnauthorized, common.RespError(errcode.NeedLogin))
|
|
c.Abort()
|
|
return
|
|
}
|
|
c.Next()
|
|
}
|
|
}
|