package main

import (
	"encoding/xml"
	"fmt"
	"math/rand"
	"sync"

	"github.com/frida/frida-go/frida"
)

// 全局变量，保持 Frida 脚本对象
var (
	fridaScript *frida.Script
	session     *frida.Session
	device      frida.DeviceInt
	taskId      = int64(0x20000000)
	myWechatId  = ""

	// 全局共享的header字段，启动时初始化一次，两个protobuf共用
	globalSessionId   = uint32(rand.Int63n(4000000000) + 100000000)
	globalDeviceId    = rand.Uint64() | (0xFFFFFFFF << 32)
	globalClientProof = []byte(fmt.Sprintf("m64%s", generateClientProof(13)))

	msgChan          = make(chan *SendMsg, 100)
	pendingResultMap sync.Map // targetId -> chan error

	config = &Config{}

	userID2NicknameMap  sync.Map
	userID2FileMsgMap   sync.Map
	videoInfoMap        sync.Map // targetId -> *VideoInfo
	buf2RespChan        = make(chan *Buf2RespData, 10)
)

type WechatMessage struct {
	GroupId     string     `json:"group_id"`
	SelfID      string     `json:"self_id"`
	UserID      string     `json:"user_id"`
	Sender      *Sender    `json:"sender"`
	Time        int64      `json:"time"`
	PostType    string     `json:"post_type"`
	MessageId   string     `json:"message_id"`
	Message     []*Message `json:"message"`
	MsgResource string     `json:"msgsource"`
	RawMessage  string     `json:"raw_message"`
	ShowContent string     `json:"show_content"`
	MessageType string     `json:"message_type"`
}

type Sender struct {
	UserID   string `json:"user_id"`
	Nickname string `json:"nickname"`
}

type SendMsg struct {
	UserId  string
	GroupID string
	Content string
	Type    string
	AtUser  string

	FIleCdnUrl string
	Md5        string
	AesKey     string
	FilePath   string
	FileType   int

	CdnKey  string
	Md5Key  string
	VideoId string

	Duration  int32
	VideoSize int32

	ReferMsgId       string
	ReferMsgSender   string
	ReferMsgType     int
	ReferCreateTime  int64
	ReferMsgsource   string
	ReferDisplayName string
	ReferContent     string

	ResultChan chan error // 发送结果回传
}

type VideoInfo struct {
	Duration  int32
	VideoSize int32
}

type Buf2RespData struct {
	MsgType string
	Data    []byte
	Err     error
}

// SendRequest 请求结构体
type SendRequest struct {
	Message []*Message `json:"message"`
	UserID  string     `json:"user_id"`
	GroupID string     `json:"group_id"`
}

type Message struct {
	Type string           `json:"type"`
	Data *SendRequestData `json:"data"`
}

type SendRequestData struct {
	Id           string         `json:"id,omitempty"`
	Text         string         `json:"text,omitempty"`
	File         string         `json:"file,omitempty"`
	URL          string         `json:"url,omitempty"`
	QQ           string         `json:"qq,omitempty"`
	Media        []byte         `json:"media,omitempty"`
	ReplyMessage *WechatMessage `json:"reply_message,omitempty"`
}

type Config struct {
	FridaType       string `json:"frida_type"`
	SendURL         string `json:"send_url"`
	ReceiveHost     string `json:"receive_host"`
	FridaGadgetAddr string `json:"frida_gadget_addr"`
	OnebotToken     string `json:"onebot_token"`
	ImagePath       string `json:"image_path"`
	ConnType        string `json:"conn_type"`
	SendInterval    int    `json:"send_interval"`
	WechatPid       int    `json:"wechat_pid"`

	WechatConf string `json:"wechat_conf"`
}

// VoiceMsg 对应外层的 <msg> 标签
type VoiceMsg struct {
	XMLName  xml.Name      `xml:"msg"`
	VoiceMsg *VoiceMsgInfo `xml:"voicemsg"`
}

