cchat-gtk/internal/ui/service/service.go

package service

import (
	"fmt"

	"github.com/diamondburned/cchat"
	"github.com/diamondburned/cchat-gtk/internal/keyring"
	"github.com/diamondburned/cchat-gtk/internal/log"
	"github.com/diamondburned/cchat-gtk/internal/ui/primitives"
	"github.com/diamondburned/cchat-gtk/internal/ui/primitives/drag"
	"github.com/diamondburned/cchat-gtk/internal/ui/primitives/roundimage"
	"github.com/diamondburned/cchat-gtk/internal/ui/rich"
	"github.com/diamondburned/cchat-gtk/internal/ui/rich/parser/markup"
	"github.com/diamondburned/cchat-gtk/internal/ui/service/session"
	"github.com/diamondburned/cchat-gtk/internal/ui/service/session/server"
	"github.com/diamondburned/cchat-gtk/internal/ui/service/session/server/traverse"
	"github.com/gotk3/gotk3/gtk"
)

const IconSize = 48

type ListController interface {
	// RowSelected is called when a server message row is clicked.
	RowSelected(*session.Row, *server.ServerRow, cchat.ServerMessage)
	// SessionSelected tells the view to change the session view.
	SessionSelected(*Service, *session.Row)
	// AuthenticateSession tells View to call to the parent's authenticator.
	AuthenticateSession(*Service)
	// MoveService tells the view to shift the service to before the target.
	MoveService(id, targetID string)

	OnSessionRemove(*Service, *session.Row)
	OnSessionDisconnect(*Service, *session.Row)
}

// Service holds everything that a single service has.
type Service struct {
	*gtk.Box
	Button *gtk.ToggleButton
	Icon   *rich.Icon

	BodyRev  *gtk.Revealer // revealed
	BodyList *session.List // not really supposed to be here

	svclctrl ListController
	service  cchat.Service // state
}

var serviceCSS = primitives.PrepareClassCSS("service", `
	.service {
		box-shadow: 0 0 2px 0 alpha(@theme_bg_color, 0.75);
		margin: 6px 8px;
		margin-bottom: 0;
		border-radius: 14px;
	}

	.service:first-child { margin-top: 8px; }
	.service:last-child  { margin-bottom: 8px; }
`)

var serviceButtonCSS = primitives.PrepareClassCSS("service-button", `
	.service-button {
		padding: 2px;
		margin:  0;
	}

	.service-button:not(:checked) {
		border-radius: 14px;
		transition: linear 80ms border-radius; /* TODO add delay */
	}

	.service-button:checked {
		border-radius: 14px 14px 0 0;
		background-color: alpha(@theme_fg_color, 0.2);
	}
`)

var serviceIconCSS = primitives.PrepareClassCSS("service-icon", `
	.service-icon { padding: 4px }
`)

func NewService(svc cchat.Service, svclctrl ListController) *Service {
	service := &Service{
		service:  svc,
		svclctrl: svclctrl,
	}

	service.BodyList = session.NewList(service)
	service.BodyList.Show()

	service.BodyRev, _ = gtk.RevealerNew()
	service.BodyRev.SetRevealChild(false) // TODO persistent state
	service.BodyRev.SetTransitionDuration(50)
	service.BodyRev.SetTransitionType(gtk.REVEALER_TRANSITION_TYPE_SLIDE_DOWN)
	service.BodyRev.Add(service.BodyList)
	service.BodyRev.Show()

	// TODO: have it so the button changes to the session avatar when collapsed

	avatar := roundimage.NewAvatar(IconSize)
	avatar.SetText(svc.Name().String())
	avatar.Show()

	service.Icon = rich.NewCustomIcon(avatar, IconSize)
	service.Icon.Show()
	// potentially nonstandard
	service.Icon.SetPlaceholderIcon("text-html-symbolic", IconSize)
	// TODO: hover for name. We use tooltip for now.
	service.Icon.SetTooltipMarkup(markup.Render(svc.Name()))
	serviceIconCSS(service.Icon)

	if iconer, ok := svc.(cchat.Icon); ok {
		service.Icon.AsyncSetIconer(iconer, "Failed to set service icon")
	}

	service.Button, _ = gtk.ToggleButtonNew()
	service.Button.Add(service.Icon)
	service.Button.SetRelief(gtk.RELIEF_NONE)
	service.Button.Show()
	service.Button.Connect("clicked", func(tb *gtk.ToggleButton) {
		revealed := !service.GetRevealChild()
		service.SetRevealChild(revealed)
		tb.SetActive(revealed)
	})
	serviceButtonCSS(service.Button)

	// Intermediary box to contain both the icon and the revealer.
	service.Box, _ = gtk.BoxNew(gtk.ORIENTATION_VERTICAL, 0)
	service.Box.PackStart(service.Button, false, false, 0)
	service.Box.PackStart(service.BodyRev, false, false, 0)
	service.Box.Show()
	serviceCSS(service.Box)

	// Bind a drag and drop on the button instead of the entire box.
	drag.BindDraggable(service, "network-workgroup", svclctrl.MoveService, service.Button)

	return service
}

