types.go

package main

import (
	"encoding/json"
)

// Query is the request body for /endpoints/verify-captcha
// Matches the TS type Query
// request_query.signature is the HMAC signature

type Query struct {
	Token        string      `json:"token"`
	TGLogin      TGLogin     `json:"tglogin"`
	RequestQuery RequestData `json:"request_query"`
}

type RequestData struct {
	ChatID    int64  `json:"chat_id,string"`
	MsgID     int64  `json:"msg_id,string"`
	Timestamp int64  `json:"timestamp,string"`
	Signature string `json:"signature"`
}

type TGUser struct {
	ID        int64  `json:"id"`
	FirstName string `json:"first_name"`
	LastName  string `json:"last_name"`
	Username  string `json:"username"`
}

// TGLogin 支持动态字段
type TGLogin struct {
	QueryID  string                 `json:"query_id"`
	User     string                 `json:"user"`
	AuthDate int64                  `json:"auth_date,string"`
	Hash     string                 `json:"hash"`
	Extra    map[string]interface{} `json:"-"`
}

func (t *TGLogin) UnmarshalJSON(data []byte) error {
	type Alias TGLogin
	temp := &struct {
		*Alias
	}{Alias: (*Alias)(t)}
	if err := json.Unmarshal(data, &temp); err != nil {
		return err
	}
	var raw map[string]interface{}
	if err := json.Unmarshal(data, &raw); err != nil {
		return err
	}
	t.Extra = make(map[string]interface{})
	for k, v := range raw {
		if k != "query_id" && k != "user" && k != "auth_date" && k != "hash" {
			t.Extra[k] = v
		}
	}
	return nil
}