From 1aa4b48d4b87e40fb99fc59f3893d620f36afe91 Mon Sep 17 00:00:00 2001
From: jaywcjlove <398188662@qq.com>
Date: Thu, 5 Sep 2024 14:37:22 +0800
Subject: [PATCH] feat: add callback support for test parameter.

---
 src/index.ts       | 10 +++++++---
 test/index.test.ts | 32 ++++++++++++++++++++++++++++++++
 2 files changed, 39 insertions(+), 3 deletions(-)

diff --git a/src/index.ts b/src/index.ts
index 33de8d4..691ff6d 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -9,7 +9,7 @@ export type RehypeVideoOptions = {
    * URL suffix verification.
    * @default /\/(.*)(.mp4|.mov)$/
    */
-  test?: RegExp;
+  test?: RegExp | ((url: string) => boolean);
   /**
    * Support `<details>` tag to wrap <video>.
    * @default true
@@ -65,10 +65,14 @@ function reElement(node: Element, details: boolean, track: boolean, href: string
 }
 
 const RehypeVideo: Plugin<[RehypeVideoOptions?], Root> = (options) => {
-  const { test = /\/(.*)(.mp4|.mov)$/, details = true, track = true } = options || {};
+  const { test = /\/(.*)(.mp4|.mov)$/i, details = true, track = true } = options || {};
   return (tree) => {
     visit(tree, 'element', (node, index, parent) => {
-      const isChecked = (str: string) => test.test(str.replace(/(\?|!|\#|$).+/g, '').toLocaleLowerCase())
+      const isChecked = (str: string) => {
+        if (test instanceof RegExp) return test.test(str.replace(/(\?|!|\#|$).+/g, '').toLocaleLowerCase())
+        if (typeof test === 'function') return test(str)
+        return false;
+      }
       const child = node.children[0];
 
       if (node.tagName === 'p' && node.children.length === 1) {
diff --git a/test/index.test.ts b/test/index.test.ts
index 754f122..f0a3161 100644
--- a/test/index.test.ts
+++ b/test/index.test.ts
@@ -6,6 +6,37 @@ import remarkParse from 'remark-parse';
 import stringify from 'rehype-stringify';
 import rehypeVideo from '../src/index.js';
 
+it('rehype-video `test:((url: string) => boolean)` test case', () => {
+  const mrkStr = `https://github.com/user-attachments/assets/0d808e2e-84c7-46ca-a220-440fa9f34118?title=rehype-video&rehype=video`;
+  const expected = `<details open class=\"octicon octicon-video\"><summary><svg aria-hidden height=\"16\" width=\"16\" viewBox=\"0 0 16 16\" version=\"1.1\" data-view-component class=\"octicon octicon-device-camera-video\"><path fill-rule=\"evenodd\" d=\"M16 3.75a.75.75 0 00-1.136-.643L11 5.425V4.75A1.75 1.75 0 009.25 3h-7.5A1.75 1.75 0 000 4.75v6.5C0 12.216.784 13 1.75 13h7.5A1.75 1.75 0 0011 11.25v-.675l3.864 2.318A.75.75 0 0016 12.25v-8.5zm-5 5.075l3.5 2.1v-5.85l-3.5 2.1v1.65zM9.5 6.75v-2a.25.25 0 00-.25-.25h-7.5a.25.25 0 00-.25.25v6.5c0 .138.112.25.25.25h7.5a.25.25 0 00.25-.25v-4.5z\"></path></svg><span aria-label=\"Video description rehype-video\">rehype-video</span><span class=\"dropdown-caret\"></span></summary><video muted controls style=\"max-height:640px;\" src=\"https://github.com/user-attachments/assets/0d808e2e-84c7-46ca-a220-440fa9f34118?title=rehype-video&#x26;rehype=video\"></video></details>`
+  const htmlStr = unified()
+    .use(remarkParse)
+    .use(remark2rehype, { allowDangerousHtml: true })
+    .use(rehypeVideo, {
+      test: (url) => {
+        return /\.(mp4|mov)|[?&]rehype=video/i.test(url);
+      }
+    })
+    .use(stringify)
+    .processSync(mrkStr)
+    .toString();
+  expect(htmlStr).toEqual(expected);
+});
+
+it('rehype-video `test:((url: string) => boolean)` boundary test case', () => {
+  const mrkStr = `https://github.com/user-attachments/assets/0d808e2e-84c7-46ca-a220-440fa9f34118?title=rehype-video&rehype=video`;
+  const htmlStr2 = unified()
+    .use(remarkParse)
+    .use(remark2rehype, { allowDangerousHtml: true })
+    // This is a boundary test, not the correct `rehypeVideo` option.
+    // @ts-ignore
+    .use(rehypeVideo, { test: false })
+    .use(stringify)
+    .processSync(mrkStr)
+    .toString();
+  expect(htmlStr2).toEqual("<p>https://github.com/user-attachments/assets/0d808e2e-84c7-46ca-a220-440fa9f34118?title=rehype-video&#x26;rehype=video</p>");
+});
+
 it('rehype-video title test case', () => {
   const mrkStr = `https://github.com/002.mp4?title=rehype-video`;
   const expected = `<details open class=\"octicon octicon-video\"><summary><svg aria-hidden height=\"16\" width=\"16\" viewBox=\"0 0 16 16\" version=\"1.1\" data-view-component class=\"octicon octicon-device-camera-video\"><path fill-rule=\"evenodd\" d=\"M16 3.75a.75.75 0 00-1.136-.643L11 5.425V4.75A1.75 1.75 0 009.25 3h-7.5A1.75 1.75 0 000 4.75v6.5C0 12.216.784 13 1.75 13h7.5A1.75 1.75 0 0011 11.25v-.675l3.864 2.318A.75.75 0 0016 12.25v-8.5zm-5 5.075l3.5 2.1v-5.85l-3.5 2.1v1.65zM9.5 6.75v-2a.25.25 0 00-.25-.25h-7.5a.25.25 0 00-.25.25v6.5c0 .138.112.25.25.25h7.5a.25.25 0 00.25-.25v-4.5z\"></path></svg><span aria-label=\"Video description rehype-video\">rehype-video</span><span class=\"dropdown-caret\"></span></summary><video muted controls style=\"max-height:640px;\" src=\"https://github.com/002.mp4?title=rehype-video\"></video></details>`
@@ -19,6 +50,7 @@ it('rehype-video title test case', () => {
     expect(htmlStr).toEqual(expected);
 });
 
+
 it('rehype-video 1 test case', () => {
   const mrkStr = `
   https://github.com/001.mp4 hi!