Notification.cpp

Go to the documentation of this file.

 
 
#include "Notification.h"
#include <cstdlib>
#include <sstream>
#include <iostream>
#include <iomanip>
#include <functional>
#include <unistd.h>
using namespace std;
 
std::string notify_ntfy_topic;
std::string notify_discord_webhook;
 
static std::string format_time(double seconds) {
    std::ostringstream ss;
    if (seconds < 60.0) {
        ss << std::fixed << std::setprecision(1) << seconds << "s";
    } else if (seconds < 3600.0) {
        int m = static_cast<int>(seconds) / 60;
        double s = seconds - m * 60;
        ss << m << "m " << std::fixed << std::setprecision(0) << s << "s";
    } else {
        int h = static_cast<int>(seconds) / 3600;
        int m = (static_cast<int>(seconds) % 3600) / 60;
        ss << h << "h " << m << "m";
    }
    return ss.str();
}
 
static std::string project_name(const std::string& path) {
    auto pos = path.find_last_of('/');
    if (pos != std::string::npos && pos + 1 < path.size())
        return path.substr(pos + 1);
    return path;
}
 
static void send_ntfy(const std::string& topic, const RunSummary& summary) {
    std::string name = project_name(summary.project);
    std::string time_str = format_time(summary.wall_seconds);
 
    std::ostringstream body;
    body << name << ": " << summary.sequences << " sequences, "
         << time_str << " wall time";
    if (summary.ranks > 1)
        body << " (" << summary.ranks << " ranks)";
 
    // curl -s --max-time 10 -H "Title: ADS-IDAC Complete" -H "Tags: white_check_mark"
    //      -d "<body>" ntfy.sh/<topic>
    std::ostringstream cmd;
    cmd << "curl -s --max-time 10"
        << " -H \"Title: ADS-IDAC Complete\""
        << " -H \"Tags: white_check_mark\""
        << " -d \"" << body.str() << "\""
        << " ntfy.sh/" << topic
        << " > /dev/null 2>&1";
 
    int rc = std::system(cmd.str().c_str());
    if (rc != 0)
        std::cerr << "[notify] ntfy.sh notification failed (curl exit " << rc << ")" << std::endl;
    else
        std::cout << "[notify] Sent ntfy.sh notification to topic '" << topic << "'" << std::endl;
}
 
static void send_discord(const std::string& webhook_url, const RunSummary& summary) {
    std::string name = project_name(summary.project);
    std::string time_str = format_time(summary.wall_seconds);
 
    // Build Discord embed JSON — all field values are controlled strings, no escaping needed
    std::ostringstream json;
    json << "{\"embeds\":[{\"title\":\"ADS-IDAC Simulation Complete\","
         << "\"color\":5763719,"  // green (#57F287)
         << "\"fields\":["
         << "{\"name\":\"Project\",\"value\":\"" << name << "\",\"inline\":true},"
         << "{\"name\":\"Sequences\",\"value\":\"" << summary.sequences << "\",\"inline\":true},"
         << "{\"name\":\"Wall Time\",\"value\":\"" << time_str << "\",\"inline\":true},"
         << "{\"name\":\"Ranks\",\"value\":\"" << summary.ranks << "\",\"inline\":true},"
         << "{\"name\":\"Solver\",\"value\":\"" << summary.solver << "\",\"inline\":true},"
         << "{\"name\":\"Truncation\",\"value\":\"T=" << std::fixed << std::setprecision(1)
         << summary.truncation_time << "\",\"inline\":true}"
         << "]}]}";
 
    // curl -s --max-time 10 -H "Content-Type: application/json" -d '<json>' <url>
    std::ostringstream cmd;
    cmd << "curl -s --max-time 10"
        << " -H \"Content-Type: application/json\""
        << " -d '" << json.str() << "'"
        << " " << webhook_url
        << " > /dev/null 2>&1";
 
    int rc = std::system(cmd.str().c_str());
    if (rc != 0)
        std::cerr << "[notify] Discord notification failed (curl exit " << rc << ")" << std::endl;
    else
        std::cout << "[notify] Sent Discord notification" << std::endl;
}
 
std::string auto_ntfy_topic() {
    char buf[256];
    if (gethostname(buf, sizeof(buf)) != 0)
        std::snprintf(buf, sizeof(buf), "unknown");
    size_t h = std::hash<std::string>{}(buf);
    char topic[16];
    std::snprintf(topic, sizeof(topic), "ads%04x", static_cast<unsigned>(h & 0xFFFF));
    return topic;
}
 
void send_completion_notification(const RunSummary& summary) {
    if (!notify_ntfy_topic.empty())
        send_ntfy(notify_ntfy_topic, summary);
    if (!notify_discord_webhook.empty())
        send_discord(notify_discord_webhook, summary);
}