Created API endpoint for ai translations

Author: Aaron-Detre

pom.xml

@@ -557,6 +557,11 @@
       <artifactId>lzstring4java</artifactId>
       <version>0.1</version>
     </dependency>
+    <dependency>
+      <groupId>software.amazon.awssdk</groupId>
+      <artifactId>translate</artifactId>
+      <version>2.40.15</version>
+    </dependency>
   </dependencies>
   <properties>
     <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>

src/main/java/org/wise/portal/presentation/web/controllers/author/project/TranslateProjectAPIController.java

@@ -2,14 +2,14 @@
 
 import java.io.IOException;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.security.access.annotation.Secured;
 import org.springframework.security.core.Authentication;
-import org.springframework.stereotype.Controller;
 import org.springframework.web.bind.annotation.PathVariable;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.ResponseBody;
+import org.springframework.web.bind.annotation.RestController;
 import org.wise.portal.domain.project.impl.ProjectImpl;
 import org.wise.portal.domain.user.User;
 import org.wise.portal.service.project.ProjectService;
@@ -18,7 +18,14 @@
 
 import com.fasterxml.jackson.databind.node.ObjectNode;
 
-@Controller
+import software.amazon.awssdk.auth.credentials.AwsBasicCredentials;
+import software.amazon.awssdk.auth.credentials.StaticCredentialsProvider;
+import software.amazon.awssdk.regions.Region;
+import software.amazon.awssdk.services.translate.TranslateClient;
+import software.amazon.awssdk.services.translate.model.TranslateTextRequest;
+import software.amazon.awssdk.services.translate.model.TranslateTextResponse;
+
+@RestController
 @RequestMapping("/api/author/project/translate")
 @Secured({ "ROLE_AUTHOR" })
 public class TranslateProjectAPIController {
@@ -32,8 +39,16 @@ public class TranslateProjectAPIController {
   @Autowired
   protected TranslateProjectService translateProjectService;
 
+  @Value("${aws.accessKeyId}")
+  private String accessKey;
+
+  @Value("${aws.secretAccessKey}")
+  private String secretKey;
+
+  @Value("${aws.region}")
+  private Region region;
+
   @PostMapping("{projectId}/{locale}")
-  @ResponseBody
   protected void saveTranslations(Authentication auth,
       @PathVariable("projectId") ProjectImpl project, @PathVariable("locale") String locale,
       @RequestBody ObjectNode translations) throws IOException {
@@ -42,4 +57,51 @@ protected void saveTranslations(Authentication auth,
       translateProjectService.saveTranslations(project, locale, translations.toString());
     }
   }
+
+  @PostMapping("translationSuggestions")
+  protected String getSuggestedTranslation(Authentication auth, @RequestBody ObjectNode objectNode) throws IOException, IllegalArgumentException {
+    String srcLang = objectNode.get("srcLang").asText();
+    String targetLang = objectNode.get("targetLang").asText(); 
+    String srcText = objectNode.get("srcText").asText(); 
+    String srcLangCode = this.convertLanguageToAWSCode(srcLang);
+    String targetLangCode = this.convertLanguageToAWSCode(targetLang);
+    TranslateClient translateClient = buildTranslateClient();
+    TranslateTextRequest request = buildTranslateTextRequest(srcText, srcLangCode, targetLangCode);
+    TranslateTextResponse textResponse = translateClient.translateText(request);
+    return textResponse.translatedText();
+  }
+
+  private TranslateClient buildTranslateClient() {
+    AwsBasicCredentials credentials = AwsBasicCredentials.create(accessKey, secretKey);
+    return TranslateClient.builder()
+                .region(region)
+                .credentialsProvider(StaticCredentialsProvider.create(credentials))
+                .build();
+  }
+
+  private TranslateTextRequest buildTranslateTextRequest(String srcText, String srcLangCode,
+      String targetLangCode) {
+    return TranslateTextRequest.builder()
+                .text(srcText)
+                .sourceLanguageCode(srcLangCode)
+                .targetLanguageCode(targetLangCode)
+                .build();
+  }
+
+  private String convertLanguageToAWSCode(String language) throws IllegalArgumentException {
+    return switch (language) {
+      case "English" -> "en";
+      case "Spanish" -> "es";
+      case "Italian" -> "it";
+      case "Japanese" -> "ja";
+      case "German" -> "de";
+      case "Chinese (Simplified)" -> "zh";
+      case "Chinese (Traditional)" -> "zh-TW";
+      case "Dutch" -> "nl";
+      case "Korean" -> "ko";
+      case "Vietnamese" -> "vi";
+      default -> throw new IllegalArgumentException("Invalid language provided");
+    };
+  }
+  
 }