mirror of
https://github.com/syssecfsu/witty.git
synced 2024-11-22 03:49:26 +01:00
96 lines
2.1 KiB
Go
96 lines
2.1 KiB
Go
|
package web
|
||
|
|
|
||
|
|
import (
|
||
|
|
"html/template"
|
||
|
|
"io/fs"
|
||
|
|
"net/http"
|
||
|
|
"os"
|
||
|
|
"strconv"
|
||
|
|
|
||
|
|
"github.com/dchest/uniuri"
|
||
|
|
"github.com/gin-gonic/contrib/sessions"
|
||
|
|
"github.com/gin-gonic/gin"
|
||
|
|
"github.com/gorilla/csrf"
|
||
|
|
adapter "github.com/gwatts/gin-adapter"
|
||
|
|
"github.com/syssecfsu/witty/term_conn"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Options struct {
|
||
|
|
Wait uint
|
||
|
|
Port uint
|
||
|
|
NoAuth bool
|
||
|
|
CmdToExec []string
|
||
|
|
Assets fs.FS
|
||
|
|
LogFile *os.File
|
||
|
|
}
|
||
|
|
|
||
|
|
var options Options
|
||
|
|
|
||
|
|
func StartWeb(opt *Options) {
|
||
|
|
options = *opt
|
||
|
|
|
||
|
|
if options.LogFile != nil {
|
||
|
|
gin.DefaultWriter = options.LogFile
|
||
|
|
}
|
||
|
|
|
||
|
|
rt := gin.Default()
|
||
|
|
|
||
|
|
// We randomly generate a key for now, should use a fixed key
|
||
|
|
// so login can survive server reboot
|
||
|
|
store := sessions.NewCookieStore([]byte(uniuri.NewLen(32)))
|
||
|
|
rt.Use(sessions.Sessions("witty-session", store))
|
||
|
|
|
||
|
|
csrfHttp := csrf.Protect([]byte(uniuri.NewLen(32)), csrf.Path("/"))
|
||
|
|
csrfGin := adapter.Wrap(csrfHttp)
|
||
|
|
rt.Use(csrfGin)
|
||
|
|
|
||
|
|
rt.SetTrustedProxies(nil)
|
||
|
|
|
||
|
|
templ := template.Must(template.New("assets").ParseFS(options.Assets, "template/*.html"))
|
||
|
|
rt.SetHTMLTemplate(templ)
|
||
|
|
|
||
|
|
// handle static files
|
||
|
|
rt.StaticFS("/assets", http.FS(options.Assets))
|
||
|
|
rt.Static("/records", "./records")
|
||
|
|
|
||
|
|
rt.GET("/login", loginPage)
|
||
|
|
rt.POST("/login", login)
|
||
|
|
|
||
|
|
g1 := rt.Group("/")
|
||
|
|
|
||
|
|
if !options.NoAuth {
|
||
|
|
g1.Use(AuthRequired)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Fill in the index page
|
||
|
|
g1.GET("/", indexPage)
|
||
|
|
g1.GET("/logout", logout)
|
||
|
|
|
||
|
|
// to update the tabs of current interactive and saved sessions
|
||
|
|
g1.GET("/update/:active", updateIndex)
|
||
|
|
|
||
|
|
// create a new interactive session
|
||
|
|
g1.POST("/new", newInteractive)
|
||
|
|
g1.GET("/ws_new/:id", newTermConn)
|
||
|
|
|
||
|
|
// create a viewer of an interactive session
|
||
|
|
g1.GET("/view/:id", viewPage)
|
||
|
|
g1.GET("/ws_view/:id", newViewWS)
|
||
|
|
|
||
|
|
// start/stop recording the session
|
||
|
|
g1.POST("/record/:id", startRecord)
|
||
|
|
g1.POST("/stop/:id", stopRecord)
|
||
|
|
|
||
|
|
// create a viewer of an interactive session
|
||
|
|
g1.GET("/replay/:id", replayPage)
|
||
|
|
|
||
|
|
// delete a recording
|
||
|
|
g1.POST("/delete/:fname", delRec)
|
||
|
|
// Rename a recording
|
||
|
|
g1.POST("/rename/:oldname/:newname", renameRec)
|
||
|
|
|
||
|
|
term_conn.Init()
|
||
|
|
port := strconv.FormatUint(uint64(uint16(options.Port)), 10)
|
||
|
|
rt.RunTLS(":"+port, "./tls/cert.pem", "./tls/private-key.pem")
|
||
|
|
}
|