move gotd fork into repo. (#111)
- update to latest telegram layer - remove some references to fields in tg.Entities that don't exist in the schema - originally added here: https://github.com/beeper/td/commit/820929062a2ba0104397bc01235ab58a9cff780e - referenced here - https://github.com/mautrix/telegramgo/commit/124f0967ed195b5a380c9bd02e170ada9710dde3 - https://github.com/mautrix/telegramgo/commit/4205047aab2e0639217148b5d125bfaab668bd8e
This commit is contained in:
@@ -0,0 +1,43 @@
|
||||
package main
|
||||
|
||||
const entityTmpl = `// Code generated by mkentity, DO NOT EDIT.
|
||||
package {{ $.PackageName }}
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
|
||||
)
|
||||
|
||||
var (
|
||||
_ = tg.Invoker(nil)
|
||||
_ = context.Context(nil)
|
||||
)
|
||||
|
||||
{{- /*gotype: go.mau.fi/mautrix-telegram/pkg/gotd/telegram/message/internal/mkrun.Config*/ -}}
|
||||
{{- range $typ := $.Data }}
|
||||
{{ $helperName := trimPrefix ( trimPrefix $typ.Name "Input" ) "MessageEntity" -}}
|
||||
// {{ $helperName }} creates Formatter of {{ $helperName }} message entity.
|
||||
//
|
||||
// See https://core.telegram.org/constructor/{{ $typ.SchemaType.Name }}.
|
||||
func {{ $helperName }}({{- range $f := $typ.Fields }}{{ lowerFirst $f.Name }} {{ $f.Type }}{{- end }}) Formatter {
|
||||
return func(offset, length int) tg.MessageEntityClass {
|
||||
return &tg.{{ $typ.Name }}{
|
||||
Offset: offset,
|
||||
Length: length,
|
||||
{{- range $f := $typ.Fields }}
|
||||
{{ $f.Name }}: {{ lowerFirst $f.Name }},
|
||||
{{- end }}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// {{ $helperName }} adds and formats message as {{ $helperName }} message entity.
|
||||
//
|
||||
// See https://core.telegram.org/constructor/{{ $typ.SchemaType.Name }}.
|
||||
func (b *Builder) {{ $helperName }}(s string,
|
||||
{{- range $f := $typ.Fields }}{{ lowerFirst $f.Name }} {{ $f.Type }}{{- end }}) *Builder {
|
||||
return b.Format(s, {{ $helperName }}({{- range $f := $typ.Fields }}{{ lowerFirst $f.Name }},{{- end }}))
|
||||
}
|
||||
{{- end }}
|
||||
`
|
||||
@@ -0,0 +1,103 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"reflect"
|
||||
|
||||
"github.com/go-faster/errors"
|
||||
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/tdp"
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/telegram/message/internal/mkrun"
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
|
||||
)
|
||||
|
||||
// Field represents type field.
|
||||
type Field struct {
|
||||
// Name is Go name of field.
|
||||
Name string
|
||||
// Type is Go type of field.
|
||||
Type string
|
||||
}
|
||||
|
||||
// Type represents generated type.
|
||||
type Type struct {
|
||||
// Name is Go name of type.
|
||||
Name string
|
||||
// Fields is slice of type fields.
|
||||
Fields []Field
|
||||
// SchemaType is related schema type.
|
||||
SchemaType tdp.Type
|
||||
}
|
||||
|
||||
var (
|
||||
constructors = tg.ClassConstructorsMap()
|
||||
create = tg.TypesConstructorMap()
|
||||
templates = map[string]string{
|
||||
"entity": entityTmpl,
|
||||
"styling": stylingTmpl,
|
||||
}
|
||||
)
|
||||
|
||||
type generator struct {
|
||||
template string
|
||||
}
|
||||
|
||||
func (g *generator) Name() string {
|
||||
return "mkentity"
|
||||
}
|
||||
|
||||
func (g *generator) Flags(set *flag.FlagSet) {
|
||||
set.StringVar(&g.template, "template", "entity", "template to use")
|
||||
}
|
||||
|
||||
func (g *generator) Template() string {
|
||||
return templates[g.template]
|
||||
}
|
||||
|
||||
func (g *generator) Data() (interface{}, error) {
|
||||
var types []Type
|
||||
for _, typeID := range constructors[tg.MessageEntityClassName] {
|
||||
v, ok := create[typeID]().(tdp.Object)
|
||||
if !ok {
|
||||
return nil, errors.Errorf("bad type %#x", typeID)
|
||||
}
|
||||
schemaType := v.TypeInfo()
|
||||
// Skip messageEntityMentionName because we should use inputMessageEntityMentionName.
|
||||
if schemaType.Name == "messageEntityMentionName" {
|
||||
continue
|
||||
}
|
||||
|
||||
tv := reflect.TypeOf(v).Elem()
|
||||
|
||||
var fields []Field
|
||||
for _, field := range schemaType.Fields {
|
||||
// These fields set by Formatter callee.
|
||||
if field.Name == "Offset" || field.Name == "Length" {
|
||||
continue
|
||||
}
|
||||
|
||||
rf, ok := tv.FieldByName(field.Name)
|
||||
if !ok {
|
||||
return nil, errors.Errorf(
|
||||
"field of %q type %q not found",
|
||||
field.Name, schemaType.Name,
|
||||
)
|
||||
}
|
||||
fields = append(fields, Field{
|
||||
Name: field.Name,
|
||||
Type: rf.Type.String(),
|
||||
})
|
||||
}
|
||||
types = append(types, Type{
|
||||
Name: tv.Name(),
|
||||
Fields: fields,
|
||||
SchemaType: v.TypeInfo(),
|
||||
})
|
||||
}
|
||||
|
||||
return types, nil
|
||||
}
|
||||
|
||||
func main() {
|
||||
mkrun.Main(&generator{})
|
||||
}
|
||||
@@ -0,0 +1,31 @@
|
||||
package main
|
||||
|
||||
const stylingTmpl = `// Code generated by mkentity, DO NOT EDIT.
|
||||
package {{ $.PackageName }}
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
|
||||
)
|
||||
|
||||
var (
|
||||
_ = tg.Invoker(nil)
|
||||
_ = context.Context(nil)
|
||||
)
|
||||
|
||||
{{- /*gotype: go.mau.fi/mautrix-telegram/pkg/gotd/telegram/message/internal/mkrun.Config*/ -}}
|
||||
{{- range $typ := $.Data }}
|
||||
{{ $helperName := trimPrefix ( trimPrefix $typ.Name "Input" ) "MessageEntity" -}}
|
||||
// {{ $helperName }} formats text as {{ $helperName }} entity.
|
||||
//
|
||||
// See https://core.telegram.org/constructor/{{ $typ.SchemaType.Name }}.
|
||||
func {{ $helperName }}(s string,
|
||||
{{- range $f := $typ.Fields }}{{ lowerFirst $f.Name }} {{ $f.Type }}{{- end }}) StyledTextOption {
|
||||
return styledTextOption(s, func(b *textBuilder) error {
|
||||
b.{{ $helperName }}(s, {{- range $f := $typ.Fields }}{{ lowerFirst $f.Name }},{{- end }})
|
||||
return nil
|
||||
})
|
||||
}
|
||||
{{- end }}
|
||||
`
|
||||
@@ -0,0 +1,15 @@
|
||||
package mkrun
|
||||
|
||||
import "flag"
|
||||
|
||||
// Generator represents generator script.
|
||||
type Generator interface {
|
||||
// Name is generator name.
|
||||
Name() string
|
||||
// Flags sets generator flags.
|
||||
Flags(set *flag.FlagSet)
|
||||
// Template returns generation template.
|
||||
Template() string
|
||||
// Data returns associated generation data.
|
||||
Data() (interface{}, error)
|
||||
}
|
||||
@@ -0,0 +1,115 @@
|
||||
// Package mkrun contains some helpers for generation scripts.
|
||||
package mkrun
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"flag"
|
||||
"fmt"
|
||||
"go/format"
|
||||
"io"
|
||||
"os"
|
||||
"strings"
|
||||
"text/template"
|
||||
"time"
|
||||
"unicode"
|
||||
"unicode/utf8"
|
||||
|
||||
"github.com/go-faster/errors"
|
||||
"go.uber.org/multierr"
|
||||
)
|
||||
|
||||
// Config is generation config.
|
||||
type Config struct {
|
||||
PackageName string
|
||||
Data interface{}
|
||||
}
|
||||
|
||||
func generate(w io.Writer, pkgName string, g Generator) error {
|
||||
start := time.Now()
|
||||
data, err := g.Data()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
collectInfoTime := time.Since(start)
|
||||
|
||||
start = time.Now()
|
||||
buf := bytes.Buffer{}
|
||||
t, err := template.New("gen").Funcs(template.FuncMap{
|
||||
"trimPrefix": strings.TrimPrefix,
|
||||
"trimSuffix": strings.TrimSuffix,
|
||||
"lowerFirst": func(s string) string {
|
||||
r, size := utf8.DecodeRuneInString(s)
|
||||
if r == utf8.RuneError || unicode.IsLower(r) {
|
||||
return s
|
||||
}
|
||||
return string(unicode.ToLower(r)) + s[size:]
|
||||
},
|
||||
}).Parse(g.Template())
|
||||
if err != nil {
|
||||
return errors.Errorf("parse: %w", err)
|
||||
}
|
||||
|
||||
if err := t.Execute(&buf, Config{
|
||||
PackageName: pkgName,
|
||||
Data: data,
|
||||
}); err != nil {
|
||||
return errors.Errorf("execute: %w", err)
|
||||
}
|
||||
|
||||
formatted, err := format.Source(buf.Bytes())
|
||||
if err != nil {
|
||||
_, _ = os.Stderr.Write(buf.Bytes())
|
||||
return errors.Errorf("format: %w", err)
|
||||
}
|
||||
|
||||
if _, err := w.Write(formatted); err != nil {
|
||||
return errors.Errorf("write: %w", err)
|
||||
}
|
||||
writeTime := time.Since(start)
|
||||
|
||||
fmt.Printf("Generation %s complete, collect time: %s, write time: %s\n",
|
||||
g.Name(),
|
||||
collectInfoTime,
|
||||
writeTime,
|
||||
)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func run(g Generator) (rErr error) {
|
||||
set := flag.NewFlagSet(os.Args[0], flag.ExitOnError)
|
||||
var (
|
||||
o = set.String("output", "", "output file")
|
||||
pkgName = set.String("package", os.Getenv("GOPACKAGE"), "package name")
|
||||
)
|
||||
g.Flags(set)
|
||||
if err := set.Parse(os.Args[1:]); err != nil {
|
||||
return errors.Wrap(err, "parse")
|
||||
}
|
||||
|
||||
if *pkgName == "" {
|
||||
if *o != "" {
|
||||
return errors.New("package name is empty")
|
||||
}
|
||||
*pkgName = "pkg"
|
||||
}
|
||||
|
||||
var w io.Writer = os.Stdout
|
||||
if path := *o; path != "" {
|
||||
f, err := os.Create(path) // #nosec G304
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer multierr.AppendInvoke(&rErr, multierr.Close(f))
|
||||
w = f
|
||||
}
|
||||
|
||||
return generate(w, *pkgName, g)
|
||||
}
|
||||
|
||||
// Main is generation main function.
|
||||
func Main(g Generator) {
|
||||
if err := run(g); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,115 @@
|
||||
// Binary mktyping generates TypingActionBuilder.
|
||||
package main
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"reflect"
|
||||
|
||||
"github.com/go-faster/errors"
|
||||
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/tdp"
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/telegram/message/internal/mkrun"
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
|
||||
)
|
||||
|
||||
// Field represents type field.
|
||||
type Field struct {
|
||||
// Name is Go name of field.
|
||||
Name string
|
||||
// Type is Go type of field.
|
||||
Type string
|
||||
}
|
||||
|
||||
// Type represents generated type.
|
||||
type Type struct {
|
||||
// Name is Go name of type.
|
||||
Name string
|
||||
// Fields is slice of type fields.
|
||||
Fields []Field
|
||||
// SchemaType is related schema type.
|
||||
SchemaType tdp.Type
|
||||
}
|
||||
|
||||
const rawTemplate = `// Code generated by mktyping, DO NOT EDIT.
|
||||
package {{ $.PackageName }}
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
|
||||
)
|
||||
|
||||
var (
|
||||
_ = tg.Invoker(nil)
|
||||
_ = context.Context(nil)
|
||||
)
|
||||
|
||||
{{- /*gotype: go.mau.fi/mautrix-telegram/pkg/gotd/telegram/message/internal/mkrun.Config*/ -}}
|
||||
{{- range $typ := $.Data }}
|
||||
{{ $helperName := trimSuffix (trimPrefix $typ.Name "SendMessage") "Action" -}}
|
||||
// {{ $helperName }} sends {{ $typ.Name }}.
|
||||
func (b *TypingActionBuilder) {{ $helperName }}(ctx context.Context,
|
||||
{{- range $f := $typ.Fields }}{{ lowerFirst $f.Name }} {{ $f.Type }},{{ end }}) error {
|
||||
return b.send(ctx, &tg.{{ $typ.Name }}{
|
||||
{{- range $f := $typ.Fields }}
|
||||
{{ $f.Name }}: {{ lowerFirst $f.Name }},
|
||||
{{- end }}
|
||||
})
|
||||
}
|
||||
{{- end }}
|
||||
`
|
||||
|
||||
var (
|
||||
constructors = tg.ClassConstructorsMap()
|
||||
create = tg.TypesConstructorMap()
|
||||
)
|
||||
|
||||
type generator struct{}
|
||||
|
||||
func (g generator) Name() string {
|
||||
return "mktyping"
|
||||
}
|
||||
|
||||
func (g generator) Flags(set *flag.FlagSet) {}
|
||||
|
||||
func (g generator) Template() string {
|
||||
return rawTemplate
|
||||
}
|
||||
|
||||
func (g generator) Data() (interface{}, error) {
|
||||
var types []Type
|
||||
for _, typeID := range constructors[tg.SendMessageActionClassName] {
|
||||
v, ok := create[typeID]().(tdp.Object)
|
||||
if !ok {
|
||||
return nil, errors.Errorf("bad type %#x", typeID)
|
||||
}
|
||||
schemaType := v.TypeInfo()
|
||||
tv := reflect.TypeOf(v).Elem()
|
||||
|
||||
var fields []Field
|
||||
for _, field := range schemaType.Fields {
|
||||
rf, ok := tv.FieldByName(field.Name)
|
||||
if !ok {
|
||||
return nil, errors.Errorf(
|
||||
"field of %q type %q not found",
|
||||
field.Name, schemaType.Name,
|
||||
)
|
||||
}
|
||||
fields = append(fields, Field{
|
||||
Name: field.Name,
|
||||
Type: rf.Type.String(),
|
||||
})
|
||||
}
|
||||
types = append(types, Type{
|
||||
Name: tv.Name(),
|
||||
Fields: fields,
|
||||
SchemaType: v.TypeInfo(),
|
||||
})
|
||||
}
|
||||
|
||||
return types, nil
|
||||
}
|
||||
|
||||
func main() {
|
||||
mkrun.Main(generator{})
|
||||
}
|
||||
Reference in New Issue
Block a user