napi.cc

/**
 * @file napi.cc
 * @author @2kai2kai2
 *
 * @copyright Copyright (c) 2025 Kai Orita
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */

#include "lib.h"
#include "lib_nvidia.h"
#include <napi.h>

Napi::Number cpuUserTime(const Napi::CallbackInfo &info) {
    using namespace Napi;
    // Check that the correct number arguments were passed
    if (info.Length() != 1)
        throw Error::New(info.Env(), "Invalid number of arguments.");
    const size_t processid = info[0].ToNumber().Int64Value();
    // Run through platform-specific implementations; return 0 if not found.
    size_t user, kernel;
    if (!cpuTimeCpp(processid, user, kernel))
        return Number::New(info.Env(), 0);
    // Return if found
    return Number::New(info.Env(), user);
}

Napi::Number cpuKernelTime(const Napi::CallbackInfo &info) {
    using namespace Napi;
    // Check that the correct number arguments were passed
    if (info.Length() != 1)
        throw Error::New(info.Env(), "Invalid number of arguments.");
    const size_t processid = info[0].ToNumber().Int64Value();
    // Run through platform-specific implementations; return 0 if not found.
    size_t user, kernel;
    if (!cpuTimeCpp(processid, user, kernel))
        return Number::New(info.Env(), 0);
    // Return if found
    return Number::New(info.Env(), kernel);
}

Napi::Number cpuTime(const Napi::CallbackInfo &info) {
    using namespace Napi;
    // Check that the correct number arguments were passed
    if (info.Length() != 1)
        throw Error::New(info.Env(), "Invalid number of arguments.");
    const size_t processid = info[0].ToNumber().Int64Value();
    // Run through platform-specific implementations; return 0 if not found.
    size_t user, kernel;
    if (!cpuTimeCpp(processid, user, kernel))
        return Number::New(info.Env(), 0);
    // Return if found
    return Number::New(info.Env(), user + kernel);
}

Napi::Number memInfo(const Napi::CallbackInfo &info) {
    using namespace Napi;
    // Check that the correct number arguments were passed
    if (info.Length() != 1)
        throw Error::New(info.Env(), "Invalid number of arguments.");
    const size_t processid = info[0].ToNumber().Int64Value();
    // Run through platform-specific implementations; return 0 if not found.
    size_t total, rss;
    if (!memInfoCpp(processid, total, rss))
        return Number::New(info.Env(), 0);
    // Return if found
    return Number::New(info.Env(), total);
}

Napi::Number memRSS(const Napi::CallbackInfo &info) {
    using namespace Napi;
    // Check that the correct number arguments were passed
    if (info.Length() != 1)
        throw Error::New(info.Env(), "Invalid number of arguments.");
    const size_t processid = info[0].ToNumber().Int64Value();
    // Run through platform-specific implementations; return 0 if not found.
    size_t total, rss;
    if (!memInfoCpp(processid, total, rss))
        return Number::New(info.Env(), 0);
    // Return if found
    return Number::New(info.Env(), rss);
}

Napi::Number fileReadBytes(const Napi::CallbackInfo &info) {
    using namespace Napi;
    // Check that the correct number arguments were passed
    if (info.Length() != 1)
        throw Error::New(info.Env(), "Invalid number of arguments.");
    const size_t processid = info[0].ToNumber().Int64Value();
    // Run platform-specific implementations; return 0 if not found.
    size_t readSize, readCount, writeSize, writeCount;
    if (!fileInfoCpp(processid, readSize, readCount, writeSize, writeCount))
        return Number::New(info.Env(), 0);
    // Return if found
    return Number::New(info.Env(), readSize);
}

Napi::Number fileWriteBytes(const Napi::CallbackInfo &info) {
    using namespace Napi;
    // Check that the correct number arguments were passed
    if (info.Length() != 1)
        throw Error::New(info.Env(), "Invalid number of arguments.");
    const size_t processid = info[0].ToNumber().Int64Value();
    // Run platform-specific implementations; return 0 if not found.
    size_t readSize, readCount, writeSize, writeCount;
    if (!fileInfoCpp(processid, readSize, readCount, writeSize, writeCount))
        return Number::New(info.Env(), 0);
    // Return if found
    return Number::New(info.Env(), writeSize);
}

Napi::Object Initialize(Napi::Env env, Napi::Object exports) {
    using namespace Napi;
    exports.Set(
        String::New(env, "cpuUserTime"),
        Function::New(env, cpuUserTime)
    );
    exports.Set(
        String::New(env, "cpuKernelTime"),
        Function::New(env, cpuKernelTime)
    );
    exports.Set(String::New(env, "cpuTime"), Function::New(env, cpuTime));
    exports.Set(String::New(env, "memInfo"), Function::New(env, memInfo));
    exports.Set(String::New(env, "memRSS"), Function::New(env, memRSS));
    exports.Set(
        String::New(env, "fileRead"),
        Function::New(env, fileReadBytes)
    );
    exports.Set(
        String::New(env, "fileWrite"),
        Function::New(env, fileWriteBytes)
    );
    NvidiaGPU::Init(env, exports);
    return exports;
}

NODE_API_MODULE(node_ps_data, Initialize)