// SetRevealChild sets whether or not the service should reveal all sessions.
func (s *Service) SetRevealChild(reveal bool) {
	s.BodyRev.SetRevealChild(reveal)
}

// GetRevealChild gets whether or not the service is revealing all sessions.
func (s *Service) GetRevealChild() bool {
	return s.BodyRev.GetRevealChild()
}

func (s *Service) SessionSelected(srow *session.Row) {
	s.svclctrl.SessionSelected(s, srow)
}

func (s *Service) AuthenticateSession() {
	s.svclctrl.AuthenticateSession(s)
}

func (s *Service) AddLoadingSession(id, name string) *session.Row {
	srow := session.NewLoading(s, id, name, s)
	srow.Show()

	s.BodyList.AddSessionRow(id, srow)
	return srow
}

func (s *Service) AddSession(ses cchat.Session) *session.Row {
	srow := session.New(s, ses, s)
	srow.Show()

	s.BodyList.AddSessionRow(ses.ID(), srow)
	s.SaveAllSessions()
	return srow
}

func (s *Service) ID() string {
	return s.service.Name().Content
}

func (s *Service) Service() cchat.Service {
	return s.service
}

func (s *Service) OnSessionDisconnect(row *session.Row) {
	// Unselect if selected.
	if cur := s.BodyList.GetSelectedRow(); cur.GetIndex() == row.GetIndex() {
		s.BodyList.UnselectAll()
	}

	s.svclctrl.OnSessionDisconnect(s, row)

	// WHY WAS THIS HERE?!?!?!
	// s.BodyList.RemoveSessionRow(row.Session.ID())
	// s.SaveAllSessions()
}

func (s *Service) RowSelected(r *session.Row, sv *server.ServerRow, m cchat.ServerMessage) {
	s.svclctrl.RowSelected(r, sv, m)
}

func (s *Service) RemoveSession(row *session.Row) {
	s.svclctrl.OnSessionRemove(s, row)
	s.BodyList.RemoveSessionRow(row.Session.ID())
	s.SaveAllSessions()
}

func (s *Service) MoveSession(id, movingID string) {
	s.BodyList.MoveSession(id, movingID)
	s.SaveAllSessions()
}

func (s *Service) Breadcrumb() string {
	return s.service.Name().Content
}

func (s *Service) ParentBreadcrumb() traverse.Breadcrumber {
	return nil
}

func (s *Service) SaveAllSessions() {
	var sessions = s.BodyList.Sessions()
	var keyrings = make([]keyring.Session, 0, len(sessions))

	for _, s := range sessions {
		if k := keyring.ConvertSession(s.Session); k != nil {
			keyrings = append(keyrings, *k)
		}
	}

	keyring.SaveSessions(s.service, keyrings)
}

func (s *Service) RestoreSession(row *session.Row, id string) {
	rs, ok := s.service.(cchat.SessionRestorer)
	if !ok {
		return
	}

	if k := keyring.RestoreSession(s.service, id); k != nil {
		restoreAsync(row, rs, *k)
		return
	}

	log.Error(fmt.Errorf(
		"Missing keyring for service %s, session ID %s",
		s.service.Name().Content, id,
	))
}

// restoreAll restores all sessions.
func (s *Service) restoreAll() {
	rs, ok := s.service.(cchat.SessionRestorer)
	if !ok {
		return
	}

	// Session is not a pointer, so we can pass it into arguments safely.
	for _, ses := range keyring.RestoreSessions(s.service) {
		row := s.AddLoadingSession(ses.ID, ses.Name)
		restoreAsync(row, rs, ses)
	}
}

// restoreAsync asynchronously restores a single session.
func restoreAsync(r *session.Row, res cchat.SessionRestorer, k keyring.Session) {
	r.RestoreSession(res, k)
}