kbtui/main.go

package main

import (
	"fmt"
	"github.com/jroimartin/gocui"
	"log"
	"samhofi.us/x/keybase"
	"strings"
	"time"
)

var k = keybase.NewKeybase()
var teamOrUser = "home"
var channel = ""
var myUsername = ""

func main() {
	if k.LoggedIn == true {
		kbtui, err := gocui.NewGui(gocui.OutputNormal)
		if err != nil {
			log.Panicln(err)
		}
		defer kbtui.Close()

		kbtui.SetManagerFunc(layout)
		go loginGreeter(kbtui)
		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)
		}
	} else {
		fmt.Println("You are not logged in.")
	}

}
func loginGreeter(g *gocui.Gui) {
	myUsername = k.Username
	messg := "Welcome " + myUsername+ "!"
	printToView(g, "Chat", messg)
}
func sendToUser(msg string) {
	chat := k.NewChat(keybase.Channel{Name:teamOrUser})
	chat.Send(msg)
}

func sendToTeam(msg string) {
	chat := k.NewChat(keybase.Channel{
		Name:        teamOrUser,
		MembersType: "team",
		TopicName:   channel,
	})
	chat.Send(msg)
}
func populateList(g *gocui.Gui) {
	for {
		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)
			}
		}
		time.Sleep(5 * time.Second)
	}

}

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", 12, 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", 12, 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", 12, 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, 10, 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 handleMessage(api keybase.ChatAPI, g *gocui.Gui) {
	if api.Msg.Content.Type == "text" {
		if strings.Contains(api.Msg.Content.Text.Body, myUsername) || strings.Contains(api.Msg.Channel.Name, myUsername) {
			if api.Msg.Sender.Username != myUsername {
				message := api.Msg.Content.Text.Body
				sender := api.Msg.Sender.Username
				team := api.Msg.Channel.Name
				channel := api.Msg.Channel.TopicName
				printMe := "@" + team + "#" + channel + " " + sender + ": " + message
				printToView(g, "Feed", printMe)
			}
		}
		if strings.Contains(api.Msg.Channel.Name, teamOrUser) {
			if channel != "" && api.Msg.Channel.TopicName == channel {

			} else {
				message := api.Msg.Content.Text.Body
				sender := api.Msg.Sender.Username
				printToView(g, "Chat", sender+": "+message)
			}
		}
	}
}
func handleInput(g *gocui.Gui) error {
	inputString, _ := getInputString(g)
	command := ""
	if len(inputString) > 2 {
		command = inputString[:2]
	}
	if "/q" == command {
		return gocui.ErrQuit
	} else if "/j" == command {
		clearView(g, "Chat")
		arr := strings.Split(inputString, " ")
		if len(arr) > 2 {
			teamOrUser = arr[1]
			channel = arr[2]
			printToView(g, "Feed", "You have joined: @" + teamOrUser + "#" + channel)
		} else {
			teamOrUser = arr[1]
			channel = ""
			printToView(g, "Feed", "You have joined: @" + teamOrUser)
		}
	} else {
		if channel == "" {
			sendToUser(inputString)
		} else {
			sendToTeam(inputString)
		}
	}
	clearView(g, "Input")
	return nil
}
func quit(g *gocui.Gui, v *gocui.View) error {
	return gocui.ErrQuit
}