daemon.h

//  Copyright (c) 2010-2013 David Caldwell <david@porkrind.org> -*- c++ -*-
//  Licenced under the GPL 3.0 or any later version. See LICENSE file for details.
#ifndef __DAEMON_H__
#define __DAEMON_H__

#include "user.h"
#include "passwd.h"
#include <string>
#include <list>
#include <time.h>

enum run_state { stopped, stopping, running, coolingdown };
const std::string _state_str[] = { "stopped", "stopping", "running", "coolingdown" };

const map<string,string> the_empty_map;

class daemon {
  public:
    std::string id;
    std::string name;
    std::string config_file;
    time_t config_file_stamp;
    //int socket;
    class user *user;

    // From config file:
    struct config {
        std::string working_dir;
        pwent run_as;
        std::string start_command;
        bool autostart;
        bool log_output;
        std::map<std::string,std::string> environment;
    } config;

    // state:
    struct current {
        int pid;
        run_state state;
        time_t cooldown;
        time_t cooldown_start;
        size_t respawns;
        time_t start_time;
        time_t respawn_time;
    } current;

    // Something important to warn the user about.
    std::list<string> whine_list;
    string get_and_clear_whines();

    daemon(std::string config_file, class user *user);

    void load_config();
    bool exists();
    std::string log_file();
    std::string state_str() { return _state_str[current.state]; }
    int fork_setuid_exec(string command, map<string,string> env = the_empty_map);

    void start(bool respawn=false);
    void stop();
    void respawn();
    void reap();

    time_t cooldown_remaining();

    std::map<std::string,std::string> to_map();
    void from_map(map<string,string> data);
};

bool daemon_compare(class daemon *a, class daemon *b);

// Use as a unary predicate for find_if and similar.
class daemon_id_match {
    public: string &id;
            daemon_id_match(string &id) : id(id) {}
            bool operator()(class daemon *d) { return d->id == id; }
};

#endif /* __DAEMON_H__ */