ui/go/gothings/uistate.go

package gothings

import (
	"cwtch.im/ui/go/constants"
	"cwtch.im/ui/go/cwutil"
	"cwtch.im/ui/go/gobjects"
	"cwtch.im/ui/go/the"
	"encoding/base32"
	"git.openprivacy.ca/openprivacy/libricochet-go/log"
	"runtime/debug"
	"strings"
)

type InterfaceState struct {
	parentGcd *GrandCentralDispatcher
	contacts  map[string]*gobjects.Contact
	messages  map[string][]*gobjects.Message
}

func NewUIState(gcd *GrandCentralDispatcher) (uis InterfaceState) {
	uis = InterfaceState{gcd, make(map[string]*gobjects.Contact), make(map[string][]*gobjects.Message)}
	return
}

func (this *InterfaceState) AddContact(c *gobjects.Contact) {
	if len(c.Handle) == 32 { // ADD GROUP
		if _, found := this.contacts[c.Handle]; !found {
			this.parentGcd.AddContact(c.Handle, c.DisplayName, c.Image, c.Server, c.Badge, c.Status, c.Trusted)
			this.contacts[c.Handle] = c
		}
		return
	} else if len(c.Handle) != 56 {
		log.Errorf("sorry, unable to handle AddContact(%v)", c.Handle)
		debug.PrintStack()
		return
	}

	if _, found := this.contacts[c.Handle]; !found {
		this.contacts[c.Handle] = c
		this.parentGcd.AddContact(c.Handle, c.DisplayName, c.Image, c.Server, c.Badge, c.Status, c.Trusted)
		if the.Peer.GetContact(c.Handle) == nil {
			decodedPub, _ := base32.StdEncoding.DecodeString(strings.ToUpper(c.Handle))
			the.Peer.AddContact(c.DisplayName, c.Handle, decodedPub, c.Trusted)
			go the.Peer.PeerWithOnion(c.Handle)
		}
	}
}

func (this *InterfaceState) GetContact(handle string) *gobjects.Contact {
	if _, found := this.contacts[handle]; !found {
		if len(handle) == 32 {
			this.AddContact(&gobjects.Contact{
				handle,
				handle,
				cwutil.RandomGroupImage(handle),
				"",
				0,
				0,
				false,
			})
			group := the.Peer.GetGroup(handle)
			go the.Peer.JoinServer(group.GroupServer)
		} else {
			this.AddContact(&gobjects.Contact{
				handle,
				handle,
				cwutil.RandomProfileImage(handle),
				"",
				0,
				0,
				false,
			})
			go the.Peer.PeerWithOnion(handle)
		}
	}

	return this.contacts[handle]
}

func (this *InterfaceState) AddMessage(m *gobjects.Message) {
	_, found := this.contacts[m.Handle]
	if !found {
		this.AddContact(&gobjects.Contact{
			m.Handle,
			m.Handle,
			cwutil.RandomProfileImage(m.Handle),
			"",
			0,
			0,
			false,
		})
	}

	_, found = this.messages[m.Handle]
	if !found {
		this.messages[m.Handle] = make([]*gobjects.Message, 0)
	}

	if m.Message == "ack" {
		// If an ack, swallow the message and ack from the list.
		acklist := the.AcknowledgementIDs[m.From]
		for _,ack := range acklist {
			ack.Ack = true
			this.parentGcd.Acknowledged(ack.ID)
		}
	} else {
		this.messages[m.Handle] = append(this.messages[m.Handle], m)

		if this.parentGcd.CurrentOpenConversation() == m.Handle {
			this.parentGcd.AppendMessage(m.Handle, m.From, m.DisplayName, m.Message, m.Image, uint(m.MessageID), m.FromMe, m.Timestamp.Format(constants.TIME_FORMAT))
		} else {
			c := this.GetContact(m.Handle)
			c.Badge++
			this.UpdateContact(c.Handle)
		}
	}

}

func (this *InterfaceState) GetMessages(handle string) []*gobjects.Message {
	_, found := this.messages[handle]
	if !found {
		this.messages[handle] = make([]*gobjects.Message, 0)
	}
	return this.messages[handle]
}

func (this *InterfaceState) UpdateContact(handle string) {
	c, found := this.contacts[handle]
	if found {
		this.parentGcd.UpdateContact(c.Handle, c.DisplayName, c.Image, c.Server, c.Badge, c.Status, c.Trusted)
	}
}