@automattic/jetpack-ai-client 0.12.2 → 0.12.4

package/package.json

@@ -1,7 +1,7 @@
 {
 	"private": false,
 	"name": "@automattic/jetpack-ai-client",
-	"version": "0.12.2",
+	"version": "0.12.4",
 	"description": "A JS client for consuming Jetpack AI services",
 	"homepage": "https://github.com/Automattic/jetpack/tree/HEAD/projects/js-packages/ai-client/#readme",
 	"bugs": {
@@ -25,6 +25,7 @@
 	"devDependencies": {
 		"@storybook/addon-actions": "8.0.6",
 		"@storybook/blocks": "8.0.6",
+		"@storybook/preview-api": "8.0.6",
 		"@storybook/react": "8.0.6",
 		"@types/markdown-it": "14.0.0",
 		"@types/turndown": "5.0.4",
@@ -41,7 +42,7 @@
 	"main": "./build/index.js",
 	"types": "./build/index.d.ts",
 	"dependencies": {
-		"@automattic/jetpack-base-styles": "^0.6.22",
+		"@automattic/jetpack-base-styles": "^0.6.23",
 		"@automattic/jetpack-connection": "^0.33.8",
 		"@automattic/jetpack-shared-extension-utils": "^0.14.10",
 		"@microsoft/fetch-event-source": "2.0.1",

package/src/components/ai-control/ai-control.tsx

@@ -0,0 +1,79 @@
+/**
+ * External dependencies
+ */
+import { PlainText } from '@wordpress/block-editor';
+import classNames from 'classnames';
+import React from 'react';
+/**
+ * Internal dependencies
+ */
+import AiStatusIndicator from '../ai-status-indicator/index.js';
+import './style.scss';
+/**
+ * Types
+ */
+import type { RequestingStateProp } from '../../types.js';
+import type { ReactElement } from 'react';
+
+type AIControlProps = {
+	disabled?: boolean;
+	value: string;
+	placeholder?: string;
+	isTransparent?: boolean;
+	state?: RequestingStateProp;
+	onChange?: ( newValue: string ) => void;
+	banner?: ReactElement;
+	error?: ReactElement;
+	actions?: ReactElement;
+	message?: ReactElement;
+	promptUserInputRef?: React.MutableRefObject< HTMLInputElement >;
+};
+
+/**
+ * Base AIControl component. Contains the main structure of the control component and slots for banner, error, actions and message.
+ *
+ * @param {AIControlProps} props - Component props
+ * @returns {ReactElement}       Rendered component
+ */
+export default function AIControl( {
+	disabled = false,
+	value = '',
+	placeholder = '',
+	isTransparent = false,
+	state = 'init',
+	onChange,
+	banner = null,
+	error = null,
+	actions = null,
+	message = null,
+	promptUserInputRef = null,
+}: AIControlProps ): ReactElement {
+	return (
+		<div className="jetpack-components-ai-control__container-wrapper">
+			{ error }
+			<div className="jetpack-components-ai-control__container">
+				{ banner }
+				<div
+					className={ classNames( 'jetpack-components-ai-control__wrapper', {
+						'is-transparent': isTransparent,
+					} ) }
+				>
+					<AiStatusIndicator state={ state } />
+
+					<div className="jetpack-components-ai-control__input-wrapper">
+						<PlainText
+							value={ value }
+							onChange={ onChange }
+							placeholder={ placeholder }
+							className="jetpack-components-ai-control__input"
+							disabled={ disabled }
+							ref={ promptUserInputRef }
+						/>
+					</div>
+					{ actions }
+				</div>
+				{ message }
+			</div>
+		</div>
+	);
+}

package/src/components/ai-control/block-ai-control.tsx

@@ -0,0 +1,278 @@
+/**
+ * External dependencies
+ */
+import { Button, ButtonGroup } from '@wordpress/components';
+import { useKeyboardShortcut } from '@wordpress/compose';
+import { useImperativeHandle, useRef, useEffect, useCallback, useState } from '@wordpress/element';
+import { __ } from '@wordpress/i18n';
+import {
+	Icon,
+	closeSmall,
+	check,
+	arrowUp,
+	trash,
+	reusableBlock as regenerate,
+} from '@wordpress/icons';
+import debugFactory from 'debug';
+import React, { forwardRef } from 'react';
+/**
+ * Internal dependencies
+ */
+import { GuidelineMessage } from '../message/index.js';
+import AIControl from './ai-control.js';
+import './style.scss';
+/**
+ * Types
+ */
+import type { RequestingStateProp } from '../../types.js';
+import type { ReactElement } from 'react';
+
+type BlockAIControlProps = {
+	disabled?: boolean;
+	value: string;
+	placeholder?: string;
+	showAccept?: boolean;
+	acceptLabel?: string;
+	showButtonLabels?: boolean;
+	isTransparent?: boolean;
+	state?: RequestingStateProp;
+	showGuideLine?: boolean;
+	customFooter?: ReactElement;
+	onChange?: ( newValue: string ) => void;
+	onSend?: ( currentValue: string ) => void;
+	onStop?: () => void;
+	onAccept?: () => void;
+	onDiscard?: () => void;
+	showRemove?: boolean;
+	banner?: ReactElement;
+	error?: ReactElement;
+};
+
+const debug = debugFactory( 'jetpack-ai-client:block-ai-control' );
+
+/**
+ * BlockAIControl component. Used by the AI Assistant block, adding logic and components to the base AIControl component.
+ *
+ * @param {BlockAIControlProps} props  - Component props
+ * @param {React.MutableRefObject} ref - Ref to the component
+ * @returns {ReactElement}             Rendered component
+ */
+export function BlockAIControl(
+	{
+		disabled = false,
+		value = '',
+		placeholder = '',
+		showAccept = false,
+		acceptLabel = __( 'Accept', 'jetpack-ai-client' ),
+		showButtonLabels = true,
+		isTransparent = false,
+		state = 'init',
+		showGuideLine = false,
+		customFooter = null,
+		onChange,
+		onSend,
+		onStop,
+		onAccept,
+		onDiscard,
+		showRemove = false,
+		banner = null,
+		error = null,
+	}: BlockAIControlProps,
+	ref: React.MutableRefObject< HTMLInputElement >
+): ReactElement {
+	const loading = state === 'requesting' || state === 'suggesting';
+	const [ editRequest, setEditRequest ] = useState( false );
+	const [ lastValue, setLastValue ] = useState( value || null );
+	const promptUserInputRef = useRef( null );
+
+	// Pass the ref to forwardRef.
+	useImperativeHandle( ref, () => promptUserInputRef.current );
+
+	useEffect( () => {
+		if ( editRequest ) {
+			promptUserInputRef?.current?.focus();
+		}
+	}, [ editRequest ] );
+
+	const sendHandler = useCallback( () => {
+		setLastValue( value );
+		setEditRequest( false );
+		onSend?.( value );
+	}, [ value ] );
+
+	const changeHandler = useCallback(
+		( newValue: string ) => {
+			onChange?.( newValue );
+			if ( state === 'init' ) {
+				return;
+			}
+
+			if ( ! lastValue ) {
+				// here we're coming from a one-click action
+				setEditRequest( newValue.length > 0 );
+			} else {
+				// here we're coming from an edit action
+				setEditRequest( newValue !== lastValue );
+			}
+		},
+		[ lastValue, state ]
+	);
+
+	const discardHandler = useCallback( () => {
+		onDiscard?.();
+	}, [] );
+
+	const cancelEdit = useCallback( () => {
+		debug( 'cancelEdit, revert to last value', lastValue );
+		onChange?.( lastValue || '' );
+		setEditRequest( false );
+	}, [ lastValue ] );
+
+	useKeyboardShortcut(
+		'mod+enter',
+		() => {
+			if ( showAccept ) {
+				onAccept?.();
+			}
+		},
+		{
+			target: promptUserInputRef,
+		}
+	);
+
+	useKeyboardShortcut(
+		'enter',
+		e => {
+			e.preventDefault();
+			sendHandler();
+		},
+		{
+			target: promptUserInputRef,
+		}
+	);
+
+	const actions = (
+		<>
+			{ ( ! showAccept || editRequest ) && (
+				<div className="jetpack-components-ai-control__controls-prompt_button_wrapper">
+					{ ! loading ? (
+						<>
+							{ editRequest && (
+								<Button
+									className="jetpack-components-ai-control__controls-prompt_button"
+									onClick={ cancelEdit }
+									variant="secondary"
+									label={ __( 'Cancel', 'jetpack-ai-client' ) }
+								>
+									{ showButtonLabels ? (
+										__( 'Cancel', 'jetpack-ai-client' )
+									) : (
+										<Icon icon={ closeSmall } />
+									) }
+								</Button>
+							) }
+
+							{ showRemove && ! editRequest && ! value?.length && onDiscard && (
+								<Button
+									className="jetpack-components-ai-control__controls-prompt_button"
+									onClick={ discardHandler }
+									variant="secondary"
+									label={ __( 'Cancel', 'jetpack-ai-client' ) }
+								>
+									{ showButtonLabels ? (
+										__( 'Cancel', 'jetpack-ai-client' )
+									) : (
+										<Icon icon={ closeSmall } />
+									) }
+								</Button>
+							) }
+
+							{ value?.length > 0 && (
+								<Button
+									className="jetpack-components-ai-control__controls-prompt_button"
+									onClick={ sendHandler }
+									variant="primary"
+									disabled={ ! value?.length || disabled }
+									label={ __( 'Send request', 'jetpack-ai-client' ) }
+								>
+									{ showButtonLabels ? (
+										__( 'Generate', 'jetpack-ai-client' )
+									) : (
+										<Icon icon={ arrowUp } />
+									) }
+								</Button>
+							) }
+						</>
+					) : (
+						<Button
+							className="jetpack-components-ai-control__controls-prompt_button"
+							onClick={ onStop }
+							variant="secondary"
+							label={ __( 'Stop request', 'jetpack-ai-client' ) }
+						>
+							{ showButtonLabels ? (
+								__( 'Stop', 'jetpack-ai-client' )
+							) : (
+								<Icon icon={ closeSmall } />
+							) }
+						</Button>
+					) }
+				</div>
+			) }
+			{ showAccept && ! editRequest && (
+				<div className="jetpack-components-ai-control__controls-prompt_button_wrapper">
+					{ ( value?.length > 0 || lastValue === null ) && (
+						<ButtonGroup>
+							<Button
+								className="jetpack-components-ai-control__controls-prompt_button"
+								label={ __( 'Discard', 'jetpack-ai-client' ) }
+								onClick={ discardHandler }
+								tooltipPosition="top"
+							>
+								<Icon icon={ trash } />
+							</Button>
+							<Button
+								className="jetpack-components-ai-control__controls-prompt_button"
+								label={ __( 'Regenerate', 'jetpack-ai-client' ) }
+								onClick={ () => onSend?.( value ) }
+								tooltipPosition="top"
+								disabled={ ! value?.length || value === null || disabled }
+							>
+								<Icon icon={ regenerate } />
+							</Button>
+						</ButtonGroup>
+					) }
+					<Button
+						className="jetpack-components-ai-control__controls-prompt_button"
+						onClick={ onAccept }
+						variant="primary"
+						label={ acceptLabel }
+					>
+						{ showButtonLabels ? acceptLabel : <Icon icon={ check } /> }
+					</Button>
+				</div>
+			) }
+		</>
+	);
+
+	const message =
+		showGuideLine && ! loading && ! editRequest && ( customFooter || <GuidelineMessage /> );
+
+	return (
+		<AIControl
+			disabled={ disabled || loading }
+			value={ value }
+			placeholder={ placeholder }
+			isTransparent={ isTransparent }
+			state={ state }
+			onChange={ changeHandler }
+			banner={ banner }
+			error={ error }
+			actions={ actions }
+			message={ message }
+			promptUserInputRef={ promptUserInputRef }
+		/>
+	);
+}
+
+export default forwardRef( BlockAIControl );

package/src/components/ai-control/extension-ai-control.tsx

@@ -0,0 +1,217 @@
+/**
+ * External dependencies
+ */
+import { Button, ButtonGroup } from '@wordpress/components';
+import { useKeyboardShortcut } from '@wordpress/compose';
+import { useImperativeHandle, useRef, useEffect, useCallback, useState } from '@wordpress/element';
+import { __ } from '@wordpress/i18n';
+import { Icon, closeSmall, arrowUp, undo } from '@wordpress/icons';
+import React, { forwardRef } from 'react';
+/**
+ * Internal dependencies
+ */
+import { GuidelineMessage, ErrorMessage, UpgradeMessage } from '../message/index.js';
+import AIControl from './ai-control.js';
+import './style.scss';
+/**
+ * Types
+ */
+import type { RequestingStateProp } from '../../types.js';
+import type { ReactElement } from 'react';
+
+type ExtensionAIControlProps = {
+	disabled?: boolean;
+	value: string;
+	placeholder?: string;
+	showButtonLabels?: boolean;
+	isTransparent?: boolean;
+	state?: RequestingStateProp;
+	showGuideLine?: boolean;
+	error?: string;
+	requestsRemaining?: number;
+	showUpgradeMessage?: boolean;
+	onChange?: ( newValue: string ) => void;
+	onSend?: ( currentValue: string ) => void;
+	onStop?: () => void;
+	onClose?: () => void;
+	onUndo?: () => void;
+	onUpgrade?: () => void;
+};
+
+/**
+ * ExtensionAIControl component. Used by the AI Assistant inline extensions, adding logic and components to the base AIControl component.
+ *
+ * @param {ExtensionAIControlProps} props  - Component props
+ * @param {React.MutableRefObject} ref     - Ref to the component
+ * @returns {ReactElement}                 Rendered component
+ */
+export function ExtensionAIControl(
+	{
+		disabled = false,
+		value = '',
+		placeholder = '',
+		showButtonLabels = true,
+		isTransparent = false,
+		state = 'init',
+		showGuideLine = false,
+		error,
+		requestsRemaining,
+		showUpgradeMessage = false,
+		onChange,
+		onSend,
+		onStop,
+		onClose,
+		onUndo,
+		onUpgrade,
+	}: ExtensionAIControlProps,
+	ref: React.MutableRefObject< HTMLInputElement >
+): ReactElement {
+	const loading = state === 'requesting' || state === 'suggesting';
+	const [ editRequest, setEditRequest ] = useState( false );
+	const [ lastValue, setLastValue ] = useState( value || null );
+	const promptUserInputRef = useRef( null );
+
+	// Pass the ref to forwardRef.
+	useImperativeHandle( ref, () => promptUserInputRef.current );
+
+	useEffect( () => {
+		if ( editRequest ) {
+			promptUserInputRef?.current?.focus();
+		}
+	}, [ editRequest ] );
+
+	const sendHandler = useCallback( () => {
+		setLastValue( value );
+		setEditRequest( false );
+		onSend?.( value );
+	}, [ onSend, value ] );
+
+	const changeHandler = useCallback(
+		( newValue: string ) => {
+			onChange?.( newValue );
+			if ( state === 'init' ) {
+				return;
+			}
+
+			if ( ! lastValue ) {
+				// here we're coming from a one-click action
+				setEditRequest( newValue.length > 0 );
+			} else {
+				// here we're coming from an edit action
+				setEditRequest( newValue !== lastValue );
+			}
+		},
+		[ onChange, lastValue, state ]
+	);
+
+	const stopHandler = useCallback( () => {
+		onStop?.();
+	}, [ onStop ] );
+
+	const closeHandler = useCallback( () => {
+		onClose?.();
+	}, [ onClose ] );
+
+	const undoHandler = useCallback( () => {
+		onUndo?.();
+	}, [ onUndo ] );
+
+	const upgradeHandler = useCallback( () => {
+		onUpgrade?.();
+	}, [ onUpgrade ] );
+
+	useKeyboardShortcut(
+		'enter',
+		e => {
+			e.preventDefault();
+			sendHandler();
+		},
+		{
+			target: promptUserInputRef,
+		}
+	);
+
+	const actions = (
+		<>
+			{ loading ? (
+				<Button
+					className="jetpack-components-ai-control__controls-prompt_button"
+					onClick={ stopHandler }
+					variant="secondary"
+					label={ __( 'Stop request', 'jetpack-ai-client' ) }
+				>
+					{ showButtonLabels ? __( 'Stop', 'jetpack-ai-client' ) : <Icon icon={ closeSmall } /> }
+				</Button>
+			) : (
+				<>
+					{ value?.length > 0 && (
+						<div className="jetpack-components-ai-control__controls-prompt_button_wrapper">
+							<Button
+								className="jetpack-components-ai-control__controls-prompt_button"
+								onClick={ sendHandler }
+								variant="primary"
+								disabled={ ! value?.length || disabled }
+								label={ __( 'Send request', 'jetpack-ai-client' ) }
+							>
+								{ showButtonLabels ? (
+									__( 'Generate', 'jetpack-ai-client' )
+								) : (
+									<Icon icon={ arrowUp } />
+								) }
+							</Button>
+						</div>
+					) }
+					{ value?.length <= 0 && state === 'done' && (
+						<div className="jetpack-components-ai-control__controls-prompt_button_wrapper">
+							<ButtonGroup>
+								<Button
+									className="jetpack-components-ai-control__controls-prompt_button"
+									label={ __( 'Undo', 'jetpack-ai-client' ) }
+									onClick={ undoHandler }
+									tooltipPosition="top"
+								>
+									<Icon icon={ undo } />
+								</Button>
+								<Button
+									className="jetpack-components-ai-control__controls-prompt_button"
+									label={ __( 'Close', 'jetpack-ai-client' ) }
+									onClick={ closeHandler }
+									variant="tertiary"
+								>
+									{ __( 'Close', 'jetpack-ai-client' ) }
+								</Button>
+							</ButtonGroup>
+						</div>
+					) }
+				</>
+			) }
+		</>
+	);
+
+	let message = null;
+	if ( error ) {
+		message = <ErrorMessage error={ error } onTryAgainClick={ sendHandler } />;
+	} else if ( showUpgradeMessage ) {
+		message = (
+			<UpgradeMessage requestsRemaining={ requestsRemaining } onUpgradeClick={ upgradeHandler } />
+		);
+	} else if ( showGuideLine ) {
+		message = <GuidelineMessage />;
+	}
+
+	return (
+		<AIControl
+			disabled={ disabled || loading }
+			value={ value }
+			placeholder={ placeholder }
+			isTransparent={ isTransparent }
+			state={ state }
+			onChange={ changeHandler }
+			actions={ actions }
+			message={ message }
+			promptUserInputRef={ promptUserInputRef }
+		/>
+	);
+}
+
+export default forwardRef( ExtensionAIControl );