#include"TcpClient.hpp"
#include"Protocol.hpp"
#include"Sock.hpp"

using namespace Protocol_ns;


static void usage(std::string proc)
{
    std::cout << "Usage:\n\t" << proc << " serverip serverport\n"
            << std::endl;
}


enum
{
    LEFT,
    OPER,
    RIGHT
};


// 10+20 ===> 把输入填充到Request中
Request Parseline(string&line)
{
    string left,right;
    char op;
    int status=LEFT;

    int i=0;
    while(i<line.size())
    {
        switch(status)
        {
        case LEFT:
            if(isdigit(line[i]))
                left.push_back(line[i++]);
            else
                status=OPER;
            break;
        case OPER:
            op=line[i++];
            status=RIGHT;
            break;
        case RIGHT:
            if(isdigit(line[i]))
                right.push_back(line[i++]);
            break;
        }
    }

    Request req;
    cout<<"left: " << left <<endl;
    cout<<"rght: " << right <<endl;
    cout<<"op: " << op <<endl;

    req._x=stoi(left);
    req._y=stoi(right);
    req._op=op;

    return req;
}


// tcpclient serverip serverport

int main(int argc,char*argv[])
{
    if (argc != 3)
    {
        usage(argv[0]);
        exit(USAGE_ERR);
    }

    string serverip=argv[1];
    uint16_t serverport=atoi(argv[2]);

    Sock sock;
    sock.Socket();

    int n=sock.Connect(serverip,serverport);
    if(n!=0)
       return 1;


    string buffer;

    while(true)
    {

        cout<<"Enter# ";    // 12+9
        string line;
        getline(cin,line);

        Request req=Parseline(line);
        // cout<<"data1# ";
        // cin>>req._x;
        // cout<<"data2# ";
        // cin>>req._y;
        //  cout<<"op# ";
        // cin>>req._op;

        std::cout << "test: " << req._x << req._op << req._y << std::endl;


        // 1. 序列化
        string Sendstring;
        req.Serialize(&Sendstring);

        // 2. 添加报头
        Sendstring=AddHeader(Sendstring);

        // 3. send
        send(sock.Fd(),Sendstring.c_str(),Sendstring.size(),0);

        // 4. 获取响应
        string package;
        int n=0;
    START:
        n=Readpackage(sock.Fd(),buffer,&package);
        if(n==0)
            goto START;
        else if(n<0)
            break;
        else
        {
        }

        // 5. 去掉报头
        package=RemoveHeader(package,n);

        // 6. 反序列化
        Response resp;
        resp.Deserialize(package);

        cout<<"result: "<< resp._result <<"[code: "<<resp._code<< "]"<<endl;
    }

    sock.Close();

    return 0;
}