#include <am.h>
#include <klib-macros.h>
#include <klib.h>
#include <stdarg.h>

#if !defined(__ISA_NATIVE__) || defined(__NATIVE_USE_KLIB__)

void print_int(int num, int width, char pad) {
  int reverse = 0;
  int count = 0;

  if (num == 0) {
    reverse = 0;
    count = 1;
  } else {
    if (num < 0) {
      putch('-');
      num = -num;
    }
    while (num != 0) {
      reverse = reverse * 10 + (num % 10);
      num /= 10;
      count++;
    }
  }

  while (width > count) {
    putch(pad);
    width--;
  }

  if (reverse == 0) {
    putch('0');
  } else {
    while (reverse != 0) {
      putch('0' + (reverse % 10));
      reverse /= 10;
    }
  }
}

int vprintf(const char *format, va_list args) {
  const char *p = format;

  while (*p) {
    if (*p == '%') {
      p++; // Skip the '%'
      char pad = ' ';
      int width = 0;

      if (*p == '0') {
        pad = '0';
        p++;
      }

      while (*p >= '0' && *p <= '9') {
        width = width * 10 + (*p - '0');
        p++;
      }

      switch (*p) {
      case 'd': { // Integer
        int ival = va_arg(args, int);
        print_int(ival, width, pad);
        break;
      }
      case 'c': { // Character
        char c = (char)va_arg(args, int);
        putch(c);
        break;
      }
      case 's': { // String
        char *s = va_arg(args, char *);
        putstr(s);
        break;
      }
      case '%': {
        putch('%');
        break;
      }
      default:
        panic("Wrong formatter provided to printf");
      }
    } else {
      putch(*p);
    }
    p++;
  }
}

int printf(const char *fmt, ...) {
  va_list args;
  va_start(args, fmt);
  vprintf(fmt, args);
  va_end(args);
  return 0;
}

void append_to_buffer(char **buf, int *pos, char c) { (*buf)[(*pos)++] = c; }

void print_int_to_buf(char **buf, int *pos, int num, int width, char pad) {
  int reverse = 0, count = 0, neg = 0;

  if (num == 0) {
    reverse = 0;
    count = 1;
  } else {
    if (num < 0) {
      append_to_buffer(buf, pos, '-');
      num = -num;
      neg = 1;
    }
    while (num != 0) {
      reverse = reverse * 10 + (num % 10);
      num /= 10;
      count++;
    }
  }

  width -= neg;
  while (width > count) {
    append_to_buffer(buf, pos, pad);
    width--;
  }

  for (int i = 0; i < count; i++) {
    append_to_buffer(buf, pos, '0' + (reverse % 10));
    reverse /= 10;
  }
}

int vsprintf(char *buf, const char *format, va_list args) {
  const char *p = format;
  int pos = 0; // Position in buf

  while (*p) {
    if (*p == '%') {
      p++; // Skip the '%'
      char pad = ' ';
      int width = 0;

      if (*p == '0') {
        pad = '0';
        p++;
      }

      while (*p >= '0' && *p <= '9') {
        width = width * 10 + (*p - '0');
        p++;
      }

      switch (*p) {
      case 'd': { // Integer
        int ival = va_arg(args, int);
        print_int_to_buf(&buf, &pos, ival, width, pad);
        break;
      }
      case 'c': { // Character
        char c = (char)va_arg(args, int);
        append_to_buffer(&buf, &pos, c);
        break;
      }
      case 's': { // String
        char *s = va_arg(args, char *);
        while (*s) {
          append_to_buffer(&buf, &pos, *s++);
        }
        break;
      }
      case '%': {
        append_to_buffer(&buf, &pos, '%');
        break;
      }
      default:
        panic("Unsupported format specifier");
      }
    } else {
      append_to_buffer(&buf, &pos, *p);
    }
    p++;
  }
  buf[pos] = '\0'; // Null-terminate the string
  return pos;
}

int sprintf(char *out, const char *fmt, ...) {
  va_list args;
  va_start(args, fmt);
  vsprintf(out, fmt, args);
  va_end(args);
  return 0;
}

int snprintf(char *out, size_t n, const char *fmt, ...) {
  panic("Not implemented");
}

int vsnprintf(char *out, size_t n, const char *fmt, va_list ap) {
  panic("Not implemented");
}

#endif