/*!The Make-like Build Utility based on Lua
 *
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.
 * 
 * Copyright (C) 2015 - 2018, TBOOX Open Source Group.
 *
 * @author      ruki
 * @file        waitlist.c
 *
 */

/* //////////////////////////////////////////////////////////////////////////////////////
 * trace
 */
#define TB_TRACE_MODULE_NAME                "process.waitlist"
#define TB_TRACE_MODULE_DEBUG               (0)

/* //////////////////////////////////////////////////////////////////////////////////////
 * includes
 */
#include "prefix.h"

/* //////////////////////////////////////////////////////////////////////////////////////
 * implementation
 */

/* count, list = process.waitlist(proclist, timeout)
 *
 * count:
 *
 * the finished count: > 0
 * timeout: 0
 * failed: -1
 *
 * for _, procinfo in ipairs(list) do
 *     print("proc: ", procinfo[1])
 *     print("index: ", procinfo[2])
 *     print("status: ", procinfo[3])
 * end
 */
tb_int_t xm_process_waitlist(lua_State* lua)
{
    // check
    tb_assert_and_check_return_val(lua, 0);

    // is table?
    if (!lua_istable(lua, 1)) 
    { 
        // error
        lua_pushfstring(lua, "invalid argument type(%s) for process.waitlist", luaL_typename(lua, 1));
        lua_error(lua);
        return 0;
    }

    // get the processes count
    tb_long_t count = lua_objlen(lua, 1);
    if (count <= 0 || count > 64)
    {
        // error
        lua_pushfstring(lua, "invalid process count(%d) for process.waitlist", count);
        lua_error(lua);
        return 0;
    }

    // get the processes
    tb_int_t            i = 0;
    tb_process_ref_t    processes[64 + 1];
    for (i = 0; i < count; i++)
    {
        // get proclist[i]
        lua_pushinteger(lua, i + 1);
        lua_gettable(lua, 1);

        // is userdata?
        if (lua_isuserdata(lua, -1))
        {
            // save this process
            processes[i] = (tb_process_ref_t)lua_touserdata(lua, -1);
            if (!processes[i])
            {
                // error
                lua_pushfstring(lua, "process[%d] is null for process.waitlist", i);
                lua_error(lua);
            }
        }
        else
        {
            // error
            lua_pushfstring(lua, "invalid process[%d] type(%s) for process.waitlist", i, luaL_typename(lua, -1));
            lua_error(lua);
        }

        // pop it
        lua_pop(lua, 1);
    }

    // end
    processes[count] = tb_null;

    // get the timeout
    tb_long_t timeout = (tb_long_t)luaL_checkinteger(lua, 2);

    // wait it
    tb_process_waitinfo_t infolist[64];
    tb_long_t infosize = tb_process_waitlist(processes, infolist, tb_arrayn(infolist), timeout);

    // save process info count
    lua_pushinteger(lua, infosize);
    if (infosize >= 0)
    {
        // save process info list
        lua_newtable(lua);
        for (i = 0; i < infosize; i++)
        {
            // save one process info
            lua_newtable(lua);
            lua_pushlightuserdata(lua, infolist[i].process);
            lua_rawseti(lua, -2, 1);
            lua_pushinteger(lua, infolist[i].index + 1);
            lua_rawseti(lua, -2, 2);
            lua_pushinteger(lua, infolist[i].status);
            lua_rawseti(lua, -2, 3);

            lua_rawseti(lua, -2, i + 1);
        }
    }
    else lua_pushnil(lua);

    // ok
    return 2;
}