third_party/WebKit/Source/core/inspector/InspectedFrames.cpp

// Copyright 2015 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.

#include "core/inspector/InspectedFrames.h"

#include "core/dom/Document.h"
#include "core/frame/LocalFrame.h"

namespace blink {

InspectedFrames::InspectedFrames(LocalFrame* root)
    : m_root(root)
{
}

InspectedFrames::Iterator InspectedFrames::begin()
{
    return Iterator(m_root, m_root);
}

InspectedFrames::Iterator InspectedFrames::end()
{
    return Iterator(m_root, nullptr);
}

bool InspectedFrames::contains(LocalFrame* frame) const
{
    return frame->instrumentingAgents() == m_root->instrumentingAgents();
}

LocalFrame* InspectedFrames::frameWithSecurityOrigin(const String& originRawString)
{
    for (LocalFrame* frame : *this) {
        if (frame->document()->securityOrigin()->toRawString() == originRawString)
            return frame;
    }
    return nullptr;
}

InspectedFrames::Iterator::Iterator(LocalFrame* root, LocalFrame* current)
    : m_root(root)
    , m_current(current)
{
}

InspectedFrames::Iterator& InspectedFrames::Iterator::operator++()
{
    if (!m_current)
        return *this;
    Frame* frame = m_current->tree().traverseNext(m_root);
    m_current = nullptr;
    for (; frame; frame = frame->tree().traverseNext(m_root)) {
        if (!frame->isLocalFrame())
            continue;
        LocalFrame* local = toLocalFrame(frame);
        if (local->instrumentingAgents() == m_root->instrumentingAgents()) {
            m_current = local;
            break;
        }
    }
    return *this;
}

InspectedFrames::Iterator InspectedFrames::Iterator::operator++(int)
{
    LocalFrame* old = m_current;
    ++*this;
    return Iterator(m_root, old);
}

bool InspectedFrames::Iterator::operator==(const Iterator& other)
{
    return m_current == other.m_current && m_root == other.m_root;
}

bool InspectedFrames::Iterator::operator!=(const Iterator& other)
{
    return !(*this == other);
}

} // namespace blink