package main import ( "fmt" "log" "strings" "github.com/jroimartin/gocui" "samhofi.us/x/keybase" ) var k = keybase.NewKeybase() var channel keybase.Channel func main() { if !k.LoggedIn { fmt.Println("You are not logged in.") return } kbtui, err := gocui.NewGui(gocui.OutputNormal) if err != nil { log.Panicln(err) } defer kbtui.Close() kbtui.SetManagerFunc(layout) printToView(kbtui, "Chat", fmt.Sprintf("Welcome %s!", k.Username)) go populateList(kbtui) go updateChatWindow(kbtui) if err := initKeybindings(kbtui); err != nil { log.Fatalln(err) } if err := kbtui.MainLoop(); err != nil && err != gocui.ErrQuit { log.Panicln(err) } } func populateChat(g *gocui.Gui) { chat := k.NewChat(channel) if api, err := chat.Read(15); err != nil { log.Fatal(err) } else { var printMe []string for _, message := range api.Result.Messages { if message.Msg.Content.Type == "text" { msgSender := message.Msg.Sender.Username msgBody := message.Msg.Content.Text.Body newMessage := fmt.Sprintf("%s: %s", msgSender, msgBody) printMe = append(printMe, newMessage) } } for i := len(printMe) - 1; i >= 0; i-- { printToView(g, "Chat", printMe[i]) } } } func sendChat(message string) { chat := k.NewChat(channel) chat.Send(message) } func populateList(g *gocui.Gui) { if testVar, err := k.ChatList(); err != nil { log.Fatalln(err) } else { clearView(g, "List") for _, s := range testVar.Result.Conversations { printToView(g, "List", s.Channel.Name) } } } func clearView(kbtui *gocui.Gui, viewName string) { kbtui.Update(func(g *gocui.Gui) error { inputView, err := kbtui.View(viewName) if err != nil { return err } else { inputView.Clear() inputView.SetCursor(0, 0) inputView.SetOrigin(0, 0) } return nil }) } func printToView(kbtui *gocui.Gui, viewName string, message string) { kbtui.Update(func(g *gocui.Gui) error { updatingView, err := kbtui.View(viewName) if err != nil { return err } else { fmt.Fprintf(updatingView, message+"\n") } return nil }) } func layout(g *gocui.Gui) error { maxX, maxY := g.Size() if feedView, err := g.SetView("Feed", maxX/2-maxX/3, 0, maxX-1, maxY/5); err != nil { if err != gocui.ErrUnknownView { return err } feedView.Autoscroll = true feedView.Wrap = true fmt.Fprintln(feedView, "Feed Window - If you are mentioned or receive a PM it will show here") } if chatView, err2 := g.SetView("Chat", maxX/2-maxX/3, maxY/5+1, maxX-1, maxY-5); err2 != nil { if err2 != gocui.ErrUnknownView { return err2 } chatView.Autoscroll = true chatView.Wrap = true fmt.Fprintf(chatView, "Your chats will appear here.\nSupported commands are as follows:\n") fmt.Fprintln(chatView, "/j $username - Open your chat with $username") fmt.Fprintln(chatView, "/j $team $channel - Open $channel from $team") fmt.Fprintln(chatView, " Please note: small teams only have #general") fmt.Fprintln(chatView, "/q - Exit") } if inputView, err3 := g.SetView("Input", maxX/2-maxX/3, maxY-4, maxX-1, maxY-1); err3 != nil { if err3 != gocui.ErrUnknownView { return err3 } if _, err := g.SetCurrentView("Input"); err != nil { return err } inputView.Editable = true inputView.Wrap = true } if listView, err4 := g.SetView("List", 0, 0, maxX/2-maxX/3-1, maxY-1); err4 != nil { if err4 != gocui.ErrUnknownView { return err4 } listView.Wrap = true fmt.Fprintf(listView, "Lists\nWindow\nTo view\n activity") } return nil } func getInputString(g *gocui.Gui) (string, error) { inputView, _ := g.View("Input") return inputView.Line(0) } func initKeybindings(g *gocui.Gui) error { if err := g.SetKeybinding("", gocui.KeyCtrlC, gocui.ModNone, func(g *gocui.Gui, v *gocui.View) error { return gocui.ErrQuit }); err != nil { return err } if err := g.SetKeybinding("Input", gocui.KeyEnter, gocui.ModNone, func(g *gocui.Gui, v *gocui.View) error { return handleInput(g) }); err != nil { return err } return nil } func updateChatWindow(g *gocui.Gui) { k.Run(func(api keybase.ChatAPI) { handleMessage(api, g) }) } func cleanChannelName(c string) string { newChannelName := strings.Replace(c, fmt.Sprintf("%s,", k.Username), "", 1) return strings.Replace(newChannelName, fmt.Sprintf(",%s", k.Username), "", 1) } func handleMessage(api keybase.ChatAPI, g *gocui.Gui) { go populateList(g) if api.Msg.Content.Type == "text" { msgBody := api.Msg.Content.Text.Body msgSender := api.Msg.Sender.Username channelName := api.Msg.Channel.Name if msgSender != k.Username { if api.Msg.Channel.MembersType == keybase.TEAM { topicName := api.Msg.Channel.TopicName for _, m := range api.Msg.Content.Text.UserMentions { if m.Text == k.Username { printToView(g, "Feed", fmt.Sprintf("[ %s#%s ] %s: %s", channelName, topicName, msgSender, msgBody)) break } } } else { printToView(g, "Feed", fmt.Sprintf("[ %s ] %s: %s", cleanChannelName(channelName), msgSender, msgBody)) } } if api.Msg.Channel.MembersType == channel.MembersType && cleanChannelName(api.Msg.Channel.Name) == channel.Name { printToView(g, "Chat", fmt.Sprintf("%s: %s", msgSender, msgBody)) } } } func handleInput(g *gocui.Gui) error { inputString, _ := getInputString(g) command := strings.Split(inputString, " ") switch strings.ToLower(command[0]) { case "/q": return gocui.ErrQuit case "/j": if len(command) == 3 { channel.MembersType = keybase.TEAM channel.Name = command[1] channel.TopicName = command[2] printToView(g, "Feed", fmt.Sprintf("You have joined: @%s#%s", channel.Name, channel.TopicName)) clearView(g, "Chat") go populateChat(g) } else if len(command) == 2 { channel.MembersType = keybase.USER channel.Name = command[1] printToView(g, "Feed", fmt.Sprintf("You have joined: @%s", channel.Name)) clearView(g, "Chat") go populateChat(g) } else { printToView(g, "Feed", "To join a team use /j ") printToView(g, "Feed", "To join a PM use /j ") } default: go sendChat(inputString) } clearView(g, "Input") return nil } func quit(g *gocui.Gui, v *gocui.View) error { return gocui.ErrQuit }