package main

import (
	"fmt"
	"os"

	"github.com/AynaLivePlayer/blivedm-go/client"
	_ "github.com/AynaLivePlayer/blivedm-go/utils"
	log "github.com/sirupsen/logrus"
	"github.com/tidwall/gjson"
)

const roomId = 3819533

var dumps = []string{"GUARD_BUY", "USER_TOAST_MSG"}

func main() {
	log.SetLevel(log.DebugLevel)
	c := client.NewClient(roomId)
	c.SetCookie(os.Getenv("bilibili_cookie"))
	for _, v := range dumps {
		vv := v
		c.RegisterCustomEventHandler(vv, func(s string) {
			data := gjson.Get(s, "data").String()
			fmt.Printf("[%s] %s\n", vv, data)
		})
	}
	err := c.Start()
	if err != nil {
		log.Fatal(err)
	}
	log.Println("started bili dumper")
	select {}
}