index.tsx
78 lines
| 1 | /** |
| 2 | * External dependencies |
| 3 | */ |
| 4 | import { createBlock, getSaveContent } from '@wordpress/blocks'; |
| 5 | import TurndownService from 'turndown'; |
| 6 | /** |
| 7 | * Internal dependencies |
| 8 | */ |
| 9 | import { blockName } from '..'; |
| 10 | import { EXTENDED_BLOCKS, isPossibleToExtendBlock } from '../extensions/ai-assistant'; |
| 11 | /** |
| 12 | * Types |
| 13 | */ |
| 14 | import type { ExtendedBlockProp } from '../extensions/ai-assistant'; |
| 15 | import type { PromptItemProps } from '../lib/prompt'; |
| 16 | |
| 17 | const turndownService = new TurndownService( { emDelimiter: '_', headingStyle: 'atx' } ); |
| 18 | |
| 19 | const from = []; |
| 20 | |
| 21 | /** |
| 22 | * Return an AI Assistant block instance from a given block type. |
| 23 | * |
| 24 | * @param {ExtendedBlockProp} blockType - Block type. |
| 25 | * @param {object} attrs - Block attributes. |
| 26 | * @returns {object} AI Assistant block instance. |
| 27 | */ |
| 28 | export function transformToAIAssistantBlock( blockType: ExtendedBlockProp, attrs ) { |
| 29 | const { content, ...restAttrs } = attrs; |
| 30 | let htmlContent = content; |
| 31 | |
| 32 | // core/heading custom transform handling. |
| 33 | if ( blockType === 'core/heading' && attrs?.level ) { |
| 34 | // Replace the HTML tags with the block level. |
| 35 | htmlContent = htmlContent.replace( /<(\/?)h\d([^>]*)>/g, `<$1h${ attrs.level }$2>` ); |
| 36 | } |
| 37 | |
| 38 | // Convert the content to markdown. |
| 39 | const aiAssistantBlockcontent = turndownService.turndown( htmlContent ); |
| 40 | |
| 41 | // Create a pair of user/assistant messages. |
| 42 | const messages: Array< PromptItemProps > = [ |
| 43 | { |
| 44 | role: 'user', |
| 45 | content: 'Tell me some content for this block, please.', |
| 46 | }, |
| 47 | { |
| 48 | role: 'assistant', |
| 49 | content: aiAssistantBlockcontent, |
| 50 | }, |
| 51 | ]; |
| 52 | |
| 53 | return createBlock( blockName, { |
| 54 | ...restAttrs, |
| 55 | content: aiAssistantBlockcontent, |
| 56 | originalContent: aiAssistantBlockcontent, |
| 57 | messages, |
| 58 | originalMessages: messages, |
| 59 | } ); |
| 60 | } |
| 61 | |
| 62 | /* |
| 63 | * Create individual transform handler for each block type. |
| 64 | */ |
| 65 | for ( const blockType of EXTENDED_BLOCKS ) { |
| 66 | from.push( { |
| 67 | type: 'block', |
| 68 | blocks: [ blockType ], |
| 69 | isMatch: () => isPossibleToExtendBlock(), |
| 70 | transform: ( attrs, innerBlocks ) => { |
| 71 | const content = getSaveContent( blockType, attrs, innerBlocks ); |
| 72 | return transformToAIAssistantBlock( blockType, { ...attrs, content } ); |
| 73 | }, |
| 74 | } ); |
| 75 | } |
| 76 | |
| 77 | export default { from }; |
| 78 |