arikawa/state/store/defaultstore/me.go

package defaultstore

import (
	"sync"

	"github.com/diamondburned/arikawa/v3/discord"
	"github.com/diamondburned/arikawa/v3/state/store"
)

type Me struct {
	mut  sync.RWMutex
	self discord.User
}

var _ store.MeStore = (*Me)(nil)

func NewMe() *Me {
	return &Me{}
}

func (m *Me) Reset() error {
	m.mut.Lock()
	m.self = discord.User{}
	m.mut.Unlock()

	return nil
}

func (m *Me) Me() (*discord.User, error) {
	m.mut.RLock()
	self := m.self
	m.mut.RUnlock()

	if !self.ID.IsValid() {
		return nil, store.ErrNotFound
	}

	return &self, nil
}

func (m *Me) MyselfSet(me discord.User, update bool) error {
	m.mut.Lock()
	if !m.self.ID.IsValid() || update {
		m.self = me
	}
	m.mut.Unlock()

	return nil
}