Added enums and querying + resetting status functions

Author: dankmolot

source/async_postgres.hpp

@@ -145,6 +145,7 @@ namespace async_postgres {
 
     // misc.cpp
     void register_misc_connection_functions(GLua::ILuaInterface* lua);
+    void register_enums(GLua::ILuaInterface* lua);
 
     // util.cpp
     std::string_view get_string(GLua::ILuaInterface* lua, int index = -1);

source/main.cpp

@@ -249,6 +249,20 @@ namespace async_postgres::lua {
                       state->query);  // busy if reset or query are in progress
         return 1;
     }
+
+    lua_protected_fn(querying) {
+        lua->CheckType(1, async_postgres::connection_meta);
+        auto state = lua_connection_state();
+        lua->PushBool(state->query.has_value());
+        return 1;
+    }
+
+    lua_protected_fn(resetting) {
+        lua->CheckType(1, async_postgres::connection_meta);
+        auto state = lua_connection_state();
+        lua->PushBool(state->reset_event.has_value());
+        return 1;
+    }
 }  // namespace async_postgres::lua
 
 #define register_lua_fn(name)                      \
@@ -272,6 +286,8 @@ void register_connection_mt(GLua::ILuaInterface* lua) {
     register_lua_fn(setNotifyCallback);
     register_lua_fn(wait);
     register_lua_fn(isBusy);
+    register_lua_fn(querying);
+    register_lua_fn(resetting);
 
     async_postgres::register_misc_connection_functions(lua);
 
@@ -283,6 +299,8 @@ void make_global_table(GLua::ILuaInterface* lua) {
 
     register_lua_fn(connect);
 
+    async_postgres::register_enums(lua);
+
     lua->SetField(GLua::INDEX_GLOBAL, "async_postgres");
 }
 

source/misc.cpp

@@ -1,3 +1,5 @@
+
+
 #include "async_postgres.hpp"
 
 using namespace async_postgres;
@@ -195,3 +197,41 @@ void async_postgres::register_misc_connection_functions(
     register_lua_fn(escapeBytea);
     register_lua_fn(unescapeBytea);
 }
+
+#define enum_value(name) {#name, name}
+
+void async_postgres::register_enums(GLua::ILuaInterface* lua) {
+    using enum_array = std::vector<std::pair<const char*, int>>;
+
+    std::vector<enum_array> enums = {
+        {
+            enum_value(CONNECTION_OK),
+            enum_value(CONNECTION_BAD),
+        },
+        {
+            enum_value(PQTRANS_IDLE),
+            enum_value(PQTRANS_ACTIVE),
+            enum_value(PQTRANS_INTRANS),
+            enum_value(PQTRANS_INERROR),
+            enum_value(PQTRANS_UNKNOWN),
+        },
+        {
+            enum_value(PQERRORS_TERSE),
+            enum_value(PQERRORS_DEFAULT),
+            enum_value(PQERRORS_VERBOSE),
+            enum_value(PQERRORS_SQLSTATE),
+        },
+        {
+            enum_value(PQSHOW_CONTEXT_NEVER),
+            enum_value(PQSHOW_CONTEXT_ERRORS),
+            enum_value(PQSHOW_CONTEXT_ALWAYS),
+        },
+    };
+
+    for (const auto& e : enums) {
+        for (const auto& [name, value] : e) {
+            lua->PushNumber(value);
+            lua->SetField(-2, name);
+        }
+    }
+}