# -*- coding: utf-8
__author__="lreqc"
__date__ ="$2009-07-14 01:07:32$"

from sunshine.lqsoft.pygadu.packets import Resolver, inpacket, outpacket

from sunshine.lqsoft.cstruct.common import CStruct
from sunshine.lqsoft.cstruct.fields import complex, numeric

from sunshine.lqsoft.utils import Enum

PACKET_HEADER_LENGTH = 8

class GaduPacketHeader(CStruct):
    """Struktura opisująca nagłówek pakietu w GG"""
    msg_type        = numeric.UIntField(0)
    msg_length      = numeric.UIntField(1)

    def __str__(self):
        return '[GGHDR: type=%d, length %d]' % (self.msg_type, self.msg_length)

class GaduPacket(CStruct):
    """Wspólna nadklasa dla wszystkich wiadomości w GG"""
    def as_packet(self):
        data = self.pack()
        hdr = GaduPacketHeader(msg_type=self.packet_id, msg_length=len(data))
        return hdr.pack() + data

    def __str__(self):
        return self.__class__.__name__

#
# INCOMING PACKETS
#
class WelcomePacket(GaduPacket):
    seed = numeric.IntField(0)    
WelcomePacket = inpacket(0x01)(WelcomePacket)

class MessageAckPacket(GaduPacket): #SendMsgAck
    MSG_STATUS = Enum({
        'BLOCKED': 0x0001, 'DELIVERED': 0x0002,
        'QUEUED': 0x0003, 'MBOXFULL': 0x0004,
        'NOT_DELIVERED': 0x0006
    })

    msg_status  = numeric.IntField(0)
    recipient   = numeric.IntField(1)
    seq         = numeric.IntField(2)
MessageAckPacket = inpacket(0x05)(MessageAckPacket)

class LoginFailedPacket(GaduPacket):
    pass
LoginFailedPacket = inpacket(0x09)(LoginFailedPacket)

class DisconnectPacket(GaduPacket):
    pass
DisconnectPacket = inpacket(0x0b)(DisconnectPacket)

class NeedEmailPacket(GaduPacket):
    pass
NeedEmailPacket = inpacket(0x14)(NeedEmailPacket)

class UnavailbleAckPacket(GaduPacket):
    pass
UnavailbleAckPacket = inpacket(0x0d)(UnavailbleAckPacket)

class PongPacket(GaduPacket):
    pass
PongPacket = inpacket(0x07)(PongPacket)

#
# OUTGOING PACKETS
#
class StructNotice(CStruct): # Notify
    TYPE = Enum({
        'BUDDY':    0x01,
        'FRIEND':   0x02,
        'IGNORE':  0x04
    })
    
    uin             = numeric.UIntField(0)
    type            = numeric.UByteField(1, default=0x03)

    def __str__(self):
        return "%d[%d]" (self.uin, self.type)

class NoticeFirstPacket(GaduPacket): #NotifyFirst
    contacts        = complex.ArrayField(0, complex.StructField(0, struct=StructNotice), length=-1)
NoticeFirstPacket = outpacket(0x0f)(NoticeFirstPacket)

class NoticeLastPacket(GaduPacket): #NotifyLast
    contacts        = complex.ArrayField(0, complex.StructField(0, struct=StructNotice), length=-1)
NoticeLastPacket = outpacket(0x10)(NoticeLastPacket)

class NoNoticesPacket(GaduPacket):
    pass
NoNoticesPacket = outpacket(0x12)(NoNoticesPacket)

class AddNoticePacket(GaduPacket):
    contact        = complex.StructField(0, struct=StructNotice)
AddNoticePacket = outpacket(0x0d)(AddNoticePacket)

class RemoveNoticePacket(GaduPacket):
    contact        = complex.StructField(0, struct=StructNotice)
RemoveNoticePacket = outpacket(0x0e)(RemoveNoticePacket)

class PingPacket(GaduPacket):
    pass
PingPacket = outpacket(0x08)(PingPacket)