/*
    * ===========================================================================
    * Copyright (C) 2011-2025 by the Okapi Framework contributors
    * -----------------------------------------------------------------------------
    * Licensed under the Apache License, Version 2.0 (the "License");
    * you may not use this file except in compliance with the License.
    * You may obtain a copy of the License at
    * 
    * http://www.apache.org/licenses/LICENSE-2.0
   * 
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   * ===========================================================================
   */
  
  package net.sf.okapi.connectors.google.v3;
  
  import java.util.ArrayList;
  import java.util.List;
  
  import com.acumenvelocity.ath.common.AthUtil;
  import com.acumenvelocity.ath.common.ControllerUtil;
  import com.acumenvelocity.ath.common.Log;
  import com.acumenvelocity.ath.gct.v3.AthTranslation;
  import com.google.api.gax.rpc.ApiException;
  import com.google.cloud.translate.v3.GetSupportedLanguagesRequest;
  import com.google.cloud.translate.v3.GlossaryName;
  import com.google.cloud.translate.v3.LocationName;
  import com.google.cloud.translate.v3.SupportedLanguage;
  import com.google.cloud.translate.v3.TranslateTextGlossaryConfig;
  import com.google.cloud.translate.v3.TranslateTextRequest;
  import com.google.cloud.translate.v3.TranslateTextResponse;
  import com.google.cloud.translate.v3.Translation;
  import com.google.cloud.translate.v3.TranslationServiceClient;
  
  import net.sf.okapi.common.Util;
  
  /**
   * Implementation of Google Cloud Translation API v3 using the official Google Cloud Java Client.
   */
  public class GoogleMTv3APIImpl implements GoogleMTv3API {
  
    private final GoogleMTv3Parameters params;
    private TranslationServiceClient client;
  
    public GoogleMTv3APIImpl(GoogleMTv3Parameters params) {
      this.params = params;
      this.client = AthTranslation.getClient();
    }
  
    @Override
    public <T> List<TranslationResponse> translate(GoogleQueryBuilder<T> qb)
        throws GoogleMTv3ErrorException {
      try {
        LocationName parent = LocationName.of(ControllerUtil.getProjectId(),
            params.getProjectLocation());
  
        TranslateTextRequest.Builder requestBuilder = TranslateTextRequest.newBuilder()
            .setParent(parent.toString())
            .setMimeType(params.getMimeType())
            .setSourceLanguageCode(qb.getSourceLanguage())
            .setTargetLanguageCode(qb.getTargetLanguage())
            .addAllContents(qb.getQueries());
  
        // Add Model
        if (!Util.isEmpty(params.getModelId())) {
          // requestBuilder.setModel(String.format("projects/%s/locations/%s/models/%s",
          // params.getModelProjectId(), params.getModelProjectLocation(), params.getModelId()));
  
          String modelProjectId = AthUtil.fallback(params.getModelProjectId(),
              ControllerUtil.getProjectId());
  
          String modelProjectLocation = AthUtil.fallback(params.getModelProjectLocation(),
              params.getProjectLocation());
  
          // ModelName modelName = ModelName.of(modelProjectId, modelProjectLocation,
          // params.getModelId());
  
          String modelName = Log.format("projects/{}/locations/{}/models/{}", modelProjectId,
              modelProjectLocation, params.getModelId());
  
          requestBuilder.setModel(modelName);
        }
  
        // Add Glossary
        if (!Util.isEmpty(params.getGlossaryId())) {
          String glossaryProjectId = AthUtil.fallback(params.getGlossaryProjectId(),
              ControllerUtil.getProjectId());
  
          String glossaryProjectLocation = AthUtil.fallback(params.getGlossaryProjectLocation(),
              params.getProjectLocation());
  
          GlossaryName glossaryName = GlossaryName.of(glossaryProjectId, glossaryProjectLocation,
              params.getGlossaryId());
  
          TranslateTextGlossaryConfig glossaryConfig = TranslateTextGlossaryConfig.newBuilder()
             // .setGlossary(String.format("projects/%s/locations/%s/glossaries/%s",
             // params.getGlossaryProjectId(), params.getGlossaryProjectLocation(),
             // params.getGlossaryId()))
             .setGlossary(glossaryName.toString())
             .build();
 
         requestBuilder.setGlossaryConfig(glossaryConfig);
       }
 
       TranslateTextResponse response = client.translateText(requestBuilder.build());
       return parseTranslations(response);
 
     } catch (ApiException e) {
       throw new GoogleMTv3ErrorException(e.getStatusCode().getCode().getHttpStatusCode(),
           e.getMessage(), qb.getQuery());
 
     } catch (Exception e) {
       throw new GoogleMTv3ErrorException(500, e.getMessage(), qb.getQuery());
     }
   }
 
   @Override
   public <T> TranslationResponse translateSingleSegment(GoogleQueryBuilder<T> qb, String sourceText)
       throws GoogleMTv3ErrorException {
     // Updated: Explicitly specify <T> and remove 'params' to match the updated GoogleQueryBuilder
     // constructor
     GoogleQueryBuilder<T> singleQb = new GoogleQueryBuilder<T>(qb.getSourceLanguage(),
         qb.getTargetLanguage());
 
     singleQb.addQuery(sourceText, null);
 
     List<TranslationResponse> responses = translate(singleQb);
 
     if (responses != null && !responses.isEmpty()) {
       return responses.get(0);
     }
 
     return null;
   }
 
   @Override
   public List<String> getLanguages() throws GoogleMTv3ErrorException {
     try {
       LocationName parent = LocationName.of(ControllerUtil.getProjectId(),
           params.getProjectLocation());
 
       GetSupportedLanguagesRequest request = GetSupportedLanguagesRequest.newBuilder()
           .setParent(parent.toString())
           .build();
 
       List<SupportedLanguage> languages = client.getSupportedLanguages(request)
           .getLanguagesList();
 
       List<String> codes = new ArrayList<>();
 
       for (SupportedLanguage lang : languages) {
         codes.add(lang.getLanguageCode());
       }
 
       return codes;
 
     } catch (ApiException e) {
       throw new GoogleMTv3ErrorException(e.getStatusCode().getCode().getHttpStatusCode(),
           e.getMessage(), "getLanguages");
     }
   }
 
   private List<TranslationResponse> parseTranslations(TranslateTextResponse response) {
     List<TranslationResponse> results = new ArrayList<>();
 
     for (Translation translation : response.getTranslationsList()) {
       String translatedText = translation.getTranslatedText();
       String detectedLanguage = translation.getDetectedLanguageCode();
 
       String finalTranslation = translatedText;
 
       // Check for glossary specific translation info if available in the response object
       if (translation.hasGlossaryConfig()) {
         // The V3 java client handles glossary application within the translated text usually,
         // but specific glossary metadata handling would go here if needed.
       }
 
       results.add(new TranslationResponse(finalTranslation, detectedLanguage));
     }
 
     return results;
   }
 
   @Override
   public void close() {
     // The GCT v3 client is closed in AthTranslation#done()
   }
 }