From bc4da963e5ebc651d1d7839a4da39ba59dfa4ea0 Mon Sep 17 00:00:00 2001
From: aunsane <aunsane@gmail.com>
Date: Sat, 2 Sep 2017 22:08:32 +0300
Subject: MirLua: small refactoring

---
 plugins/MirLua/src/m_database.cpp | 72 ++++++++++++++++-----------------------
 1 file changed, 30 insertions(+), 42 deletions(-)

(limited to 'plugins/MirLua/src/m_database.cpp')

diff --git a/plugins/MirLua/src/m_database.cpp b/plugins/MirLua/src/m_database.cpp
index 79a07c898a..9619b36584 100644
--- a/plugins/MirLua/src/m_database.cpp
+++ b/plugins/MirLua/src/m_database.cpp
@@ -53,54 +53,42 @@ static int db_Contacts(lua_State *L)
 	return 1;
 }
 
+static const char *mods[] =
+{
+	"FirstName",
+	"LastName",
+	"Nick",
+	"CustomNick",
+	"Email",
+	"City",
+	"State",
+	"Country",
+	"Phone",
+	"Homepage",
+	"About",
+	"Gender",
+	"Age",
+	"FullName",
+	"Uid",
+	"DisplayName",
+	NULL
+};
+
 static int db_GetContactInfo(lua_State *L)
 {
 	MCONTACT hContact = luaL_checkinteger(L, 1);
 
 	int type = 0;
-	if (lua_isnumber(L, 2))
-		type = lua_tointeger(L, 2);
-	else if (lua_isstring(L, 2))
+	switch (lua_type(L, 2))
 	{
-		const char *key = luaL_checkstring(L, 2);
-
-		if (mir_strcmpi(key, "FirstName") == 0)
-			type = CNF_FIRSTNAME;
-		else if (mir_strcmpi(key, "LastName") == 0)
-			type = CNF_LASTNAME;
-		else if (mir_strcmpi(key, "Nick") == 0)
-			type = CNF_NICK;
-		else if (mir_strcmpi(key, "CustomNick") == 0)
-			type = CNF_CUSTOMNICK;
-		else if (mir_strcmpi(key, "FullName") == 0)
-			type = CNF_FIRSTLAST;
-		else if (mir_strcmpi(key, "DisplayName") == 0)
-			type = CNF_DISPLAY;
-		else if (mir_strcmpi(key, "Uid") == 0)
-			type = CNF_UNIQUEID;
-		else if (mir_strcmpi(key, "Email") == 0)
-			type = CNF_EMAIL;
-		else if (mir_strcmpi(key, "City") == 0)
-			type = CNF_CITY;
-		else if (mir_strcmpi(key, "State") == 0)
-			type = CNF_STATE;
-		else if (mir_strcmpi(key, "Country") == 0)
-			type = CNF_COUNTRY;
-		else if (mir_strcmpi(key, "Phone") == 0)
-			type = CNF_PHONE;
-		else if (mir_strcmpi(key, "Homepage") == 0)
-			type = CNF_HOMEPAGE;
-		else if (mir_strcmpi(key, "About") == 0)
-			type = CNF_ABOUT;
-		else if (mir_strcmpi(key, "Age") == 0)
-			type = CNF_AGE;
-		else if (mir_strcmpi(key, "Gender") == 0)
-			type = CNF_GENDER;
-		else
-		{
-			lua_pushnil(L);
-			return 1;
-		}
+	case LUA_TNUMBER:
+		type = luaL_checkinteger(L, 2);
+		break;
+	case LUA_TSTRING:
+		type = luaL_checkoption(L, 2, NULL, mods) + 1;
+		break;
+	default:
+		luaL_argerror(L, 1, luaL_typename(L, 2));
 	}
 
 	ptrW value(Contact_GetInfo(type, hContact));
-- 
cgit v1.2.3