/* $Revision: 1.2 $ */

#include <common.h>
#include <kern.h>
#include <sem.h>
#include <bsp.h>
#include <string.h>


#define I_AM_UGLY

#ifdef I_AM_UGLY
#include <varargs.h>
extern void   do_printf(void (*do_putc) (char), va_list );
#else
#include <stdio.h>
#endif


extern volatile char *tty;

void stdio_init(void);
int task_stdio_init(void);
void record_char(char ch);
void putchar(char ch);
int getchar(void);
char *gets(void);

char cinput[BUFSIZ];

int cinput_i = 0;
int coutput_i = 0;

sem_t stdio_sem;

void stdio_init(void){
  sem_init(&stdio_sem);
}

extern tcb_t *cur_task;
extern tcb_t *fake_task;

int task_stdio_init(void) {
  static char buffer[BUFSIZ];

  if (!cur_task || cur_task == fake_task)
    return -1;
  cur_task->tcb_stdio_buf = buffer;	/* XXX malloc it! */
  return 1;
}

/* called by BSP-specific tty_rx_handle() */
void record_char(char ch){
  cinput[cinput_i] = ch;
  cinput_i = (cinput_i + 1) % BUFSIZ;
  sem_v(&stdio_sem);
}

extern int paniced;
extern char *panic_str;

void putchar(char ch) {
  if (paniced) {
    kputchar(ch);
    *panic_str++ = ch;
    
    return;
  }
  if (ch == '\n')
    tty_write(tty, '\r');
  if (ch == '\r')
    tty_write(tty, '\n');
  tty_write(tty, ch);
}

void puts(char * str){
  char * cnt;
  while( *cnt != 0 ){
    putchar( *cnt );
    cnt ++ ;
  }
}

int getchar(void){
  int ch;

  for (;;) {
    while (cinput_i == coutput_i)
      sem_p(&stdio_sem);
    if ((ch = kgetchar()) < 0)
      continue;
    return ch;
  }
}

char * gets(void)
{
#warning fixme i am possibly broken
  char *str;
  char ch;
  char *org = str;

  if (task_stdio_init() < 0)
    return 0;
  str = cur_task->tcb_stdio_buf;
  do {
    if ((ch = getchar()) < 0)
      continue;
    *str = ch;
    str++;
  }
  while (!isnewline(ch));
  *str = '\0';
  return org;
}

#ifdef I_AM_UGLY

void printf( va_alist )
     va_dcl
{
  va_list pvar;

  va_start(pvar);
  do_printf(putchar, pvar);
  va_end(pvar);
}

extern lcd_write( UBYTE ch ) ;

void lprintf( va_alist )
     va_dcl
{
  va_list pvar;

  va_start(pvar);
  do_printf(lcd_write, pvar);
  va_end(pvar);
}


#else

void printf(char * fmt, ...)
{
  va_list pvar;

  va_start(pvar,fmt);
  do_printf(putchar, fmt , pvar);
  va_end(pvar);
}

#endif /* I_AM_UGLY */