v1.0.0
This commit is contained in:
commit
a157afe625
21 changed files with 4623 additions and 0 deletions
201
includes/class-ai-translator-translator.php
Normal file
201
includes/class-ai-translator-translator.php
Normal file
|
|
@ -0,0 +1,201 @@
|
|||
<?php
|
||||
/**
|
||||
* Translation engine for the AI Translator plugin.
|
||||
*
|
||||
* Wraps the WordPress AI client (https://github.com/WordPress/ai) so the rest of the
|
||||
* plugin does not need to know which AI provider is configured.
|
||||
*
|
||||
* @package ROBOTSTXT\AI_Translator
|
||||
* @since 1.0.0
|
||||
*/
|
||||
|
||||
if ( ! defined( 'ABSPATH' ) ) {
|
||||
exit;
|
||||
}
|
||||
|
||||
if ( ! class_exists( 'AI_Translator_Translator' ) ) {
|
||||
|
||||
/**
|
||||
* Talks to the WordPress AI client to translate strings.
|
||||
*
|
||||
* @since 1.0.0
|
||||
*/
|
||||
class AI_Translator_Translator {
|
||||
|
||||
/**
|
||||
* Indicates whether the WordPress AI client is available.
|
||||
*
|
||||
* @since 1.0.0
|
||||
*
|
||||
* @return bool True if the dependency is active.
|
||||
*/
|
||||
public function is_available() {
|
||||
return function_exists( 'wp_ai_client_prompt' );
|
||||
}
|
||||
|
||||
/**
|
||||
* Indicates whether the WordPress AI client supports text generation right now.
|
||||
*
|
||||
* The check uses a tiny dummy prompt so the AI client can answer whether at
|
||||
* least one configured provider can serve text generation requests.
|
||||
*
|
||||
* @since 1.0.0
|
||||
*
|
||||
* @return bool True if text generation is supported.
|
||||
*/
|
||||
public function is_supported() {
|
||||
if ( ! $this->is_available() ) {
|
||||
return false;
|
||||
}
|
||||
|
||||
try {
|
||||
$builder = wp_ai_client_prompt( 'test' );
|
||||
|
||||
return (bool) $builder->is_supported_for_text_generation();
|
||||
} catch ( Exception $e ) {
|
||||
return false;
|
||||
} catch ( Throwable $e ) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Translates a single piece of text to the target locale.
|
||||
*
|
||||
* @since 1.0.0
|
||||
*
|
||||
* @param string $text Source text. Plain text or HTML; preserved as-is.
|
||||
* @param string $target_locale WordPress locale code (for example: es_ES, en_US, fr_FR).
|
||||
*
|
||||
* @return string|WP_Error Translated text, or WP_Error on failure.
|
||||
*/
|
||||
public function translate( $text, $target_locale ) {
|
||||
$text = (string) $text;
|
||||
|
||||
if ( '' === trim( $text ) ) {
|
||||
return '';
|
||||
}
|
||||
|
||||
if ( ! $this->is_available() ) {
|
||||
return new WP_Error(
|
||||
'ai_translator_unavailable',
|
||||
__( 'The WordPress AI client is not available. Install and activate the WordPress AI plugin.', 'robotstxt-ai-translator' )
|
||||
);
|
||||
}
|
||||
|
||||
if ( ! $this->is_supported() ) {
|
||||
return new WP_Error(
|
||||
'ai_translator_unsupported',
|
||||
__( 'The configured AI providers cannot serve text generation requests. Open the WordPress AI settings to configure a provider.', 'robotstxt-ai-translator' )
|
||||
);
|
||||
}
|
||||
|
||||
$language_name = $this->locale_to_language_name( $target_locale );
|
||||
|
||||
if ( '' === $language_name ) {
|
||||
return new WP_Error(
|
||||
'ai_translator_invalid_locale',
|
||||
__( 'The target language is not installed on this site.', 'robotstxt-ai-translator' )
|
||||
);
|
||||
}
|
||||
|
||||
$system_instruction = sprintf(
|
||||
/* translators: %s: target language name. */
|
||||
__( 'You are a professional translator. Translate the user message into %s. Preserve any HTML, shortcodes, line breaks, and Markdown exactly as they appear. Return only the translated text, without explanations, prefaces, or quotation marks.', 'robotstxt-ai-translator' ),
|
||||
$language_name
|
||||
);
|
||||
|
||||
try {
|
||||
$result = wp_ai_client_prompt( $text )
|
||||
->using_system_instruction( $system_instruction )
|
||||
->generate_text();
|
||||
} catch ( Exception $e ) {
|
||||
return new WP_Error( 'ai_translator_exception', $e->getMessage() );
|
||||
} catch ( Throwable $e ) {
|
||||
return new WP_Error( 'ai_translator_exception', $e->getMessage() );
|
||||
}
|
||||
|
||||
if ( is_wp_error( $result ) ) {
|
||||
return $result;
|
||||
}
|
||||
|
||||
if ( ! is_string( $result ) ) {
|
||||
return new WP_Error(
|
||||
'ai_translator_invalid_response',
|
||||
__( 'The AI service returned an unexpected response.', 'robotstxt-ai-translator' )
|
||||
);
|
||||
}
|
||||
|
||||
return trim( $result );
|
||||
}
|
||||
|
||||
/**
|
||||
* Converts a WordPress locale code into a human-readable language name.
|
||||
*
|
||||
* Only locales installed on the site (en_US plus any installed .po files) are
|
||||
* accepted, so the editor cannot request translations to languages that the
|
||||
* site has not declared as available.
|
||||
*
|
||||
* @since 1.0.0
|
||||
*
|
||||
* @param string $locale WordPress locale code.
|
||||
*
|
||||
* @return string Native language name, or empty string when the locale is not installed.
|
||||
*/
|
||||
public function locale_to_language_name( $locale ) {
|
||||
$locale = (string) $locale;
|
||||
|
||||
if ( '' === $locale ) {
|
||||
return '';
|
||||
}
|
||||
|
||||
$installed = $this->get_installed_locales();
|
||||
|
||||
if ( ! in_array( $locale, $installed, true ) ) {
|
||||
return '';
|
||||
}
|
||||
|
||||
if ( 'en_US' === $locale ) {
|
||||
return 'English (United States)';
|
||||
}
|
||||
|
||||
if ( ! function_exists( 'wp_get_available_translations' ) ) {
|
||||
require_once ABSPATH . 'wp-admin/includes/translation-install.php';
|
||||
}
|
||||
|
||||
if ( function_exists( 'wp_get_available_translations' ) ) {
|
||||
$translations = wp_get_available_translations();
|
||||
if ( isset( $translations[ $locale ]['native_name'] ) && '' !== $translations[ $locale ]['native_name'] ) {
|
||||
return (string) $translations[ $locale ]['native_name'];
|
||||
}
|
||||
if ( isset( $translations[ $locale ]['english_name'] ) && '' !== $translations[ $locale ]['english_name'] ) {
|
||||
return (string) $translations[ $locale ]['english_name'];
|
||||
}
|
||||
}
|
||||
|
||||
return $locale;
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns the locales installed on the site, including en_US.
|
||||
*
|
||||
* @since 1.0.0
|
||||
*
|
||||
* @return array<int,string> List of locale codes.
|
||||
*/
|
||||
public function get_installed_locales() {
|
||||
$locales = array( 'en_US' );
|
||||
|
||||
if ( function_exists( 'get_available_languages' ) ) {
|
||||
$installed = get_available_languages();
|
||||
if ( is_array( $installed ) ) {
|
||||
foreach ( $installed as $locale ) {
|
||||
$locales[] = (string) $locale;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return array_values( array_unique( $locales ) );
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in a new issue