/* Simple shell to run on SOS */
/* Written by Gernot Heiser   */

#include <assert.h>
#include <l4/ipc.h>
#include <l4/syscalls.h>
#include <l4/sigma0.h>
#include <string.h>

#include "sos.h"
#include "printf.h"

extern l4_threadid_t syscall_server_id;	/* testing only */

#define BUF_SIZ 128
#define N_STACK 128

#define MAX_PROCESSES 10

static process_t process[MAX_PROCESSES];
static int       n_processes = 0;

static
void ps(int print) {
  int i;

  n_processes = process_status(process, MAX_PROCESSES);
  if (!print) {
    return;
  }
  printf ("TID SIZE   STIME   CTIME COMMAND\n");
  for (i=0; i<n_processes; i++) {
    printf ("%3x %4x %7d %7d %s\n", process[i].tid, process[i].size,
	    process[i].stime, process[i].ctime, process[i].command);
  }
}

static dword_t stack[N_STACK];


static stat_t   sbuf;


static
void prstat(const char *name) {
  /* print out stat buf */
  printf("%c%c%c%c 0x%06x 0x%x 0x%06x %s\n",
	 sbuf.st_type ==ST_SPECIAL ? 's' : '-',
	 sbuf.st_fmode&FM_READ     ? 'r' : '-',
	 sbuf.st_fmode&FM_WRITE    ? 'w' : '-',
	 sbuf.st_fmode&FM_EXEC     ? 'x' : '-',
	 sbuf.st_size, sbuf.st_ctime, sbuf.st_atime, name);
}

static
void dir(const char *path) {
  int  i=0, r;
  char buf[BUF_SIZ];

  if (path) {
    r = stat(path, &sbuf);
    if (r<0) {
      printf("stat(%s) failed: %d\n", buf, r);
      return;
    }
    prstat(path);
    return;
  }

  while (1) {
    r = getdirent(i, buf, BUF_SIZ);
    if (r<0) {
      printf("dirent(%d) failed: %d\n", i, r);
      break;
    } else if (!r) {
      break;
    }
#if 0
    printf("dirent(%d): \"%s\"\n", i, buf);
#endif
    r = stat(buf, &sbuf);
    if (r<0) {
      printf("stat(%s) failed: %d\n", buf, r);
      break;
    }
    prstat(buf);
    i++;
  }
}


void main_(void) {
  tid_t            tid;
  char             buf[BUF_SIZ];
  int              r, count, i, done, found, new;
  fildes_t         in;
  char             *bp, *p;

  in = open("console", FM_READ);
  assert (in >= 0);

  bp   = buf;
  done = 0;
  new  = 1;
  while (!done) {
    if (new) {
      printf("$ ");
    }
    new   = 0;
    found = 0;
    while (!found && !done) {
      r = read(in, bp, BUF_SIZ-1+buf-bp);
      if (r<0) {
	printf("Console read failed!\n");
	done=1;
	break;
      } else if (!r) {
	continue;
      }
      bp[r] = 0;			/* terminate */
      for (p=bp; p<bp+r; p++) {
	if (*p == '\03') { 		/* ^C */
	  printf("^C\n");
	  p   = buf;
	  new = 1;
	  break;
	} else if (*p == '\04') { 	/* ^D */
	  p++;
	  found = 1;
	} else if (*p == '\010') { 	/* ^H */
	  if (p>buf) {
	    printf("\010 \010",*p);
	    p--;
	    r--;
	  }
	  p--;
	  r--;
	} else if (*p == '\n') {	/* ^J */
	  printf("%c",*p);
	  *p    = 0;
	  found = p>buf;
	  p     = buf;
	  new   = 1;
	  break;
	} else {
	  printf("%c",*p);
	}
      }
      bp = p;
      if (bp == buf) {
	break;
      }
    }
    if (!found) {
      continue;
    }
    p = buf;
    while (*p == ' ') {
      p++;
    }
    if (!*p) {
      found = 0;
      continue;
    }
    if (strcmp(p, "dir") == 0  ||  strcmp(p, "ls") == 0) {
      dir(0);
    } else if (strcmp(p, "ps") == 0) {
      ps(1);
    } else if (strcmp(p, "exit") == 0  ||  strcmp(p, "\04") == 0) {/* ^D */
      done=1;
      break;
    } else if (stat(p, &sbuf) != 0) {
      printf("Command \"%s\" not found\n", p);
    } else if (!(sbuf.st_fmode & FM_EXEC)) {
      printf("File \"%s\" not executable\n", p);
    } else {
      r = close(in);
      assert(!r);
      tid = process_create(p);
      if (tid>=0) {
	printf("Child tid=%d\n", tid);
	do {
	  ps(0);
	} while (n_processes>1);
      } else {
	printf("Failed!\n");
      }
      in = open("console", FM_READ);
      assert(in>=0);
    }
    new = 1;
  }
  printf("Exit\n", buf);
}