from construct import *
from construct.protocols.layer3.ipv4 import IpAddress
import charserv
from protocol import *
from utils import *
from common import *
from loggers import netlog

server = None


@extendable
def smsg_server_version(data):
    netlog.info("SMSG_SERVER_VERSION {}.{}".format(data.hi, data.lo))
    cmsg_login_register(server.username, server.password)


@extendable
def smsg_update_host(data):
    netlog.info("SMSG_UPDATE_HOST {}".format(data.host))


@extendable
def smsg_login_data(data):
    netlog.info("SMSG_LOGIN_DATA {}".format(data))
    server.close()

    charserv.connect(data.worlds[0].address, data.worlds[0].port)
    charserv.server.char_name = server.char_name
    charserv.cmsg_char_server_connect(data.account, data.session1,
                                      data.session2, 1, data.gender)


@extendable
def smsg_login_error(data):
    error_codes = {
        0: "Unregistered ID",
        1: "Wrong password",
        2: "Account expired",
        3: "Rejected from server",
        4: "Permban",
        5: "Client too old",
        6: "Temporary ban until {}".format(data.date),
        7: "Server overpopulated",
        9: "Username already taken",
        10: "Wrong name",
        11: "Incurrect email",
        99: "Username permanently erased" }

    err_msg = "SMSG_LOGIN_ERROR {}".format(
        error_codes.get(data.code, "Unknown error"))
    netlog.error(err_msg)
    server.close()
    raise Exception(err_msg)


protodef = {
    0x7531 : (smsg_server_version,
              Struct("data",
                     ULInt32("hi"),
                     ULInt32("lo"))),
    0x0063 : (smsg_update_host,
              Struct("data",
                     ULInt16("length"),
                     StringZ("host",
                            lambda ctx: ctx.length - 4))),
    0x0069 : (smsg_login_data,
              Struct("data",
                     ULInt16("length"),
                     ULInt32("session1"),
                     ULInt32("account"),
                     ULInt32("session2"),
                     IpAddress("oldip"),
                     StringZ("lastlogin", 24),
                     Padding(2),
                     Gender("gender"),
                     Array(lambda ctx: (ctx.length - 47) / 32,
                           Struct("worlds",
                                  IpAddress("address"),
                                  ULInt16("port"),
                                  StringZ("name", 20),
                                  ULInt16("onlineusers"),
                                  Padding(4))))),

    0x006a : (smsg_login_error,
              Struct("data",
                     Byte("code"),
                     StringZ("date", 20)))
}


def cmsg_server_version_request():
    netlog.info("CMSG_SERVER_VERSION_REQUEST")
    ULInt16("opcode").build_stream(0x7530, server)


def cmsg_login_register(username, password):
    netlog.info("CMSG_LOGIN_REGISTER username={} password={}".format(
        username, password))
    send_packet(server, CMSG_LOGIN_REGISTER,
                (ULInt32("clientversion"),  6),
                (StringZ("username", 24),   username),
                (StringZ("password", 24),   password),
                (Byte("flags"),             3))


def connect(host, port):
    global server
    server = SocketWrapper(host=host, port=port, protodef=protodef)