status-console-client/contacts.go

package main

import (
	"fmt"

	"github.com/jroimartin/gocui"
)

// Types of contacts.
const (
	ContactPublicChat int = iota + 1
)

// Contact is a single contact which has a type and name.
type Contact struct {
	Name string
	Type int
}

// String returns a string representation.
func (c Contact) String() string {
	switch c.Type {
	case ContactPublicChat:
		return fmt.Sprintf("#%s", c.Name)
	default:
		return c.Name
	}
}

// ContactsViewController manages contacts view.
type ContactsViewController struct {
	*ViewController
	items []Contact
}

// NewContactsViewController returns a new contact view controller.
func NewContactsViewController(vm *ViewController, items []Contact) *ContactsViewController {
	return &ContactsViewController{vm, items}
}

// ContactByIdx allows to retrieve a contact for a given index.
func (c *ContactsViewController) ContactByIdx(idx int) (Contact, bool) {
	if idx > -1 && idx < len(c.items) {
		return c.items[idx], true
	}
	return Contact{}, false
}

// Refresh repaints the current list of contacts.
func (c *ContactsViewController) Refresh() {
	c.g.Update(func(*gocui.Gui) error {
		if err := c.Clear(); err != nil {
			return err
		}

		for _, item := range c.items {
			if _, err := fmt.Fprintln(c.ViewController, item); err != nil {
				return err
			}
		}
		return nil
	})
}