#include <errno.h>

#include "dbus.h"

#include "log.h"

#include "dbus-job.h"

static const char introspection[] =

DBUS_INTROSPECT_1_0_XML_DOCTYPE_DECL_NODE

"<node>"

" <interface name=\"org.freedesktop.systemd1.Job\">"

" <method name=\"Cancel\"/>"

" <signal name=\"Changed\"/>"

" <property name=\"Id\" type=\"u\" access=\"read\"/>"

" <property name=\"Unit\" type=\"(so)\" access=\"read\"/>"

" <property name=\"JobType\" type=\"s\" access=\"read\"/>"

" <property name=\"State\" type=\"s\" access=\"read\"/>"

" </interface>"

BUS_PROPERTIES_INTERFACE

BUS_INTROSPECTABLE_INTERFACE

"</node>";

static DEFINE_BUS_PROPERTY_APPEND_ENUM(bus_job_append_state, job_state, JobState);

static DEFINE_BUS_PROPERTY_APPEND_ENUM(bus_job_append_type, job_type, JobType);

static int bus_job_append_unit(Manager *m, DBusMessageIter *i, const char *property, void *data) {

Job *j = data;

DBusMessageIter sub;

char *p;

assert(m);

assert(i);

assert(property);

assert(j);

if (!dbus_message_iter_open_container(i, DBUS_TYPE_STRUCT, NULL, &sub))

return -ENOMEM;

if (!(p = unit_dbus_path(j->unit)))

return -ENOMEM;

if (!dbus_message_iter_append_basic(&sub, DBUS_TYPE_STRING, &j->unit->meta.id) ||

!dbus_message_iter_append_basic(&sub, DBUS_TYPE_OBJECT_PATH, &p)) {

free(p);

return -ENOMEM;

}

free(p);

if (!dbus_message_iter_close_container(i, &sub))

return -ENOMEM;

return 0;

}

static DBusHandlerResult bus_job_message_dispatch(Job *j, DBusMessage *message) {

const BusProperty properties[] = {

{ "org.freedesktop.systemd1.Job", "Id", bus_property_append_uint32, "u", &j->id },

{ "org.freedesktop.systemd1.Job", "State", bus_job_append_state, "s", &j->state },

{ "org.freedesktop.systemd1.Job", "JobType", bus_job_append_type, "s", &j->type },

{ "org.freedesktop.systemd1.Job", "Unit", bus_job_append_unit, "(so)", j },

{ NULL, NULL, NULL, NULL, NULL }

};

DBusMessage *reply = NULL;

Manager *m = j->manager;

if (dbus_message_is_method_call(message, "org.freedesktop.systemd1.Job", "Cancel")) {

if (!(reply = dbus_message_new_method_return(message)))

goto oom;

job_free(j);

} else

return bus_default_message_handler(j->manager, message, introspection, properties);

if (reply) {

if (!dbus_connection_send(m->api_bus, reply, NULL))

goto oom;

dbus_message_unref(reply);

}

return DBUS_HANDLER_RESULT_HANDLED;

oom:

if (reply)

dbus_message_unref(reply);

return DBUS_HANDLER_RESULT_NEED_MEMORY;

}

static DBusHandlerResult bus_job_message_handler(DBusConnection *connection, DBusMessage *message, void *data) {

Manager *m = data;

Job *j;

int r;

assert(connection);

assert(message);

assert(m);

log_debug("Got D-Bus request: %s.%s() on %s",

dbus_message_get_interface(message),

dbus_message_get_member(message),

dbus_message_get_path(message));

if ((r = manager_get_job_from_dbus_path(m, dbus_message_get_path(message), &j)) < 0) {

if (r == -ENOMEM)

return DBUS_HANDLER_RESULT_NEED_MEMORY;

if (r == -ENOENT)

return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;

return bus_send_error_reply(m, message, NULL, r);

}

return bus_job_message_dispatch(j, message);

}

const DBusObjectPathVTable bus_job_vtable = {

.message_function = bus_job_message_handler

};

void bus_job_send_change_signal(Job *j) {

char *p = NULL;

DBusMessage *m = NULL;

assert(j);

assert(j->in_dbus_queue);

LIST_REMOVE(Job, dbus_queue, j->manager->dbus_job_queue, j);

j->in_dbus_queue = false;

if (set_isempty(j->manager->subscribed)) {

j->sent_dbus_new_signal = true;

return;

}

if (!(p = job_dbus_path(j)))

goto oom;

if (j->sent_dbus_new_signal) {

/* Send a change signal */

if (!(m = dbus_message_new_signal(p, "org.freedesktop.systemd1.Job", "Changed")))

goto oom;

} else {

/* Send a new signal */

if (!(m = dbus_message_new_signal("/org/freedesktop/systemd1", "org.freedesktop.systemd1.Manager", "JobNew")))

goto oom;

if (!dbus_message_append_args(m,

DBUS_TYPE_UINT32, &j->id,

DBUS_TYPE_OBJECT_PATH, &p,

DBUS_TYPE_INVALID))

goto oom;

}

if (!dbus_connection_send(j->manager->api_bus, m, NULL))

goto oom;

free(p);

dbus_message_unref(m);

j->sent_dbus_new_signal = true;

return;

oom:

free(p);

if (m)

dbus_message_unref(m);

log_error("Failed to allocate job change signal.");

}

void bus_job_send_removed_signal(Job *j) {

char *p = NULL;

DBusMessage *m = NULL;

assert(j);

if (set_isempty(j->manager->subscribed) || !j->sent_dbus_new_signal)

return;

if (!(p = job_dbus_path(j)))

goto oom;

if (!(m = dbus_message_new_signal("/org/freedesktop/systemd1", "org.freedesktop.systemd1.Manager", "JobRemoved")))

goto oom;

if (!dbus_message_append_args(m,

DBUS_TYPE_UINT32, &j->id,

DBUS_TYPE_OBJECT_PATH, &p,

DBUS_TYPE_INVALID))

goto oom;

if (!dbus_connection_send(j->manager->api_bus, m, NULL))

goto oom;

free(p);

dbus_message_unref(m);

return;

oom:

free(p);

if (m)

dbus_message_unref(m);

log_error("Failed to allocate job remove signal.");

}