cchat-mock/server.go

package mock

import (
	"math/rand"
	"strconv"
	"sync/atomic"

	"github.com/Pallinder/go-randomdata"
	"github.com/diamondburned/cchat"
)

type Server struct {
	session  *Service
	id       uint32
	name     string
	children []cchat.Server
}

var (
	_ cchat.Server     = (*Server)(nil)
	_ cchat.ServerList = (*Server)(nil)
)

func (sv *Server) ID() string {
	return strconv.Itoa(int(sv.id))
}

func (sv *Server) Name() (string, error) {
	return sv.name, nil
}

func (sv *Server) Servers(container cchat.ServersContainer) error {
	container.SetServers(sv.children)
	return nil
}

func GenerateServers(s *Service) []cchat.Server {
	return generateServers(s, rand.Intn(45))
}

func generateServers(s *Service, amount int) []cchat.Server {
	var channels = make([]cchat.Server, amount)
	for i := range channels {
		channels[i] = &Server{
			session:  s,
			id:       atomic.AddUint32(&s.lastid, 1),
			name:     randomdata.Noun(),
			children: generateChannels(s, rand.Intn(12)),
		}
	}
	return channels
}