@diplodoc/transform 4.64.1 → 4.65.0

package/src/transform/utils.ts

@@ -74,7 +74,7 @@ export function headingInfo(tokens: Token[], idx: number) {
 }
 
 export function isExternalHref(href: string) {
-    return href.startsWith('http') || href.startsWith('//');
+    return /^(\w{1,10}:)?\/\//.test(href) || /^([+\w]{1,10}:)/.test(href);
 }
 
 export function transformLinkToOriginalArticle(opts: {root: string; currentPath: string}) {
@@ -113,3 +113,40 @@ export function defaultTransformLink(href: string) {
 
     return href;
 }
+
+type TokenWalker = (
+    token: Token,
+    state: {commented: boolean; index: number},
+) => void | undefined | {skip: number};
+
+export function filterTokens(tokens: Token[], type: string, handler: TokenWalker) {
+    let commented = false;
+
+    if (!tokens || !tokens.length) {
+        return;
+    }
+
+    for (let index = 0; index < tokens.length; index++) {
+        const token = tokens[index];
+
+        if (token.type === 'html_block') {
+            const commentStart = token.content.match('<!--');
+            const commentEnd = token.content.match('-->');
+
+            if (commentStart && !commentEnd) {
+                commented = true;
+            }
+
+            if (!commentStart && commentEnd) {
+                commented = false;
+            }
+        }
+
+        if (token.type === type) {
+            const result = handler(token, {commented, index});
+            if (result?.skip) {
+                index += result?.skip - 1;
+            }
+        }
+    }
+}