/* SPDX-License-Identifier: LGPL-2.1-or-later */

#include "bus-error.h"
#include "bus-locator.h"
#include "systemctl-daemon-reload.h"
#include "systemctl-util.h"
#include "systemctl.h"

int daemon_reload(enum action action, bool graceful) {
        _cleanup_(sd_bus_error_free) sd_bus_error error = SD_BUS_ERROR_NULL;
        _cleanup_(sd_bus_message_unrefp) sd_bus_message *m = NULL;
        const char *method;
        sd_bus *bus;
        int r;

        r = acquire_bus(BUS_MANAGER, &bus);
        if (r < 0)
                return r;

        polkit_agent_open_maybe();

        switch (action) {

        case ACTION_RELOAD:
                method = "Reload";
                break;

        case ACTION_REEXEC:
                method = "Reexecute";
                break;

        default:
                return -EINVAL;
        }

        r = bus_message_new_method_call(bus, &m, bus_systemd_mgr, method);
        if (r < 0)
                return bus_log_create_error(r);

        /* Note we use an extra-long timeout here. This is because a reload or reexec means generators are
         * rerun which are timed out after DEFAULT_TIMEOUT_USEC. Let's use twice that time here, so that the
         * generators can have their timeout, and for everything else there's the same time budget in
         * place. */

        r = sd_bus_call(bus, m, DEFAULT_TIMEOUT_USEC * 2, &error, NULL);

        /* On reexecution, we expect a disconnect, not a reply */
        if (IN_SET(r, -ETIMEDOUT, -ECONNRESET) && action == ACTION_REEXEC)
                return 1;
        if (r < 0) {
                if (graceful) { /* If graceful mode is selected, debug log, but don't fail */
                        log_debug_errno(r, "Failed to reload daemon via the bus, ignoring: %s", bus_error_message(&error, r));
                        return 0;
                }

                return log_error_errno(r, "Failed to reload daemon: %s", bus_error_message(&error, r));
        }

        return 1;
}

int verb_daemon_reload(int argc, char *argv[], void *userdata) {
        enum action a;
        int r;

        assert(argc >= 1);

        if (streq(argv[0], "daemon-reexec"))
                a = ACTION_REEXEC;
        else if (streq(argv[0], "daemon-reload"))
                a = ACTION_RELOAD;
        else
                assert_not_reached();

        r = daemon_reload(a, /* graceful= */ false);
        if (r < 0)
                return r;

        return 0;
}