notify.cc

// Copyright (C) 2017 - 2019, Rory Bradford <rory@dysfunctionalprogramming.com> and contributors
// MIT License

#include <node.h>
#include <systemd/sd-daemon.h>
#include <systemd/sd-journal.h>

using v8::Number;

#define READY "READY=1"
#define STOPPING "STOPPING=1"
#define WATCHDOG "WATCHDOG=1"

namespace notify {

const char* ToCString(const v8::String::Utf8Value& value) {
  return *value ? *value : "STATUS=conversion failed";
}

void ready(const v8::FunctionCallbackInfo<v8::Value>& args) {
  int pid = args[0].As<Number>()->Value();
  sd_pid_notify(pid, 0, READY);
}

void stopping(const v8::FunctionCallbackInfo<v8::Value>& args) {
  int pid = args[0].As<Number>()->Value();
  sd_pid_notify(pid, 0, STOPPING);
}

void watchdog(const v8::FunctionCallbackInfo<v8::Value>& args) {
  sd_notify(0, WATCHDOG);
}

void sendstate(const v8::FunctionCallbackInfo<v8::Value>& args) {
  v8::Isolate* isolate = args.GetIsolate();
  v8::String::Utf8Value str(isolate, args[0]);
  const char *state = ToCString(str);
  sd_notify(0, state);
}

void interval(const v8::FunctionCallbackInfo<v8::Value>& args) {
  v8::Isolate* isolate = args.GetIsolate();
  uint64_t interval;
  int res = sd_watchdog_enabled(0, &interval);

  if (res > 0) {
    args.GetReturnValue().Set(v8::Number::New(isolate, interval / 1000));
  } else {
    args.GetReturnValue().Set(v8::Number::New(isolate, 0));
  }
}

void journal_print(const v8::FunctionCallbackInfo<v8::Value>& args) {
  int level = args[0].As<Number>()->Value();
  v8::Isolate* isolate = args.GetIsolate();
  v8::String::Utf8Value str(isolate, args[1]);
  const char *message = ToCString(str);
  sd_journal_print(level, message);
}

}  // namespace notify

void Init(v8::Local<v8::Object> exports) {
  NODE_SET_METHOD(exports, "ready", notify::ready);
  NODE_SET_METHOD(exports, "stopping", notify::stopping);
  NODE_SET_METHOD(exports, "watchdog", notify::watchdog);
  NODE_SET_METHOD(exports, "watchdogInterval", notify::interval);
  NODE_SET_METHOD(exports, "sendState", notify::sendstate);
  NODE_SET_METHOD(exports, "journalPrint", notify::journal_print);
}

NODE_MODULE(addon, Init)