refactor addLineNumberAttribute

packages/app/src/services/renderer/rehype-plugins/add-line-number-attribute.ts

@@ -1,10 +1,11 @@
+import { Schema as SanitizeOption } from 'hast-util-sanitize';
 import { Element } from 'hast-util-select';
 import { Plugin } from 'unified';
 import { visit } from 'unist-util-visit';
 
 const REGEXP_TARGET_TAGNAMES = new RegExp(/h1|h2|h3|h4|h5|h6|p|img|pre|blockquote|hr|ol|ul/);
 
-export const addLineNumberAttribute: Plugin = () => {
+export const rehypePlugin: Plugin = () => {
   return (tree) => {
     visit(tree, 'element', (node: Element) => {
       if (REGEXP_TARGET_TAGNAMES.test(node.tagName as string)) {
@@ -20,3 +21,9 @@ export const addLineNumberAttribute: Plugin = () => {
     });
   };
 };
+
+export const sanitizeOption: SanitizeOption = {
+  attributes: {
+    '*': ['data-line'],
+  },
+};

packages/app/src/services/renderer/renderer.tsx

@@ -28,7 +28,7 @@ import { RendererConfig } from '~/interfaces/services/renderer';
 import loggerFactory from '~/utils/logger';
 
 import { addClass } from './rehype-plugins/add-class';
-import { addLineNumberAttribute } from './rehype-plugins/add-line-number-attribute';
+import * as addLineNumberAttribute from './rehype-plugins/add-line-number-attribute';
 import { relativeLinks } from './rehype-plugins/relative-links';
 import { relativeLinksByPukiwikiLikeLinker } from './rehype-plugins/relative-links-by-pukiwiki-like-linker';
 import { pukiwikiLikeLinker } from './remark-plugins/pukiwiki-like-linker';
@@ -439,7 +439,7 @@ export const generatePreviewOptions = (pagePath: string, config: RendererConfig)
   rehypePlugins.push(
     katex,
     [lsxGrowiPlugin.rehypePlugin, { pagePath }],
-    addLineNumberAttribute,
+    addLineNumberAttribute.rehypePlugin,
     // [autoLinkHeadings, {
     //   behavior: 'append',
     // }]
@@ -448,11 +448,7 @@ export const generatePreviewOptions = (pagePath: string, config: RendererConfig)
   const sanitizeOption = deepmerge(
     commonSanitizeOption,
     lsxGrowiPlugin.sanitizeOption,
-    {
-      attributes: {
-        '*': ['data-line'],
-      },
-    },
+    addLineNumberAttribute.sanitizeOption,
   );
   rehypePlugins.push([sanitize, sanitizeOption]);