struct_tool/msgid_rw.cpp at master · javapretty/struct_tool · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
#include "msgid_rw.h"

MsgId_RW::MsgId_RW(char *path, char *srcfile, char *dstfile):
	srcfile_(""),
	dstfile_(dstfile),
	path_(),
	msgIdDefines_()
{
	char temp[128] = {};
	srcfile_ = get_string_field(temp, srcfile, 1);
	memcpy(path_, path, strlen(path));
	memcpy(path_ + strlen(path), srcfile, strlen(srcfile));
}

MsgId_RW::~MsgId_RW(){

}

int MsgId_RW::readFile(){
	FILE *fp = fopen(path_, "rb+");
	if(fp == NULL){
		std::cout<<"Can not open the file "<<srcfile_<<std::endl;
		exit(0);
	}
	get_enum(fp);
	fclose(fp);
	return 0;
}

int MsgId_RW::get_enum(FILE *fp){
	char temp[128] = {};
	char dst[128] = {};
	char slip[8] = "\x20\t\n,={";
	while(fgets(temp, 128, fp)){
		if(strncmp("enum", temp, 4)){
			continue;
		}
		MsgId_Define *msgid_def = new MsgId_Define(get_string_field(dst, temp, 2, slip));
		msgid_def->readAttr(fp);
		msgIdDefines_.push_back(msgid_def);
	}
	return 0;
}

int MsgId_RW::write_to_h(){
	FILE *fp = fopen("CPP/Message.h", "w+");
	if(fp == NULL){
		std::cout<<"Can not open the .h file "<<dstfile_<<std::endl;
		exit(0);
	}
	write_to_h_head(fp);
	write_to_h_enum(fp);
	write_to_h_end(fp);
	fclose(fp);
	return 0;
}

int MsgId_RW::write_to_h_head(FILE *fp){
	char temp[256] = {};
	sprintf(temp, HEAD_FILE_COMMENT, srcfile_.c_str(), srcfile_.c_str());
	fputs(temp, fp);
	return 0;
}

int MsgId_RW::write_to_h_end(FILE *fp){
	char temp[256] = {};
	sprintf(temp, HEAD_FILE_END);
	fputs(temp, fp);
	return 0;
}

int MsgId_RW::write_to_h_enum(FILE *fp){
	for(MSGIDDEFINES::iterator iter = msgIdDefines_.begin(); iter != msgIdDefines_.end(); iter++){
		(*iter)->write_to_h(fp);
	}
	return 0;
}

int MsgId_RW::write_to_js(){
	FILE *fp = fopen("../js/message.js", "w+");
	if(fp == NULL){
		std::cout<<"Can not open the js file "<<dstfile_<<std::endl;
		exit(0);
	}
	long pos = ftell(fp);
	if(pos == 0)
		write_to_js_head(fp);
	write_to_js_enum(fp);
	fclose(fp);
	return 0;
}

int MsgId_RW::write_to_js_head(FILE *fp){
	return 0;
}

int MsgId_RW::write_to_js_enum(FILE *fp){
	std::string js_name = "Msg";
	char temp[256] = {};
	sprintf(temp, ENUM_JS_HEAD, js_name.c_str(), js_name.c_str());
	fputs(temp, fp);
	for(MSGIDDEFINES::iterator iter = msgIdDefines_.begin(); iter != msgIdDefines_.end(); iter++){
		(*iter)->write_to_js(js_name, fp);
	}
	sprintf(temp, JS_STRUCT_END);
	fputs(temp, fp);
	return 0;
}