refactor code for checking condition.

1 files changed, 28 insertions, 22 deletions

diff --git a/lua/lua-plugin-init.c b/lua/lua-plugin-init.c
index 3965452..ca4fb7e 100644
--- a/lua/lua-plugin-init.c
+++ b/lua/lua-plugin-init.c
@@ -66,44 +66,54 @@ static gboolean ime_is_white_space(const char c){
   return FALSE;
 }
 
-static int ime_trim_string_left(lua_State* L){
-  size_t l; int start, end;
-  const char * s = luaL_checklstring(L, 1, &l);
-  if (NULL == s){
+#define IME_TRIM_PRECHECK                       \
+  if (NULL == s){                               \
+  lua_pushliteral(L, "");                       \
+  return 1;                                     \
+  }
+
+
+static int ime_push_string(lua_State* L, const char * s, 
+                                int start, int end){
+  if (start >= end ){
     lua_pushliteral(L, "");
     return 1;
   }
+  lua_pushlstring(L, s + start, end -start);
+  return 1; 
+}
+
+static int ime_trim_string_left(lua_State* L){
+  size_t l; int start, end;
+  const char * s = luaL_checklstring(L, 1, &l);
+
+  IME_TRIM_PRECHECK;
   start = 0; end = strlen(s);
   while( ime_is_white_space(s[start])){
     start++;
   }
 
-  lua_pushlstring(L, s + start, end - start);
-  return 1;
+  return ime_push_string(L, s, start, end);
 }
 
 static int ime_trim_string_right(lua_State* L){
   size_t l; int start, end;
   const char * s = luaL_checklstring(L, 1, &l);
-  if ( NULL == s){
-    lua_pushliteral(L, "");
-    return 1;
-  }
+
+  IME_TRIM_PRECHECK;
   start = 0; end = strlen(s);
   while( ime_is_white_space(s[end - 1])){
     end--;
   }
-  lua_pushlstring(L, s + start, end -start);
-  return 1;
+
+  return ime_push_string(L, s, start, end);
 }
 
 static int ime_trim_string(lua_State* L){
   size_t l; int start, end;
   const char * s = luaL_checklstring(L, 1, &l);
-  if ( NULL == s){
-    lua_pushliteral(L, "");
-    return 1;
-  }
+
+  IME_TRIM_PRECHECK;
   start = 0; end = strlen(s);
   while( ime_is_white_space(s[start])){
     start++;
@@ -111,12 +121,8 @@ static int ime_trim_string(lua_State* L){
   while( ime_is_white_space(s[end - 1])){
     end--;
   }
-  if (start >= end ){
-    lua_pushliteral(L, "");
-    return 1;
-  }
-  lua_pushlstring(L, s + start, end -start);
-  return 1;
+
+  return ime_push_string(L, s, start, end);
 }