123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167 |
- package main
- import (
- "errors"
- "fmt"
- "io/ioutil"
- "os"
- "strings"
- log "github.com/Sirupsen/logrus"
- haikunator "github.com/atrox/haikunatorgo"
- "github.com/gin-gonic/contrib/sessions"
- "github.com/gin-gonic/gin"
- )
- const sessionHeader = "slide-session"
- func NewApp() *gin.Engine {
- r := gin.Default()
- store := sessions.NewCookieStore([]byte("secret"))
- r.Use(sessions.Sessions(sessionHeader, store))
- r.LoadHTMLGlob("templates/*.tmpl")
- r.Static("/static", "./static")
- r.GET("/", func(c *gin.Context) {
- fname := c.Param("name")
- log.WithFields(log.Fields{
- "name": fname,
- }).Info("Restore?")
- haikunator := haikunator.NewHaikunator()
- haikunator.TokenLength = 0
- name := haikunator.Haikunate()
- path := fmt.Sprintf("slides/%s.md", name)
- log.WithFields(log.Fields{
- "path": path,
- }).Info("A new session")
- session := sessions.Default(c)
- session.Set("name", path)
- session.Save()
- c.HTML(200, "index.tmpl", gin.H{
- "pubTo": path,
- })
- })
- mustHaveSession := func(c *gin.Context) (string, error) {
- session := sessions.Default(c)
- val := session.Get("name")
- emptySession := errors.New("Emtpy session")
- if val == nil {
- c.String(400, "No context")
- return "", emptySession
- }
- log.WithFields(log.Fields{
- "path": val,
- }).Info("Got session")
- path, ok := val.(string)
- if !ok {
- c.String(400, "No context")
- return "", emptySession
- }
- return path, nil
- }
- r.GET("/slides.md", func(c *gin.Context) {
- path, err := mustHaveSession(c)
- if err != nil {
- return
- }
- if _, err := os.Stat(path); err != nil {
- // copy sample markdown file to the path
- body, err := ioutil.ReadFile("initial-slides.md")
- if err != nil {
- panic(err)
- }
- ioutil.WriteFile(path, body, 0644)
- c.String(200, string(body))
- return
- }
- body, err := ioutil.ReadFile(path)
- if err != nil {
- panic(err)
- }
- c.String(200, string(body))
- })
- r.PUT("/slides.md", func(c *gin.Context) {
- path, err := mustHaveSession(c)
- if err != nil {
- return
- }
- body, _ := ioutil.ReadAll(c.Request.Body)
- ioutil.WriteFile(path, body, 0644)
- log.WithFields(log.Fields{
- "size": len(body),
- "file": path,
- }).Info("Wrote to file")
- c.String(200, "")
- })
- r.GET("/stash", func(c *gin.Context) {
- files, err := ioutil.ReadDir("slides")
- if err != nil {
- log.Fatal(err)
- }
- var stash []string
- for _, file := range files {
- stash = append(stash, file.Name())
- }
- c.JSON(200, gin.H{
- "data": stash,
- })
- })
- r.GET("/stash/edit/:name", func(c *gin.Context) {
- name := c.Param("name")
- log.WithFields(log.Fields{
- "name": name,
- }).Info("Restore session?")
- if strings.HasSuffix(name, ".md") {
- name = name[0 : len(name)-3]
- }
- path := fmt.Sprintf("slides/%s.md", name)
- session := sessions.Default(c)
- session.Set("name", path)
- session.Save()
- c.HTML(200, "index.tmpl", gin.H{
- "pubTo": path,
- })
- })
- r.GET("/published/slides/:name", func(c *gin.Context) {
- name := c.Param("name")
- log.WithFields(log.Fields{
- "name": name,
- }).Info("Published")
- if strings.HasSuffix(name, ".md") {
- name = name[0 : len(name)-3]
- }
- path := fmt.Sprintf("slides/%s.md", name)
- session := sessions.Default(c)
- session.Set("name", path)
- session.Save()
- c.HTML(200, "slides.tmpl", gin.H{
- "pubTo": path,
- })
- })
- return r
- }
- func main() {
- r := NewApp()
- r.Run(":8080")
- }
|