/*
* Copyright 2012-2014 Aerospike, Inc.
*
* Portions may be licensed to Aerospike, Inc. under one or more contributor
* license agreements.
*
* Licensed under the Apache License, Version 2.0 (the "License"); you may not
* use this file except in compliance with the License. You may obtain a copy of
* the License at http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
* License for the specific language governing permissions and limitations under
* the License.
*/
package com.aerospike.client.lua;
import java.util.ArrayList;
import java.util.Iterator;
import org.luaj.vm2.LuaTable;
import org.luaj.vm2.LuaValue;
import org.luaj.vm2.Varargs;
import org.luaj.vm2.lib.OneArgFunction;
import org.luaj.vm2.lib.ThreeArgFunction;
import org.luaj.vm2.lib.TwoArgFunction;
import org.luaj.vm2.lib.VarArgFunction;
import org.luaj.vm2.lib.ZeroArgFunction;
public final class LuaListLib extends OneArgFunction {
private final LuaInstance instance;
public LuaListLib(LuaInstance instance) {
this.instance = instance;
instance.load(new MetaLib(instance));
}
public LuaValue call(LuaValue env) {
LuaTable meta = new LuaTable(0,2);
meta.set("__call", new create(instance));
LuaTable table = new LuaTable(0,8);
table.setmetatable(meta);
table.set("size", new len());
table.set("iterator", new iterator());
table.set("append", new append());
table.set("prepend", new prepend());
table.set("take", new take());
table.set("drop", new drop());
// list.clone is defined in as.lua
// table.set("clone", new clone());
instance.registerPackage("list", table);
return table;
}
public static final class MetaLib extends OneArgFunction {
private final LuaInstance instance;
public MetaLib(LuaInstance instance) {
this.instance = instance;
}
public LuaValue call(LuaValue env) {
LuaTable meta = new LuaTable(0, 5);
meta.set("__len", new len());
meta.set("__tostring", new tostring());
meta.set("__index", new index());
meta.set("__newindex", new newindex());
instance.registerPackage("List", meta);
return meta;
}
}
public static final class create extends VarArgFunction {
private final LuaInstance instance;
public create(LuaInstance instance) {
this.instance = instance;
}
@Override
public Varargs invoke(Varargs args) {
LuaList list = new LuaList(instance, new ArrayList<LuaValue>());
if (args.istable(2)) {
LuaTable table = args.checktable(2);
LuaValue k = LuaValue.NIL;
while (true) {
Varargs n = table.next(k);
if ((k = n.arg1()).isnil())
break;
LuaValue v = n.arg(2);
list.append(v);
}
}
return LuaValue.varargsOf(new LuaValue[] {list});
}
}
public static final class iterator extends OneArgFunction {
@Override
public LuaValue call(LuaValue l) {
LuaList list = (LuaList)l;
return new nextLuaValue(list.iterator());
}
}
public static final class nextLuaValue extends ZeroArgFunction {
private final Iterator<LuaValue> iter;
public nextLuaValue(Iterator<LuaValue> iter) {
this.iter = iter;
}
@Override
public LuaValue call() {
return (iter.hasNext())? iter.next() : LuaValue.NIL;
}
}
public static final class append extends TwoArgFunction {
@Override
public LuaValue call(LuaValue l, LuaValue value) {
LuaList list = (LuaList)l;
list.append(value);
return NIL;
}
}
public static final class prepend extends TwoArgFunction {
@Override
public LuaValue call(LuaValue l, LuaValue value) {
LuaList list = (LuaList)l;
list.prepend(value);
return NIL;
}
}
public static final class take extends TwoArgFunction {
@Override
public LuaValue call(LuaValue l, LuaValue count) {
LuaList list = (LuaList)l;
return list.take(count);
}
}
public static final class drop extends TwoArgFunction {
@Override
public LuaValue call(LuaValue l, LuaValue count) {
LuaList list = (LuaList)l;
return list.drop(count);
}
}
public static final class len extends OneArgFunction {
@Override
public LuaValue call(LuaValue l) {
LuaList list = (LuaList)l;
return list.size();
}
}
public static final class tostring extends OneArgFunction {
@Override
public LuaValue call(LuaValue l) {
LuaList list = (LuaList)l;
return list.toLuaString();
}
}
public static final class index extends TwoArgFunction {
@Override
public LuaValue call(LuaValue l, LuaValue index) {
LuaList list = (LuaList)l;
return list.get(index);
}
}
public static final class newindex extends ThreeArgFunction {
@Override
public LuaValue call(LuaValue l, LuaValue index, LuaValue value) {
LuaList list = (LuaList)l;
list.set(index, value);
return NIL;
}
}
}