// VoiceMsgInfo 对应内部的 <voicemsg> 标签及其属性
type VoiceMsgInfo struct {
	EndFlag      int    `xml:"endflag,attr"`
	CancelFlag   int    `xml:"cancelflag,attr"`
	ForwardFlag  int    `xml:"forwardflag,attr"`
	VoiceFormat  int    `xml:"voiceformat,attr"`
	VoiceLength  int    `xml:"voicelength,attr"`
	Length       int    `xml:"length,attr"`
	BufID        int    `xml:"bufid,attr"`
	AESKey       string `xml:"aeskey,attr"`
	VoiceURL     string `xml:"voiceurl,attr"`
	VoiceMD5     string `xml:"voicemd5,attr"`
	ClientMsgID  string `xml:"clientmsgid,attr"`
	FromUserName string `xml:"fromusername,attr"`
}

// FileMsg 对应 <msg> 标签
type FileMsg struct {
	XMLName       xml.Name      `xml:"msg"`
	Image         Image         `xml:"img"`
	Emoji         Emoji         `xml:"emoji"`
	Video         VideoMsg      `xml:"videomsg"`
	GameExt       GameExt       `xml:"gameext"`
	AppMsg        AppMsg        `xml:"appmsg"`
	ExtCommonInfo ExtCommonInfo `xml:"extcommoninfo"`
	FromUsername  string        `xml:"fromusername"`
	Scene         string        `xml:"scene"`
	AppInfo       AppInfo       `xml:"appinfo"`
	CommentURL    string        `xml:"commenturl"`
}

// Image 对应 <img> 标签及其属性和子节点
type Image struct {
	// 属性（Attributes）
	AesKey      string `xml:"aeskey,attr"`
	EncryVer    int    `xml:"encryver,attr"`
	ThumbAesKey string `xml:"cdnthumbaeskey,attr"`
	ThumbURL    string `xml:"cdnthumburl,attr"`
	Length      int    `xml:"length,attr"`
	Md5         string `xml:"md5,attr"`
	HDHeight    int    `xml:"cdnhdheight,attr"`
	HDWidth     int    `xml:"cdnhdwidth,attr"`
	MidImgURL   string `xml:"cdnmidimgurl,attr"`

	// 子节点
	SecHashInfo string `xml:"secHashInfoBase64"`
	Live        Live   `xml:"live"`
}

// Live 对应 <live> 标签
type Live struct {
	Duration int    `xml:"duration"`
	Size     int    `xml:"size"`
	FileID   string `xml:"fileid"`
}

type DownloadRequest struct {
	FileID         string `json:"file_id"`
	Media          []byte `json:"media"`
	CDNURL         string `json:"cdn_url"`
	LastAppendTime int64  `json:"last_append_time"`
	FilePath       string `json:"file_path"`
}

type ScriptMessage struct {
	Type    string      `json:"type"`
	Payload interface{} `json:"payload"`
}

type Emoji struct {
	FromUsername      string `xml:"fromusername,attr"`
	ToUserName        string `xml:"tousername,attr"`
	Type              string `xml:"type,attr"`
	IdBuffer          string `xml:"idbuffer,attr"`
	Md5               string `xml:"md5,attr"`
	Len               string `xml:"len,attr"`
	ProductId         string `xml:"productid,attr"`
	AndroidMd5        string `xml:"androidmd5,attr"`
	AndroidLen        string `xml:"androidlen,attr"`
	S60v3Md5          string `xml:"s60v3md5,attr"`
	S60v3Len          string `xml:"s60v3len,attr"`
	S60v5Md5          string `xml:"s60v5md5,attr"`
	S60v5Len          string `xml:"s60v5len,attr"`
	CdnUrl            string `xml:"cdnurl,attr"`
	DesignerId        string `xml:"designerid,attr"`
	ThumbUrl          string `xml:"thumburl,attr"`
	EncryptUrl        string `xml:"encrypturl,attr"`
	AesKey            string `xml:"aeskey,attr"`
	ExternUrl         string `xml:"externurl,attr"`
	ExternMd5         string `xml:"externmd5,attr"`
	Width             string `xml:"width,attr"`
	Height            string `xml:"height,attr"`
	TpUrl             string `xml:"tpurl,attr"`
	TpAuthKey         string `xml:"tpauthkey,attr"`
	AttachedText      string `xml:"attachedtext,attr"`
	AttachedTextColor string `xml:"attachedtextcolor,attr"`
	LensId            string `xml:"lensid,attr"`
	EmojiAttr         string `xml:"emojiattr,attr"`
	LinkId            string `xml:"linkid,attr"`
	Desc              string `xml:"desc,attr"`
}

