gecko-dev/dom/worklet/Worklet.h

/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
/* 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/. */

#ifndef mozilla_dom_Worklet_h
#define mozilla_dom_Worklet_h

#include "mozilla/Attributes.h"
#include "mozilla/ErrorResult.h"
#include "nsRefPtrHashtable.h"
#include "nsWrapperCache.h"
#include "nsCOMPtr.h"

class nsPIDOMWindowInner;
class nsIPrincipal;

namespace mozilla {
namespace dom {

class Promise;
class WorkletGlobalScope;
class WorkletFetchHandler;
enum class CallerType : uint32_t;

class Worklet final : public nsISupports
                    , public nsWrapperCache
{
public:
  NS_DECL_CYCLE_COLLECTING_ISUPPORTS
  NS_DECL_CYCLE_COLLECTION_SCRIPT_HOLDER_CLASS(Worklet)

  enum WorkletType {
    eAudioWorklet,
    ePaintWorklet,
  };

  Worklet(nsPIDOMWindowInner* aWindow, nsIPrincipal* aPrincipal,
          WorkletType aWorkletType);

  nsPIDOMWindowInner* GetParentObject() const
  {
    return mWindow;
  }

  virtual JSObject*
  WrapObject(JSContext* aCx, JS::Handle<JSObject*> aGivenProto) override;

  already_AddRefed<Promise>
  Import(const nsAString& aModuleURL, CallerType aCallerType,
         ErrorResult& aRv);

  WorkletGlobalScope*
  GetOrCreateGlobalScope(JSContext* aCx);

private:
  ~Worklet();

  WorkletFetchHandler*
  GetImportFetchHandler(const nsACString& aURI);

  void
  AddImportFetchHandler(const nsACString& aURI, WorkletFetchHandler* aHandler);

  nsCOMPtr<nsPIDOMWindowInner> mWindow;
  nsCOMPtr<nsIPrincipal> mPrincipal;

  WorkletType mWorkletType;

  RefPtr<WorkletGlobalScope> mScope;
  nsRefPtrHashtable<nsCStringHashKey, WorkletFetchHandler> mImportHandlers;

  friend class WorkletFetchHandler;
};

} // namespace dom
} // namespace mozilla

#endif // mozilla_dom_Worklet_h