/* 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/. */

#include "mozilla/dom/StyledRange.h"

#include "AbstractRange.h"
#include "nsCycleCollectionParticipant.h"

namespace mozilla::dom {

StyledRange::StyledRange(AbstractRange* aRange, TextRangeStyle aStyle)
    : mRange(aRange), mTextRangeStyle(aStyle) {}

void ImplCycleCollectionTraverse(nsCycleCollectionTraversalCallback& aCallback,
                                 StyledRangeCollection& aField,
                                 const char* aName, uint32_t aFlags) {
  for (size_t i = 0; i < aField.Length(); ++i) {
    CycleCollectionNoteChild(aCallback, aField.GetAbstractRangeAt(i),
                             "mRanges[i]", aFlags);
  }
}
void StyledRangeCollection::AppendElement(StyledRange&& aRange) {
  SetTextRangeStyle(*mRanges.AppendElement(std::move(aRange.mRange)),
                    aRange.mTextRangeStyle);
}
void StyledRangeCollection::AppendElement(const StyledRange& aRange) {
  SetTextRangeStyle(*mRanges.AppendElement(aRange.mRange),
                    aRange.mTextRangeStyle);
}

void StyledRangeCollection::InsertElementAt(size_t aIndex,
                                            StyledRange&& aRange) {
  SetTextRangeStyle(*mRanges.InsertElementAt(aIndex, std::move(aRange.mRange)),
                    aRange.mTextRangeStyle);
}

void StyledRangeCollection::InsertElementAt(size_t aIndex,
                                            const StyledRange& aRange) {
  SetTextRangeStyle(*mRanges.InsertElementAt(aIndex, aRange.mRange),
                    aRange.mTextRangeStyle);
}

void StyledRangeCollection::InsertElementsAt(
    size_t aIndex, const nsTArray<StyledRange>& aStyledRanges) {
  for (size_t i = 0; i < aStyledRanges.Length(); ++i) {
    InsertElementAt(aIndex + i, aStyledRanges[i]);
  }
}

bool StyledRangeCollection::RemoveElement(const AbstractRange* aRange) {
  for (size_t i = 0; i < mRanges.Length(); ++i) {
    if (mRanges[i] == aRange) {
      RemoveElementAt(i);
      return true;
    }
  }
  return false;
}

void StyledRangeCollection::RemoveElementAt(size_t aIndex) {
  mozilla::dom::AbstractRange* range = mRanges[aIndex];
  RemoveStyle(range);
  mRanges.RemoveElementAt(aIndex);
}

void StyledRangeCollection::RemoveElementsAt(size_t aStart, size_t aCount) {
  for (size_t i = aStart; i < aStart + aCount; ++i) {
    RemoveStyle(mRanges[i]);
  }
  mRanges.RemoveElementsAt(aStart, aCount);
}

void StyledRangeCollection::Clear() {
  mRangeStyleData.Clear();
  mRanges.Clear();
}

StyledRange StyledRangeCollection::ExtractElementAt(size_t aIndex) {
  mozilla::dom::AbstractRange* range = mRanges[aIndex];
  StyledRange result(range);
  if (const auto* style = GetTextRangeStyleIfNotDefault(range)) {
    result.mTextRangeStyle = *style;
  }
  RemoveElementAt(aIndex);
  return result;
}

void StyledRangeCollection::SetTextRangeStyle(const AbstractRange* aRange,
                                              const TextRangeStyle& aStyle) {
  static const TextRangeStyle defaultStyle{};
  if (aStyle != defaultStyle) {
    mRangeStyleData.InsertOrUpdate(aRange, aStyle);
  }
}

void StyledRangeCollection::RemoveStyle(const AbstractRange* aRange) {
  mRangeStyleData.Remove(aRange);
}

const mozilla::TextRangeStyle*
StyledRangeCollection::GetTextRangeStyleIfNotDefault(
    const AbstractRange* aRange) {
  if (!aRange) {
    return nullptr;
  }

  return mRangeStyleData.Lookup(aRange).DataPtrOrNull();
}
}  // namespace mozilla::dom