feat: add user emoji status and custom verification methods

SetUserEmojiStatus (bots.updateUserEmojiStatus) and VerifyUser/VerifyChat/
RemoveUserVerification/RemoveChatVerification (bots.setCustomVerification).

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>

Author: ernado

emoji_status.go

@@ -0,0 +1,61 @@
+package botapi
+
+import (
+	"context"
+	"strconv"
+
+	"github.com/gotd/td/tg"
+)
+
+// EmojiStatusOption configures a SetUserEmojiStatus call.
+type EmojiStatusOption func(*emojiStatusConfig)
+
+type emojiStatusConfig struct {
+	until int
+}
+
+// WithEmojiStatusExpiration sets the Unix time when the emoji status will
+// expire. By default the status does not expire.
+func WithEmojiStatusExpiration(unixTime int) EmojiStatusOption {
+	return func(c *emojiStatusConfig) { c.until = unixTime }
+}
+
+// SetUserEmojiStatus sets the emoji status of a user that previously allowed the
+// bot to manage it. An empty customEmojiID removes the current emoji status.
+func (b *Bot) SetUserEmojiStatus(ctx context.Context, userID int64, customEmojiID string, opts ...EmojiStatusOption) error {
+	var cfg emojiStatusConfig
+
+	for _, o := range opts {
+		o(&cfg)
+	}
+
+	user, err := b.resolveInputUser(ctx, userID)
+	if err != nil {
+		return err
+	}
+
+	var status tg.EmojiStatusClass = &tg.EmojiStatusEmpty{}
+
+	if customEmojiID != "" {
+		id, err := strconv.ParseInt(customEmojiID, 10, 64)
+		if err != nil {
+			return errInvalidCustomEmojiID()
+		}
+
+		s := &tg.EmojiStatus{DocumentID: id}
+		if cfg.until != 0 {
+			s.SetUntil(cfg.until)
+		}
+
+		status = s
+	}
+
+	if _, err := b.raw.BotsUpdateUserEmojiStatus(ctx, &tg.BotsUpdateUserEmojiStatusRequest{
+		UserID:      user,
+		EmojiStatus: status,
+	}); err != nil {
+		return asAPIError(err)
+	}
+
+	return nil
+}

emoji_status_test.go

@@ -0,0 +1,47 @@
+package botapi
+
+import (
+	"context"
+	"testing"
+
+	"github.com/gotd/td/tg"
+)
+
+func TestSetUserEmojiStatus(t *testing.T) {
+	inv := newMockInvoker()
+	inv.reply(tg.BotsUpdateUserEmojiStatusRequestTypeID, &tg.BoolTrue{})
+
+	if err := newMockBot(inv).SetUserEmojiStatus(context.Background(), 99, "555", WithEmojiStatusExpiration(1000)); err != nil {
+		t.Fatalf("SetUserEmojiStatus: %v", err)
+	}
+
+	var req tg.BotsUpdateUserEmojiStatusRequest
+
+	inv.decode(t, tg.BotsUpdateUserEmojiStatusRequestTypeID, &req)
+
+	status, ok := req.EmojiStatus.(*tg.EmojiStatus)
+	if !ok || status.DocumentID != 555 {
+		t.Fatalf("status = %#v", req.EmojiStatus)
+	}
+
+	if until, ok := status.GetUntil(); !ok || until != 1000 {
+		t.Fatalf("until = %d, ok=%v", until, ok)
+	}
+}
+
+func TestSetUserEmojiStatusClear(t *testing.T) {
+	inv := newMockInvoker()
+	inv.reply(tg.BotsUpdateUserEmojiStatusRequestTypeID, &tg.BoolTrue{})
+
+	if err := newMockBot(inv).SetUserEmojiStatus(context.Background(), 99, ""); err != nil {
+		t.Fatalf("SetUserEmojiStatus: %v", err)
+	}
+
+	var req tg.BotsUpdateUserEmojiStatusRequest
+
+	inv.decode(t, tg.BotsUpdateUserEmojiStatusRequestTypeID, &req)
+
+	if _, ok := req.EmojiStatus.(*tg.EmojiStatusEmpty); !ok {
+		t.Fatalf("status = %#v, want empty", req.EmojiStatus)
+	}
+}

verification.go

