commit c218a0d: [Minor] Lua_url: Add comparison metamethods

[Vsevolod Stakhov]

Author: Vsevolod Stakhov
Date: 2021-02-22 16:26:44 +0000
URL: https://github.com/rspamd/rspamd/commit/c218a0dcefc2afcd82c8aff016630d5fe34aaee1 (HEAD -> master)

[Minor] Lua_url: Add comparison metamethods

---
 src/lua/lua_url.c | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)

diff --git a/src/lua/lua_url.c b/src/lua/lua_url.c
index 7909444d5..13a75b40b 100644
--- a/src/lua/lua_url.c
+++ b/src/lua/lua_url.c
@@ -70,6 +70,8 @@ LUA_FUNCTION_DEF (url, get_visible);
 LUA_FUNCTION_DEF (url, create);
 LUA_FUNCTION_DEF (url, init);
 LUA_FUNCTION_DEF (url, all);
+LUA_FUNCTION_DEF (url, lt);
+LUA_FUNCTION_DEF (url, eq);
 
 static const struct luaL_reg urllib_m[] = {
 	LUA_INTERFACE_DEF (url, get_length),
@@ -98,6 +100,8 @@ static const struct luaL_reg urllib_m[] = {
 	{"get_redirected", lua_url_get_phished},
 	LUA_INTERFACE_DEF (url, set_redirected),
 	{"__tostring", lua_url_tostring},
+	{"__eq", lua_url_eq},
+	{"__lt", lua_url_lt},
 	{NULL, NULL}
 };
 
@@ -1233,6 +1237,41 @@ lua_url_adjust_skip_prob (gdouble timestamp,
 	return sz;
 }
 
+static gint
+lua_url_eq (lua_State *L)
+{
+	LUA_TRACE_POINT;
+	struct rspamd_lua_url *u1 = lua_check_url (L, 1),
+			*u2 = lua_check_url (L, 2);
+
+	if (u1 && u2) {
+		lua_pushboolean (L, (rspamd_url_cmp (u1->url, u2->url) == 0));
+	}
+	else {
+		lua_pushboolean (L, false);
+	}
+
+	return 1;
+}
+
+static gint
+lua_text_lt (lua_State *L)
+{
+	LUA_TRACE_POINT;
+	struct rspamd_lua_url *u1 = lua_check_url (L, 1),
+			*u2 = lua_check_url (L, 2);
+
+	if (u1 && u2) {
+		lua_pushinteger (L, rspamd_url_cmp (u1->url, u2->url));
+	}
+	else {
+		return luaL_error (L, "invalid arguments");
+	}
+
+	return 1;
+}
+
+
 static gint
 lua_load_url (lua_State * L)
 {