#include "position.h"

#include "serializer.h"

#include <limits.h>
#include <stdio.h>
#include <stdlib.h>

Position Position::FromString(const std::string& encoded) {
  char* p = const_cast<char*>(encoded.c_str());
  int16_t line = int16_t(strtol(p, &p, 10)) - 1;
  assert(*p == ':');
  p++;
  int16_t column = int16_t(strtol(p, &p, 10)) - 1;
  return {line, column};
}

std::string Position::ToString() {
  char buf[99];
  snprintf(buf, sizeof buf, "%d:%d", line + 1, column + 1);
  return buf;
}

Range Range::FromString(const std::string& encoded) {
  Position start, end;
  char* p = const_cast<char*>(encoded.c_str());
  start.line = int16_t(strtol(p, &p, 10)) - 1;
  assert(*p == ':');
  p++;
  start.column = int16_t(strtol(p, &p, 10)) - 1;
  assert(*p == '-');
  p++;

  end.line = int16_t(strtol(p, &p, 10)) - 1;
  assert(*p == ':');
  p++;
  end.column = int16_t(strtol(p, nullptr, 10)) - 1;
  return {start, end};
}

bool Range::Contains(int line, int column) const {
  if (line > INT16_MAX)
    return false;
  Position p{int16_t(line), int16_t(std::min(column, INT16_MAX))};
  return !(p < start) && p < end;
}

Range Range::RemovePrefix(Position position) const {
  return {std::min(std::max(position, start), end), end};
}

std::string Range::ToString() {
  char buf[99];
  snprintf(buf, sizeof buf, "%d:%d-%d:%d", start.line + 1, start.column + 1,
           end.line + 1, end.column + 1);
  return buf;
}

// Position
void Reflect(Reader& visitor, Position& value) {
  if (visitor.Format() == SerializeFormat::Json) {
    value = Position::FromString(visitor.GetString());
  } else {
    Reflect(visitor, value.line);
    Reflect(visitor, value.column);
  }
}
void Reflect(Writer& visitor, Position& value) {
  if (visitor.Format() == SerializeFormat::Json) {
    std::string output = value.ToString();
    visitor.String(output.c_str(), output.size());
  } else {
    Reflect(visitor, value.line);
    Reflect(visitor, value.column);
  }
}

// Range
void Reflect(Reader& visitor, Range& value) {
  if (visitor.Format() == SerializeFormat::Json) {
    value = Range::FromString(visitor.GetString());
  } else {
    Reflect(visitor, value.start.line);
    Reflect(visitor, value.start.column);
    Reflect(visitor, value.end.line);
    Reflect(visitor, value.end.column);
  }
}
void Reflect(Writer& visitor, Range& value) {
  if (visitor.Format() == SerializeFormat::Json) {
    std::string output = value.ToString();
    visitor.String(output.c_str(), output.size());
  } else {
    Reflect(visitor, value.start.line);
    Reflect(visitor, value.start.column);
    Reflect(visitor, value.end.line);
    Reflect(visitor, value.end.column);
  }
}