package main import ( "context" "errors" "flag" "fmt" "io" "log" "strings" "time" "github.com/gdamore/tcell/v2" "github.com/rivo/tview" "github.com/vilmibm/hermeticum/proto" "google.golang.org/grpc" "google.golang.org/grpc/credentials/insecure" ) var ( tls = flag.Bool("tls", false, "Connection uses TLS if true, else plain TCP") caFile = flag.String("ca_file", "", "The file containing the CA root cert file") serverAddr = flag.String("addr", "localhost:6666", "The server address in the format of host:port") serverHostOverride = flag.String("server_host_override", "x.test.example.com", "The server name used to verify the hostname returned by the TLS handshake") ) type ClientState struct { App *tview.Application Client proto.GameWorldClient SessionInfo *proto.SessionInfo MaxMessages int messagesView *tview.TextView messages []*proto.ClientMessage cmdStream proto.GameWorld_CommandsClient } func (cs *ClientState) Messages() error { ctx, cancel := context.WithCancel(context.Background()) defer cancel() stream, err := cs.Client.Messages(ctx, cs.SessionInfo) if err != nil { return err } for { msg, err := stream.Recv() if err == io.EOF { break } if err != nil { return err } cs.AddMessage(msg) } return nil } func (cs *ClientState) HandleInput(input string) error { var verb string rest := input if strings.HasPrefix(input, "/") { input = input[1:] parts := strings.SplitN(input, " ", 1) verb = parts[0] if len(parts) > 1 { rest = parts[1] } } else { verb = "say" } cmd := &proto.Command{ SessionInfo: cs.SessionInfo, Verb: verb, Rest: rest, } // TODO I'm punting on handling CommandAcks for now but it will be a nice UX thing later for showing connectivity problems err := cs.cmdStream.Send(cmd) if err != nil { return err } if verb == "quit" || verb == "q" { cs.App.Stop() } return nil } func (cs *ClientState) InitCommandStream() error { ctx := context.Background() stream, err := cs.Client.Commands(ctx) if err != nil { return err } cs.cmdStream = stream return nil } func (cs *ClientState) AddMessage(msg *proto.ClientMessage) { // TODO i don't like this function cs.messages = append(cs.messages, msg) if len(cs.messages) > cs.MaxMessages { cs.messages = cs.messages[1 : len(cs.messages)-1] } // TODO look into using the SetChangedFunc thing. cs.App.QueueUpdateDraw(func() { // TODO trim content of messagesView /or/ see if tview has a buffer size that does it for me. use cs.messages to re-constitute. fmt.Fprintf(cs.messagesView, "%s: %s\n", msg.GetSpeaker(), msg.GetText()) cs.messagesView.ScrollToEnd() }) } func _main() error { var opts []grpc.DialOption if *tls { return errors.New("TODO tls unsupported") /* if *caFile == "" { *caFile = data.Path("x509/ca_cert.pem") } creds, err := credentials.NewClientTLSFromFile(*caFile, *serverHostOverride) if err != nil { log.Fatalf("Failed to create TLS credentials %v", err) } opts = append(opts, grpc.WithTransportCredentials(creds)) */ } else { opts = append(opts, grpc.WithTransportCredentials(insecure.NewCredentials())) } conn, err := grpc.Dial(*serverAddr, opts...) if err != nil { return fmt.Errorf("fail to dial: %w", err) } defer conn.Close() client := proto.NewGameWorldClient(conn) app := tview.NewApplication() // TODO make a NewClientState // TODO rename this, like, UI cs := &ClientState{ App: app, SessionInfo: &proto.SessionInfo{}, Client: client, MaxMessages: 15, // TODO for testing messages: []*proto.ClientMessage{}, } err = cs.InitCommandStream() if err != nil { return fmt.Errorf("could not create command stream: %w", err) } ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second) defer cancel() pong, err := cs.Client.Ping(ctx, cs.SessionInfo) if err != nil { log.Fatalf("%v.Ping -> %v", cs.Client, err) } log.Printf("%#v", pong) pages := tview.NewPages() pages.AddPage("splash", tview.NewModal(). AddButtons([]string{"hey. let's go"}). SetDoneFunc(func(_ int, _ string) { pages.SwitchToPage("main") app.ResizeToFullScreen(pages) }).SetText("h e r m e t i c u m"), true, true) commandInput := tview.NewInputField().SetLabel("> ") handleInput := func(_ tcell.Key) { input := commandInput.GetText() // TODO do i need to clear the input's text? go cs.HandleInput(input) } commandInput.SetDoneFunc(handleInput) mainPage := tview.NewList(). AddItem("jack in", "connect using an existing account", '1', func() { pages.SwitchToPage("login") }). AddItem("rez a toon", "create a new account", '2', func() { pages.SwitchToPage("register") }). AddItem("open the hood", "client configuration", '3', nil). AddItem("get outta here", "quit the client", '4', func() { app.Stop() }) pages.AddPage("main", mainPage, true, false) lunfi := tview.NewInputField().SetLabel("account name") lpwfi := tview.NewInputField().SetLabel("password").SetMaskCharacter('~') loginPage := tview.NewForm().AddFormItem(lunfi).AddFormItem(lpwfi). SetCancelFunc(func() { pages.SwitchToPage("main") }) loginSubmitFn := func() { ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second) defer cancel() si, err := cs.Client.Login(ctx, &proto.AuthInfo{ Username: lunfi.GetText(), Password: lpwfi.GetText(), }) if err != nil { panic(err.Error()) } cs.SessionInfo = si pages.SwitchToPage("game") app.SetFocus(commandInput) // TODO error handle go cs.Messages() } // TODO login and register pages should refuse blank entries // TODO password should have rules loginPage.AddButton("gimme that shit", loginSubmitFn) loginPage.AddButton("nah get outta here", func() { pages.SwitchToPage("main") }) pages.AddPage("login", loginPage, true, false) runfi := tview.NewInputField().SetLabel("account name") rpwfi := tview.NewInputField().SetLabel("password").SetMaskCharacter('~') registerPage := tview.NewForm().AddFormItem(runfi).AddFormItem(rpwfi). SetCancelFunc(func() { pages.SwitchToPage("main") }) registerSubmitFn := func() { ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second) defer cancel() si, err := cs.Client.Register(ctx, &proto.AuthInfo{ Username: runfi.GetText(), Password: rpwfi.GetText(), }) if err != nil { panic(err.Error()) } cs.SessionInfo = si pages.SwitchToPage("game") app.SetFocus(commandInput) // TODO error handle go cs.Messages() } registerPage.AddButton("gimme that shit", registerSubmitFn) registerPage.AddButton("nah get outta here", func() { pages.SwitchToPage("main") }) pages.AddPage("register", registerPage, true, false) msgView := tview.NewTextView().SetScrollable(true).SetWrap(true).SetWordWrap(true) cs.messagesView = msgView gamePage := tview.NewGrid(). SetRows(1, 40, 3). SetColumns(-1, -1). SetBorders(true). AddItem( tview.NewTextView().SetTextAlign(tview.AlignLeft).SetText("h e r m e t i c u m"), 0, 0, 1, 1, 1, 1, false). AddItem( tview.NewTextView().SetTextAlign(tview.AlignRight).SetText("TODO server status"), 0, 1, 1, 1, 1, 1, false). AddItem( msgView, 1, 0, 1, 1, 10, 20, false). AddItem( tview.NewTextView().SetText("TODO details"), 1, 1, 1, 1, 10, 10, false). AddItem( commandInput, 2, 0, 1, 2, 1, 30, false) pages.AddPage("game", gamePage, true, false) return app.SetRoot(pages, true).SetFocus(pages).Run() } func main() { err := _main() if err != nil { log.Fatal(err.Error()) } }