53 lines
998 B
Go
53 lines
998 B
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
|
|
}
|