witty/web/interactive.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)
}