[people/ms/systemd.git] / systemctl.vala

/***
  This file is part of systemd.

  Copyright 2010 Lennart Poettering

  systemd is free software; you can redistribute it and/or modify it
  under the terms of the GNU General Public License as published by
  the Free Software Foundation; either version 2 of the License, or
  (at your option) any later version.

  systemd is distributed in the hope that it will be useful, but
  WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with systemd; If not, see <http://www.gnu.org/licenses/>.
***/

using DBus;
using GLib;

static string type = null;
static bool all = false;
static bool replace = false;
static bool session = false;

public static int job_info_compare(void* key1, void* key2) {
        Manager.JobInfo *j1 = (Manager.JobInfo*) key1;
        Manager.JobInfo *j2 = (Manager.JobInfo*) key2;

        return j1->id < j2->id ? -1 : (j1->id > j2->id ? 1 : 0);
}

public static int unit_info_compare(void* key1, void* key2) {
        Manager.UnitInfo *u1 = (Manager.UnitInfo*) key1;
        Manager.UnitInfo *u2 = (Manager.UnitInfo*) key2;

        int r = Posix.strcmp(Posix.strrchr(u1->id, '.'), Posix.strrchr(u2->id, '.'));
        if (r != 0)
                return r;

        return Posix.strcmp(u1->id, u2->id);
}

public void on_unit_new(string id, ObjectPath path) {
        stdout.printf("Unit %s added.\n", id);
}

public void on_job_new(uint32 id, ObjectPath path) {
        stdout.printf("Job %u added.\n", id);
}

public void on_unit_removed(string id, ObjectPath path) {
        stdout.printf("Unit %s removed.\n", id);
}

public void on_job_removed(uint32 id, ObjectPath path) {
        stdout.printf("Job %u removed.\n", id);
}

static const OptionEntry entries[] = {
        { "type",    't', 0,                   OptionArg.STRING, out type,    "List only particular type of units", "TYPE" },
        { "all",     'a', 0,                   OptionArg.NONE,   out all,     "Show all units, including dead ones", null  },
        { "replace", 0,   0,                   OptionArg.NONE,   out replace, "When installing a new job, replace existing conflicting ones", null },
        { "session", 0,   0,                   OptionArg.NONE,   out session, "Connect to session bus", null },
        { "system",  0,   OptionFlags.REVERSE, OptionArg.NONE,   out session, "Connect to system bus", null },
        { null }
};

