mirror of
https://github.com/syssecfsu/witty.git
synced 2024-11-24 21:09:29 +01:00
102 lines
2.0 KiB
Go
102 lines
2.0 KiB
Go
package web
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/dchest/uniuri"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/gorilla/csrf"
|
|
"github.com/syssecfsu/witty/term_conn"
|
|
)
|
|
|
|
type InteractiveSession struct {
|
|
Ip string
|
|
Cmd string
|
|
Id string
|
|
}
|
|
|
|
func collectSessions(c *gin.Context, cmd string) (players []InteractiveSession) {
|
|
term_conn.ForEachSession(func(tc *term_conn.TermConn) {
|
|
players = append(players, InteractiveSession{
|
|
Id: tc.Name,
|
|
Ip: tc.Ip,
|
|
Cmd: cmd,
|
|
})
|
|
})
|
|
|
|
return
|
|
}
|
|
|
|
func indexPage(c *gin.Context) {
|
|
var disabled = ""
|
|
|
|
if options.NoAuth {
|
|
disabled = "disabled"
|
|
}
|
|
|
|
c.HTML(http.StatusOK, "index.html",
|
|
gin.H{
|
|
"disabled": disabled,
|
|
"csrfField": csrf.TemplateField(c.Request),
|
|
"csrfToken": csrf.Token(c.Request),
|
|
})
|
|
}
|
|
|
|
func updateIndex(c *gin.Context) {
|
|
var active0, active1 string
|
|
|
|
// setup which tab is active, it is hard to do in javascript at
|
|
// client side due to timing issues.
|
|
which := c.Param("active")
|
|
if which == "0" {
|
|
active0 = "active"
|
|
active1 = ""
|
|
} else {
|
|
active0 = ""
|
|
active1 = "active"
|
|
}
|
|
|
|
players := collectSessions(c, options.CmdToExec[0])
|
|
records := collectRecords(c)
|
|
|
|
c.HTML(http.StatusOK, "tab.html", gin.H{
|
|
"players": players,
|
|
"records": records,
|
|
"active0": active0,
|
|
"active1": active1,
|
|
})
|
|
}
|
|
|
|
func newInteractive(c *gin.Context) {
|
|
id := uniuri.New()
|
|
|
|
c.HTML(http.StatusOK, "term.html", gin.H{
|
|
"title": "interactive terminal",
|
|
"path": "/ws_new/" + id,
|
|
"id": id,
|
|
"logo": "keyboard",
|
|
"csrfToken": csrf.Token(c.Request),
|
|
})
|
|
}
|
|
|
|
func newTermConn(c *gin.Context) {
|
|
id := c.Param("id")
|
|
term_conn.ConnectTerm(c.Writer, c.Request, false, id, options.CmdToExec)
|
|
}
|
|
|
|
func viewPage(c *gin.Context) {
|
|
id := c.Param("id")
|
|
c.HTML(http.StatusOK, "term.html", gin.H{
|
|
"title": "viewer terminal",
|
|
"path": "/ws_view/" + id,
|
|
"id": id,
|
|
"logo": "view",
|
|
"csrfToken": csrf.Token(c.Request),
|
|
})
|
|
}
|
|
|
|
func newViewWS(c *gin.Context) {
|
|
id := c.Param("id")
|
|
term_conn.ConnectTerm(c.Writer, c.Request, true, id, nil)
|
|
}
|