1 files changed, 88 insertions, 0 deletions
diff --git a/dom/l10n/DocumentL10n.h b/dom/l10n/DocumentL10n.h
new file mode 100644
index 0000000000..e7b2fd5189
--- /dev/null
+++ b/dom/l10n/DocumentL10n.h
@@ -0,0 +1,88 @@
+/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* vim:set ts=2 sw=2 sts=2 et cindent: */
+/* 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_l10n_DocumentL10n_h
+#define mozilla_dom_l10n_DocumentL10n_h
+
+#include "mozilla/dom/DOMLocalization.h"
+
+class nsIContentSink;
+
+namespace mozilla::dom {
+
+class Document;
+
+enum class DocumentL10nState {
+  // State set when the DocumentL10n gets constructed.
+  Constructed = 0,
+
+  // State set when the initial translation got triggered. This happens
+  // if DocumentL10n was constructed during parsing of the document.
+  //
+  // If the DocumentL10n gets constructed later, we'll skip directly to
+  // Ready state.
+  InitialTranslationTriggered,
+
+  // State set the DocumentL10n has been fully initialized, potentially
+  // with initial translation being completed.
+  Ready,
+};
+
+/**
+ * This class maintains localization status of the document.
+ *
+ * The document will initialize it lazily when a link with a localization
+ * resource is added to the document.
+ *
+ * Once initialized, DocumentL10n relays all API methods to an
+ * instance of mozILocalization and maintains a single promise
+ * which gets resolved the first time the document gets translated.
+ */
+class DocumentL10n final : public DOMLocalization {
+ public:
+  NS_DECL_ISUPPORTS_INHERITED
+  NS_DECL_CYCLE_COLLECTION_CLASS_INHERITED(DocumentL10n, DOMLocalization)
+
+  static RefPtr<DocumentL10n> Create(Document* aDocument, bool aSync);
+
+ protected:
+  explicit DocumentL10n(Document* aDocument, bool aSync);
+  virtual ~DocumentL10n() = default;
+
+  RefPtr<Document> mDocument;
+  RefPtr<Promise> mReady;
+  DocumentL10nState mState;
+  nsCOMPtr<nsIContentSink> mContentSink;
+
+ public:
+  virtual JSObject* WrapObject(JSContext* aCx,
+                               JS::Handle<JSObject*> aGivenProto) override;
+
+  Promise* Ready();
+
+  void TriggerInitialTranslation();
+  already_AddRefed<Promise> TranslateDocument(ErrorResult& aRv);
+
+  void InitialTranslationCompleted(bool aL10nCached);
+
+  Document* GetDocument() const { return mDocument; };
+  void OnCreatePresShell();
+
+  void ConnectRoot(nsINode& aNode, bool aTranslate, ErrorResult& aRv);
+
+  DocumentL10nState GetState() { return mState; };
+
+  void MaybeRecordTelemetry();
+
+  bool mBlockingLayout = false;
+
+  mozilla::TimeStamp mInitialTranslationStart;
+  static bool mIsFirstBrowserWindow;
+};
+
+}  // namespace mozilla::dom
+
+#endif  // mozilla_dom_l10n_DocumentL10n_h