int main (string[] args) {

        OptionContext context = new OptionContext("[OPTION...] [COMMAND [ARGUMENT...]]");
        context.add_main_entries(entries, null);
        context.set_description(
                        "Commands:\n" +
                        "  list-units          List units\n" +
                        "  list-jobs           List jobs\n" +
                        "  clear-jobs          Cancel all jobs\n" +
                        "  load [NAME...]      Load one or more units\n" +
                        "  cancel [JOB...]     Cancel one or more jobs\n" +
                        "  start [NAME...]     Start on or more units\n" +
                        "  stop [NAME...]      Stop on or more units\n" +
                        "  restart [NAME...]   Restart on or more units\n" +
                        "  reload [NAME...]    Reload on or more units\n" +
                        "  monitor             Monitor unit/job changes\n" +
                        "  dump                Dump servier status\n" +
                        "  snapshot [NAME]     Create a snapshot\n" +
                        "  daemon-reload       Reload daemon configuration\n" +
                        "  daemon-reexecute    Reexecute daemon\n");

        try {
                context.parse(ref args);
        } catch (GLib.OptionError e) {
                message("Failed to parse command line: %s".printf(e.message));
        }

        try {
                Connection bus = Bus.get(session ? BusType.SESSION : BusType.SYSTEM);

                Manager manager = bus.get_object (
                                "org.freedesktop.systemd1",
                                "/org/freedesktop/systemd1",
                                "org.freedesktop.systemd1") as Manager;

                if (args[1] == "list-units" || args.length <= 1) {
                        var list = manager.list_units();
                        uint n = 0;
                        Posix.qsort(list, list.length, sizeof(Manager.UnitInfo), unit_info_compare);

                        stdout.printf("%-45s %-6s %-12s %-12s %-17s\n", "UNIT", "LOAD", "ACTIVE", "SUB", "JOB");

                        foreach (var i in list) {

                                if (type != null && !i.id.has_suffix(".%s".printf(type)))
                                        continue;

                                if (!all && i.active_state == "inactive")
                                        continue;

                                stdout.printf("%-45s %-6s %-12s %-12s", i.id, i.load_state, i.active_state, i.sub_state);

                                if (i.job_id != 0)
                                        stdout.printf(" → %-15s", i.job_type);

                                stdout.puts("\n");
                                n++;
                        }

                        if (all)
                                stdout.printf("\n%u units listed.\n", n);
                        else
                                stdout.printf("\n%u live units listed. Pass --all to see dead units, too.\n", n);


                } else if (args[1] == "list-jobs") {
                        var list = manager.list_jobs();
                        Posix.qsort(list, list.length, sizeof(Manager.JobInfo), job_info_compare);

                        stdout.printf("%4s %-45s %-17s %-7s\n", "JOB", "UNIT", "TYPE", "STATE");

                        foreach (var i in list)
                                stdout.printf("%4u %-45s → %-15s %-7s\n", i.id, i.name, i.type, i.state);

                        stdout.printf("\n%u jobs listed.\n", list.length);

                } else if (args[1] == "clear-jobs") {

                        manager.clear_jobs();

                } else if (args[1] == "load") {

                        if (args.length < 3) {
                                stderr.printf("Missing argument.\n");
                                return 1;
                        }

                        for (int i = 2; i < args.length; i++)
                                manager.load_unit(args[i]);

                } else if (args[1] == "cancel") {

                        if (args.length < 3) {
                                stderr.printf("Missing argument.\n");
                                return 1;
                        }

                        for (int i = 2; i < args.length; i++) {
                                uint32 id;

                                if (args[i].scanf("%u", out id) != 1) {
                                        stderr.printf("Failed to parse argument.\n");
                                        return 1;
                                }

                                ObjectPath p = manager.get_job(id);

                                Job j = bus.get_object (
                                                "org.freedesktop.systemd1",
                                                p,
                                                "org.freedesktop.systemd1.Job") as Job;

                                j.cancel();
                        }

                } else if (args[1] == "start" ||
                           args[1] == "stop" ||
                           args[1] == "reload" ||
                           args[1] == "restart") {

                        if (args.length < 3) {
                                stderr.printf("Missing argument.\n");
                                return 1;
                        }

                        for (int i = 2; i < args.length; i++) {

                                ObjectPath p = manager.get_unit(args[i]);

                                Unit u = bus.get_object(
                                                "org.freedesktop.systemd1",
                                                p,
                                                "org.freedesktop.systemd1.Unit") as Unit;

                                string mode = replace ? "replace" : "fail";

                                if (args[1] == "start")
                                        u.start(mode);
                                else if (args[1] == "stop")
                                        u.stop(mode);
                                else if (args[1] == "restart")
                                        u.restart(mode);
                                else if (args[1] == "reload")
                                        u.reload(mode);
                        }

                } else if (args[1] == "monitor") {

                        manager.subscribe();

                        manager.unit_new += on_unit_new;
                        manager.unit_removed += on_unit_removed;
                        manager.job_new += on_job_new;
                        manager.job_removed += on_job_removed;

                        MainLoop l = new MainLoop();
                        l.run();

                } else if (args[1] == "dump")
                        stdout.puts(manager.dump());
                else if (args[1] == "snapshot") {

                        ObjectPath p = manager.create_snapshot(args.length > 2 ? args[2] : "");

                        Unit u = bus.get_object(
                                        "org.freedesktop.systemd1",
                                        p,
                                        "org.freedesktop.systemd1.Unit") as Unit;

                        stdout.printf("%s\n", u.id);
                } else if (args[1] == "daemon-reload")
                        manager.reload();
                else if (args[1] == "daemon-reexecute" || args[1] == "daemon-reexec")
                        manager.reexecute();
                else if (args[1] == "daemon-exit")
                        manager.exit();
                else {
                        stderr.printf("Unknown command %s.\n", args[1]);
                        return 1;
                }

        } catch (DBus.Error e) {
                stderr.printf("%s\n".printf(e.message));
        }

        return 0;
}
Commit	Line	Data
a7334b09 LP	1	/***
	2	This file is part of systemd.
	3
	4	Copyright 2010 Lennart Poettering
	5
	6	systemd is free software; you can redistribute it and/or modify it
	7	under the terms of the GNU General Public License as published by
	8	the Free Software Foundation; either version 2 of the License, or
	9	(at your option) any later version.
	10
	11	systemd is distributed in the hope that it will be useful, but
	12	WITHOUT ANY WARRANTY; without even the implied warranty of
	13	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
	14	General Public License for more details.
	15
	16	You should have received a copy of the GNU General Public License
	17	along with systemd; If not, see <http://www.gnu.org/licenses/>.
	18	***/
	19
5630af71 LP	20	using DBus;
	21	using GLib;
	22
