mirror of https://github.com/syssecfsu/witty.git
130 lines
2.8 KiB
Go
130 lines
2.8 KiB
Go
package main
|
|
|
|
import (
|
|
"embed"
|
|
"flag"
|
|
"fmt"
|
|
"io/fs"
|
|
"log"
|
|
"os"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/syssecfsu/witty/cmd"
|
|
"github.com/syssecfsu/witty/web"
|
|
)
|
|
|
|
const (
|
|
subcmds = "witty (adduser|deluser|listusers|replay|merge|run)"
|
|
)
|
|
|
|
//go:embed assets/*
|
|
var fullAssets embed.FS
|
|
|
|
func main() {
|
|
if len(os.Args) < 2 {
|
|
fmt.Println(subcmds)
|
|
return
|
|
}
|
|
|
|
switch os.Args[1] {
|
|
case "adduser":
|
|
if len(os.Args) != 3 {
|
|
fmt.Println("witty adduser <username>")
|
|
return
|
|
}
|
|
cmd.AddUser(os.Args[2])
|
|
|
|
case "deluser":
|
|
if len(os.Args) != 3 {
|
|
fmt.Println("witty deluser <username>")
|
|
return
|
|
}
|
|
cmd.DelUser(os.Args[2])
|
|
|
|
case "listusers":
|
|
cmd.ListUsers()
|
|
|
|
case "replay":
|
|
var wait uint
|
|
replayCmd := flag.NewFlagSet("replay", flag.ExitOnError)
|
|
replayCmd.UintVar(&wait, "w", 1000, "Max wait time between outputs")
|
|
replayCmd.UintVar(&wait, "wait", 1000, "Max wait time between outputs")
|
|
|
|
replayCmd.Parse(os.Args[2:])
|
|
|
|
if len(replayCmd.Args()) != 1 {
|
|
fmt.Println("witty replay <recorded>")
|
|
return
|
|
}
|
|
|
|
cmd.Replay(replayCmd.Arg(0), wait)
|
|
|
|
case "merge":
|
|
var output string
|
|
|
|
defName := "merged_" + strconv.FormatInt(time.Now().Unix(), 16) + ".scr"
|
|
|
|
mergeCmd := flag.NewFlagSet("merge", flag.ExitOnError)
|
|
mergeCmd.StringVar(&output, "o", defName, "Set the name of merged files")
|
|
mergeCmd.StringVar(&output, "output", defName, "Set the name of merged files")
|
|
|
|
mergeCmd.Parse(os.Args[2:])
|
|
|
|
if len(mergeCmd.Args()) < 2 {
|
|
fmt.Println("witty merge -o output_file file1 file2 ... (at least two files)")
|
|
return
|
|
}
|
|
|
|
cmd.Merge(mergeCmd.Args(), output)
|
|
|
|
case "run":
|
|
// setup the web options
|
|
var options web.Options
|
|
runCmd := flag.NewFlagSet("run", flag.ExitOnError)
|
|
runCmd.BoolVar(&options.NoAuth, "n", false, "Run WiTTY without user authentication")
|
|
runCmd.BoolVar(&options.NoAuth, "naked", false, "Run WiTTY without user authentication")
|
|
runCmd.UintVar(&options.Port, "p", 8080, "Port number to listen on")
|
|
runCmd.UintVar(&options.Port, "port", 8080, "Port number to listen on")
|
|
runCmd.UintVar(&options.Wait, "w", 1000, "Max wait time between outputs")
|
|
runCmd.UintVar(&options.Wait, "wait", 1000, "Max wait time between outputs")
|
|
|
|
fp, err := os.OpenFile("witty.log", os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0644)
|
|
|
|
if err == nil {
|
|
defer fp.Close()
|
|
log.SetOutput(fp)
|
|
}
|
|
|
|
options.LogFile = fp
|
|
|
|
runCmd.Parse(os.Args[2:])
|
|
|
|
var cmdToExec []string
|
|
args := runCmd.Args()
|
|
if len(args) > 0 {
|
|
cmdToExec = args
|
|
} else {
|
|
cmdToExec = []string{"bash"}
|
|
}
|
|
|
|
options.CmdToExec = cmdToExec
|
|
|
|
// we need to strip the top level directory for Gin to find the files
|
|
assets, err := fs.Sub(fullAssets, "assets")
|
|
|
|
if err != nil {
|
|
log.Fatal("Failed to load assets", err)
|
|
}
|
|
|
|
options.Assets = assets
|
|
|
|
web.StartWeb(&options)
|
|
|
|
default:
|
|
fmt.Println(subcmds)
|
|
return
|
|
}
|
|
|
|
}
|