comm.cc

#include "comm.hh"

CommMessage::CommMessage(){
	message = "";
	command = "";
	clientID = "";
	serverID = "";
	timestamp = "";
	version = "1.0";
	count = 0;
	size = 0;
}

void CommMessage::updateMessage(std::string msg){
	message = msg;
}

bool CommMessage::parse(){
	char tempa[30],tempb[30];
	deviceIDs.clear();
	std::stringstream ss(message);
	std::string line;
	while(std::getline(ss,line)){
		if(sscanf(line.c_str(),"%s IoTPS\\%s",tempa,tempb) == 2){
			command = tempa;
			version = tempb;		
			continue;
		}
		else if(sscanf(line.c_str(),"ClientID: %s",tempa) == 1){
			clientID = tempa;
			continue;
		}
		else if(sscanf(line.c_str(),"ServerID: %s",tempa) == 1){
			serverID = tempa;
			continue;		
		}
		else if(sscanf(line.c_str(),"DeviceID: %s",tempa) == 1){
			deviceIDs.push_back(tempa);
			continue;		
		}
		else if(sscanf(line.c_str(),"Count: %s",tempa) == 1){
			count = (size_t)strtol(tempa,NULL,10);
			continue;
		}
		else if(sscanf(line.c_str(),"Size: %s",tempa) == 1){
			size = (size_t)strtol(tempa,NULL,10);
			continue;	
		}
		else if(sscanf(line.c_str(),"SeqNumber: %s",tempa) == 1){
			seqno = (size_t)strtol(tempa,NULL,10);
			continue;
		}
		else if(sscanf(line.c_str(),"TimeStamp: %s",tempa) == 1){
			timestamp = tempa;
			continue;		
		}
	}
	return true;
}

bool CommMessage::sanityCheck(){
	// check for errors 
}

void CommMessage::print(){
	//std::vector<std::string>::iterator itr;
	//std::cout<<"Command: "<<command<<std::endl<<"Version: "<<version<<std::endl<<"ClientId: "<<clientID<<std::endl \
	<<"ServerID: "<<serverID<<std::endl<<"Count: "<<count<<std::endl<<"Size: "<<size<<std::endl;
	//for(itr = deviceIDs.begin();itr != deviceIDs.end();itr++)
		//std::cout<<*itr<<std::endl;
	std::cout<<message<<std::endl;
}

// Get
std::string CommMessage::getCommand(){
	return command;
}

std::string CommMessage::getVersion(){
	return version;
}

std::string CommMessage::getClientID(){
	return clientID;
}

std::string CommMessage::getServerID(){
	return serverID;
}

size_t CommMessage::getSeqNumber(){
	return seqno;
}

std::string CommMessage::getTimeStamp(){
	return timestamp;
}

size_t CommMessage::getCount(){
	return count;
}

size_t CommMessage::getSize(){
	return size;
}

std::vector<std::string> CommMessage::getDeviceIDs(){
	return deviceIDs;
}

std::string CommMessage::getMessage(){
	return message;
}

//Updates
void CommMessage::updateVersion(std::string ver){
	version = ver;
}

void CommMessage::updateClientID(std::string cID){
	clientID = cID;
}

void CommMessage::updateServerID(std::string sID){
	serverID = sID;
}

void CommMessage::updateCount(size_t c){
	count = c;
}

void CommMessage::updateSize(size_t s){
	size = s;
}

void CommMessage::updateSeqNo(size_t sn){
	seqno = sn;
}

void CommMessage::updateTimeStamp(std::string t){
	timestamp = t;
}

void CommMessage::updateDeviceIDs(std::vector<std::string> dIDs){
	std::vector<std::string>::iterator itr;
	deviceIDs.clear();	
	for(itr = dIDs.begin();itr != dIDs.end();itr++)
		deviceIDs.push_back(*itr);
}

//Requests
std::string CommMessage::createListRequest(){
	std::string str;
	str = "LIST IoTPS\\"+version+"\r\nClientID: "+clientID+"\r\n\r\n";
	return str;
}

std::string CommMessage::createSubsRequest(){
	std::string str;
	std::vector<std::string>::iterator itr;
	std::stringstream cc;
	cc << count;	
	str = "SUBSCRIBE IoTPS\\"+version+"\r\nCount: "+cc.str()+"\r\n";
	for(itr = deviceIDs.begin();itr != deviceIDs.end();itr++)
		str+="DeviceID: "+(*itr)+"\r\n";
	str+="ClientID: "+clientID+"\r\n\r\n";
	return str;
}

std::string CommMessage::createUnsubsRequest(){
	std::string str;
	std::vector<std::string>::iterator itr;
	std::stringstream cc;
	cc << count;	
	str = "UNSUBSCRIBE IoTPS\\"+version+"\r\nCount: "+cc.str()+"\r\n";
	for(itr = deviceIDs.begin();itr != deviceIDs.end();itr++)
		str+="DeviceID: "+(*itr)+"\r\n";
	str+="ClientID: "+clientID+"\r\n\r\n";
	return str;
	
}

//Good Replies
std::string CommMessage::createListReply(){
	std::string str;
	std::vector<std::string>::iterator itr;	
	str = "OK IoTPS\\"+version+"\r\n";
	for(itr = deviceIDs.begin();itr != deviceIDs.end();itr++)
		str+="DeviceID: "+(*itr)+"\r\n";	
	str += "ServerID: "+serverID+"\r\n\r\n";
	return str;
}

std::string CommMessage::createSubscribeReply(){
	std::string str;
	std::stringstream ss,cc;
	ss << size;
	cc << count;
	std::vector<std::string>::iterator itr;	
	str = "OK IoTPS\\"+version+"\r\nCount: "+cc.str()+"\r\n";
	for(itr = deviceIDs.begin();itr != deviceIDs.end();itr++)
		str+="DeviceID: "+(*itr)+"\r\n";
	str+="ServerID: "+serverID+"\r\n\r\n";
	return str;
}

std::string CommMessage::createUnsubscribeReply(){
	std::string str;
	std::vector<std::string>::iterator itr;	
	std::stringstream cc;
 	cc << count;
	str = "OK IoTPS\\"+version+"\r\nCount: "+cc.str()+"\r\n";
	for(itr = deviceIDs.begin();itr != deviceIDs.end();itr++)
		str+="DeviceID: "+(*itr)+"\r\n";
	str+="ServerID: "+serverID+"\r\n\r\n";
	return str;
}

std::string CommMessage::createUpdatesMessage(){
	std::string str;
	std::vector<std::string>::iterator itr;	
	std::stringstream ss,cc,snsn;
	ss << size;
	cc << count;
	snsn << seqno;
	str = "UPDATES IoTPS\\"+version+"\r\nCount: "+cc.str()+"\r\nSize: "+ss.str()+"\r\n";
	str += "SeqNumber: "+snsn.str()+"\r\n";
	str += "TimeStamp: "+timestamp+"\r\n";
	for(itr = deviceIDs.begin();itr != deviceIDs.end();itr++)
		str+="DeviceID: "+(*itr)+"\r\n";
	str+="ServerID: "+serverID+"\r\n\r\n";
	return str;
}


//Bad Replies 
std::string CommMessage::createErrorReply(){
	std::string str;
	str = "ERROR IoTPS\\"+version+"\r\nServerID: "+serverID+"\r\n\r\n";
	return str;
}
std::string CommMessage::createInvalidReply(){
	std::string str;
	str = "INVALID IoTPS\\"+version+"\r\nServerID: "+serverID+"\r\n\r\n";
	return str;
}