jcs.org
Serenity Operating System
1/*
2 * Copyright (c) 2021-2022, Matthew Olsson <mattco@serenityos.org>
3 *
4 * SPDX-License-Identifier: BSD-2-Clause
5 */
6
7#pragma once
8
9#include <AK/SourceLocation.h>
10#include <AK/WeakPtr.h>
11#include <LibPDF/Object.h>
12#include <LibPDF/Operator.h>
13#include <LibPDF/Reader.h>
14#include <LibPDF/XRefTable.h>
15
16namespace PDF {
17
18template<typename T, typename... Args>
19static NonnullRefPtr<T> make_object(Args... args)
20requires(IsBaseOf<Object, T>)
21{
22 return adopt_ref(*new T(forward<Args>(args)...));
23}
24
25class Document;
26
27class Parser {
28public:
29 static PDFErrorOr<Vector<Operator>> parse_operators(Document*, ReadonlyBytes);
30
31 Parser(ReadonlyBytes);
32 Parser(Document*, ReadonlyBytes);
33
34 void set_document(WeakPtr<Document> const&);
35
36 DeprecatedString parse_comment();
37
38 void move_by(size_t count) { m_reader.move_by(count); }
39 void move_to(size_t offset) { m_reader.move_to(offset); }
40
41 enum class CanBeIndirectValue {
42 No,
43 Yes
44 };
45
46 PDFErrorOr<Value> parse_value(CanBeIndirectValue = CanBeIndirectValue::Yes);
47 PDFErrorOr<Value> parse_possible_indirect_value_or_ref();
48 PDFErrorOr<NonnullRefPtr<IndirectValue>> parse_indirect_value(u32 index, u32 generation);
49 PDFErrorOr<NonnullRefPtr<IndirectValue>> parse_indirect_value();
50 PDFErrorOr<Value> parse_number();
51 PDFErrorOr<NonnullRefPtr<NameObject>> parse_name();
52 NonnullRefPtr<StringObject> parse_string();
53 DeprecatedString parse_literal_string();
54 DeprecatedString parse_hex_string();
55 PDFErrorOr<NonnullRefPtr<ArrayObject>> parse_array();
56 PDFErrorOr<NonnullRefPtr<DictObject>> parse_dict();
57 PDFErrorOr<NonnullRefPtr<StreamObject>> parse_stream(NonnullRefPtr<DictObject> dict);
58 PDFErrorOr<Vector<Operator>> parse_operators();
59
60protected:
61 void push_reference(Reference const& ref) { m_current_reference_stack.append(ref); }
62 void pop_reference() { m_current_reference_stack.take_last(); }
63
64 Error error(
65 DeprecatedString const& message
66#ifdef PDF_DEBUG
67 ,
68 SourceLocation loc = SourceLocation::current()
69#endif
70 ) const;
71
72 Reader m_reader;
73 WeakPtr<Document> m_document;
74 Vector<Reference> m_current_reference_stack;
75 bool m_disable_encryption { false };
76};
77
78};