acl/app/wizard/http_creator.cpp
2023-06-22 19:38:36 +08:00

216 lines
5.2 KiB
C++
Raw Blame History

#include "stdafx.h"
#include "file_tmpl.h"
#include "http_creator.h"
//////////////////////////////////////////////////////////////////////////
static bool set_cookies(tpl_t* tpl)
{
printf("Please enter cookie name: "); fflush(stdout);
char line[256];
int n = acl_vstream_gets_nonl(ACL_VSTREAM_IN, line, sizeof(line));
if (n == ACL_VSTREAM_EOF) {
printf("enter cookie name error!\r\n");
return false;
}
string buf;
buf.format("const char* %s = req.getCookieValue(\"%s\");\r\n", line, line);
buf.format_append("\tif (%s == NULL) {\r\n\t\tres.addCookie(\"%s\", \"{xxx}\");\r\n\t}\r\n", line, line);
tpl_set_field_fmt_global(tpl, "GET_COOKIES", "%s", buf.c_str());
return true;
}
//////////////////////////////////////////////////////////////////////////
static bool create_threads(file_tmpl& tmpl)
{
string file(tmpl.get_project_name());
file << ".cf";
if (tmpl.copy_and_replace("master_threads.cf", file.c_str()) == false) {
return false;
}
const char* name = "master_threads";
const FILE_FROM_TO tab[] = {
{ "main_threads.cpp", "main.cpp" },
{ "master_threads.h", "master_service.h" },
{ "master_threads.cpp", "master_service.cpp" },
{ "http_service.h", "http_service.h" },
{ "http_service.cpp", "http_service.cpp" },
{ "http_servlet.h", "http_servlet.h" },
{ "websocket.cpp", "websocket.cpp" },
{ "websocket.h", "websocket.h" },
{ NULL, NULL }
};
return tmpl.files_copy(name, tab);
}
static bool create_fiber(file_tmpl& tmpl)
{
string file(tmpl.get_project_name());
file << ".cf";
if (tmpl.copy_and_replace("master_fiber.cf", file.c_str()) == false)
return false;
const char* name = "master_fiber";
const FILE_FROM_TO tab[] = {
{ "stdafx_fiber.h", "stdafx.h" },
{ "main_fiber.cpp", "main.cpp" },
{ "master_fiber.h", "master_service.h" },
{ "master_fiber.cpp", "master_service.cpp" },
{ "http_service.h", "http_service.h" },
{ "http_service.cpp", "http_service.cpp" },
{ "http_servlet.h", "http_servlet.h" },
{ "websocket.cpp", "websocket.cpp" },
{ "websocket.h", "websocket.h" },
{ NULL, NULL }
};
return tmpl.files_copy(name, tab)
&& tmpl.copy_and_replace("Makefile_fiber", "Makefile")
&& tmpl.file_copy("tmpl/Makefile_fiber.in",
"Makefile.in");
}
static bool create_proc(file_tmpl& tmpl)
{
string file(tmpl.get_project_name());
file << ".cf";
if (tmpl.copy_and_replace("master_proc.cf", file.c_str()) == false) {
return false;
}
const char* name = "master_proc";
const FILE_FROM_TO tab[] = {
{ "main_proc.cpp", "main.cpp" },
{ "master_proc.h", "master_service.h" },
{ "master_proc.cpp", "master_service.cpp" },
{ "http_service.h", "http_service.h" },
{ "http_service.cpp", "http_service.cpp" },
{ "http_servlet.h", "http_servlet.h" },
{ "websocket.cpp", "websocket.cpp" },
{ "websocket.h", "websocket.h" },
{ NULL, NULL }
};
return tmpl.files_copy(name, tab);
}
static bool create_service(file_tmpl& tmpl, const char* type)
{
char buf[256];
if (type == NULL || *type == 0) {
printf("choose master_service type:\r\n");
printf(" t: for master_threads\r\n"
" f: for master_fiber\r\n"
" p: for master_proc\r\n");
printf(">");
fflush(stdout);
int n = acl_vstream_gets_nonl(ACL_VSTREAM_IN, buf, sizeof(buf));
if (n == ACL_VSTREAM_EOF) {
return false;
}
type = buf;
}
if (strcasecmp(type, "t") == 0) {
create_threads(tmpl);
} else if (strcasecmp(type, "p") == 0) {
create_proc(tmpl);
} else if (strcasecmp(type, "f") == 0) {
create_fiber(tmpl);
} else {
printf("invalid type: %s\r\n", type);
return false;
}
return true;
}
static bool create_http_servlet(file_tmpl& tmpl, const char* type)
{
tpl_t* tpl = tmpl.open_tpl("http_servlet.cpp");
if (tpl == NULL) {
return false;
}
printf("Do you want add cookie? [y/n]: ");
fflush(stdout);
char buf[64];
int n = acl_vstream_gets_nonl(ACL_VSTREAM_IN, buf, sizeof(buf));
if (n != ACL_VSTREAM_EOF && strcasecmp(buf, "y") == 0) {
set_cookies(tpl);
}
string filepath;
filepath.format("%s/http_servlet.cpp", tmpl.get_project_name());
if (tpl_save_as(tpl, filepath.c_str()) != TPL_OK) {
printf("save to %s error: %s\r\n", filepath.c_str(),
last_serror());
tpl_free(tpl);
return false;
}
printf("create %s ok.\r\n", filepath.c_str());
tpl_free(tpl);
// <20><><EFBFBD>÷<EFBFBD><C3B7><EFBFBD><EFBFBD><EFBFBD>ģ<EFBFBD><C4A3><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
return create_service(tmpl, type);
}
void http_creator(const char* name, const char* type)
{
bool loop;
file_tmpl tmpl;
if (name && *name && type && *type) {
loop = true;
} else {
loop = false;
}
// <20><><EFBFBD><EFBFBD>Դ<EFBFBD><D4B4><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ŀ¼
tmpl.set_path_from("tmpl/http");
while (true) {
char buf[256];
int n;
if (name == NULL || *name == 0) {
printf("please input your program name: ");
fflush(stdout);
n = acl_vstream_gets_nonl(ACL_VSTREAM_IN, buf, sizeof(buf));
if (n == ACL_VSTREAM_EOF) {
break;
}
if (n == 0) {
acl::safe_snprintf(buf, sizeof(buf), "http_demo");
}
name = buf;
}
tmpl.set_project_name(name);
// <20><><EFBFBD><EFBFBD>Ŀ¼
tmpl.create_dirs();
tmpl.create_common();
create_http_servlet(tmpl, type);
if (!loop) {
break;
}
}
}