vis-lua: make vis:count assignable

mikaoP · Feb 23, 2017 · dddf34d · dddf34d
1 parent bfd007c
commit dddf34d
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/vis-lua.c b/vis-lua.c
@@ -1147,6 +1147,16 @@ static int vis_newindex(lua_State *L) {
 			vis_mode_switch(vis, mode);
 			return 0;
 		}
+
+		if (strcmp(key, "count") == 0) {
+			int count;
+			if (lua_isnil(L, 3))
+				count = VIS_COUNT_UNKNOWN;
+			else
+				count = luaL_checkunsigned(L, 3);
+			vis_count_set(vis, count);
+			return 1;
+		}
 	}
 	return newindex_common(L);
 }