#include "src/app/ls.h"
#include "src/fs.h"
#include "src/string.h"
#include "src/util.h"

#include <stdint.h>

static stream_t *stream_in;
static stream_t *stream_out;

#define WRITE_S(s) stream_out->write(stream_out, s, sizeof(s) - 1);
#define WRITE_D(s) stream_out->write(stream_out, s, string_length(s));

void ls(stream_t *stdin, stream_t *stdout, uint8_t argc, char **argv) {
  stream_in = stdin;
  stream_out = stdout;

  if (argc != 2) {
    WRITE_S("ls: requires a single path\n");
    return;
  }

  char *path_components[16];
  uint64_t path_length = string_split(argv[1], '/', 16, path_components);
  if (!string_empty(path_components[0])) {
    WRITE_S("ls: relative paths not supported\n");
    return;
  }

  fs_node_t *root = fs_mount();

  fs_node_t *prev = NUL;
  fs_node_t *curr = root;
  uint8_t i = 1;
  while (i < path_length) {
    if (string_empty(path_components[i])) {
      i++;
      continue;
    }

    if (!curr->is_dir) {
      WRITE_S("ls: can not navigate into a file\n");
      fs_close(curr);
      return;
    }

    prev = curr;
    curr = fs_open_by(curr, path_components[i]);
    if (prev != root) {
      fs_close(prev);
    }

    if (!curr) {
      WRITE_S("ls: invalid path\n");
      return;
    }
    i++;
  }

  if (!curr->is_dir) {
    WRITE_S("ls: can not list file\n");
    fs_close(curr);
    return;
  }

  uint8_t j = 0;
  fs_node_t *item = fs_open_at(curr, j++);
  while (item) {
    WRITE_D(item->name);
    WRITE_S("\n");
    fs_close(item);
    item = fs_open_at(curr, j++);
  }
  if (curr != root) {
    fs_close(curr);
  }

  fs_unmount(root);

  stream_in = stream_out = NUL;
}