third_party/WebKit/Source/platform/mojo/ReferrerStructTraits.h

// Copyright 2017 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef ReferrerStructTraits_h
#define ReferrerStructTraits_h

#include "platform/weborigin/Referrer.h"
#include "platform/weborigin/ReferrerPolicy.h"
#include "platform/wtf/Assertions.h"
#include "public/platform/ReferrerPolicyEnumTraits.h"
#include "public/platform/WebReferrerPolicy.h"

namespace mojo {

template <>
struct StructTraits<blink::mojom::ReferrerDataView, blink::Referrer> {
  static blink::KURL url(const blink::Referrer& referrer) {
    if (referrer.referrer == blink::Referrer::NoReferrer())
      return blink::KURL();

    return blink::KURL(blink::KURL(), referrer.referrer);
  }

  // Equality of values is asserted in //Source/web/AssertMatchingEnums.cpp.
  static blink::WebReferrerPolicy policy(const blink::Referrer& referrer) {
    return static_cast<blink::WebReferrerPolicy>(referrer.referrer_policy);
  }

  static bool Read(blink::mojom::ReferrerDataView data, blink::Referrer* out) {
    blink::KURL referrer;
    blink::WebReferrerPolicy webReferrerPolicy;
    if (!data.ReadUrl(&referrer) || !data.ReadPolicy(&webReferrerPolicy))
      return false;

    out->referrer_policy =
        static_cast<blink::ReferrerPolicy>(webReferrerPolicy);
    out->referrer = AtomicString(referrer.GetString());

    // Mimics the DCHECK() done in the blink::Referrer constructor.
    return referrer.IsValid() || out->referrer == blink::Referrer::NoReferrer();
  }
};

}  // namespace mojo

#endif  // ReferrerStructTraits_h