5630af71 LP	23	static string type = null;
5630af71 LP	24	static bool all = false;
501c7d0b	25	static bool replace = false;
edddf4ff	26	static bool session = false;
5630af71 LP	27
	28	public static int job_info_compare(void* key1, void* key2) {
	29	Manager.JobInfo j1 = (Manager.JobInfo) key1;
	30	Manager.JobInfo j2 = (Manager.JobInfo) key2;
	31
501c7d0b	32	return j1->id < j2->id ? -1 : (j1->id > j2->id ? 1 : 0);
5630af71 LP	33	}
	34
	35	public static int unit_info_compare(void* key1, void* key2) {
	36	Manager.UnitInfo u1 = (Manager.UnitInfo) key1;
	37	Manager.UnitInfo u2 = (Manager.UnitInfo) key2;
	38
	39	int r = Posix.strcmp(Posix.strrchr(u1->id, '.'), Posix.strrchr(u2->id, '.'));
	40	if (r != 0)
	41	return r;
	42
	43	return Posix.strcmp(u1->id, u2->id);
	44	}
	45
c1e1601e LP	46	public void on_unit_new(string id, ObjectPath path) {
	47	stdout.printf("Unit %s added.\n", id);
	48	}
	49
	50	public void on_job_new(uint32 id, ObjectPath path) {
	51	stdout.printf("Job %u added.\n", id);
	52	}
	53
	54	public void on_unit_removed(string id, ObjectPath path) {
	55	stdout.printf("Unit %s removed.\n", id);
	56	}
	57
	58	public void on_job_removed(uint32 id, ObjectPath path) {
	59	stdout.printf("Job %u removed.\n", id);
	60	}
	61
