Просмотр исходного кода

Add default useragent and identify properties

ayn2op 3 лет назад
Родитель
Сommit
1e1c0c0d2d
1 измененных файлов с 18 добавлено и 0 удалено
  1. 18 0
      state.go

+ 18 - 0
state.go

@@ -1,13 +1,25 @@
 package main
 
 import (
+	"fmt"
 	"log"
+	"runtime"
 
+	"github.com/diamondburned/arikawa/v3/api"
 	"github.com/diamondburned/arikawa/v3/gateway"
 	"github.com/diamondburned/arikawa/v3/state"
 	"github.com/rivo/tview"
 )
 
+func init() {
+	api.UserAgent = fmt.Sprintf("%s/%s %s/%s", name, "0.1", "arikawa", "v3")
+	gateway.DefaultIdentity = gateway.IdentifyProperties{
+		OS:      runtime.GOOS,
+		Browser: name,
+		Device:  "",
+	}
+}
+
 type State struct {
 	*state.State
 }
@@ -27,6 +39,12 @@ func (s *State) onReady(r *gateway.ReadyEvent) {
 	dmNode := tview.NewTreeNode("Direct Messages")
 	guildsTree.root.AddChild(dmNode)
 
+	for _, g := range r.Guilds {
+		log.Println(g.Name)
+	}
+
+	log.Printf("%#v\n", r.UserSettings.GuildFolders)
+
 	for _, gf := range r.UserSettings.GuildFolders {
 		/// If the ID of the guild folder is zero, the guild folder only contains single guild.
 		if gf.ID == 0 {