YourSelfHosted/memos
1
0
Fork 0
mirror of https://github.com/usememos/memos.git synced 2025-06-05 22:09:59 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat: const session secret key in dev

Browse Source
This commit is contained in:
email
2022-03-29 07:30:29 +08:00
parent 8f76120e4e
commit 1cebf1dbd0
5 changed files with 9 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
server/server.go
View File

@ -23,7 +23,7 @@ type Server struct {
port int
}
func NewServer(port int) *Server {
func NewServer(port int, mode string) *Server {
e := echo.New()
e.Debug = true
e.HideBanner = true
@ -46,7 +46,11 @@ func NewServer(port int) *Server {
HTML5: true,
}))
e.Use(session.Middleware(sessions.NewCookieStore([]byte(securecookie.GenerateRandomKey(16)))))
secret := []byte("justmemos")
if mode != "dev" {
secret = securecookie.GenerateRandomKey(16)
}
e.Use(session.Middleware(sessions.NewCookieStore(secret)))
s := &Server{
e: e,

2
server/user.go
View File

@ -11,8 +11,8 @@ import (
)
func (s *Server) registerUserRoutes(g *echo.Group) {
// GET /api/user/me is used to check if the user is logged in.
g.GET("/user/me", func(c echo.Context) error {
// /api/user/me is used to check if the user is logged in,
userSessionId := c.Get(getUserIdContextKey())
if userSessionId == nil {
return echo.NewHTTPError(http.StatusUnauthorized, "Missing session")

3
server/webhook.go
View File

@ -103,9 +103,8 @@ func (s *Server) registerWebhookRoutes(g *echo.Group) {
}
c.Response().Writer.WriteHeader(http.StatusOK)
c.Response().Writer.Header().Set("Content-Type", "application/octet-stream")
c.Response().Writer.Header().Set("Content-Type", resource.Type)
c.Response().Writer.Write(resource.Blob)
return nil
})
}