type GameExt struct {
	Type    string `xml:"type,attr"`
	Content string `xml:"content,attr"`
}

type AppMsg struct {
	AppID         string        `xml:"appid,attr"`
	SDKVer        string        `xml:"sdkver,attr"`
	Title         string        `xml:"title"`
	Type          string        `xml:"type"`
	Action        string        `xml:"action"`
	AppAttach     AppAttach     `xml:"appattach"`
	MD5           string        `xml:"md5"`
	WebViewShared WebViewShared `xml:"webviewshared"`
}

type AppAttach struct {
	TotalLen          string `xml:"totallen"`
	FileExt           string `xml:"fileext"`
	AttachID          string `xml:"attachid"`
	CdnAttachURL      string `xml:"cdnattachurl"`
	CdnThumbAesKey    string `xml:"cdnthumbaeskey"`
	AesKey            string `xml:"aeskey"`
	EncryVer          string `xml:"encryver"`
	FileKey           string `xml:"filekey"`
	OverwriteNewMsgID string `xml:"overwrite_newmsgid"`
	FileUploadToken   string `xml:"fileuploadtoken"`
}

type WebViewShared struct {
	JsAppID        string `xml:"jsAppId"`
	PublisherReqID string `xml:"publisherReqId"`
}

type ExtCommonInfo struct {
	MediaExpireAt string `xml:"media_expire_at"`
}

type AppInfo struct {
	Version string `xml:"version"`
	AppName string `xml:"appname"`
}

type VideoMsg struct {
	AesKey            string `xml:"aeskey,attr"`            // 视频解密 Key
	CdnVideoUrl       string `xml:"cdnvideourl,attr"`       // 视频 CDN 地址
	CdnThumbAesKey    string `xml:"cdnthumbaeskey,attr"`    // 缩略图解密 Key
	CdnThumbUrl       string `xml:"cdnthumburl,attr"`       // 缩略图 CDN 地址
	Length            int64  `xml:"length,attr"`            // 视频文件大小 (字节)
	PlayLength        int    `xml:"playlength,attr"`        // 播放时长 (秒)
	CdnThumbLength    int    `xml:"cdnthumblength,attr"`    // 缩略图大小
	CdnThumbWidth     int    `xml:"cdnthumbwidth,attr"`     // 缩略图宽度
	CdnThumbHeight    int    `xml:"cdnthumbheight,attr"`    // 缩略图高度
	FromUsername      string `xml:"fromusername,attr"`      // 发送者 ID
	Md5               string `xml:"md5,attr"`               // 视频 MD5
	NewMd5            string `xml:"newmd5,attr"`            // 新版 MD5
	IsPlaceholder     int    `xml:"isplaceholder,attr"`     // 是否占位符
	RawMd5            string `xml:"rawmd5,attr"`            // 原始 MD5
	RawLength         int64  `xml:"rawlength,attr"`         // 原始长度
	CdnRawVideoUrl    string `xml:"cdnrawvideourl,attr"`    // 原始视频 CDN 地址
	CdnRawVideoAesKey string `xml:"cdnrawvideoaeskey,attr"` // 原始视频解密 Key
	OverwriteNewMsgId int64  `xml:"overwritenewmsgid,attr"` // 覆盖消息 ID
	OriginSourceMd5   string `xml:"originsourcemd5,attr"`   // 原始源 MD5
	IsAd              int    `xml:"isad,attr"`              // 是否为广告
}