/* 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 "SVGAnimatedString.h"

#include <utility>

#include "SMILStringType.h"
#include "SVGAttrTearoffTable.h"
#include "mozilla/SMILValue.h"
#include "mozilla/dom/TrustedTypeUtils.h"
#include "mozilla/dom/TrustedTypesConstants.h"

using namespace mozilla::dom;

namespace mozilla {

/* Implementation */

void SVGAnimatedString::SetBaseValue(const nsAString& aValue,
                                     SVGElement* aSVGElement, bool aDoSetAttr) {
  NS_ASSERTION(aSVGElement, "Null element passed to SetBaseValue");

  mIsBaseSet = true;
  if (aDoSetAttr) {
    aSVGElement->SetStringBaseValue(mAttrEnum, aValue);
  }
  if (!mAnimVal.IsVoid()) {
    aSVGElement->AnimationNeedsResample();
  }

  aSVGElement->DidChangeString(mAttrEnum);
}

void SVGAnimatedString::GetAnimValue(nsAString& aResult,
                                     const SVGElement* aSVGElement) const {
  if (!mAnimVal.IsVoid()) {
    aResult = mAnimVal;
    return;
  }

  aSVGElement->GetStringBaseValue(mAttrEnum, aResult);
}

void SVGAnimatedString::SetAnimValue(const nsAString& aValue,
                                     SVGElement* aSVGElement) {
  if (aSVGElement->IsStringAnimatable(mAttrEnum)) {
    if (!mAnimVal.IsVoid() && mAnimVal.Equals(aValue)) {
      return;
    }
    mAnimVal = aValue;
    aSVGElement->DidAnimateString(mAttrEnum);
  }
}

std::unique_ptr<SMILAttr> SVGAnimatedString::ToSMILAttr(
    SVGElement* aSVGElement) {
  return std::make_unique<SMILString>(this, aSVGElement);
}

nsresult SVGAnimatedString::SMILString::ValueFromString(
    const nsAString& aStr, const dom::SVGAnimationElement* /*aSrcElement*/,
    SMILValue& aValue, bool& aPreventCachingOfSandwich) const {
  SMILValue val(SMILStringType::Singleton());

  *static_cast<nsAString*>(val.mU.mPtr) = aStr;
  aValue = std::move(val);
  return NS_OK;
}

SMILValue SVGAnimatedString::SMILString::GetBaseValue() const {
  SMILValue val(SMILStringType::Singleton());
  mSVGElement->GetStringBaseValue(mVal->mAttrEnum,
                                  *static_cast<nsAString*>(val.mU.mPtr));
  return val;
}

void SVGAnimatedString::SMILString::ClearAnimValue() {
  if (!mVal->mAnimVal.IsVoid()) {
    mVal->mAnimVal = VoidString();
    mSVGElement->DidAnimateString(mVal->mAttrEnum);
  }
}

nsresult SVGAnimatedString::SMILString::SetAnimValue(const SMILValue& aValue) {
  NS_ASSERTION(aValue.mType == SMILStringType::Singleton(),
               "Unexpected type to assign animated value");
  if (aValue.mType == SMILStringType::Singleton()) {
    mVal->SetAnimValue(*static_cast<nsAString*>(aValue.mU.mPtr), mSVGElement);
  }
  return NS_OK;
}

void SVGAnimatedScriptHrefString::SetBaseValue(
    const TrustedScriptURLOrString& aValue, SVGElement* aSVGElement,
    bool aDoSetAttr, nsIPrincipal* aSubjectPrincipal, ErrorResult& aRv) {
  // https://svgwg.org/svg2-draft/single-page.html#types-InterfaceSVGAnimatedString
  // See https://github.com/w3c/svgwg/pull/934
  MOZ_ASSERT(aSVGElement->IsSVGElement(nsGkAtoms::script));
  nsCOMPtr<SVGElement> svgElement = aSVGElement;
  constexpr nsLiteralString sink = u"SVGScriptElement href"_ns;
  Maybe<nsAutoString> compliantStringHolder;
  const nsAString* compliantString =
      TrustedTypeUtils::GetTrustedTypesCompliantString(
          aValue, sink, kTrustedTypesOnlySinkGroup, *svgElement,
          aSubjectPrincipal, compliantStringHolder, aRv);
  if (aRv.Failed()) {
    return;
  }
  SVGAnimatedString::SetBaseValue(*compliantString, aSVGElement, aDoSetAttr);
}

}  // namespace mozilla
