mirror/cowyo
1
0
Fork 0
mirror of https://github.com/schollz/cowyo.git synced 2023-08-10 21:13:00 +03:00
Code Issues Projects Releases Wiki Activity

Allow setting session secret cookie code

Browse Source
This commit is contained in:
Daniel Heath 2018-01-18 20:51:08 +11:00
parent 89d58f5a22
commit 76ec1c1acb
2 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
handlers.go
View File

@ -33,10 +33,11 @@ func serve(
defaultPassword string, defaultPassword string,
debounce int, debounce int,
diary bool, diary bool,
secret string,
) { ) {
gin.SetMode(gin.ReleaseMode) gin.SetMode(gin.ReleaseMode)
router := gin.Default() router := gin.Default()
store := sessions.NewCookieStore([]byte("secret")) store := sessions.NewCookieStore([]byte(secret))
router.Use(sessions.Sessions("mysession", store)) router.Use(sessions.Sessions("mysession", store))
router.HTMLRender = loadTemplates("index.tmpl") router.HTMLRender = loadTemplates("index.tmpl")
// router.Use(static.Serve("/static/", static.LocalFile("./static", true))) // router.Use(static.Serve("/static/", static.LocalFile("./static", true)))

6
main.go
View File

@ -51,6 +51,7 @@ func main() {
c.GlobalString("lock"), c.GlobalString("lock"),
c.GlobalInt("debounce"), c.GlobalInt("debounce"),
c.GlobalBool("diary"), c.GlobalBool("diary"),
c.GlobalString("cookie-secret"),
) )
return nil return nil
} }
@ -117,6 +118,11 @@ func main() {
Name: "diary", Name: "diary",
Usage: "turn diary mode (doing New will give a timestamped page)", Usage: "turn diary mode (doing New will give a timestamped page)",
}, },
cli.StringFlag{
Name: "cookie-secret",
Value: "secret",
Usage: "random data to use for cookies; changing it will invalidate all sessions",
},
} }
app.Commands = []cli.Command{ app.Commands = []cli.Command{
{ {