mockit.coverage.reporting.sourceFiles.NeutralOutput Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2006-2015 Rogério Liesenfeld
 * This file is subject to the terms of the MIT license (see LICENSE.txt).
 */
package mockit.coverage.reporting.sourceFiles;

import java.io.*;
import javax.annotation.*;

import mockit.coverage.reporting.parsing.*;

final class NeutralOutput
{
   @Nonnull private final PrintWriter output;
   private boolean previousLineInImports;
   private boolean previousLineInComments;
   @Nullable private String lineIndentation;
   private boolean blankLinesPending;

   NeutralOutput(@Nonnull PrintWriter output) { this.output = output; }

   boolean writeLineWithoutCoverageInfo(@Nonnull LineParser lineParser)
   {
      if (previousLineInComments || !previousLineInImports) {
         if (writeLineInComments(lineParser) || writeLineInImports(lineParser)) {
            return true;
         }
      }
      else if (writeLineInImports(lineParser) || writeLineInComments(lineParser)) {
         return true;
      }

      if (lineParser.isBlankLine()) {
         blankLinesPending = true;
         return true;
      }

      writeBlankLineIfPending();
      return false;
   }

   private boolean writeLineInComments(@Nonnull LineParser lineParser)
   {
      LineElement initialElement = lineParser.getInitialElement();

      if (
         lineParser.isInComments() ||
         previousLineInComments && initialElement.isComment() && initialElement.getNext() == null
      ) {
         String lineText = initialElement.toString();

         if (previousLineInComments) {
            output.println();
         }
         else {
            writeOpeningForCollapsibleBlockOfLines();
            output.write("      ");
            extractLineIndentation(lineText);
            previousLineInComments = true;
         }

         output.write(lineText);
         return true;
      }
      else if (previousLineInComments) {
         output.append("
").append(lineIndentation).println("/*...*/");
         output.println("    ");
         previousLineInComments = false;
      }

      return false;
   }

   private void writeOpeningForCollapsibleBlockOfLines()
   {
      writeBlankLineIfPending();
      output.println("    ");
      output.println("       ");
   }

   private void writeBlankLineIfPending()
   {
      if (blankLinesPending) {
         output.println("     ");
         blankLinesPending = false;
      }
   }

   private void extractLineIndentation(@Nonnull String lineText)
   {
      int indentationSize = 0;

      for (int i = 0; i < lineText.length(); i++, indentationSize++) {
         if (lineText.charAt(i) > ' ') break;
      }

      lineIndentation = lineText.substring(0, indentationSize);
   }

   private boolean writeLineInImports(@Nonnull LineParser lineParser)
   {
      LineElement initialElement = lineParser.getInitialElement();
      boolean isImport = initialElement.isKeyword("import");

      if (!previousLineInImports && isImport || previousLineInImports && (isImport || lineParser.isBlankLine())) {
         String lineText = initialElement.toString();

         if (previousLineInImports) {
            output.println();
            blankLinesPending = !isImport;
         }
         else {
            writeOpeningForCollapsibleBlockOfLines();
            output.write("      ");
            previousLineInImports = true;
         }

         output.write(lineText);
         return true;
      }
      else if (previousLineInImports) {
         output.println("
import ...");
         output.println("    ");
         previousLineInImports = false;
      }

      return false;
   }
}