Added heredoc support.

Author: prwater

src/CodeStore.php

@@ -49,6 +49,14 @@ abstract class CodeStore
    */
   const C_INDENT_DECREMENT_AFTER = 8;
 
+  /**
+   * No indentation, heredoc.
+   *
+   * @since 1.0.0
+   * @api
+   */
+  const C_INDENT_HEREDOC = 32;
+
   /**
    * String for separating parts of the generated code. In most cases a comment with one character repeated many times.
    *
@@ -206,8 +214,15 @@ public function getCode(): string
         $line = $this->shortenSeparator($this->width - $this->indentation * $indentLevel);
       }
 
-      // Append the line with indentation.
-      $lines[] = $this->addIndentation($line, $indentLevel);
+      if ($mode & self::C_INDENT_HEREDOC)
+      {
+        $lines[] = $line;
+      }
+      else
+      {
+        // Append the line with indentation.
+        $lines[] = $this->addIndentation($line, $indentLevel);
+      }
 
       // Increment or decrement indentation level.
       if ($mode & self::C_INDENT_INCREMENT_AFTER)

test/CodeStoreTest.php

@@ -136,6 +136,11 @@ public function testIndentationLevels1(): void
               'statement',
               'middle',
               'statement',
+              'heredoc-start',
+              'line1',
+              'line2',
+              'line3',
+              'heredoc-end',
               'statement',
               'end',
               'begin',
@@ -162,6 +167,11 @@ public function testIndentationLevels1(): void
     statement
   middle
     statement
+    heredoc-start
+line1
+line2
+line3
+heredoc-end
     statement
   end
   begin

test/ConcreteCodeStore.php

@@ -10,17 +10,25 @@
  */
 class ConcreteCodeStore extends CodeStore
 {
+  //--------------------------------------------------------------------------------------------------------------------
+  /**
+   * If true the current lines are part of a heredoc.
+   *
+   * @var bool
+   */
+  private $isHeredoc = false;
+
   //--------------------------------------------------------------------------------------------------------------------
   /**
    * {@inheritdoc}
    */
-  public function __construct(int $indentation=2, int $width=15)
+  public function __construct(int $indentation = 2, int $width = 15)
   {
     parent::__construct($indentation, $width);
-    
+
     $this->separator = '#'.str_repeat('-', $width - 1);
   }
-  
+
   //--------------------------------------------------------------------------------------------------------------------
   /**
    * {@inheritdoc}
@@ -50,11 +58,21 @@ protected function indentationMode(string $line): int
       case 'end-end':
         return self::C_INDENT_DECREMENT_BEFORE_DOUBLE;
 
-      default:
+      case 'heredoc-start':
+        $this->isHeredoc = true;
+
         return 0;
+
+      case 'heredoc-end':
+        $this->isHeredoc = false;
+
+        return self::C_INDENT_HEREDOC;
+
+      default:
+        return ($this->isHeredoc===false) ? 0 : self::C_INDENT_HEREDOC;
     }
   }
-  
+
   //--------------------------------------------------------------------------------------------------------------------
 }