#include <LaunchRoster.h>
#include <StringList.h>
#include <TextTable.h>
#include <errno.h>
#include <getopt.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#define COLOR_RED "\033[31;1m"
#define COLOR_GREEN "\033[32;1m"
#define COLOR_BLUE "\033[34;1m"
#define COLOR_BOLD "\033[;1m"
#define COLOR_RESET "\033[0m"
static struct option const kLongOptions[] = {
{"verbose", no_argument, 0, 'v'},
{"help", no_argument, 0, 'h'},
{NULL}
};
static struct option const kLogLongOptions[] = {
{"help", no_argument, 0, 'h'},
{"raw", no_argument, 0, 'r'},
{"user", no_argument, 0, 'u'},
{"system", no_argument, 0, 's'},
{"event", required_argument, 0, 'e'},
{"limit", required_argument, 0, 'l'},
{NULL}
};
extern const char *__progname;
static const char *kProgramName = __progname;
static void
print_log(const BMessage& log)
{
time_t now = time(NULL);
bigtime_t runtime = system_time();
for (int32 index = 0;; index++) {
BMessage item;
if (log.FindMessage("item", index, &item) != B_OK)
break;
uint64 when;
const char* message;
if (item.FindUInt64("when", &when) != B_OK
|| item.FindString("message", &message) != B_OK)
break;
time_t at = now - (runtime - when) / 1000000l;
struct tm tm;
localtime_r(&at, &tm);
char label[256];
strftime(label, sizeof(label), "%F %X", &tm);
printf("%s %s\n", label, message);
}
}
static void
log_usage(int status)
{
fprintf(stderr, "Usage: %s log [-rusel] [<job-name>]\n"
"Where the following options are allowed:\n"
" -u --user List only user log entries\n"
" -s --system List only system log entries\n"
" -e --event Filter by event name (partial names accepted)\n"
" -l --limit <n> Limit output to <n> events\n"
"<job-name>, if given, filters the jobs by name.\n",
kProgramName);
exit(status);
}
static void
get_log(int argCount, char** args)
{
bool raw = false;
bool userOnly = false;
bool systemOnly = false;
int32 limit = 0;
const char* event = NULL;
const char* job = NULL;
optind = 0;
int c;
while ((c = getopt_long(argCount, args, "hruse:l:", kLogLongOptions, NULL))
!= -1) {
switch (c) {
case 0:
break;
case 'h':
log_usage(0);
break;
case 'r':
raw = true;
break;
case 'u':
userOnly = true;
break;
case 's':
systemOnly = true;
break;
case 'e':
event = optarg;
break;
case 'l':
limit = strtol(optarg, NULL, 0);
break;
}
}
if (argCount - optind >= 1)
job = args[optind];
BLaunchRoster roster;
BMessage filter;
if (userOnly)
filter.AddBool("userOnly", true);
if (systemOnly)
filter.AddBool("systemOnly", true);
if (event != NULL)
filter.AddString("event", event);
if (job != NULL)
filter.AddString("job", job);
if (limit != 0)
filter.AddInt32("limit", limit);
BMessage info;
status_t status = roster.GetLog(filter, info);
if (status != B_OK) {
fprintf(stderr, "%s: Could not get log: %s\n", kProgramName,
strerror(status));
exit(EXIT_FAILURE);
}
if (raw) {
info.PrintToStream();
return;
}
print_log(info);
BMessage user;
if (info.FindMessage("user", &user) == B_OK) {
if (user.HasMessage("item"))
puts("User log:");
print_log(user);
}
}
static void
print_summary(TextTable& table, BMessage info, bool target)
{
status_t result = B_OK;
const char* name;
result = info.FindString("name", &name);
if (result != B_OK) {
fprintf(stderr, "%s: Could not find target or job name.\n", kProgramName);
exit(EXIT_FAILURE);
}
const int row = table.CountRows();
table.SetTextAt(row, 0, BString(COLOR_BOLD).Append(name).Append(COLOR_RESET));
if (target)
return;
bool service = false;
result = info.FindBool("service", &service);
if (result == B_OK)
table.SetTextAt(row, 1, service ? "service" : "job");
bool running = false, launched = false;
result = info.FindBool("running", &running);
if (result == B_OK) {
result = info.FindBool("launched", &launched);
table.SetTextAt(row, 2,
BString(running ? COLOR_GREEN : launched ? COLOR_BLUE : COLOR_RED)
.Append(running ? "running" : launched ? "idle" : "stopped")
.Append(COLOR_RESET));
}
bool enabled = false;
result = info.FindBool("enabled", &enabled);
if (result == B_OK)
table.SetTextAt(row, 3, enabled ? "yes" : "no");
return;
}
static status_t
get_info(TextTable& table, const char* name, bool verbose)
{
BLaunchRoster roster;
BMessage info;
status_t status = roster.GetTargetInfo(name, info);
if (status == B_OK) {
print_summary(table, info, true);
if (table.CountColumns() < 1)
table.AddColumn("Name", B_ALIGN_RIGHT);
} else {
info.MakeEmpty();
status = roster.GetJobInfo(name, info);
if (status == B_OK)
print_summary(table, info, false);
if (table.CountColumns() < 4) {
if (table.CountColumns() < 1)
table.AddColumn("Name", B_ALIGN_RIGHT);
table.AddColumn("Type");
table.AddColumn("State");
table.AddColumn("Enabled");
}
}
if (status != B_OK) {
fprintf(stderr, "%s: Could not get target or job info for \"%s\": "
"%s\n", kProgramName, name, strerror(status));
return status;
}
if (verbose) {
table.Print(INT32_MAX);
printf("\nDetails: ");
info.PrintToStream();
}
return B_OK;
}
static void
list_targets(bool verbose)
{
BLaunchRoster roster;
BStringList targets;
status_t status = roster.GetTargets(targets);
if (status != B_OK) {
fprintf(stderr, "%s: Could not get target listing: %s\n", kProgramName,
strerror(status));
exit(EXIT_FAILURE);
}
TextTable table;
for (int32 i = 0; i < targets.CountStrings(); i++)
get_info(table, targets.StringAt(i).String(), verbose);
table.Print(INT32_MAX);
}
static void
list_jobs(bool verbose)
{
BLaunchRoster roster;
BStringList jobs;
status_t status = roster.GetJobs(NULL, jobs);
if (status != B_OK) {
fprintf(stderr, "%s: Could not get job listing: %s\n", kProgramName,
strerror(status));
exit(EXIT_FAILURE);
}
TextTable table;
for (int32 i = 0; i < jobs.CountStrings(); i++)
get_info(table, jobs.StringAt(i).String(), verbose);
table.Print(INT32_MAX);
}
static void
start_job(const char* name)
{
BLaunchRoster roster;
status_t status = roster.Start(name);
if (status == B_NAME_NOT_FOUND)
status = roster.Target(name);
if (status != B_OK) {
fprintf(stderr, "%s: Starting job \"%s\" failed: %s\n", kProgramName,
name, strerror(status));
exit(EXIT_FAILURE);
}
}
static void
stop_job(const char* name)
{
BLaunchRoster roster;
status_t status = roster.Stop(name);
if (status == B_NAME_NOT_FOUND)
status = roster.StopTarget(name);
if (status != B_OK) {
fprintf(stderr, "%s: Stopping job \"%s\" failed: %s\n", kProgramName,
name, strerror(status));
exit(EXIT_FAILURE);
}
}
static void
restart_job(const char* name)
{
stop_job(name);
start_job(name);
}
static void
enable_job(const char* name, bool enable)
{
BLaunchRoster roster;
status_t status = roster.SetEnabled(name, enable);
if (status != B_OK) {
fprintf(stderr, "%s: %s job \"%s\" failed: %s\n", kProgramName,
enable ? "Enabling" : "Disabling", name, strerror(status));
exit(EXIT_FAILURE);
}
}
static void
usage(int status)
{
fprintf(stderr, "Usage: %s <command>\n"
"Where <command> is one of:\n"
" list - Lists all jobs (the default command)\n"
" list-targets - Lists all targets\n"
" log - Displays the event log\n"
"The following <command>s have a <name> argument:\n"
" start - Starts a job/target\n"
" stop - Stops a running job/target\n"
" restart - Restarts a running job/target\n"
" info - Shows info for a job/target\n",
kProgramName);
exit(status);
}
int
main(int argc, char** argv)
{
const char* command = "list";
bool verbose = false;
int c;
while ((c = getopt_long(argc, argv, "+hv", kLongOptions, NULL)) != -1) {
switch (c) {
case 0:
break;
case 'h':
usage(0);
break;
case 'v':
verbose = true;
break;
default:
usage(1);
break;
}
}
if (argc - optind >= 1)
command = argv[optind];
if (strcmp(command, "list") == 0) {
list_jobs(verbose);
} else if (strcmp(command, "list-targets") == 0) {
list_targets(verbose);
} else if (strcmp(command, "log") == 0) {
get_log(argc - optind, &argv[optind]);
} else if (argc == optind + 1) {
TextTable table;
get_info(table, command, true);
} else {
const char* name = argv[argc - 1];
if (strcmp(command, "info") == 0) {
TextTable table;
get_info(table, name, true);
} else if (strcmp(command, "start") == 0) {
start_job(name);
} else if (strcmp(command, "stop") == 0) {
stop_job(name);
} else if (strcmp(command, "restart") == 0) {
restart_job(name);
} else if (strcmp(command, "enable") == 0) {
enable_job(name, true);
} else if (strcmp(command, "disable") == 0) {
enable_job(name, false);
} else {
fprintf(stderr, "%s: Unknown command \"%s\".\n", kProgramName,
command);
}
}
return 0;
}