5630af71	62	static const OptionEntry entries[] = {
edddf4ff LP	63	{ "type", 't', 0, OptionArg.STRING, out type, "List only particular type of units", "TYPE" },
	64	{ "all", 'a', 0, OptionArg.NONE, out all, "Show all units, including dead ones", null },
	65	{ "replace", 0, 0, OptionArg.NONE, out replace, "When installing a new job, replace existing conflicting ones", null },
	66	{ "session", 0, 0, OptionArg.NONE, out session, "Connect to session bus", null },
	67	{ "system", 0, OptionFlags.REVERSE, OptionArg.NONE, out session, "Connect to system bus", null },
5630af71 LP	68	{ null }
	69	};
	70
	71	int main (string[] args) {
	72
edddf4ff	73	OptionContext context = new OptionContext("[OPTION...] [COMMAND [ARGUMENT...]]");
5630af71	74	context.add_main_entries(entries, null);
c1e1601e LP	75	context.set_description(
	76	"Commands:\n" +
	77	" list-units List units\n" +
	78	" list-jobs List jobs\n" +
	79	" clear-jobs Cancel all jobs\n" +
	80	" load [NAME...] Load one or more units\n" +
	81	" cancel [JOB...] Cancel one or more jobs\n" +
	82	" start [NAME...] Start on or more units\n" +
	83	" stop [NAME...] Stop on or more units\n" +
	84	" restart [NAME...] Restart on or more units\n" +
	85	" reload [NAME...] Reload on or more units\n" +
41447faf LP	86	" monitor Monitor unit/job changes\n" +
41447faf LP	87	" dump Dump servier status\n" +
a16e1123 LP	88	" snapshot [NAME] Create a snapshot\n" +
	89	" daemon-reload Reload daemon configuration\n" +
	90	" daemon-reexecute Reexecute daemon\n");
5630af71 LP	91
	92	try {
	93	context.parse(ref args);
	94	} catch (GLib.OptionError e) {
	95	message("Failed to parse command line: %s".printf(e.message));
	96	}
	97
	98	try {
edddf4ff	99	Connection bus = Bus.get(session ? BusType.SESSION : BusType.SYSTEM);
5630af71 LP	100
	101	Manager manager = bus.get_object (
	102	"org.freedesktop.systemd1",
	103	"/org/freedesktop/systemd1",
	104	"org.freedesktop.systemd1") as Manager;
	105
	106	if (args[1] == "list-units" \|\| args.length <= 1) {
501c7d0b	107	var list = manager.list_units();
5630af71 LP	108	uint n = 0;
	109	Posix.qsort(list, list.length, sizeof(Manager.UnitInfo), unit_info_compare);
	110
10a94420	111	stdout.printf("%-45s %-6s %-12s %-12s %-17s\n", "UNIT", "LOAD", "ACTIVE", "SUB", "JOB");
5630af71 LP	112
	113	foreach (var i in list) {
	114
	115	if (type != null && !i.id.has_suffix(".%s".printf(type)))
	116	continue;
	117
	118	if (!all && i.active_state == "inactive")
	119	continue;
	120
10a94420	121	stdout.printf("%-45s %-6s %-12s %-12s", i.id, i.load_state, i.active_state, i.sub_state);
5630af71 LP	122
5630af71 LP	123	if (i.job_id != 0)
4f320e4d	124	stdout.printf(" → %-15s", i.job_type);
5630af71 LP	125
	126	stdout.puts("\n");
	127	n++;
	128	}
	129
	130	if (all)
	131	stdout.printf("\n%u units listed.\n", n);
	132	else
	133	stdout.printf("\n%u live units listed. Pass --all to see dead units, too.\n", n);
	134
	135
	136	} else if (args[1] == "list-jobs") {
501c7d0b	137	var list = manager.list_jobs();
5630af71 LP	138	Posix.qsort(list, list.length, sizeof(Manager.JobInfo), job_info_compare);
5630af71 LP	139
501c7d0b	140	stdout.printf("%4s %-45s %-17s %-7s\n", "JOB", "UNIT", "TYPE", "STATE");
dcc47b17	141
5630af71	142	foreach (var i in list)
501c7d0b	143	stdout.printf("%4u %-45s → %-15s %-7s\n", i.id, i.name, i.type, i.state);
5630af71	144
dcc47b17 LP	145	stdout.printf("\n%u jobs listed.\n", list.length);
	146
	147	} else if (args[1] == "clear-jobs") {
	148
501c7d0b	149	manager.clear_jobs();
dcc47b17	150
5630af71 LP	151	} else if (args[1] == "load") {
	152
	153	if (args.length < 3) {
	154	stderr.printf("Missing argument.\n");
	155	return 1;
	156	}
	157
17d37c28	158	for (int i = 2; i < args.length; i++)
501c7d0b LP	159	manager.load_unit(args[i]);
	160
	161	} else if (args[1] == "cancel") {
	162
	163	if (args.length < 3) {
	164	stderr.printf("Missing argument.\n");
	165	return 1;
	166	}
	167
17d37c28	168	for (int i = 2; i < args.length; i++) {
501c7d0b LP	169	uint32 id;
	170
	171	if (args[i].scanf("%u", out id) != 1) {
	172	stderr.printf("Failed to parse argument.\n");
	173	return 1;
	174	}
	175
	176	ObjectPath p = manager.get_job(id);
	177
	178	Job j = bus.get_object (
	179	"org.freedesktop.systemd1",
	180	p,
	181	"org.freedesktop.systemd1.Job") as Job;
	182
	183	j.cancel();
	184	}
	185
	186	} else if (args[1] == "start" \|\|
	187	args[1] == "stop" \|\|
	188	args[1] == "reload" \|\|
	189	args[1] == "restart") {
	190
	191	if (args.length < 3) {
	192	stderr.printf("Missing argument.\n");
	193	return 1;
	194	}
	195
ab8ea244	196	for (int i = 2; i < args.length; i++) {
501c7d0b LP	197
	198	ObjectPath p = manager.get_unit(args[i]);
	199
	200	Unit u = bus.get_object(
	201	"org.freedesktop.systemd1",
	202	p,
	203	"org.freedesktop.systemd1.Unit") as Unit;
	204
	205	string mode = replace ? "replace" : "fail";
	206
	207	if (args[1] == "start")
	208	u.start(mode);
	209	else if (args[1] == "stop")
	210	u.stop(mode);
	211	else if (args[1] == "restart")
	212	u.restart(mode);
	213	else if (args[1] == "reload")
	214	u.reload(mode);
	215	}
	216
c1e1601e LP	217	} else if (args[1] == "monitor") {
	218
	219	manager.subscribe();
	220
	221	manager.unit_new += on_unit_new;
	222	manager.unit_removed += on_unit_removed;
	223	manager.job_new += on_job_new;
	224	manager.job_removed += on_job_removed;
	225
	226	MainLoop l = new MainLoop();
	227	l.run();
	228
b152adec LP	229	} else if (args[1] == "dump")
b152adec LP	230	stdout.puts(manager.dump());
41447faf LP	231	else if (args[1] == "snapshot") {
	232
	233	ObjectPath p = manager.create_snapshot(args.length > 2 ? args[2] : "");
	234
	235	Unit u = bus.get_object(
	236	"org.freedesktop.systemd1",
	237	p,
	238	"org.freedesktop.systemd1.Unit") as Unit;
	239
	240	stdout.printf("%s\n", u.id);
a16e1123 LP	241	} else if (args[1] == "daemon-reload")
	242	manager.reload();
	243	else if (args[1] == "daemon-reexecute" \|\| args[1] == "daemon-reexec")
	244	manager.reexecute();
	245	else if (args[1] == "daemon-exit")
	246	manager.exit();
	247	else {
5630af71 LP	248	stderr.printf("Unknown command %s.\n", args[1]);
	249	return 1;
	250	}
	251
	252	} catch (DBus.Error e) {
	253	stderr.printf("%s\n".printf(e.message));
	254	}
	255
	256	return 0;
	257	}