#ifndef SRC_SPELLCHECKER_H_
#define SRC_SPELLCHECKER_H_

#include <string>
#include <vector>
#include <memory>
#include <stdint.h>

namespace spellchecker {

const int USE_SYSTEM_DEFAULTS = 0;
const int ALWAYS_USE_SYSTEM = 1;
const int ALWAYS_USE_HUNSPELL = 2;

struct MisspelledRange {
  size_t start;
  size_t end;
};

class SpellcheckerImplementation;

class SpellcheckerThreadView {
public:
  SpellcheckerThreadView(SpellcheckerImplementation *impl) : impl{impl}
  {
    //
  }

  virtual ~SpellcheckerThreadView()
  {
    //
  }

  virtual std::vector<MisspelledRange> CheckSpelling(const uint16_t *text, size_t length);

private:
  SpellcheckerImplementation *impl;
};

class SpellcheckerImplementation {
public:
  virtual bool SetDictionary(const std::string& language, const std::string& path) = 0;
  virtual std::vector<std::string> GetAvailableDictionaries(const std::string& path) = 0;

  // Returns an array containing possible corrections for the word.
  virtual std::vector<std::string> GetCorrectionsForMisspelling(const std::string& word) = 0;

  // Returns true if the word is misspelled.
  virtual bool IsMisspelled(const std::string& word) = 0;

  virtual std::vector<MisspelledRange> CheckSpelling(const uint16_t *text, size_t length) = 0;

  // Adds a new word to the dictionary.
  // NB: When using Hunspell, this will not modify the .dic file; custom words must be added each
  // time the spellchecker is created. Use a custom dictionary file.
  virtual void Add(const std::string& word) = 0;

  // Removes a word from the custom dictionary added by Add.
  // NB: When using Hunspell, this will not modify the .dic file; custom words must be added each
  // time the spellchecker is created. Use a custom dictionary file.
  virtual void Remove(const std::string& word) = 0;

  virtual std::unique_ptr<SpellcheckerThreadView> CreateThreadView()
  {
    return std::unique_ptr<SpellcheckerThreadView>(new SpellcheckerThreadView(this));
  }

  virtual ~SpellcheckerImplementation() {}
};

class SpellcheckerFactory {
public:
  static SpellcheckerImplementation* CreateSpellchecker(int spellcheckerType);
};

inline std::vector<MisspelledRange> SpellcheckerThreadView::CheckSpelling(const uint16_t *text, size_t length)
{
  return impl->CheckSpelling(text, length);
}

}  // namespace spellchecker

#endif  // SRC_SPELLCHECKER_H_