/*
    * ===========================================================================
    * 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 org.slf4j.Logger;
  import org.slf4j.LoggerFactory;
  
  import net.sf.okapi.common.query.MatchType;
  import net.sf.okapi.common.query.QueryResult;
  import net.sf.okapi.common.resource.InvalidContentException;
  import net.sf.okapi.common.resource.TextFragment;
  import net.sf.okapi.lib.translation.QueryUtil;
  
  /**
   * Builder for creating QueryResult objects from TextFragment translations.
   */
  public class FragmentQueryResultBuilder implements QueryResultBuilder<TextFragment> {
  
    private final Logger logger = LoggerFactory.getLogger(getClass());
    private final String origin;
    private final int weight;
    private final QueryUtil util;
  
    public FragmentQueryResultBuilder(String origin, int weight) {
      this.origin = origin;
      this.weight = weight;
      this.util = new QueryUtil();
    }
  
    @Override
    public List<QueryResult> convertResponses(List<TranslationResponse> responses,
        TextFragment originalSource) {
  
      List<QueryResult> results = new ArrayList<>();
  
      for (TranslationResponse response : responses) {
        QueryResult qr = new QueryResult();
        qr.source = originalSource;
  
        try {
          // Convert the translated HTML back to a TextFragment
          String translatedHtml = response.getTranslatedText();
          qr.target = makeFragment(translatedHtml, originalSource);
  
          qr.setFuzzyScore(100); // MT is always 100% match
          qr.setCombinedScore(100);
          qr.matchType = MatchType.MT;
          qr.origin = origin;
          qr.weight = weight;
          qr.setQuality(QueryResult.QUALITY_UNDEFINED);
  
        } catch (InvalidContentException e) {
          // Something went wrong in the resulting MT candidate
          // We fall back on no candidate with a zero score
          logger.error("This MT candidate will be ignored.\n{}\n{}", originalSource.toString(),
              e.getMessage());
  
          qr = createDummyResponse(originalSource);
        }
  
        results.add(qr);
      }
  
      return results;
    }
  
    @Override
    public QueryResult createDummyResponse(TextFragment originalSource) {
      QueryResult qr = new QueryResult();
  
      qr.source = originalSource;
      qr.target = originalSource.clone(); // Return source as target
      qr.setFuzzyScore(0); // No match
      qr.setCombinedScore(0);
      qr.matchType = MatchType.MT;
      qr.origin = origin;
      qr.weight = weight;
      qr.setQuality(QueryResult.QUALITY_UNDEFINED);
  
      return qr;
   }
 
   private TextFragment makeFragment(String codedHtml, TextFragment sourceFragment) {
     return sourceFragment.hasCode()
         ? new TextFragment(util.fromCodedHTML(codedHtml, sourceFragment, false),
             sourceFragment.getClonedCodes())
 
         : new TextFragment(util.fromCodedHTML(codedHtml, sourceFragment, false));
   }
 }