#include <sys/auxv.h>
#include <sys/sysinfo.h>

#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>

#include <proc/readproc.h>

static void
print_human(intmax_t i)
{
        double d = i;
        const char *u = "\0\0K\0M\0G\0T\0P\0E\0Z\0Y\0";
        while (d >= 1024) {
                u += 2;
                d /= 1024.0;
        }

        if (!*u)
                printf("%5.0f", d);
        else if (d < 10.0)
                printf("%4.1f%s", d, u);
        else
                printf("%4.0f%s", d, u);
}

int
main(int argc, char *argv[])
{
	proc_t **result;
	int clktck;

	if ((clktck = getauxval(AT_CLKTCK)) <= 0)
		if ((clktck = sysconf(_SC_CLK_TCK)) <= 0)
			clktck = 100; 	/* educated guess */

	struct sysinfo si;
	sysinfo(&si);

	time_t now = time(0);

	result = readproctab(PROC_FILLMEM | PROC_FILLCOM | PROC_FILLUSR |
	    PROC_FILLSTAT | PROC_FILLSTATUS);

	int matched = 0;

	for (; *result; result++) {
		proc_t *p = *result;

		if (argc > 1) {
			for (int i = 1; i < argc; i++)
				if (strstr(p->cmd, argv[i]))
					goto match;
			continue;
match:
			matched++;
		} else {
			matched++;
		}

		if (matched == 1)
			printf("  PID USER     %%CPU  VSZ  RSS START ELAPSED   TIME COMMAND\n");

		printf("%5d", p->tid);
		printf(" %-8.8s", p->euser);

		double pcpu = 0;
		time_t seconds;

		seconds = si.uptime - p->start_time / clktck;
		if (seconds)
			pcpu = ((p->utime + p->stime) * 100.0 / clktck) / seconds;

		printf(" %4.1f", pcpu);
		print_human(p->vm_size*1024);
		print_human(p->vm_rss*1024);

		time_t start;
		time_t seconds_ago;
		start = (now - si.uptime) + p->start_time / clktck;
		seconds_ago = now - start;
		if (seconds_ago < 0)
			seconds_ago=0;

		char buf[7];
		if (seconds_ago > 3600*24)
			strftime(buf, sizeof buf, "%b%d", localtime(&start));
		else
			strftime(buf, sizeof buf, "%H:%M", localtime(&start));
		printf(" %s", buf);

		time_t t = seconds_ago;
		unsigned dd,hh,mm,ss;
		ss = t%60;
		t /= 60;
		mm = t%60;
		t /= 60;
		hh = t%24;
		t /= 24;
		dd = t;

		if (dd)
			printf(" %3ud%02uh", dd, hh);
		else if (hh)
			printf("  %2uh%02um", hh, mm);
		else
			printf("   %2u:%02u", mm, ss);


		t = (p->utime + p->stime) / clktck;
		ss = t%60;
		t /= 60;
		mm = t%60;
		t /= 60;
		hh = t%24;

		if (hh)
			printf(" %3uh%02u", hh, mm);
		else
			printf("  %2u:%02u", mm, ss);

		if (p->cmdline) {
			for (int i = 0; p->cmdline[i]; i++) {
				printf(" %s", p->cmdline[i]);
			}
		} else {
			// kernel threads
			printf(" [%s]", p->cmd);
		}

		if (p->state == 'Z')
			printf(" <defunct>");
		printf("\n");
	}

	exit(!matched);
}