#ifndef DEBUG_H
#define DEBUG_H

#include "config-check.h"

enum DebugSink {
  DEBUG_SINK_SERIAL,
  DEBUG_SINK_SOFTWARE_SERIAL,
  DEBUG_SINK_VOID,
};

template <enum DebugSink sink>
class Debug {
public:
  inline void begin(unsigned int baud_rate);

  inline operator bool();

  inline bool available();

  size_t write(uint8_t);

  template<typename T>
  inline
  void print(T t, int format = DEC);

  inline
  void print(const char *t);

  inline
  void print(const __FlashStringHelper *t);

  template<typename T>
  inline
  void println(T t, int format = DEC);

  template<typename T>
  inline
  void println(const T *t);

  inline
  void println(const __FlashStringHelper *t);

  inline
  void println();

  inline
  void flush();
};

template<enum DebugSink sink>
inline void Debug<sink>::begin(unsigned int baud_rate) {
  if (sink == DEBUG_SINK_SERIAL) {
    Serial.begin(baud_rate);
  }
}

template<enum DebugSink sink>
inline Debug<sink>::operator bool() {
  if (sink == DEBUG_SINK_SERIAL) {
    return Serial;
  }

  return false;
}

template<enum DebugSink sink>
inline bool Debug<sink>::available() {
  if (sink == DEBUG_SINK_SERIAL) {
    return Serial.available();
  }

  return false;
}

template<enum DebugSink sink>
inline size_t Debug<sink>::write(uint8_t value) {
  if (sink == DEBUG_SINK_SERIAL) {
    return Serial.write(value);
  }

  return 0;
}

template<enum DebugSink sink>
template<typename T>
inline
void Debug<sink>::print(T t, int format) {
  if (sink == DEBUG_SINK_SERIAL) {
    Serial.print(t, format);
  }
}

template<enum DebugSink sink>
inline
void Debug<sink>::print(const char *t) {
  if (sink == DEBUG_SINK_SERIAL) {
    Serial.print(t);
  }
}
template<enum DebugSink sink>
inline
void Debug<sink>::print(const __FlashStringHelper *t) {
  if (sink == DEBUG_SINK_SERIAL) {
    Serial.print(t);
  }
}

template<enum DebugSink sink>
template<typename T>
inline
void Debug<sink>::println(T t, int format) {
  if (sink == DEBUG_SINK_SERIAL) {
    Serial.println(t, format);
  }
}

template<enum DebugSink sink>
template<typename T>
inline
void Debug<sink>::println(const T *t) {
  if (sink == DEBUG_SINK_SERIAL) {
    Serial.println(t);
  }
}

template<enum DebugSink sink>
inline
void Debug<sink>::println(const __FlashStringHelper *t) {
  if (sink == DEBUG_SINK_SERIAL) {
    Serial.println(t);
  }
}

template<enum DebugSink sink>
inline
void Debug<sink>::println() {
  if (sink == DEBUG_SINK_SERIAL) {
    Serial.println();
  }
}

#if DEBUG_SINK == 1
static Debug<DEBUG_SINK_VOID> debug;
#elif DEBUG_SINK == 2
static Debug<DEBUG_SINK_SERIAL> debug;
#else
#error Invalid value for DEBUG_SINK, must be one of 1 (for no output), 2 for serial port
// , or 3 for software serial.
#endif

#endif