1. 1
2. 2
3. 3
4. 4
5. 5
6. 6
7. 7
8. 8
9. 9
10. 10
11. 11
12. 12
13. 13
14. 14
15. 15
16. 16
17. 17
18. 18
19. 19
20. 20
21. 21
22. 22
23. 23
24. 24
25. 25
26. 26
27. 27
28. 28
29. 29
30. 30
31. 31
32. 32
33. 33
34. 34
35. 35
36. 36
37. 37
38. 38
39. 39
40. 40
41. 41
42. 42
43. 43
44. 44
45. 45
46. 46
47. 47
48. 48
49. 49
50. 50
51. 51
52. 52
53. 53
54. 54
55. 55
56. 56
57. 57
58. 58
59. 59
60. 60
61. 61
62. 62
63. 63
64. 64
65. 65
66. 66
67. 67
68. 68
69. 69
70. 70
71. 71
72. 72
73. 73
74. 74
75. 75
76. 76
77. 77
78. 78
79. 79
80. 80
81. 81
82. 82
83. 83
84. 84
85. 85
86. 86
87. 87
88. 88
89. 89
90. 90
91. 91
92. 92
93. 93
94. 94
95. 95
96. 96
97. 97
98. 98
99. 99
100. 100
101. 101
102. 102
103. 103
104. 104
105. 105
106. 106
107. 107
108. 108
109. 109
110. 110
111. 111
112. 112
113. 113
114. 114
115. 115
116. 116
117. 117
118. 118
119. 119
120. 120
121. 121

// SPDX-FileCopyrightText: 2024 Shota FUJI <pockawoooh@gmail.com>
//
// SPDX-License-Identifier: Apache-2.0

import type * as Mdast from "../../deps/esm.sh/mdast/types.ts";
import { SKIP, visit } from "../../deps/esm.sh/unist-util-visit/mod.ts";
import { definitions } from "../../deps/esm.sh/mdast-util-definitions/mod.ts";

import type { ParseParameters } from "../interface.ts";
import type { DocumentToken } from "../../types.ts";

import type { OfmWikilink } from "./mdast_util_ofm_wikilink.ts";

const SEPARATOR = "/";
const FRAGMENT_PREFIX = "#";
const IGNORE_REGEXP_PATTERN = /^([a-z0-9]+:\/\/|#)/i;

function setDocumentToken(node: Mdast.Node, token: DocumentToken): void {
	node.data ??= {};
	// @ts-expect-error: incorrect library type definition.
	node.data.macanaDocumentToken = token;
}

function parseInternalLink(
	link: string,
): { path: readonly string[]; fragments: readonly string[] } {
	const [ref, ...fragments] = link.split(FRAGMENT_PREFIX);

	return {
		path: (ref || "").split(SEPARATOR),
		fragments: fragments,
	};
}

/**
 * Searches document references and Marks thoese node by setting `macanaDocumentToken`
 * with Document Token.
 *
 * This function mutates the Mdast tree in place.
 */
export async function macanaMarkDocumentToken(
	tree: Mdast.Nodes | OfmWikilink,
	getDocumentToken: ParseParameters["getDocumentToken"],
): Promise<void> {
	const promises: Promise<unknown>[] = [];

	const defs = definitions(tree as Mdast.Nodes);

	visit(
		tree,
		(node) =>
			node.type === "link" || node.type === "linkReference" ||
			node.type === "ofmWikilink",
		(node) => {
			switch (node.type) {
				case "ofmWikilink": {
					const { path, fragments } = parseInternalLink(node.target);

					const token = getDocumentToken(path, fragments);

					if (token instanceof Promise) {
						promises.push(token.then((t) => {
							setDocumentToken(node, t);
						}));
						return SKIP;
					}

					setDocumentToken(node, token);
					return SKIP;
				}
				case "link": {
					if (IGNORE_REGEXP_PATTERN.test(node.url)) {
						return SKIP;
					}

					const { path, fragments } = parseInternalLink(node.url);

					const token = getDocumentToken(path, fragments);

					if (token instanceof Promise) {
						promises.push(token.then((t) => {
							setDocumentToken(node, t);
						}));
						return SKIP;
					}

					setDocumentToken(node, token);
					return SKIP;
				}
				case "linkReference": {
					const def = defs(node.identifier);
					if (!def) {
						return;
					}

					if (IGNORE_REGEXP_PATTERN.test(def.url)) {
						return SKIP;
					}

					const { path, fragments } = parseInternalLink(def.url);

					const token = getDocumentToken(path, fragments);

					if (token instanceof Promise) {
						promises.push(token.then((t) => {
							setDocumentToken(node, t);
						}));
						return SKIP;
					}

					setDocumentToken(node, token);
					return SKIP;
				}
			}
		},
	);

	if (promises.length > 0) {
		await Promise.all(promises);
	}
}