fastlane-plugin-translate_gpt 0.1.6 → 0.1.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: be8140845fda90144346f59a44239c4baa066fc8b75f1afc26655783be3e3686
-  data.tar.gz: 0a39de422d2b3760c4678df1053a61799cd444434399d8f3f0f22a9dfcf261b0
+  metadata.gz: 5e16882ee4e890369a5c14b579f362303a3196e4238d4d3b3f685a9e6b4fb21a
+  data.tar.gz: 59719d66698e72c02ae4c625a68a9b5fe8d7dfc34023a5c329d0625bed27b2e9
 SHA512:
-  metadata.gz: a60eb486cf4a385386e429d85d6d53e833ff3afb80c8f60dae482a6596f55fce3218e6268669a4ad4f60e7868244ffaf62a772e2e481fd91e168e18d746f1dd2
-  data.tar.gz: e4a68b0fa334c61c03dc8fdb26fcab4a1310fa9ca877f57522e21d138ea754c937d9ce11263e9dde2d184a23f05d76fc91566c4994167847dd32f2c220bd69a2
+  metadata.gz: 23f15a4ca667b24e958ca6775cd6894dc43b85ecf3e705f4270526af2428a1e7a947824b0bbf74132dbc0fa55059edd9020f1dad10946f5e345ae2d69d7997ac
+  data.tar.gz: d94703743059519b4e3c17ea988a7591c8f67b981d3d38a075ed8fc801ce952509c146e3fb3f92207102680b490cefbc103f325da85c0f7c7e1c2e0281255e0b

data/lib/fastlane/plugin/translate_gpt/actions/translate_gpt_action.rb

@@ -84,7 +84,9 @@ module Fastlane
             description: "The path to the Localizable.strings file to be translated",
             verify_block: proc do |value|
               UI.user_error!("Invalid file path: #{value}") unless File.exist?(value)
