/* 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 SystemGlobal_h_
#define SystemGlobal_h_

#include "js/loader/ModuleLoaderBase.h"
#include "mozilla/BasePrincipal.h"
#include "mozilla/SchedulerGroup.h"
#include "mozilla/StorageAccess.h"
#include "nsISupports.h"
#include "nsWeakReference.h"
#include "nsIGlobalObject.h"
#include "nsIScriptObjectPrincipal.h"
#include "nsIXPCScriptable.h"

#include "js/HeapAPI.h"

class nsICookieJarSettings;
class XPCWrappedNative;

// The shared system global (used by ChromeUtils.importESModule), and also
// the xpcshell's global.
class SystemGlobal final : public nsIGlobalObject,
                           public nsIScriptObjectPrincipal,
                           public nsIXPCScriptable,
                           public nsIClassInfo,
                           public nsSupportsWeakReference {
 public:
  SystemGlobal();

  NS_DECL_ISUPPORTS
  NS_DECL_NSIXPCSCRIPTABLE
  NS_DECL_NSICLASSINFO

  using ModuleLoaderBase = JS::loader::ModuleLoaderBase;

  nsIPrincipal* GetPrincipal() override { return mPrincipal; }

  nsIPrincipal* GetEffectiveCookiePrincipal() override { return mPrincipal; }

  nsIPrincipal* GetEffectiveStoragePrincipal() override { return mPrincipal; }

  nsIPrincipal* PartitionedPrincipal() override { return mPrincipal; }

  mozilla::Maybe<nsID> GetAgentClusterId() const override {
    return mozilla::Some(mAgentClusterId);
  }

  mozilla::OriginTrials Trials() const override { return {}; }

  JSObject* GetGlobalJSObject() override;
  JSObject* GetGlobalJSObjectPreserveColor() const override;

  ModuleLoaderBase* GetModuleLoader(JSContext* aCx) override {
    return mModuleLoader;
  }

  nsICookieJarSettings* GetCookieJarSettings() override {
    MOZ_ASSERT(NS_IsMainThread());
    return mCookieJarSettings;
  }

  mozilla::StorageAccess GetStorageAccess() final {
    MOZ_ASSERT(NS_IsMainThread());
    return mozilla::StorageAccess::eAllow;
  }

  mozilla::Result<mozilla::ipc::PrincipalInfo, nsresult> GetStorageKey()
      override;

  void ForgetGlobalObject() { mWrapper = nullptr; }

  void SetGlobalObject(JSObject* global);

  void InitModuleLoader(ModuleLoaderBase* aModuleLoader) {
    MOZ_ASSERT(!mModuleLoader);
    mModuleLoader = aModuleLoader;
  }

  nsISerialEventTarget* SerialEventTarget() const final {
    return mozilla::GetMainThreadSerialEventTarget();
  }
  nsresult Dispatch(already_AddRefed<nsIRunnable> aRunnable) const final {
    return mozilla::SchedulerGroup::Dispatch(std::move(aRunnable));
  }

  bool IsSharedMemoryAllowed() const override { return true; }

  bool ShouldResistFingerprinting(RFPTarget aTarget) const override {
    // SystemGlobal is always the System Principal
    MOZ_RELEASE_ASSERT(mPrincipal->IsSystemPrincipal());
    return false;
  }

 private:
  virtual ~SystemGlobal() = default;

  const nsID mAgentClusterId;
  nsCOMPtr<nsIPrincipal> mPrincipal;
  nsCOMPtr<nsICookieJarSettings> mCookieJarSettings;
  XPCWrappedNative* mWrapper;

  RefPtr<JS::loader::ModuleLoaderBase> mModuleLoader;
};

#endif  // SystemGlobal_h_