package luan.modules;

import java.io.IOException;
import java.util.Arrays;
import java.util.Collections;
import luan.Luan;
import luan.LuanState;
import luan.LuanTable;
import luan.LuanFunction;
import luan.LuanJavaFunction;
import luan.LuanElement;
import luan.LuanException;


public final class PackageLuan {

	public static final LuanFunction requireFn;
	static {
		try {
			requireFn = new LuanJavaFunction(PackageLuan.class.getMethod("require",LuanState.class,String.class),null);
		} catch(NoSuchMethodException e) {
			throw new RuntimeException(e);
		}
	}

	public static LuanTable loaded(LuanState luan) {
		LuanTable tbl = (LuanTable)luan.registry().get("Package.loaded");
		if( tbl == null ) {
			tbl = new LuanTable();
			luan.registry().put("Package.loaded",tbl);
		}
		return tbl;
	}

	public static Object require(LuanState luan,String modName) throws LuanException {
		Object mod = load(luan,modName);
		if( mod==null )
			throw luan.exception( "module '"+modName+"' not found" );
		return mod;
	}

	public static Object load(LuanState luan,String modName) throws LuanException {
		LuanTable loaded = loaded(luan);
		Object mod = loaded.rawGet(modName);
		if( mod == null ) {
			if( modName.startsWith("java:") ) {
				mod = JavaLuan.load(luan,modName.substring(5));
			} else {
				String src = read(luan,modName,true);
				if( src == null )
					return null;
				LuanFunction loader = BasicLuan.load(luan,src,modName,null,false);
				mod = Luan.first(
					loader.call(luan,new Object[]{modName})
				);
				if( mod == null ) {
					mod = loaded.rawGet(modName);
					if( mod != null )
						return mod;
					throw luan.exception( "module '"+modName+"' returned nil" );
				}
			}
			loaded.rawPut(modName,mod);
		}
		return mod;
	}

	static String read(LuanState luan,String uri,Boolean addExtension) throws LuanException {
		LuanTable t = IoLuan.uri(luan,uri,addExtension);
		if( t == null )
			return null;
		LuanFunction existsFn = (LuanFunction)t.get(luan.JAVA,"exists");
		boolean exists = (Boolean)Luan.first(existsFn.call(luan));
		if( !exists )
			return null;
		LuanFunction reader = (LuanFunction)t.get(luan.JAVA,"read_text");
		return (String)Luan.first(reader.call(luan));
	}

}