Add support for heredoc strings

Fixes #12

Author: smortex

lib/puppet-lint/plugins/check_strict_indent.rb

@@ -7,19 +7,22 @@ def match(tokens)
       :RBRACE => :LBRACE,
       :RBRACK => :LBRACK,
       :RPAREN => :LPAREN,
+      :HEREDOC => :HEREDOC_OPEN,
+      :HEREDOC_POST => :HEREDOC_OPEN,
     }
     open = {
       :LBRACE => [],
       :LBRACK => [],
       :LPAREN => [],
+      :HEREDOC_OPEN => [],
     }
 
     matches = {}
 
     tokens.each do |token|
-      if [:LBRACE, :LBRACK, :LPAREN].include?(token.type)
+      if [:LBRACE, :LBRACK, :LPAREN, :HEREDOC_OPEN].include?(token.type)
         open[token.type] << token
-      elsif [:RBRACE, :RBRACK, :RPAREN].include?(token.type)
+      elsif [:RBRACE, :RBRACK, :RPAREN, :HEREDOC, :HEREDOC_POST].include?(token.type)
         match = open[opening_token[token.type]].pop
         if not match.nil?
           matches[token] = match
@@ -50,6 +53,9 @@ def check
       open_groups = 0
       prev_token = token.prev_token
       while not prev_token.nil? and prev_token.type != :NEWLINE
+        if prev_token.type == :HEREDOC_OPEN
+          temp_indent += 1
+        end
         if [:LBRACE, :LBRACK, :LPAREN].include?(prev_token.type)
           if matches[prev_token].nil? or matches[prev_token].line > prev_token.line
             # left braces not matched in the same line increase indent
@@ -125,6 +131,10 @@ def check
       actual = 0
       if token.next_token.type == :INDENT
         actual = token.next_token.value.length
+      elsif token.prev_token.type == :HEREDOC
+        actual = token.prev_token.value.split("\n").last.length
+      elsif token.prev_token.type == :HEREDOC_OPEN
+        actual = next_token.prev_token.value.split("\n").last.length
       else
         actual = 0
       end

spec/fixtures/fail/heredoc_extra.pp

@@ -0,0 +1,4 @@
+$variable = @(EOT)
+    This is a multiline
+    heredoc string
+   | EOT

spec/fixtures/fail/heredoc_unsuficient.pp

@@ -0,0 +1,4 @@
+$variable = @(EOT)
+This is a multiline
+heredoc string
+| EOT

spec/fixtures/pass/heredoc.pp

@@ -0,0 +1,21 @@
+$variable = @(EOT)
+  This is a multiline
+  heredoc string
+  | EOT
+
+$variable_with_interpolation = @("EOT")
+  Another example
+  ${variable}
+  with
+  ${variable}
+  with interpolation
+  | EOT
+
+case fact('os.family') {
+  'debian': {
+    $greeting = @(EOT)
+      Hello
+      World
+      | EOT
+  }
+}