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:
Adam Van Ymeren
2025-06-27 20:03:37 -07:00
committed by GitHub
parent 0952df0244
commit 7a04f298d2
19264 changed files with 1539697 additions and 84 deletions
@@ -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{})
}