mirror of
https://github.com/diamondburned/arikawa.git
synced 2024-11-01 20:45:42 +00:00
54cadd2f45
This commit refactors the whole package gateway as well as utils/ws (formerly utils/wsutil) and voice/voicegateway. The new refactor utilizes a design pattern involving a concurrent loop and an arriving event channel. An additional change was made to the way gateway events are typed. Before, pretty much any type will satisfy a gateway event type, since the actual type was just interface{}. The new refactor defines a concrete interface that events can implement: type Event interface { Op() OpCode EventType() EventType } Using this interface, the user can easily add custom gateway events independently of the library without relying on string maps. This adds a lot of type safety into the library and makes type-switching on Event types much more reasonable. Gateway error callbacks are also almost entirely removed in favor of custom gateway events. A catch-all can easily be added like this: s.AddHandler(func(err error) { log.Println("gateway error:, err") })
87 lines
1.4 KiB
Go
87 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"flag"
|
|
"go/format"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"text/template"
|
|
|
|
_ "embed"
|
|
)
|
|
|
|
type data struct {
|
|
Package string
|
|
ImportDiscord bool
|
|
Snowflakes []snowflakeType
|
|
}
|
|
|
|
type snowflakeType struct {
|
|
TypeName string
|
|
}
|
|
|
|
//go:embed template.tmpl
|
|
var packageTmpl string
|
|
|
|
var tmpl = template.Must(template.New("").Parse(packageTmpl))
|
|
|
|
func main() {
|
|
var pkg string
|
|
var out string
|
|
|
|
log.SetFlags(0)
|
|
|
|
flag.Usage = func() {
|
|
log.Printf("usage: %s [-p package] <type names...>", filepath.Base(os.Args[0]))
|
|
flag.PrintDefaults()
|
|
}
|
|
|
|
flag.StringVar(&out, "o", "", "output, empty for stdout")
|
|
flag.StringVar(&pkg, "p", "discord", "package name")
|
|
flag.Parse()
|
|
|
|
if len(flag.Args()) == 0 {
|
|
flag.Usage()
|
|
os.Exit(1)
|
|
}
|
|
|
|
d := data{
|
|
Package: pkg,
|
|
ImportDiscord: pkg != "discord",
|
|
}
|
|
|
|
for _, arg := range flag.Args() {
|
|
d.Snowflakes = append(d.Snowflakes, snowflakeType{
|
|
TypeName: arg,
|
|
})
|
|
}
|
|
|
|
buf := bytes.Buffer{}
|
|
if err := tmpl.Execute(&buf, d); err != nil {
|
|
log.Fatalln("failed to execute template:", err)
|
|
}
|
|
|
|
b, err := format.Source(buf.Bytes())
|
|
if err != nil {
|
|
log.Fatalln("failed to fmt:", err)
|
|
}
|
|
|
|
outFile := os.Stdout
|
|
|
|
if out != "" {
|
|
f, err := os.Create(out)
|
|
if err != nil {
|
|
log.Fatalln("failed to create output file:", err)
|
|
}
|
|
defer f.Close()
|
|
|
|
outFile = f
|
|
}
|
|
|
|
if _, err := outFile.Write(b); err != nil {
|
|
log.Fatalln("failed to write to file:", err)
|
|
}
|
|
}
|