1 files changed, 62 insertions, 0 deletions
diff --git a/parser/html/nsHtml5Speculation.h b/parser/html/nsHtml5Speculation.h
new file mode 100644
index 0000000000..9e1b2c31f5
--- /dev/null
+++ b/parser/html/nsHtml5Speculation.h
@@ -0,0 +1,62 @@
+/* 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 nsHtml5Speculation_h
+#define nsHtml5Speculation_h
+
+#include "nsHtml5OwningUTF16Buffer.h"
+#include "nsAHtml5TreeBuilderState.h"
+#include "nsHtml5TreeOperation.h"
+#include "nsAHtml5TreeOpSink.h"
+#include "nsTArray.h"
+#include "mozilla/Attributes.h"
+#include "mozilla/UniquePtr.h"
+
+class nsHtml5Speculation final : public nsAHtml5TreeOpSink {
+ public:
+  nsHtml5Speculation(nsHtml5OwningUTF16Buffer* aBuffer, int32_t aStart,
+                     int32_t aStartLineNumber,
+                     nsAHtml5TreeBuilderState* aSnapshot);
+
+  ~nsHtml5Speculation();
+
+  nsHtml5OwningUTF16Buffer* GetBuffer() { return mBuffer; }
+
+  int32_t GetStart() { return mStart; }
+
+  int32_t GetStartLineNumber() { return mStartLineNumber; }
+
+  nsAHtml5TreeBuilderState* GetSnapshot() { return mSnapshot.get(); }
+
+  /**
+   * Flush the operations from the tree operations from the argument
+   * queue unconditionally.
+   */
+  [[nodiscard]] virtual bool MoveOpsFrom(
+      nsTArray<nsHtml5TreeOperation>& aOpQueue) override;
+
+  [[nodiscard]] bool FlushToSink(nsAHtml5TreeOpSink* aSink);
+
+ private:
+  /**
+   * The first buffer in the pending UTF-16 buffer queue
+   */
+  RefPtr<nsHtml5OwningUTF16Buffer> mBuffer;
+
+  /**
+   * The start index of this speculation in the first buffer
+   */
+  int32_t mStart;
+
+  /**
+   * The current line number at the start of the speculation
+   */
+  int32_t mStartLineNumber;
+
+  mozilla::UniquePtr<nsAHtml5TreeBuilderState> mSnapshot;
+
+  nsTArray<nsHtml5TreeOperation> mOpQueue;
+};
+
+#endif  // nsHtml5Speculation_h