Fix return value of table.remove()

src/addon/java/lua/addon/luacompat/LuaCompat.java

@@ -723,7 +723,8 @@ public class LuaCompat extends LFunction {
 		int n = vm.gettop();
 		LTable table = vm.totable(2);
 		int pos = (n>=3? vm.tointeger(3): 0);
-		table.luaRemovePos( pos );
+		vm.settop(0);
+		vm.pushlvalue( table.luaRemovePos( pos ) );
 	}
 
 	/** table.sort (table [, comp])

src/main/java/lua/value/LTable.java

@@ -536,16 +536,22 @@ public class LTable extends LValue {
 	 * Remove an element from the list part of the table
 	 * @param pos position to remove, or 0 to remove last element
 	 */
-	public void luaRemovePos(int pos) {
+	public LValue luaRemovePos(int pos) {
 		if ( pos > m_arrayEntries ) {
+			LValue val = get( pos );
+			if ( val != LNil.NIL )
 				put( pos, LNil.NIL );
+			return val;
 		} else {
+			final int n = m_vector.length - 1;
 			final int index = Math.max(0,pos<=0? m_arrayEntries: pos)-1;
 			if ( index < 0 )
-				return;
-			System.arraycopy(m_vector, index+1, m_vector, index, m_vector.length-1-index);
-			m_vector[m_vector.length-1] = LNil.NIL;
+				return LNil.NIL;
+			LValue val = m_vector[index];
+			System.arraycopy(m_vector, index+1, m_vector, index, n-index);
+			m_vector[n] = LNil.NIL;
 			--m_arrayEntries;
+			return val;
 		}
 	}
 

src/test/res/table.lua

@@ -39,19 +39,19 @@ print( table.maxn({}), #{} )
 print( '-- remove tests' )
 t = { "one", "two", "three", "four", "five", "six", "seven", [10]="ten", a='aaa', b='bbb', c='ccc' }
 print( table.concat(t,'-'), table.maxn(t), #t )
-table.remove(t) 
+print( table.remove(t) )
 print( table.concat(t,'-'), table.maxn(t) )
-table.remove(t,1) 
+print( table.remove(t,1) )
 print( table.concat(t,'-'), table.maxn(t) )
-table.remove(t,3) 
+print( table.remove(t,3) )
 print( table.concat(t,'-'), table.maxn(t) )
-table.remove(t,5) 
+print( table.remove(t,5) )
 print( table.concat(t,'-'), table.maxn(t), t[10] )
-table.remove(t,10) 
+print( table.remove(t,10) )
 print( table.concat(t,'-'), table.maxn(t), t[10] )
-table.remove(t,-1) 
+print( table.remove(t,-1) )
 print( table.concat(t,'-'), table.maxn(t), t[10] )
-table.remove(t,-1) 
+print( table.remove(t,-1) ) 
 print( table.concat(t,'-'), table.maxn(t), t[10] )
 
 -- sort