commit e743bd053eb727b77dd061f7203c562bbf2c37ec · kitten.sh/rn-llms-txt-gen

-7

src/types.d.ts

···

       1
       1
       -
       declare module 'remark-title' {

     

       2
       2
       -
         interface TitleOptions {

     

       3
       3
       -
           title: string;

     

       4
       4
       -
         }

     

       5
       5
       -
         export default function remarkTitle(opts: TitleOptions): (tree: Root) => undefined;

     

       6
       6
       -
         export type Root = import('mdast').Root;

     

       7
       7
       -
       }

+51 -37

src/unified.ts

···

       1
       1
       -
       import type { List, ListItem, PhrasingContent, Root } from 'mdast';

     

       1
       1
       +
       import type { Heading, List, ListItem, PhrasingContent, Root } from 'mdast';

     

       2
       2
        
       import { unified } from 'unified';

     

       3
       3
        
       import { visit } from 'unist-util-visit';

     

       4
       4
        
       import { defaultSchema as defaultSanitizeSchema } from 'hast-util-sanitize';

     
···

       12
       12
        
       import remarkGfm from 'remark-gfm';

     

       13
       13
        
       import remarkParse from 'remark-parse';

     

       14
       14
        
       import remarkSqueezeParagraphs from 'remark-squeeze-paragraphs';

     

       15
       15
       -
       import remarkTitle from 'remark-title';

     

       15
       15
       +
       

     

       16
       16
       +
       const toString = (nodes: PhrasingContent[]): string =>

     

       17
       17
       +
         nodes.map((node) => {

     

       18
       18
       +
           switch (node.type) {

     

       19
       19
       +
             case 'break':

     

       20
       20
       +
               return '\n';

     

       21
       21
       +
             case 'delete':

     

       22
       22
       +
             case 'emphasis':

     

       23
       23
       +
             case 'link':

     

       24
       24
       +
             case 'strong':

     

       25
       25
       +
               return toString(node.children);

     

       26
       26
       +
             case 'inlineCode':

     

       27
       27
       +
               return `\`${node.value}\``;

     

       28
       28
       +
             case 'text':

     

       29
       29
       +
               return node.value;

     

       30
       30
       +
             case 'footnoteReference':

     

       31
       31
       +
             case 'html':

     

       32
       32
       +
             case 'image':

     

       33
       33
       +
             case 'imageReference':

     

       34
       34
       +
             case 'linkReference':

     

       35
       35
       +
             default:

     

       36
       36
       +
               return '';

     

       37
       37
       +
           }

     

       38
       38
       +
         }).join('');

     

       16
       39
        
       

     

       17
       40
        
       export async function sanitizeHtml(html: string): Promise<string> {

     

       18
       41
        
         const vfile = await unified()

     
···

       161
       184
        
       }

     

       162
       185
        
       

     

       163
       186
        
       function extractTitle(markdown: string): string | null {

     

       164
       164
       -
         let depth: number | null = null;

     

       165
       165
       -
         let title: string | null = null;

     

       166
       166
       -
         const toString = (nodes: PhrasingContent[]): string =>

     

       167
       167
       -
           nodes.map((node) => {

     

       168
       168
       -
             switch (node.type) {

     

       169
       169
       -
               case 'break':

     

       170
       170
       -
                 return '\n';

     

       171
       171
       -
               case 'delete':

     

       172
       172
       -
               case 'emphasis':

     

       173
       173
       -
               case 'link':

     

       174
       174
       -
               case 'strong':

     

       175
       175
       -
                 return toString(node.children);

     

       176
       176
       -
               case 'inlineCode':

     

       177
       177
       -
                 return `\`${node.value}\``;

     

       178
       178
       -
               case 'text':

     

       179
       179
       -
                 return node.value;

     

       180
       180
       -
               case 'footnoteReference':

     

       181
       181
       -
               case 'html':

     

       182
       182
       -
               case 'image':

     

       183
       183
       -
               case 'imageReference':

     

       184
       184
       -
               case 'linkReference':

     

       185
       185
       -
               default:

     

       186
       186
       -
                 return '';

     

       187
       187
       -
             }

     

       188
       188
       -
           }).join('');

     

       189
       187
        
         const tree = unified()

     

       190
       188
        
           .use(remarkParse, { fragment: true })

     

       191
       189
        
           .use(remarkGfm, {

     
···

       193
       191
        
             tableCellPadding: false,

     

       194
       192
        
           })

     

       195
       193
        
           .parse(markdown);

     

       196
       196
       -
         visit(tree, function (node) {

     

       197
       197
       -
           if (node.type !== 'heading')

     

       198
       198
       -
             return;

     

       199
       199
       -
           if (!depth || node.depth < depth)

     

       200
       200
       -
             title = toString(node.children);

     

       201
       201
       -
         });

     

       202
       202
       -
         return title;

     

       194
       194
       +
         const node = tree.children[0];

     

       195
       195
       +
         if (node && node.type === 'heading' && node.depth === 1) {

     

       196
       196
       +
           return toString(node.children);

     

       197
       197
       +
         } else {

     

       198
       198
       +
           return null;

     

       199
       199
       +
         }

     

       200
       200
       +
       }

     

       201
       201
       +
       

     

       202
       202
       +
       function remarkTitle(opts: { title: string }) {

     

       203
       203
       +
         return function checkTitleTransformer(root: Root) {

     

       204
       204
       +
           const node = root.children[0]!;

     

       205
       205
       +
           const replacement: Heading = {

     

       206
       206
       +
             type: 'heading',

     

       207
       207
       +
             depth: 1,

     

       208
       208
       +
             children: [

     

       209
       209
       +
               { type: 'text', value: opts.title }

     

       210
       210
       +
             ]

     

       211
       211
       +
           };

     

       212
       212
       +
           if (node && node.type === 'heading') {

     

       213
       213
       +
             node.depth = 1;

     

       214
       214
       +
             node.children = replacement.children;

     

       215
       215
       +
           } else {

     

       216
       216
       +
             root.children?.unshift(replacement);

     

       217
       217
       +
           }

     

       218
       218
       +
         }

     

       203
       219
        
       }

     

       204
       220
        
       

     

       205
       221
        
       export async function transferTitle(from: string, to: string): Promise<string> {

     
···

       211
       227
        
             tablePipeAlign: false,

     

       212
       228
        
             tableCellPadding: false,

     

       213
       229
        
           })

     

       214
       214
       -
           .use(remarkTitle, {

     

       215
       215
       -
             title,

     

       216
       216
       -
           })

     

       230
       230
       +
           .use(remarkTitle, { title })

     

       217
       231
        
           .use(remarkStringify, {

     

       218
       232
        
             bullet: '-',

     

       219
       233
        
             incrementListMarker: false,