/*-
 * Copyright (c) 2002-2018 The UbixOS Project.
 * All rights reserved.
 *
 * This was developed by Christopher W. Olsen for the UbixOS Project.
 *
 * Redistribution and use in source and binary forms, with or without modification, are permitted
 * provided that the following conditions are met:
 *
 * 1) Redistributions of source code must retain the above copyright notice, this list of
 *    conditions, the following disclaimer and the list of authors.
 * 2) Redistributions in binary form must reproduce the above copyright notice, this list of
 *    conditions, the following disclaimer and the list of authors in the documentation and/or
 *    other materials provided with the distribution.
 * 3) Neither the name of the UbixOS Project nor the names of its contributors may be used to
 *    endorse or promote products derived from this software without specific prior written
 *    permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED
 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS
 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA,
 * OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <ubixos/sched.h>
#include <sys/thread.h>
#include <sys/sysproto_posix.h>
#include <sys/descrip.h>
#include <sys/video.h>
#include <sys/pipe.h>
#include <sys/errno.h>
#include <string.h>
#include <ufs/ufs.h>

int sys_open(struct thread *td, struct sys_open_args *args) {
  kprintf("sys_open?");
  return (kern_openat(td, AT_FDCWD, args->path, args->flags, args->mode));

}

int sys_openat(struct thread *td, struct sys_openat_args *args) {

  int error = 0x0;
  int fd = 0x0;
  struct file *nfp = 0x0;

  error = falloc(td, &nfp, &fd);

  if (error)
    return (error);

  if ((args->flag & O_WRONLY) == O_WRONLY)
    nfp->fd = fopen(args->path, "w");
  else if ((args->flag & O_RDWR) == O_RDWR)
    nfp->fd = fopen(args->path, "a");
  else
    nfp->fd = fopen(args->path, "r");

  if (nfp->fd == 0x0) {
    if (fdestroy(td, nfp, fd) != 0x0)
      kprintf("[%s:%i] fdestroy() failed.", __FILE__, __LINE__);

    td->td_retval[0] = -1;
    error = -1;
    /*

     kprintf("[sOA: 0x%X:%s:%s:]", args->flag, args->mode, args->path, td->td_retval[0]);

     if ((args->flag & O_RDONLY) == O_RDONLY)
     kprintf("O_RDONLY");

     if ((args->flag & O_WRONLY) == O_WRONLY)
     kprintf("O_WRONLY");

     if ((args->flag & O_RDWR) == O_RDWR)
     kprintf("O_RDWR");

     if ((args->flag & O_ACCMODE) == O_ACCMODE)
     kprintf("O_ACCMODE");
     */

  }
  else {
        kprintf("[%s:%i] o(%s)%i", __FILE__, __LINE__, args->path, fd);
    td->td_retval[0] = fd;
  }

  return (error);
}

int sys_close(struct thread *td, struct sys_close_args *args) {
  struct file *fd = 0x0;
  struct pipeInfo *pFD = 0x0;

  getfd(td, &fd, args->fd);

  //kprintf("[sC:%i:0x%X:0x%X]", args->fd, fd, fd->fd);

#ifdef DEBUG_VFS_CALLS
  kprintf("[sC::0x%X:0x%X]", args->fd, fd, fd->fd);
#endif

  if (fd == 0x0) {
    kprintf("COULDN'T FIND FD: ", args->fd);
    td->td_retval[0] = -1;
  }
  else {
    switch (fd->fd_type) {
      case 3:
        pFD = fd->data;
        if (args->fd == pFD->rFD) {
          if (pFD->rfdCNT < 2)
            if (fdestroy(td, fd, args->fd) != 0x0)
              kprintf("[%s:%i] fdestroy() failed.", __FILE__, __LINE__);
          pFD->rfdCNT--;
        }

        if (args->fd == pFD->wFD) {
          if (pFD->wfdCNT < 2)
            if (fdestroy(td, fd, args->fd) != 0x0)
              kprintf("[%s:%i] fdestroy() failed.", __FILE__, __LINE__);
          pFD->wfdCNT--;
        }

        break;
      default:
        if (args->fd < 3)
          td->td_retval[0] = 0;
        else {
          if (!fclose(fd->fd))
            td->td_retval[0] = -1;

          //kprintf("DESTROY: %i!", args->fd);
          if (fdestroy(td, fd, args->fd) != 0x0)
            kprintf("[%s:%i] fdestroy(0x%X, 0x%X) failed\n", __FILE__, __LINE__, fd, td->o_files[args->fd]);

          td->td_retval[0] = 0;

        }
    }
  }
  return (0);
}

