mirror of
https://github.com/syssecfsu/witty.git
synced 2024-12-25 04:02:36 +01:00
85 lines
1.6 KiB
Go
85 lines
1.6 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"flag"
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"os"
|
|
"time"
|
|
|
|
"golang.org/x/term"
|
|
)
|
|
|
|
type writeRecord struct {
|
|
Dur time.Duration `json:"Duration"`
|
|
Data []byte `json:"Data"`
|
|
}
|
|
|
|
func main() {
|
|
var wait uint
|
|
|
|
flag.UintVar(&wait, "w", 2000, "Max wait time between outputs")
|
|
flag.UintVar(&wait, "wait", 2000, "Max wait time between outputs")
|
|
flag.Parse()
|
|
|
|
left := flag.Args()
|
|
|
|
if len(left) != 1 {
|
|
fmt.Println("Usage: replay -w/wait time <recordfile>")
|
|
return
|
|
}
|
|
|
|
fp, err := os.Open(left[0])
|
|
|
|
if err != nil {
|
|
log.Fatalln("Failed to open record file", err)
|
|
}
|
|
|
|
screen := struct {
|
|
io.Reader
|
|
io.Writer
|
|
}{os.Stdin, os.Stdout}
|
|
|
|
t := term.NewTerminal(screen, "$")
|
|
|
|
if t == nil {
|
|
log.Fatalln("Failed to create terminal")
|
|
}
|
|
|
|
w, h, _ := term.GetSize(int(os.Stdout.Fd()))
|
|
|
|
if (w != 120) || (h != 36) {
|
|
log.Println("Set terminal window to 120x36 before continue")
|
|
}
|
|
|
|
decoder := json.NewDecoder(fp)
|
|
|
|
if decoder == nil {
|
|
log.Fatalln("Failed to create JSON decoder")
|
|
}
|
|
|
|
// To work with javascript decoder, we organize the file as
|
|
// an array of writeRecord. golang decode instead decode
|
|
// as individual record. Call decoder.Token to skip opening [
|
|
decoder.Token()
|
|
for decoder.More() {
|
|
var record writeRecord
|
|
|
|
if err := decoder.Decode(&record); err != nil {
|
|
log.Println("Failed to decode record", err)
|
|
continue
|
|
}
|
|
|
|
if record.Dur > time.Duration(wait)*time.Millisecond {
|
|
record.Dur = time.Duration(wait) * time.Millisecond
|
|
}
|
|
|
|
time.Sleep(record.Dur)
|
|
t.Write(record.Data)
|
|
}
|
|
|
|
t.Write([]byte("\n\n---end of replay---\n\n"))
|
|
}
|