@@ -0,0 +1,87 @@
+package botapi
+
+import (
+	"context"
+
+	"github.com/gotd/td/tg"
+)
+
+// VerificationOption configures a verify call.
+type VerificationOption func(*verificationConfig)
+
+type verificationConfig struct {
+	description string
+}
+
+// WithVerificationDescription sets a custom description for the verification
+// shown instead of the bot's default verification description.
+func WithVerificationDescription(text string) VerificationOption {
+	return func(c *verificationConfig) { c.description = text }
+}
+
+// setCustomVerification enables or disables the bot's custom verification of a
+// peer.
+func (b *Bot) setCustomVerification(ctx context.Context, peer tg.InputPeerClass, enabled bool, opts []VerificationOption) error {
+	var cfg verificationConfig
+
+	for _, o := range opts {
+		o(&cfg)
+	}
+
+	req := &tg.BotsSetCustomVerificationRequest{Peer: peer}
+	req.SetEnabled(enabled)
+
+	if enabled && cfg.description != "" {
+		req.SetCustomDescription(cfg.description)
+	}
+
+	if _, err := b.raw.BotsSetCustomVerification(ctx, req); err != nil {
+		return asAPIError(err)
+	}
+
+	return nil
+}
+
+// VerifyUser verifies a user on behalf of the organization represented by the
+// bot.
+func (b *Bot) VerifyUser(ctx context.Context, userID int64, opts ...VerificationOption) error {
+	user, err := b.resolveInputUser(ctx, userID)
+	if err != nil {
+		return err
+	}
+
+	return b.setCustomVerification(ctx, userToInputPeer(user), true, opts)
+}
+
+// VerifyChat verifies a chat on behalf of the organization represented by the
+// bot.
+func (b *Bot) VerifyChat(ctx context.Context, chat ChatID, opts ...VerificationOption) error {
+	peer, err := b.resolveInputPeer(ctx, chat)
+	if err != nil {
+		return err
+	}
+
+	return b.setCustomVerification(ctx, peer, true, opts)
+}
+
+// RemoveUserVerification removes the verification from a user that is verified on
+// behalf of the organization represented by the bot.
+func (b *Bot) RemoveUserVerification(ctx context.Context, userID int64) error {
+	user, err := b.resolveInputUser(ctx, userID)
+	if err != nil {
+		return err
+	}
+
+	return b.setCustomVerification(ctx, userToInputPeer(user), false, nil)
+}
+
+// RemoveChatVerification removes the verification from a chat that is verified on
+// behalf of the organization represented by the bot.
+func (b *Bot) RemoveChatVerification(ctx context.Context, chat ChatID) error {
+	peer, err := b.resolveInputPeer(ctx, chat)
+	if err != nil {
+		return err
+	}
+
+	return b.setCustomVerification(ctx, peer, false, nil)
+}

verification_test.go

@@ -0,0 +1,80 @@
+package botapi
+
+import (
+	"context"
+	"testing"
+
+	"github.com/gotd/td/tg"
+)
+
+func TestVerifyUser(t *testing.T) {
+	inv := newMockInvoker()
+	inv.reply(tg.BotsSetCustomVerificationRequestTypeID, &tg.BoolTrue{})
+
+	if err := newMockBot(inv).VerifyUser(context.Background(), 99, WithVerificationDescription("trusted")); err != nil {
+		t.Fatalf("VerifyUser: %v", err)
+	}
+
+	var req tg.BotsSetCustomVerificationRequest
+
+	inv.decode(t, tg.BotsSetCustomVerificationRequestTypeID, &req)
+
+	if !req.GetEnabled() {
+		t.Fatal("enabled should be true")
+	}
+
+	if desc, ok := req.GetCustomDescription(); !ok || desc != "trusted" {
+		t.Fatalf("description = %q, ok=%v", desc, ok)
+	}
+
+	if _, ok := req.Peer.(*tg.InputPeerUser); !ok {
+		t.Fatalf("peer = %#v, want user", req.Peer)
+	}
+}
+
+func TestVerifyChat(t *testing.T) {
+	inv := newMockInvoker()
+	inv.reply(tg.BotsSetCustomVerificationRequestTypeID, &tg.BoolTrue{})
+
+	if err := newMockBot(inv).VerifyChat(context.Background(), tdlibChannel(50)); err != nil {
+		t.Fatalf("VerifyChat: %v", err)
+	}
+
+	var req tg.BotsSetCustomVerificationRequest
+
+	inv.decode(t, tg.BotsSetCustomVerificationRequestTypeID, &req)
+
+	if _, ok := req.Peer.(*tg.InputPeerChannel); !ok {
+		t.Fatalf("peer = %#v, want channel", req.Peer)
+	}
+}
+
+func TestRemoveUserVerification(t *testing.T) {
+	inv := newMockInvoker()
+	inv.reply(tg.BotsSetCustomVerificationRequestTypeID, &tg.BoolTrue{})
+
+	if err := newMockBot(inv).RemoveUserVerification(context.Background(), 99); err != nil {
+		t.Fatalf("RemoveUserVerification: %v", err)
+	}
+
+	var req tg.BotsSetCustomVerificationRequest
+
+	inv.decode(t, tg.BotsSetCustomVerificationRequestTypeID, &req)
+
+	if req.GetEnabled() {
+		t.Fatal("enabled should be false")
+	}
+}
+
+func TestRemoveChatVerification(t *testing.T) {
+	inv := newMockInvoker()
+	inv.reply(tg.BotsSetCustomVerificationRequestTypeID, &tg.BoolTrue{})
+
+	if err := newMockBot(inv).RemoveChatVerification(context.Background(), tdlibChannel(50)); err != nil {
+		t.Fatalf("RemoveChatVerification: %v", err)
+	}
+
+	if !inv.called(tg.BotsSetCustomVerificationRequestTypeID) {
+		t.Fatal("verification not called")
+	}
+}