Add command line to translate repository using the AiTranslateService

Author: ja-openai

cli/src/main/java/com/box/l10n/mojito/cli/command/RepositoryAiTranslationCommand.java

@@ -0,0 +1,89 @@
+package com.box.l10n.mojito.cli.command;
+
+import com.beust.jcommander.Parameter;
+import com.beust.jcommander.Parameters;
+import com.box.l10n.mojito.cli.command.param.Param;
+import com.box.l10n.mojito.cli.console.ConsoleWriter;
+import com.box.l10n.mojito.rest.client.RepositoryAiTranslateClient;
+import com.box.l10n.mojito.rest.client.RepositoryAiTranslateClient.ProtoAiTranslateResponse;
+import com.box.l10n.mojito.rest.entity.PollableTask;
+import java.util.List;
+import java.util.stream.Collectors;
+import org.fusesource.jansi.Ansi.Color;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.context.annotation.Scope;
+import org.springframework.stereotype.Component;
+
+/**
+ * Command to machine translate untranslated strings in a repository.
+ *
+ * @author jaurambault
+ */
+@Component
+@Scope("prototype")
+@Parameters(
+    commandNames = {"repository-ai-translate"},
+    commandDescription = "Ai translate untranslated and rejected strings in a repository")
+public class RepositoryAiTranslationCommand extends Command {
+
+  /** logger */
+  static Logger logger = LoggerFactory.getLogger(RepositoryAiTranslationCommand.class);
+
+  @Autowired ConsoleWriter consoleWriter;
+
+  @Parameter(
+      names = {Param.REPOSITORY_LONG, Param.REPOSITORY_SHORT},
+      arity = 1,
+      required = true,
+      description = Param.REPOSITORY_DESCRIPTION)
+  String repositoryParam;
+
+  @Parameter(
+      names = {Param.REPOSITORY_LOCALES_LONG, Param.REPOSITORY_LOCALES_SHORT},
+      variableArity = true,
+      required = true,
+      description = "List of locales (bcp47 tags) to machine translate")
+  List<String> locales;
+
+  @Parameter(
+      names = {"--source-text-max-count"},
+      arity = 1,
+      description =
+          "Source text max count per locale sent to MT (this param is used to avoid "
+              + "sending too many strings to MT)")
+  int sourceTextMaxCount = 100;
+
+  @Autowired CommandHelper commandHelper;
+
+  @Autowired RepositoryAiTranslateClient repositoryAiTranslateClient;
+
+  @Override
+  public boolean shouldShowInCommandList() {
+    return false;
+  }
+
+  @Override
+  public void execute() throws CommandException {
+
+    consoleWriter
+        .newLine()
+        .a("Ai translate repository: ")
+        .fg(Color.CYAN)
+        .a(repositoryParam)
+        .reset()
+        .a(" for locales: ")
+        .fg(Color.CYAN)
+        .a(locales.stream().collect(Collectors.joining(", ", "[", "]")))
+        .println(2);
+
+    ProtoAiTranslateResponse protoAiTranslateResponse =
+        repositoryAiTranslateClient.translateRepository(
+            new RepositoryAiTranslateClient.ProtoAiTranslateRequest(
+                repositoryParam, locales, sourceTextMaxCount));
+
+    PollableTask pollableTask = protoAiTranslateResponse.pollableTask();
+    commandHelper.waitForPollableTask(pollableTask.getId());
+  }
+}

restclient/src/main/java/com/box/l10n/mojito/rest/client/RepositoryAiTranslateClient.java

@@ -0,0 +1,35 @@
+package com.box.l10n.mojito.rest.client;
+
+import com.box.l10n.mojito.rest.entity.PollableTask;
+import java.util.List;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.stereotype.Component;
+
+/**
+ * @author jaurambault
+ */
+@Component
+public class RepositoryAiTranslateClient extends BaseClient {
+
+  /** logger */
+  static Logger logger = LoggerFactory.getLogger(RepositoryAiTranslateClient.class);
+
+  @Override
+  public String getEntityName() {
+    return "proto-ai-translate";
+  }
+
+  /** Ai translate untranslated and rejected strings in a repository for a given list of locales */
+  public ProtoAiTranslateResponse translateRepository(
+      ProtoAiTranslateRequest protoAiTranslateRequest) {
+
+    return authenticatedRestTemplate.postForObject(
+        getBasePathForEntity(), protoAiTranslateRequest, ProtoAiTranslateResponse.class);
+  }
+
+  public record ProtoAiTranslateRequest(
+      String repositoryName, List<String> targetBcp47tags, int sourceTextMaxCountPerLocale) {}
+
+  public record ProtoAiTranslateResponse(PollableTask pollableTask) {}
+}