mirror of
https://github.com/discourse/discourse.git
synced 2025-10-03 17:21:20 +08:00
e46fe3af1e
This change ensures that oneboxes are correctly generated for localized posts. The `PostLocalizer` was using `PrettyText.cook` directly, which does not perform all the necessary post-processing steps, such as onebox generation. This commit introduces a new `LocalizedCookedPostProcessor` class that is responsible for post-processing the cooked HTML of translated posts. This new class reuses the `CookedProcessorMixin` to gain access to the `post_process_oneboxes` method. The `PostLocalizer` is updated to use this new processor, ensuring that oneboxes are correctly generated in the translated content.
64 lines
2 KiB
Ruby
64 lines
2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require_relative "localized_cooked_post_processor"
|
|
|
|
module DiscourseAi
|
|
module Translation
|
|
class PostLocalizer
|
|
def self.localize(post, target_locale = I18n.locale)
|
|
if post.blank? || target_locale.blank? ||
|
|
LocaleNormalizer.is_same?(post.locale, target_locale) || post.raw.blank?
|
|
return
|
|
end
|
|
return if post.raw.length > SiteSetting.ai_translation_max_post_length
|
|
target_locale = target_locale.to_s.sub("-", "_")
|
|
|
|
translated_raw = PostRawTranslator.new(text: post.raw, target_locale:, post:).translate
|
|
|
|
localization =
|
|
PostLocalization.find_or_initialize_by(post_id: post.id, locale: target_locale)
|
|
|
|
localization.raw = translated_raw
|
|
localization.cooked = post.post_analyzer.cook(translated_raw, post.cooking_options || {})
|
|
|
|
cooked_processor = LocalizedCookedPostProcessor.new(localization, post, {})
|
|
cooked_processor.post_process
|
|
localization.cooked = cooked_processor.html
|
|
|
|
localization.post_version = post.version
|
|
localization.localizer_user_id = Discourse.system_user.id
|
|
localization.save!
|
|
localization
|
|
end
|
|
|
|
def self.has_relocalize_quota?(post, locale, skip_incr: false)
|
|
return false if get_relocalize_quota(post, locale).to_i >= 2
|
|
|
|
incr_relocalize_quota(post, locale) unless skip_incr
|
|
true
|
|
end
|
|
|
|
private
|
|
|
|
def self.relocalize_key(post, locale)
|
|
"post_relocalized_#{post.id}_#{locale}"
|
|
end
|
|
|
|
def self.get_relocalize_quota(post, locale)
|
|
Discourse.redis.get(relocalize_key(post, locale)).to_i || 0
|
|
end
|
|
|
|
def self.incr_relocalize_quota(post, locale)
|
|
key = relocalize_key(post, locale)
|
|
|
|
if (count = get_relocalize_quota(post, locale)).zero?
|
|
Discourse.redis.set(key, 1, ex: 1.day.to_i)
|
|
else
|
|
ttl = Discourse.redis.ttl(key)
|
|
incr = count.to_i + 1
|
|
Discourse.redis.set(key, incr, ex: ttl)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
