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

pkg/gotd/telegram/updates/hook/hook.go

@@ -0,0 +1,34 @@
+// Package hook contains telegram update hook middleware.
+package hook
+
+import (
+	"context"
+
+	"github.com/go-faster/errors"
+
+	"go.mau.fi/mautrix-telegram/pkg/gotd/bin"
+	"go.mau.fi/mautrix-telegram/pkg/gotd/telegram"
+	"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
+)
+
+// UpdateHook middleware is called on each tg.UpdatesClass method result.
+//
+// Function is called before invoker return. Returned error will be wrapped
+// and returned as InvokeRaw result.
+type UpdateHook func(ctx context.Context, u tg.UpdatesClass) error
+
+// Handle implements telegram.Middleware.
+func (h UpdateHook) Handle(next tg.Invoker) telegram.InvokeFunc {
+	return func(ctx context.Context, input bin.Encoder, output bin.Decoder) error {
+		if err := next.Invoke(ctx, input, output); err != nil {
+			return err
+		}
+		if u, ok := output.(*tg.UpdatesBox); ok {
+			if err := h(ctx, u.Updates); err != nil {
+				return errors.Wrap(err, "hook")
+			}
+		}
+
+		return nil
+	}
+}

pkg/gotd/telegram/updates/hook/hook_test.go

@@ -0,0 +1,90 @@
+package hook
+
+import (
+	"context"
+	"testing"
+
+	"github.com/go-faster/errors"
+
+	"github.com/stretchr/testify/assert"
+
+	"go.mau.fi/mautrix-telegram/pkg/gotd/telegram"
+
+	"go.mau.fi/mautrix-telegram/pkg/gotd/bin"
+	"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
+)
+
+func TestUpdateHook_InvokeRaw(t *testing.T) {
+	t.Run("Success", func(t *testing.T) {
+		var invokerCalled, hookCalled bool
+		assert.NoError(t, UpdateHook(func(ctx context.Context, u tg.UpdatesClass) error {
+			assert.NotNil(t, u)
+			hookCalled = true
+			return nil
+		}).Handle(telegram.InvokeFunc(func(ctx context.Context, input bin.Encoder, output bin.Decoder) error {
+			invokerCalled = true
+			return nil
+		})).Invoke(context.TODO(), nil, &tg.UpdatesBox{
+			Updates: &tg.UpdateShortMessage{
+				ID: 100,
+			},
+		}))
+
+		assert.True(t, invokerCalled, "invoker should be called")
+		assert.True(t, hookCalled, "hook should be called")
+	})
+	t.Run("Error", func(t *testing.T) {
+		t.Run("Handler", func(t *testing.T) {
+			var invokerCalled, hookCalled bool
+			err := errors.New("failure")
+			assert.ErrorIs(t, UpdateHook(func(ctx context.Context, u tg.UpdatesClass) error {
+				assert.NotNil(t, u)
+				hookCalled = true
+				return nil
+			}).Handle(telegram.InvokeFunc(func(ctx context.Context, input bin.Encoder, output bin.Decoder) error {
+				invokerCalled = true
+				return err
+			})).Invoke(context.TODO(), nil, &tg.UpdatesBox{
+				Updates: &tg.UpdateShortMessage{
+					ID: 100,
+				},
+			}), err)
+
+			assert.True(t, invokerCalled, "invoker should be called")
+			assert.False(t, hookCalled, "hook should not be called")
+		})
+		t.Run("Hook", func(t *testing.T) {
+			var invokerCalled, hookCalled bool
+			err := errors.New("failure")
+			assert.ErrorIs(t, UpdateHook(func(ctx context.Context, u tg.UpdatesClass) error {
+				assert.NotNil(t, u)
+				hookCalled = true
+				return err
+			}).Handle(telegram.InvokeFunc(func(ctx context.Context, input bin.Encoder, output bin.Decoder) error {
+				invokerCalled = true
+				return nil
+			})).Invoke(context.TODO(), nil, &tg.UpdatesBox{
+				Updates: &tg.UpdateShortMessage{
+					ID: 100,
+				},
+			}), err)
+
+			assert.True(t, invokerCalled, "invoker should be called")
+			assert.True(t, hookCalled, "hook should be called")
+		})
+	})
+	t.Run("Not update", func(t *testing.T) {
+		var invokerCalled, hookCalled bool
+		assert.NoError(t, UpdateHook(func(ctx context.Context, u tg.UpdatesClass) error {
+			assert.NotNil(t, u)
+			hookCalled = true
+			return nil
+		}).Handle(telegram.InvokeFunc(func(ctx context.Context, input bin.Encoder, output bin.Decoder) error {
+			invokerCalled = true
+			return nil
+		})).Invoke(context.TODO(), nil, &tg.User{}))
+
+		assert.True(t, invokerCalled, "invoker should be called")
+		assert.False(t, hookCalled, "hook should not be called")
+	})
+}