request.cpp 3.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117
  1. #include "request.h"
  2. #include "net/http_reqpack.h"
  3. #include "session.h"
  4. module::request::request(network::http::request* request) :m_request(request)
  5. {
  6. }
  7. module::request::~request()
  8. {
  9. if (m_session != nullptr)
  10. delete m_session;
  11. }
  12. std::string module::request::token()
  13. {
  14. return m_request->token();
  15. }
  16. module::session* module::request::session(const std::string& token)
  17. {
  18. if (m_session == nullptr)
  19. m_session = new module::session(&m_request->session(token));
  20. return m_session;
  21. }
  22. sol::table module::request::body_param(sol::this_state s)
  23. {
  24. sol::state_view lua(s);
  25. sol::table result_table = lua.create_table();
  26. auto map = m_request->parser()->body_param();
  27. for_iter(iter,map)
  28. result_table[iter->first] = iter->second;
  29. return result_table;
  30. }
  31. sol::table module::request::url_param(sol::this_state s)
  32. {
  33. sol::state_view lua(s);
  34. sol::table result_table = lua.create_table();
  35. auto map = m_request->parser()->url_param();
  36. for_iter(iter, map)
  37. result_table[iter->first] = iter->second;
  38. return result_table;
  39. }
  40. std::string module::request::body()
  41. {
  42. return m_request->parser()->text();
  43. }
  44. void* module::request::website()
  45. {
  46. return m_request->website();
  47. }
  48. void module::request::regist(sol::state* lua)
  49. {
  50. // 绑定 Request 类到 Lua
  51. lua->new_usertype<module::request>("module_request",
  52. "header", &module::request::header,
  53. "method", &module::request::method,
  54. "filepath", &module::request::filepath,
  55. "host", &module::request::host,
  56. "remote_ipaddress", &module::request::remote_ipaddress,
  57. "remote_port", &module::request::remote_port,
  58. "param", &module::request::param,
  59. "session", &module::request::session,
  60. "token", &module::request::token,
  61. "body_param", &module::request::body_param,
  62. "url_param", &module::request::url_param,
  63. "body", &module::request::body
  64. );
  65. (*lua)["GET"] = (int)network::http::GET;
  66. (*lua)["POST"] = (int)network::http::POST;
  67. (*lua)["DEL"] = (int)network::http::DEL;
  68. (*lua)["HEAD"] = (int)network::http::HEAD;
  69. (*lua)["PUT"] = (int)network::http::PUT;
  70. }
  71. std::string module::request::header(const std::string& name)
  72. {
  73. std::string value;
  74. m_request->header(name, value);
  75. return value;
  76. }
  77. ylib::network::http::method module::request::method()
  78. {
  79. return m_request->method();
  80. }
  81. std::string module::request::filepath()
  82. {
  83. return m_request->filepath();
  84. }
  85. std::string module::request::host()
  86. {
  87. return m_request->host();
  88. }
  89. VarType module::request::param(const std::string& name, bool throw_,sol::this_state s)
  90. {
  91. std::string value;
  92. bool result = request_param(name, value);
  93. if (result == false){
  94. if (throw_)
  95. throw ylib::exception("request parameter '" + name + "' was not found");
  96. else
  97. return sol::make_object(s, sol::nil);
  98. }
  99. return sol::make_object(s,value);
  100. }
  101. std::string module::request::remote_ipaddress()
  102. {
  103. return m_request->remote_ipaddress(false);
  104. }
  105. ushort module::request::remote_port()
  106. {
  107. return m_request->remote_port();
  108. }
  109. bool module::request::request_param(const std::string& name, std::string& value)
  110. {
  111. if (m_request->parser()->url_param(name, value) == false)
  112. return m_request->parser()->body_param(name, value);
  113. return true;
  114. }