third_party/WebKit/Source/core/streams/UnderlyingSourceBase.h

// Copyright 2016 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 UnderlyingSourceBase_h
#define UnderlyingSourceBase_h

#include "bindings/core/v8/ActiveScriptWrappable.h"
#include "bindings/core/v8/ScriptPromise.h"
#include "bindings/core/v8/ScriptState.h"
#include "bindings/core/v8/ScriptValue.h"
#include "bindings/core/v8/ScriptWrappable.h"
#include "core/CoreExport.h"
#include "core/dom/ActiveDOMObject.h"
#include "platform/heap/GarbageCollected.h"
#include "platform/heap/Handle.h"

namespace blink {

class ReadableStreamController;

class CORE_EXPORT UnderlyingSourceBase : public GarbageCollectedFinalized<UnderlyingSourceBase>, public ScriptWrappable, public ActiveScriptWrappable, public ActiveDOMObject {
    DEFINE_WRAPPERTYPEINFO();
    USING_GARBAGE_COLLECTED_MIXIN(UnderlyingSourceBase);

public:
    DECLARE_VIRTUAL_TRACE();
    virtual ~UnderlyingSourceBase() {}

    ScriptPromise startWrapper(ScriptState*, ScriptValue stream);
    virtual ScriptPromise start(ScriptState*);

    virtual ScriptPromise pull(ScriptState*);

    ScriptPromise cancelWrapper(ScriptState*, ScriptValue reason);
    virtual ScriptPromise cancel(ScriptState*, ScriptValue reason);

    void notifyLockAcquired();
    void notifyLockReleased();

    // ActiveScriptWrappable
    bool hasPendingActivity() const;

    // ActiveDOMObject
    void stop() override;

protected:
    explicit UnderlyingSourceBase(ScriptState* scriptState)
        : ActiveScriptWrappable(this)
        , ActiveDOMObject(scriptState->getExecutionContext())
    {
        this->suspendIfNeeded();
    }

    ReadableStreamController* controller() const { return m_controller; }

private:
    Member<ReadableStreamController> m_controller;
    bool m_isStreamLocked = false;
};

} // namespace blink

#endif // UnderlyingSourceBase_h