/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */

/* This Source Code Form is subject to the terms of the Mozilla Public

 * License, v. 2.0. If a copy of the MPL was not distributed with this

 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#ifndef mozSpellChecker_h__

#define mozSpellChecker_h__

#include "nsCOMPtr.h"

#include "nsCOMArray.h"

#include "nsISpellChecker.h"

#include "nsString.h"

#include "mozIPersonalDictionary.h"

#include "mozISpellCheckingEngine.h"

#include "nsClassHashtable.h"

#include "nsTArray.h"

#include "nsCycleCollectionParticipant.h"

class mozEnglishWordUtils;

namespace mozilla {

class RemoteSpellcheckEngineChild;

} // namespace mozilla

class mozSpellChecker final : public nsISpellChecker

{

public:

  NS_DECL_CYCLE_COLLECTING_ISUPPORTS

  NS_DECL_CYCLE_COLLECTION_CLASS(mozSpellChecker)

  static already_AddRefed<mozSpellChecker>

  Create()

  {

    RefPtr<mozSpellChecker> spellChecker = new mozSpellChecker();

    nsresult rv = spellChecker->Init();

    NS_ENSURE_SUCCESS(rv, nullptr);

    return spellChecker.forget();

  }

  // nsISpellChecker

  NS_IMETHOD SetDocument(mozilla::TextServicesDocument* aTextServicesDocument,

                         bool aFromStartofDoc) override;

  NS_IMETHOD NextMisspelledWord(nsAString &aWord, nsTArray<nsString> *aSuggestions) override;

  NS_IMETHOD CheckWord(const nsAString &aWord, bool *aIsMisspelled, nsTArray<nsString> *aSuggestions) override;

  NS_IMETHOD Replace(const nsAString &aOldWord, const nsAString &aNewWord, bool aAllOccurrences) override;

  NS_IMETHOD IgnoreAll(const nsAString &aWord) override;

  NS_IMETHOD AddWordToPersonalDictionary(const nsAString &aWord) override;

  NS_IMETHOD RemoveWordFromPersonalDictionary(const nsAString &aWord) override;

  NS_IMETHOD GetPersonalDictionary(nsTArray<nsString> *aWordList) override;

  NS_IMETHOD GetDictionaryList(nsTArray<nsString> *aDictionaryList) override;

  NS_IMETHOD GetCurrentDictionary(nsAString &aDictionary) override;

  NS_IMETHOD SetCurrentDictionary(const nsAString &aDictionary) override;

  NS_IMETHOD_(RefPtr<mozilla::GenericPromise>)

    SetCurrentDictionaryFromList(const nsTArray<nsString>& aList) override;

  void DeleteRemoteEngine() {

    mEngine = nullptr;

  }

  mozilla::TextServicesDocument* GetTextServicesDocument();

protected:

  mozSpellChecker();

  virtual ~mozSpellChecker();

  nsresult Init();

  RefPtr<mozEnglishWordUtils> mConverter;

  RefPtr<mozilla::TextServicesDocument> mTextServicesDocument;

  nsCOMPtr<mozIPersonalDictionary> mPersonalDictionary;

  nsCOMPtr<mozISpellCheckingEngine>  mSpellCheckingEngine;

  bool mFromStart;

  nsString mCurrentDictionary;

  nsresult SetupDoc(int32_t *outBlockOffset);

  nsresult GetCurrentBlockIndex(

             mozilla::TextServicesDocument* aTextServicesDocument,

             int32_t* aOutBlockIndex);

  nsresult GetEngineList(nsCOMArray<mozISpellCheckingEngine> *aDictionaryList);

  mozilla::RemoteSpellcheckEngineChild *mEngine;

  friend class mozilla::RemoteSpellcheckEngineChild;

};

#endif // mozSpellChecker_h__