basename = plugin_basename( $entry_file ); $this->entry_file = $entry_file; $this->root_dir = dirname( $entry_file ); $this->slug = $slug; $this->options = new Options( $this->get_options() ); $this->container = Container::create(); } public function init(): void { if ( is_admin() ) { new Admin( $this->slug ); } elseif ( $this->options->get( 'modules.chatbot', false ) && class_exists( Chatbot::class ) ) { new Chatbot( $this, self::REST_NAMESPACE, $this->options, $this->get(SessionStore::class), $this->slug ); } if ( class_exists( CommentHelper::class ) ) { $commentHelper = new CommentHelper(); $commentHelper->init($this->get( \AiBuddy\AiContentGenerator::class )); } $notice = new Notice(); $notice->initNotice(); } public function get_options(): array { return (array) get_option( $this->slug, array() ); } public function update_options( array $options ): bool { $options = array_replace_recursive( $this->get_options(), $options ); return update_option( $this->slug, $options, false ); } public function update_option( string $option, $value ): bool { $options = $this->get_options(); $options[ $option ] = $value; return $this->update_options( $options ); } /** * @param string $option * @param mixed $default * * @return mixed */ public function get_option( string $option, $default = null ) { $options = $this->get_options(); return $options[ $option ] ?? $default; } /** * @template T * @psalm-param class-string $key * @return T */ public function get( string $key ) { return $this->container->get( $key ); } public function get_polished_post( $post ) { if ( is_object( $post ) ) { $post = (array) $post; } $content = apply_filters( 'ai_buddy_pre_post_content', $post['post_content'], $post['ID'] ); $content = $this->polish_text( $content ); $content = apply_filters( 'ai_buddy_post_content', $content, $post['ID'] ); $title = $post['post_title']; $excerpt = $post['post_excerpt']; $url = get_permalink( $post['ID'] ); $checksum = wp_hash( $content . $title . $url ); return array( 'postId' => $post['ID'], 'title' => $title, 'content' => $content, 'excerpt' => $excerpt, 'url' => $url, 'checksum' => $checksum, ); } public function polish_text( string $raw_text ): string { $text = html_entity_decode( $raw_text ); $text = wp_strip_all_tags( $text ); $text = preg_replace( '/[\r\n]+/', "\n", $text ); return $text . ' '; } public function remove_duplicate_sentences( string $text, $max_tokens = 512 ): string { $sentences = preg_split( '/(?<=[.?!。．！？])+/u', $text ); $hashes = array(); $unique_sentences = array(); $length = 0; foreach ( $sentences as $sentence ) { $sentence = preg_replace( '/^[\pZ\pC]+|[\pZ\pC]+$/u', '', $sentence ); $hash = md5( $sentence ); if ( ! in_array( $hash, $hashes ) ) { $tokens_count = apply_filters( 'ai_buddy_estimate_tokens', 0, $sentence ); if ( $length + $tokens_count > $max_tokens ) { continue; } $hashes[] = $hash; $unique_sentences[] = $sentence; $length += $tokens_count; } } $polished_text = implode( ' ', $unique_sentences ); $polished_text = preg_replace( '/^[\pZ\pC]+|[\pZ\pC]+$/u', '', $polished_text ); return $polished_text; } }