/*
    * Licensed to the Apache Software Foundation (ASF) under one
    * or more contributor license agreements.  See the NOTICE file
    * distributed with this work for additional information
    * regarding copyright ownership.  The ASF licenses this file
    * to you 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 org.apache.maven.plugin.surefire.report;
  
  import javax.annotation.Nonnull;
  
  import java.text.MessageFormat;
  import java.util.Collections;
  import java.util.Locale;
  import java.util.Map;
  
  import org.apache.maven.surefire.api.report.ReportEntry;
  import org.apache.maven.surefire.api.report.RunMode;
  import org.apache.maven.surefire.api.report.StackTraceWriter;
  import org.apache.maven.surefire.api.report.TestSetReportEntry;
  
  import static java.util.Collections.unmodifiableMap;
  import static org.apache.maven.surefire.api.util.internal.StringUtils.NL;
  import static org.apache.maven.surefire.shared.utils.StringUtils.isBlank;
  
  /**
   * @author Kristian Rosenvold
   */
  public class WrappedReportEntry implements TestSetReportEntry {
      private static final float ONE_SECOND = 1000.0f;
  
      private final ReportEntry original;
  
      private final ReportEntryType reportEntryType;
  
      private final long startTime;
  
      private final Integer elapsed;
  
      private final Utf8RecodingDeferredFileOutputStream stdout;
  
      private final Utf8RecodingDeferredFileOutputStream stdErr;
  
      private final Map<String, String> systemProperties;
  
      /*
       * Rationale: The idea is to always display four digits for visually consistent output
       * Important: Keep in sync with maven-surefire-report-plugin/src/main/resources/surefire-report.properties
       */
      private final MessageFormat elapsedTimeFormat = new MessageFormat(
              "{0,choice,0#0|0.0<{0,number,0.000}|10#{0,number,0.00}|100#{0,number,0.0}|1000#{0,number,0}} s",
              Locale.ROOT);
  
      public WrappedReportEntry(
              ReportEntry original,
              ReportEntryType reportEntryType,
              long startTime,
              Integer estimatedElapsed,
              Utf8RecodingDeferredFileOutputStream stdout,
              Utf8RecodingDeferredFileOutputStream stdErr,
              Map<String, String> systemProperties) {
          this.original = original;
          this.reportEntryType = reportEntryType;
          this.startTime = startTime;
          this.elapsed = estimatedElapsed;
          this.stdout = stdout;
          this.stdErr = stdErr;
          this.systemProperties = unmodifiableMap(systemProperties);
      }
  
      public WrappedReportEntry(
              ReportEntry original,
              ReportEntryType reportEntryType,
              long startTime,
              Integer estimatedElapsed,
              Utf8RecodingDeferredFileOutputStream stdout,
              Utf8RecodingDeferredFileOutputStream stdErr) {
          this(original, reportEntryType, startTime, estimatedElapsed, stdout, stdErr, Collections.emptyMap());
      }
  
      public long getStartTime() {
          return startTime;
      }
  
      @Override
      public Integer getElapsed() {
          return elapsed;
      }
 
     @Override
     public int getElapsed(int fallback) {
         return elapsed == null ? fallback : elapsed;
     }
 
     public ReportEntryType getReportEntryType() {
         return reportEntryType;
     }
 
     public Utf8RecodingDeferredFileOutputStream getStdout() {
         return stdout;
     }
 
     public Utf8RecodingDeferredFileOutputStream getStdErr() {
         return stdErr;
     }
 
     @Override
     public String getSourceName() {
         return original.getSourceName();
     }
 
     @Override
     public String getSourceText() {
         return original.getSourceText();
     }
 
     @Override
     public String getName() {
         return original.getName();
     }
 
     @Override
     public String getNameText() {
         return original.getNameText();
     }
 
     public String getClassMethodName() {
         return original.getSourceName() + "." + original.getName();
     }
 
     public String getFullName() {
         return getName() == null ? getSourceName() : getClassMethodName();
     }
 
     @Override
     public String getGroup() {
         return original.getGroup();
     }
 
     @Override
     public StackTraceWriter getStackTraceWriter() {
         return original.getStackTraceWriter();
     }
 
     @Override
     public String getMessage() {
         return original.getMessage();
     }
 
     public String getStackTrace(boolean trimStackTrace) {
         StackTraceWriter w = original.getStackTraceWriter();
         return w == null ? null : (trimStackTrace ? w.writeTrimmedTraceToString() : w.writeTraceToString());
     }
 
     public String elapsedTimeAsString() {
         return getElapsed() != null ? elapsedTimeFormat.format(new Object[] {getElapsed() / ONE_SECOND}) : null;
     }
 
     String getReportSourceName() {
         String sourceName = getSourceName();
         String sourceText = getSourceText();
         return isBlank(sourceText) ? sourceName : sourceText;
     }
 
     String getReportSourceName(String suffix) {
         return isBlank(suffix) ? getReportSourceName() : getReportSourceName() + "(" + suffix + ")";
     }
 
     String getSourceName(String suffix) {
         return isBlank(suffix) ? getSourceName() : getSourceName() + "(" + suffix + ")";
     }
 
     String getSourceQualifiedName(String suffix) {
         String qualifiedName = original.getSourceQualifiedName();
         if (qualifiedName != null) {
             return isBlank(suffix) ? qualifiedName : qualifiedName + "(" + suffix + ")";
         }
         // Fall back to pre-existing behaviour: sourceText if set, else sourceName (with suffix)
         String sourceText = getSourceText();
         return sourceText != null ? sourceText : getSourceName(suffix);
     }
 
     String getReportName() {
         String name = getName();
         String nameText = getNameText();
         return isBlank(nameText) ? name : nameText;
     }
 
     public String getOutput(boolean trimStackTrace) {
         String outputLine =
                 getElapsedTimeSummary() + " <<< " + getReportEntryType().name() + "!";
         String trimmedStackTrace = getStackTrace(trimStackTrace);
         return trimmedStackTrace == null ? outputLine : outputLine + NL + trimmedStackTrace;
     }
 
     public String getElapsedTimeVerbose() {
         return "Time elapsed: " + (getElapsed() != null ? elapsedTimeAsString() : "(unknown)");
     }
 
     public String getElapsedTimeSummary() {
         return getFullName() + " -- " + getElapsedTimeVerbose();
     }
 
     public boolean isErrorOrFailure() {
         ReportEntryType thisType = getReportEntryType();
         return ReportEntryType.FAILURE == thisType || ReportEntryType.ERROR == thisType;
     }
 
     public boolean isSkipped() {
         return ReportEntryType.SKIPPED == getReportEntryType();
     }
 
     public boolean isSucceeded() {
         return ReportEntryType.SUCCESS == getReportEntryType();
     }
 
     @Override
     public String getNameWithGroup() {
         return original.getNameWithGroup();
     }
 
     @Override
     public String getReportNameWithGroup() {
         String reportNameWithGroup = original.getReportNameWithGroup();
 
         if (isBlank(reportNameWithGroup)) {
             return getNameWithGroup();
         }
 
         return reportNameWithGroup;
     }
 
     @Nonnull
     @Override
     public RunMode getRunMode() {
         return original.getRunMode();
     }
 
     @Override
     public Long getTestRunId() {
         return original.getTestRunId();
     }
 
     @Override
     public Map<String, String> getSystemProperties() {
         return systemProperties;
     }
 }