-              UI.user_error!("Translation file must have .strings extension") unless File.extname(value) == ".strings"
+              extension = File.extname(value)
+              available_extensions = [".strings", ".xcstrings"]
+              UI.user_error!("Translation file must have any of these extensions: #{available_extensions}") unless available_extensions.include? extension
             end
           ),
           FastlaneCore::ConfigItem.new(
@@ -93,7 +95,9 @@ module Fastlane
             description: "Path to the translation file to update",
             verify_block: proc do |value|
               UI.user_error!("Invalid file path: #{value}") unless File.exist?(value)
-              UI.user_error!("Translation file must have .strings extension") unless File.extname(value) == ".strings"
+              extension = File.extname(value)
+              available_extensions = [".strings", ".xcstrings"]
+              UI.user_error!("Translation file must have any of these extensions: #{available_extensions}") unless available_extensions.include? extension
             end
           ),    
           FastlaneCore::ConfigItem.new(

data/lib/fastlane/plugin/translate_gpt/helper/translate_gpt_helper.rb

@@ -1,6 +1,7 @@
 require 'fastlane_core/ui/ui'
-require 'loco_strings'
+require 'loco_strings/parsers/xcstrings_file'
 require 'json'
+# rubocop:disable all
 
 module Fastlane
   UI = FastlaneCore::UI unless Fastlane.const_defined?("UI")
@@ -16,13 +17,50 @@ module Fastlane
         @timeout = params[:request_timeout]
       end
 
-      # Get the strings from a file
-      def prepare_hashes() 
+      def prepare_xcstrings() 
+        @xcfile = LocoStrings::XCStringsFile.new @params[:source_file]
+        @output_hash = {}
+        @to_translate = @xcfile.read
+        
+        if @params[:skip_translated] == true
+          @to_translate = @to_translate.reject { |k, original| 
+            !check_value_for_translate(
+              @xcfile.unit(k, @params[:target_language]),
+              original
+            )
+          }
+        end 
+      end
+
+      def check_value_for_translate(string, orignal_string)
+        return true unless string 
+        if string.is_a? LocoStrings::LocoString
+          return false if orignal_string.value.nil? || orignal_string.value.empty?
+          return string.value.empty?
+        elsif string.is_a? LocoStrings::LocoVariantions
+          orignal_string.strings.each do |key, _|
+            return true unless string.strings.has_key?(key)
+            return true if string.strings[key].value.empty?
+          end
+        end
+        return false
+      end
+
+      def prepare_strings() 
         @input_hash = get_strings(@params[:source_file])
         @output_hash = get_strings(@params[:target_file])
         @to_translate = filter_translated(@params[:skip_translated], @input_hash, @output_hash)
       end
 
+      # Get the strings from a file
+      def prepare_hashes() 
+        if File.extname(@params[:source_file]) == ".xcstrings"
+          prepare_xcstrings() 
+        else
+          prepare_strings() 
+        end
+      end
+
       # Log information about the input strings
       def log_input(bunch_size) 
         @translation_count = @to_translate.size
@@ -66,8 +104,13 @@ module Fastlane
       def translate_bunch_of_strings(bunch_size)
         bunch_index = 0
         number_of_bunches = (@translation_count / bunch_size.to_f).ceil
+        @keys_associations = {}
         @to_translate.each_slice(bunch_size) do |bunch|
           prompt = prepare_bunch_prompt bunch
+          if prompt.empty?
+            UI.important "Empty prompt, skipping bunch"
+            next
+          end
           max_retries = 10
           times_retried = 0
 
@@ -109,7 +152,7 @@ module Fastlane
       def prepare_bunch_prompt(strings)
         prompt = "I want you to act as a translator for a mobile application strings. " + \
             "Try to keep length of the translated text. " + \
-            "You need to response with a JSON only with the translation and nothing else until I say to stop it."
+            "You need to response with a JSON only with the translation and nothing else until I say to stop it. "
         if @params[:context] && !@params[:context].empty?
           prompt += "This app is #{@params[:context]}. "
         end
@@ -117,21 +160,46 @@ module Fastlane
 
         json_hash = []
         strings.each do |key, string|
+          UI.message "Translating #{key} - #{string}"
+          next if string.nil?
+
           string_hash = {}
           context = string.comment
-          if context && !context.empty?
-            string_hash["context"] = context
+          string_hash["context"] = context if context && !context.empty?
+
+          key = transform_string(string.key)
+          @keys_associations[key] = string.key
+          string_hash["key"] = key
+
+          if string.is_a? LocoStrings::LocoString
+            next if string.value.nil? || string.value.empty?
+            string_hash["string_to_translate"] = string.value
+          elsif string.is_a? LocoStrings::LocoVariantions
+            variants = {}
+            string.strings.each do |key, variant|
+              next if variant.nil? || variant.value.nil? || variant.value.empty?
+              variants[key] = variant.value
+            end
+            string_hash["strings_to_translate"] = variants
+          else 
+            UI.warning "Unknown type of string: #{string.key}"
           end
-          string_hash["key"] = string.key
-          string_hash["string_to_translate"] = string.value
           json_hash << string_hash
         end
+        return '' if json_hash.empty?
         prompt += "'''\n"
         prompt += json_hash.to_json
         prompt += "\n'''"
+        UI.message "Prompt: #{prompt}"
         return prompt
       end
 
+      def transform_string(input_string)
+        uppercased_string = input_string.upcase
+        escaped_string = uppercased_string.gsub(/[^0-9a-zA-Z]+/, '_')
+        return escaped_string
+      end
+
       # Request a translation from the GPT API
       def request_translate(key, string, prompt, index)
         response = @client.chat(
@@ -195,12 +263,21 @@ module Fastlane
             string_hash.delete("key")
             string_hash.delete("context")
             translated_string = string_hash.values.first
-            if key && !key.empty? && translated_string && !translated_string.empty?
-              UI.message "#{index_log} Translating #{key} - #{translated_string}"
-              string = LocoStrings::LocoString.new(key, translated_string, context)
-              @output_hash[key] = string
-              keys_to_translate.delete(key)
+            return unless key && !key.empty? 
+            real_key = @keys_associations[key]
+            if translated_string.is_a? Hash
+              strings = {}
+              translated_string.each do |pl_key, value|
+                UI.message "#{index_log} Translating #{real_key} > #{pl_key} - #{value}"
+                strings[pl_key] = LocoStrings::LocoString.new(pl_key, value, context)
+              end
+              string = LocoStrings::LocoVariantions.new(real_key, strings, context)
+            elsif translated_string && !translated_string.empty?
+              UI.message "#{index_log} Translating #{real_key} - #{translated_string}"
+              string = LocoStrings::LocoString.new(real_key, translated_string, context)
             end
+            @output_hash[real_key] = string
+            keys_to_translate.delete(key)
           end
 
           if keys_to_translate.length > 0
@@ -215,12 +292,26 @@ module Fastlane
         target_string = Colorizer::colorize(@params[:target_file], :white)
         UI.message "Writing #{number_of_strings} strings to #{target_string}..."
 
-        file = LocoStrings.load(@params[:target_file])
-        file.read
-        @output_hash.each do |key, value|
-          file.update(key, value.value, value.comment)
+        if @xcfile.nil?
+          file = LocoStrings.load(@params[:target_file])
+          file.read
+          @output_hash.each do |key, value|
+            file.update(key, value.value, value.comment)
+          end
+          file.write
+        else
+          @xcfile.update_file_path(@params[:target_file])
+          @output_hash.each do |key, value|
+            if value.is_a? LocoStrings::LocoString
+              @xcfile.update(key, value.value, value.comment, "translated", @params[:target_language])
+            elsif value.is_a? LocoStrings::LocoVariantions
+              value.strings.each do |pl_key, variant|
+                @xcfile.update_variation(key, pl_key, variant.value, variant.comment, "translated", @params[:target_language])
+              end
+            end
+          end
+          @xcfile.write
         end
-        file.write
       end
 
       # Read the strings file into a hash
@@ -298,3 +389,5 @@ module Fastlane
     end
   end
 end
+
+# rubocop:enable all

data/lib/fastlane/plugin/translate_gpt/version.rb

@@ -1,5 +1,5 @@
 module Fastlane
   module TranslateGpt
-    VERSION = "0.1.6"
+    VERSION = "0.1.8"
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: fastlane-plugin-translate_gpt
 version: !ruby/object:Gem::Version
-  version: 0.1.6
+  version: 0.1.8
 platform: ruby
 authors:
 - Aleksei Cherepanov
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-08-24 00:00:00.000000000 Z
+date: 2024-02-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ruby-openai
@@ -30,14 +30,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.1.1
+        version: 0.1.4.1
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.1.1
+        version: 0.1.4.1
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -212,7 +212,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.4.13
+rubygems_version: 3.4.19
 signing_key:
 specification_version: 4
 summary: This fastlane plugin provides an easy way to use the OpenAI GPT language