int sys_read(struct thread *td, struct sys_read_args *args) {
  int x = 0;
  char c = 0x0;
  char bf[2];
  volatile char *buf = args->buf;

  struct file *fd = 0x0;

  struct pipeInfo *pFD = 0x0;
  struct pipeInfo *rpFD = 0x0;

  size_t nbytes;

  int rpCNT = 0;

  getfd(td, &fd, args->fd);

  if (args->fd > 3) {
    switch (fd->fd_type) {
      case 3: /* XXX - Pipe2 Handling */
        pFD = fd->data;
        while (pFD->bCNT == 0 && rpCNT < 100) {
          sched_yield();
          rpCNT++;
        }

        if (rpCNT >= 100 && pFD->bCNT == 0) {
          td->td_retval[0] = 0;
        }
        else {
          nbytes = (args->nbyte - (pFD->headPB->nbytes - pFD->headPB->offset) <= 0) ? args->nbyte : (pFD->headPB->nbytes - pFD->headPB->offset);
          //kprintf("[unb: , nbs: %i, bf: 0x%X]", args->nbyte, nbytes, fd->fd->buffer);
          //kprintf("PR: []", nbytes);
          memcpy(args->buf, pFD->headPB->buffer + pFD->headPB->offset, nbytes);
          pFD->headPB->offset += nbytes;

          if (pFD->headPB->offset >= pFD->headPB->nbytes) {
            rpFD = pFD->headPB;
            pFD->headPB = pFD->headPB->next;
            kfree(rpFD);
            pFD->bCNT--;
          }

          td->td_retval[0] = nbytes;
        }
        break;
      default:
        //kprintf("[r:0x%X::%i:%s]",fd->fd, args->fd, fd->fd_type, fd->fd->fileName);
        //kprintf("[%s:%i]", __FILE__, __LINE__);
        td->td_retval[0] = fread(args->buf, args->nbyte, 1, fd->fd);
    }
  }
  else {
    bf[1] = '\0';
    if (_current->term == tty_foreground)
      c = getchar();

    for (x = 0; x < args->nbyte && c != '\n';) {
      if (_current->term == tty_foreground) {

        if (c != 0x0) {
          buf[x++] = c;
          bf[0] = c;
          kprintf(bf);
        }

        if (c == '\n') {
          buf[x++] = c;
          break;
        }

        sched_yield();
        c = getchar();
      }
      else {
        sched_yield();
      }
    }
    if (c == '\n')
      buf[x++] = '\n';

    bf[0] = '\n';
    kprintf(bf);

    td->td_retval[0] = x;
  }
  return (0);
}

int sys_pread(struct thread *td, struct sys_pread_args *args) {
  int offset = 0;
  int x = 0;
  char c = 0x0;
  char bf[2];
  volatile char *buf = args->buf;

  struct file *fd = 0x0;

  getfd(td, &fd, args->fd);

  if (args->fd > 3) {
    offset = fd->fd->offset;
    fd->fd->offset = args->offset;
    td->td_retval[0] = fread(args->buf, args->nbyte, 1, fd->fd);
    fd->fd->offset = offset;
  }
  else {
    bf[1] = '\0';
    if (_current->term == tty_foreground)
      c = getchar();

    for (x = 0; x < args->nbyte && c != '\n';) {
      if (_current->term == tty_foreground) {

        if (c != 0x0) {
          buf[x++] = c;
          bf[0] = c;
          kprintf(bf);
        }

        if (c == '\n') {
          buf[x++] = c;
          break;
        }

        sched_yield();
        c = getchar();
      }
      else {
        sched_yield();
      }
    }
    if (c == '\n')
      buf[x++] = '\n';

    bf[0] = '\n';
    kprintf(bf);

    td->td_retval[0] = x;
  }
  return (0);
}

