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

#include "mozilla/ErrorResult.h"
#include "mozilla/net/MozURL.h"
#include "mozilla/net/URLPatternGlue.h"

namespace mozilla::dom {

NS_IMPL_CYCLE_COLLECTION_WRAPPERCACHE(URLPattern, mParent)

NS_IMPL_CYCLE_COLLECTING_ADDREF(URLPattern)
NS_IMPL_CYCLE_COLLECTING_RELEASE(URLPattern)

NS_INTERFACE_MAP_BEGIN_CYCLE_COLLECTION(URLPattern)
  NS_WRAPPERCACHE_INTERFACE_MAP_ENTRY
  NS_INTERFACE_MAP_ENTRY(nsISupports)
NS_INTERFACE_MAP_END

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

void GlueToBindingInit(const UrlPatternInit& aGInit, URLPatternInit& aBInit) {
  if (aGInit.protocol.valid) {
    aBInit.mProtocol.Construct(aGInit.protocol.string);
  }
  if (aGInit.username.valid) {
    aBInit.mUsername.Construct(aGInit.username.string);
  }
  if (aGInit.password.valid) {
    aBInit.mPassword.Construct(aGInit.password.string);
  }
  if (aGInit.hostname.valid) {
    aBInit.mHostname.Construct(aGInit.hostname.string);
  }
  if (aGInit.port.valid) {
    aBInit.mPort.Construct(aGInit.port.string);
  }
  if (aGInit.pathname.valid) {
    aBInit.mPathname.Construct(aGInit.pathname.string);
  }
  if (aGInit.search.valid) {
    aBInit.mSearch.Construct(aGInit.search.string);
  }
  if (aGInit.hash.valid) {
    aBInit.mHash.Construct(aGInit.hash.string);
  }
  if (aGInit.base_url.valid) {
    aBInit.mBaseURL.Construct(aGInit.base_url.string);
  }
}

void BindingToGlueInit(const URLPatternInit& aBInit, UrlPatternInit& aGInit) {
  if (aBInit.mProtocol.WasPassed()) {
    aGInit.protocol = net::CreateMaybeString(aBInit.mProtocol.Value(), true);
  }
  if (aBInit.mUsername.WasPassed()) {
    aGInit.username = net::CreateMaybeString(aBInit.mUsername.Value(), true);
  }
  if (aBInit.mPassword.WasPassed()) {
    aGInit.password = net::CreateMaybeString(aBInit.mPassword.Value(), true);
  }
  if (aBInit.mHostname.WasPassed()) {
    aGInit.hostname = net::CreateMaybeString(aBInit.mHostname.Value(), true);
  }
  if (aBInit.mPort.WasPassed()) {
    aGInit.port = net::CreateMaybeString(aBInit.mPort.Value(), true);
  }
  if (aBInit.mPathname.WasPassed()) {
    aGInit.pathname = net::CreateMaybeString(aBInit.mPathname.Value(), true);
  }
  if (aBInit.mSearch.WasPassed()) {
    aGInit.search = net::CreateMaybeString(aBInit.mSearch.Value(), true);
  }
  if (aBInit.mHash.WasPassed()) {
    aGInit.hash = net::CreateMaybeString(aBInit.mHash.Value(), true);
  }
  if (aBInit.mBaseURL.WasPassed()) {
    aGInit.base_url = net::CreateMaybeString(aBInit.mBaseURL.Value(), true);
  }
}

// static
already_AddRefed<URLPattern> URLPattern::Constructor(
    const GlobalObject& aGlobal, const UTF8StringOrURLPatternInit& aInput,
    const URLPatternOptions& aOptions, ErrorResult& rv) {
  MOZ_LOG(gUrlPatternLog, LogLevel::Debug,
          ("URLPattern::Constructor() (without base)"));
  UrlPatternGlue pattern{};
  UrlPatternOptions options{};
  options.ignore_case = aOptions.mIgnoreCase;
  if (!aInput.IsURLPatternInit()) {
    bool res = urlpattern_parse_pattern_from_string(&aInput.GetAsUTF8String(),
                                                    nullptr, options, &pattern);
    if (!res) {
      rv.ThrowTypeError("Failed to create URLPattern (from string)");
      return nullptr;
    }
  } else {
    UrlPatternInit init{};
    URLPatternInit b_init;
    b_init = aInput.GetAsURLPatternInit();
    BindingToGlueInit(b_init, init);
    if (init.base_url.valid && init.base_url.string.Equals("")) {
      rv.ThrowTypeError("Should not provide empty base url with init");
      return nullptr;
    }
    bool res = urlpattern_parse_pattern_from_init(&init, options, &pattern);
    if (!res) {
      rv.ThrowTypeError("Failed to create URLPattern (from init)");
      return nullptr;
    }
  }

  return MakeAndAddRef<URLPattern>(aGlobal.GetAsSupports(), pattern,
                                   aOptions.mIgnoreCase);
}

// static
already_AddRefed<URLPattern> URLPattern::Constructor(
    const GlobalObject& aGlobal, const UTF8StringOrURLPatternInit& aInput,
    const nsACString& aBase, const URLPatternOptions& aOptions,
    ErrorResult& rv) {
  MOZ_LOG(gUrlPatternLog, LogLevel::Debug,
          ("UrlPattern::Constructor() (w base)"));
  UrlPatternGlue pattern{};
  UrlPatternOptions options{};
  options.ignore_case = aOptions.mIgnoreCase;
  if (!aInput.IsURLPatternInit()) {
    bool res = urlpattern_parse_pattern_from_string(&aInput.GetAsUTF8String(),
                                                    &aBase, options, &pattern);
    if (!res) {
      rv.ThrowTypeError(
          "Failed to create URLPattern with base url (from string)");
      return nullptr;
    }
  } else {
    if (!aBase.IsEmpty()) {
      rv.ThrowTypeError("Should not provide base url with init");
      return nullptr;
    }
    UrlPatternInit init{};
    URLPatternInit b_init;
    b_init = aInput.GetAsURLPatternInit();
    BindingToGlueInit(b_init, init);
    bool res = urlpattern_parse_pattern_from_init(&init, options, &pattern);
    if (!res) {
      rv.ThrowTypeError(
          "Failed to create URLPattern with base url (from init)");
      return nullptr;
    }
  }
  return MakeAndAddRef<URLPattern>(aGlobal.GetAsSupports(), pattern,
                                   aOptions.mIgnoreCase);
}

URLPattern::~URLPattern() { urlpattern_pattern_free(mPattern); }

void ConvertGroupsToRecord(
    const nsTHashMap<nsCStringHashKey, MaybeString>& aGroups,
    Optional<Record<nsCString, OwningUTF8StringOrUndefined>>& aRes) {
  Record<nsCString, OwningUTF8StringOrUndefined> record;
  for (auto iter = aGroups.ConstIter(); !iter.Done(); iter.Next()) {
    MaybeString s = iter.Data();
    OwningUTF8StringOrUndefined value;
    value.SetUndefined();  // if capture group doesn't match we leave undefined
    if (s.valid) {
      value.SetAsUTF8String().Assign(s.string);
    }
    auto* entry = record.Entries().AppendElement().get();
    entry->mKey.Assign(iter.Key());
    entry->mValue = std::move(value);
  }
  aRes.Construct(std::move(record));
}

void GlueToBindingComponent(const net::UrlPatternComponentResult& aGlueCompRes,
                            URLPatternComponentResult& aBindingCompRes) {
  aBindingCompRes.mInput.Construct(aGlueCompRes.mInput);
  ConvertGroupsToRecord(aGlueCompRes.mGroups, aBindingCompRes.mGroups);
}

void ConvertInputsToSequence(
    const CopyableTArray<UrlPatternInput>& aInputs,
    Optional<Sequence<OwningUTF8StringOrURLPatternInit>>& aRes,
    ErrorResult& rv) {
  Sequence<OwningUTF8StringOrURLPatternInit> sequence;
  for (const auto& input : aInputs) {
    OwningUTF8StringOrURLPatternInit variant;
    if (input.string_or_init_type == UrlPatternStringOrInitType::String) {
      variant.SetAsUTF8String().Assign(input.str);
    } else {
      GlueToBindingInit(input.init, variant.SetAsURLPatternInit());
    }

    if (!sequence.AppendElement(std::move(variant), fallible)) {
      aRes.Reset();
      rv.ThrowOperationError("Failed to append inputs list to sequence");
      return;
    }
  }
  aRes.Construct(std::move(sequence));
}

void GlueToBindingResult(const net::UrlPatternResult& aGlueRes,
                         URLPatternResult& aBindingRes, ErrorResult& rv) {
  if (aGlueRes.mProtocol.isSome()) {
    URLPatternComponentResult tmp;
    GlueToBindingComponent(aGlueRes.mProtocol.value(), tmp);
    aBindingRes.mProtocol.Construct(std::move(tmp));
  }
  if (aGlueRes.mUsername.isSome()) {
    URLPatternComponentResult tmp;
    GlueToBindingComponent(aGlueRes.mUsername.value(), tmp);
    aBindingRes.mUsername.Construct(std::move(tmp));
  }
  if (aGlueRes.mPassword.isSome()) {
    URLPatternComponentResult tmp;
    GlueToBindingComponent(aGlueRes.mPassword.value(), tmp);
    aBindingRes.mPassword.Construct(std::move(tmp));
  }
  if (aGlueRes.mHostname.isSome()) {
    URLPatternComponentResult tmp;
    GlueToBindingComponent(aGlueRes.mHostname.value(), tmp);
    aBindingRes.mHostname.Construct(std::move(tmp));
  }
  if (aGlueRes.mPort.isSome()) {
    URLPatternComponentResult tmp;
    GlueToBindingComponent(aGlueRes.mPort.value(), tmp);
    aBindingRes.mPort.Construct(std::move(tmp));
  }
  if (aGlueRes.mPathname.isSome()) {
    URLPatternComponentResult tmp;
    GlueToBindingComponent(aGlueRes.mPathname.value(), tmp);
    aBindingRes.mPathname.Construct(std::move(tmp));
  }
  if (aGlueRes.mSearch.isSome()) {
    URLPatternComponentResult tmp;
    GlueToBindingComponent(aGlueRes.mSearch.value(), tmp);
    aBindingRes.mSearch.Construct(std::move(tmp));
  }
  if (aGlueRes.mHash.isSome()) {
    URLPatternComponentResult tmp;
    GlueToBindingComponent(aGlueRes.mHash.value(), tmp);
    aBindingRes.mHash.Construct(std::move(tmp));
  }
  ConvertInputsToSequence(aGlueRes.mInputs, aBindingRes.mInputs, rv);
}

bool URLPattern::Test(const UTF8StringOrURLPatternInit& aInput,
                      const Optional<nsACString>& aBaseUrl, ErrorResult& rv) {
  MOZ_LOG(gUrlPatternLog, LogLevel::Debug, ("UrlPattern::Test()"));
  UrlPatternInput input;
  Maybe<nsAutoCString> execBaseUrl;
  if (aInput.IsURLPatternInit()) {
    UrlPatternInit initGlue{};
    BindingToGlueInit(aInput.GetAsURLPatternInit(), initGlue);
    input = net::CreateUrlPatternInput(initGlue);
    if (aBaseUrl.WasPassed()) {
      rv.ThrowTypeError(
          "Do not pass baseUrl separately with init, use init's baseURL "
          "property");
      return false;
    }
  } else {
    input = net::CreateUrlPatternInput(aInput.GetAsUTF8String());
    if (aBaseUrl.WasPassed()) {
      execBaseUrl.emplace(aBaseUrl.Value());
    }
  }
  return net::UrlPatternTest(mPattern, input, execBaseUrl, mIgnoreCase);
}

void URLPattern::Exec(const UTF8StringOrURLPatternInit& aInput,
                      const Optional<nsACString>& aBaseUrl,
                      Nullable<URLPatternResult>& aResult, ErrorResult& rv) {
  MOZ_LOG(gUrlPatternLog, LogLevel::Debug, ("UrlPattern::Exec()"));
  UrlPatternInput input;
  Maybe<nsAutoCString> execBaseUrl;
  if (aInput.IsURLPatternInit()) {
    UrlPatternInit initGlue{};
    BindingToGlueInit(aInput.GetAsURLPatternInit(), initGlue);
    input = net::CreateUrlPatternInput(initGlue);
    if (aBaseUrl.WasPassed()) {
      rv.ThrowTypeError(
          "Do not pass baseUrl separately with init, use init's baseURL "
          "property");
      return;
    }
  } else {
    input = net::CreateUrlPatternInput(aInput.GetAsUTF8String());
    if (aBaseUrl.WasPassed()) {
      execBaseUrl.emplace(aBaseUrl.Value());
    }
  }

  Maybe<net::UrlPatternResult> patternResult =
      net::UrlPatternExec(mPattern, input, execBaseUrl, mIgnoreCase);
  if (patternResult.isSome()) {
    URLPatternResult res;
    GlueToBindingResult(patternResult.value(), res, rv);
    if (rv.Failed()) {
      aResult.SetNull();
      return;
    }
    aResult.SetValue(std::move(res));
    return;
  }
  aResult.SetNull();
}

void URLPattern::GetProtocol(nsACString& aProtocol) const {
  aProtocol.Assign(net::UrlPatternGetProtocol(mPattern));
}

void URLPattern::GetUsername(nsACString& aUsername) const {
  aUsername.Assign(net::UrlPatternGetUsername(mPattern));
}

void URLPattern::GetPassword(nsACString& aPassword) const {
  aPassword.Assign(net::UrlPatternGetPassword(mPattern));
}

void URLPattern::GetHostname(nsACString& aHostname) const {
  aHostname.Assign(net::UrlPatternGetHostname(mPattern));
}

void URLPattern::GetPort(nsACString& aPort) const {
  aPort.Assign(net::UrlPatternGetPort(mPattern));
}

void URLPattern::GetPathname(nsACString& aPathname) const {
  aPathname.Assign(net::UrlPatternGetPathname(mPattern));
}

void URLPattern::GetSearch(nsACString& aSearch) const {
  aSearch.Assign(net::UrlPatternGetSearch(mPattern));
}

void URLPattern::GetHash(nsACString& aHash) const {
  aHash.Assign(net::UrlPatternGetHash(mPattern));
}

bool URLPattern::HasRegExpGroups() const {
  return urlpattern_get_has_regexp_groups(mPattern);
}

}  // namespace mozilla::dom