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

#include "mozilla/Components.h"
#include "mozilla/StaticPrefs_datareporting.h"
#include "mozilla/dom/BindingUtils.h"
#include "mozilla/dom/ContentChild.h"
#include "mozilla/dom/PrivateAttributionBinding.h"
#include "nsIGlobalObject.h"
#include "nsIPrivateAttributionService.h"
#include "nsURLHelper.h"
#include "nsXULAppAPI.h"

namespace mozilla::dom {

NS_IMPL_CYCLE_COLLECTION_WRAPPERCACHE(PrivateAttribution, mGlobal)

PrivateAttribution::PrivateAttribution(nsIGlobalObject* aGlobal)
    : mGlobal(aGlobal) {
  MOZ_ASSERT(aGlobal);
}

JSObject* PrivateAttribution::WrapObject(JSContext* aCx,
                                         JS::Handle<JSObject*> aGivenProto) {
  return PrivateAttribution_Binding::Wrap(aCx, this, aGivenProto);
}

PrivateAttribution::~PrivateAttribution() = default;

bool PrivateAttribution::ShouldRecord() {
#ifdef MOZ_TELEMETRY_REPORTING
  return (StaticPrefs::dom_private_attribution_submission_enabled() &&
          StaticPrefs::datareporting_healthreport_uploadEnabled());
#else
  return false;
#endif
}

bool PrivateAttribution::GetSourceHostIfNonPrivate(nsACString& aSourceHost,
                                                   ErrorResult& aRv) {
  MOZ_ASSERT(mGlobal);
  nsIPrincipal* prin = mGlobal->PrincipalOrNull();
  if (!prin || NS_FAILED(prin->GetHost(aSourceHost))) {
    aRv.ThrowInvalidStateError("Couldn't get source host");
    return false;
  }
  return !prin->GetIsInPrivateBrowsing();
}

[[nodiscard]] static bool ValidateHost(const nsACString& aHost,
                                       ErrorResult& aRv) {
  if (!net_IsValidDNSHost(aHost)) {
    aRv.ThrowSyntaxError(aHost + " is not a valid host name"_ns);
    return false;
  }
  return true;
}

void PrivateAttribution::SaveImpression(
    const PrivateAttributionImpressionOptions& aOptions, ErrorResult& aRv) {
  nsAutoCString source;
  if (!GetSourceHostIfNonPrivate(source, aRv)) {
    return;
  }

  if (!ValidateHost(aOptions.mTarget, aRv)) {
    return;
  }

  if (!ShouldRecord()) {
    return;
  }

  if (XRE_IsParentProcess()) {
    nsCOMPtr<nsIPrivateAttributionService> pa =
        components::PrivateAttribution::Service();
    if (NS_WARN_IF(!pa)) {
      return;
    }
    pa->OnAttributionEvent(source, GetEnumString(aOptions.mType),
                           aOptions.mIndex, aOptions.mAd, aOptions.mTarget);
    return;
  }

  auto* content = ContentChild::GetSingleton();
  if (NS_WARN_IF(!content)) {
    return;
  }
  content->SendAttributionEvent(source, aOptions.mType, aOptions.mIndex,
                                aOptions.mAd, aOptions.mTarget);
}

void PrivateAttribution::MeasureConversion(
    const PrivateAttributionConversionOptions& aOptions, ErrorResult& aRv) {
  nsAutoCString source;
  if (!GetSourceHostIfNonPrivate(source, aRv)) {
    return;
  }
  for (const nsACString& host : aOptions.mSources) {
    if (!ValidateHost(host, aRv)) {
      return;
    }
  }

  if (!ShouldRecord()) {
    return;
  }

  if (XRE_IsParentProcess()) {
    nsCOMPtr<nsIPrivateAttributionService> pa =
        components::PrivateAttribution::Service();
    if (NS_WARN_IF(!pa)) {
      return;
    }
    pa->OnAttributionConversion(
        source, aOptions.mTask, aOptions.mHistogramSize,
        aOptions.mLookbackDays.WasPassed() ? aOptions.mLookbackDays.Value() : 0,
        aOptions.mImpression.WasPassed()
            ? GetEnumString(aOptions.mImpression.Value())
            : EmptyCString(),
        aOptions.mAds, aOptions.mSources);
    return;
  }

  auto* content = ContentChild::GetSingleton();
  if (NS_WARN_IF(!content)) {
    return;
  }
  content->SendAttributionConversion(
      source, aOptions.mTask, aOptions.mHistogramSize,
      aOptions.mLookbackDays.WasPassed() ? Some(aOptions.mLookbackDays.Value())
                                         : Nothing(),
      aOptions.mImpression.WasPassed() ? Some(aOptions.mImpression.Value())
                                       : Nothing(),
      aOptions.mAds, aOptions.mSources);
}

}  // namespace mozilla::dom