int sys_write(struct thread *td, struct sys_write_args *uap) {
  char *buffer = 0x0;
  struct file *fd = 0x0;

  struct pipeInfo *pFD = 0x0;
  struct pipeBuf *pBuf = 0x0;

  size_t nbytes;

    kprintf("<size_t: %i:%i>", sizeof(size_t), uap->nbyte);

  if (uap->fd == 2) {
    buffer = kmalloc(1024);
        kprintf("nbyte: %i", uap->nbyte);
    memcpy(buffer, uap->buf, uap->nbyte);
    printColor += 1;
    kprintf(buffer);
    printColor = defaultColor;
    kfree(buffer);
    td->td_retval[0] = uap->nbyte;
  }
  else if (uap->fd == 1 && ((struct file*) td->o_files[1])->fd == 0x0) {
        // buffer = kmalloc(1024);
        //kprintf("nbyte: %i", uap->nbyte);
        //memcpy(buffer, uap->buf, uap->nbyte);
        //kprintf(buffer);
        // kprintf(uap->buf);
        //kfree(buffer);
        //kprintf("[%s:%i]", __FILE__, __LINE__);
        kprint(uap->buf);
    td->td_retval[0] = uap->nbyte;
  }
  else {
        return (0x0);

    getfd(td, &fd, uap->fd);

    //kprintf("[fd: %i:0x%X, fd_type: %i]", uap->fd, fd, fd->fd_type);

    switch (fd->fd_type) {
      case 3: /* XXX - Temp Pipe Stuff */

        pFD = fd->data;
        pBuf = (struct pipeBuf*) kmalloc(sizeof(struct pipeBuf));
        pBuf->buffer = kmalloc(uap->nbyte);

        memcpy(pBuf->buffer, uap->buf, uap->nbyte);

        pBuf->nbytes = uap->nbyte;

        if (pFD->tailPB)
          pFD->tailPB->next = pBuf;

        pFD->tailPB = pBuf;

        if (!pFD->headPB)
          pFD->headPB = pBuf;

        pFD->bCNT++;

        td->td_retval[0] = nbytes;

        break;
      default:
        if (fd->fd) {
          kprintf("[0x%X]", fd->fd->res);
                    td->td_retval[0] = fwrite(uap->buf, uap->nbyte, 1, fd->fd);
        }
        else {
          kprintf("[%i]", uap->nbyte);
          buffer = kmalloc(uap->nbyte);
          memcpy(buffer, uap->buf, uap->nbyte);
          kprintf("(%i) %s", uap->fd, uap->buf);
          kfree(buffer);
          td->td_retval[0] = uap->nbyte;
        }
    }

  }
  return (0x0);
}

int sys_access(struct thread *td, struct sys_access_args *args) {
  /* XXX - This is a temporary as it always returns true */

  td->td_retval[0] = 0;
  return (0);
}

int sys_getdirentries(struct thread *td, struct sys_getdirentries_args *args) {

  struct file *fd = 0x0;

  getfd(td, &fd, args->fd);

  char buf[DEV_BSIZE];
  struct dirent *d;
  char *s;
  ssize_t n;

  td->td_retval[0] = fread(args->buf, args->count, 1, fd->fd);

  return (0);
}

int sys_readlink(struct thread *thr, struct sys_readlink_args *args) {
  /* XXX - Need to implement readlink */

  kprintf("RL: %s:\n", args->path, args->count);

  //Return Error
  thr->td_retval[0] = 2;
  return (-1);
}

int kern_openat(struct thread *thr, int afd, char *path, int flags, int mode) {
  int error = 0x0;
  int fd = 0x0;
  struct file *nfp = 0x0;

  /*
   * Only one of the O_EXEC, O_RDONLY, O_WRONLY and O_RDWR flags
   * may be specified.
   */
  if (flags & O_EXEC) {
    if (flags & O_ACCMODE)
      return (EINVAL);
  }
  else if ((flags & O_ACCMODE) == O_ACCMODE) {
    return (EINVAL);
  }
  else {
    flags = FFLAGS(flags);
  }

  error = falloc(thr, &nfp, &fd);

  if (error) {
    thr->td_retval[0] = -1;
    return (error);
  }

  if (flags | O_CREAT)
    kprintf("O_CREAT\n");

  nfp->f_flag = flags & FMASK;

  nfp->fd = fopen(path, "rwb");

    kprintf("[%s:%i] o(%s)%i", __FILE__, __LINE__, path, fd);

  if (nfp->fd == 0x0) {
    if (fdestroy(thr, nfp, fd) != 0x0)
      kprintf("[%s:%i] fdestroy() failed.", __FILE__, __LINE__);

    thr->td_retval[0] = -1;

    error = -1;
  }
  else {
    thr->td_retval[0] = fd;
  }

  //kprintf("sO: 0x%X:%s:", args->mode, args->path, td->td_retval[0]);

  return (error);

}

int sys_unlink(struct thread *td, struct sys_unlink_args *uap) {
  int error = 0x0;
  td->td_retval[0] = fl_remove(uap->path);
  if (td->td_retval[0] != 0x0)
    kprintf("[%s:%i]Path: %s", __FILE__, __LINE__, uap->path);
  return (error);
}