package client

import (
	"fmt"
	"github.com/Mrs4s/MiraiGo/binary"
	"time"
)

// --- tlv decoders for qq client ---

func (c *QQClient) decodeT161(data []byte) {
	reader := binary.NewReader(data)
	reader.ReadBytes(2)
	t := reader.ReadTlvMap(2)
	if t172, ok := t[0x172]; ok {
		c.rollbackSig = t172
	}
}

func (c *QQClient) decodeT119(data []byte) {
	tea := binary.NewTeaCipher(SystemDeviceInfo.TgtgtKey)
	reader := binary.NewReader(tea.Decrypt(data))
	reader.ReadBytes(2)
	m := reader.ReadTlvMap(2)

	if t130, ok := m[0x130]; ok {
		c.decodeT130(t130)
	}
	if t113, ok := m[0x113]; ok {
		c.decodeT113(t113)
	}
	if t528, ok := m[0x528]; ok {
		c.t528 = t528
	}
	if t530, ok := m[0x530]; ok {
		c.t530 = t530
	}
	if t108, ok := m[0x108]; ok {
		c.ksid = t108
	}

	var (
		//openId   []byte
		//openKey  []byte
		//payToken []byte
		//pf       []byte
		//pfkey    []byte
		gender uint16 = 0
		age    uint16 = 0
		nick          = ""
		//a1       []byte
		//noPicSig []byte
		//ctime           = time.Now().Unix()
		//etime           = ctime + 2160000
	)

	if _, ok := m[0x125]; ok {
		//openId, openKey = readT125(t125)
	}
	if t186, ok := m[0x186]; ok {
		c.decodeT186(t186)
	}
	if t11a, ok := m[0x11a]; ok {
		nick, age, gender = readT11A(t11a)
	}
	if _, ok := m[0x199]; ok {
		//openId, payToken = readT199(t199)
	}
	if _, ok := m[0x200]; ok {
		//pf, pfkey = readT200(t200)
	}
	if _, ok := m[0x512]; ok {

	} // 暂不处理, Http api cookie
	if _, ok := m[0x531]; ok {
		//a1, noPicSig = readT531(t531)
	}

	c.sigInfo = &loginSigInfo{
		loginBitmap:        0,
		tgt:                m[0x10a],
		tgtKey:             m[0x10d],
		userStKey:          m[0x10e],
		userStWebSig:       m[0x103],
		sKey:               m[0x120],
		d2:                 m[0x143],
		d2Key:              m[0x305],
		wtSessionTicketKey: m[0x134],
		deviceToken:        m[0x322],
	}
	c.Nickname = nick
	c.Age = age
	c.Gender = gender
}

func (c *QQClient) decodeT130(data []byte) {
	reader := binary.NewReader(data)
	reader.ReadBytes(2)
	c.timeDiff = int64(reader.ReadInt32()) - time.Now().Unix()
	c.t149 = reader.ReadBytes(4)
}

func (c *QQClient) decodeT113(data []byte) {
	reader := binary.NewReader(data)
	uin := reader.ReadInt32() // ?
	fmt.Println("got t113 uin:", uin)
}

func (c *QQClient) decodeT186(data []byte) {
	c.pwdFlag = data[1] == 1
}

// --- tlv readers ---

func readT125(data []byte) (openId, openKey []byte) {
	reader := binary.NewReader(data)
	openId = reader.ReadBytesShort()
	openKey = reader.ReadBytesShort()
	return
}

func readT11A(data []byte) (nick string, age, gender uint16) {
	reader := binary.NewReader(data)
	reader.ReadUInt16()
	age = uint16(reader.ReadByte())
	gender = uint16(reader.ReadByte())
	nick = reader.ReadStringLimit(int(reader.ReadByte()) & 0xff)
	return
}

func readT199(data []byte) (openId, payToken []byte) {
	reader := binary.NewReader(data)
	openId = reader.ReadBytesShort()
	payToken = reader.ReadBytesShort()
	return
}

func readT200(data []byte) (pf, pfKey []byte) {
	reader := binary.NewReader(data)
	pf = reader.ReadBytesShort()
	pfKey = reader.ReadBytesShort()
	return
}

func readT531(data []byte) (a1, noPicSig []byte) {
	reader := binary.NewReader(data)
	m := reader.ReadTlvMap(2)
	if m.Exists(0x103) && m.Exists(0x16a) && m.Exists(0x113) && m.Exists(0x10c) {
		a1 = append(m[0x106], m[0x10c]...)
		noPicSig = m[0x16a]
	}
	return
}