diff --git a/lib/slither/generator.rb b/lib/slither/generator.rb index fbd3032..658338a 100644 --- a/lib/slither/generator.rb +++ b/lib/slither/generator.rb @@ -37,8 +37,8 @@ def stream(data, io) private def generate_file_output - output = @builder.join("\n") - output.concat("\n") if add_trailing_newline? + output = @builder.join(newline) + output.concat(newline) if add_trailing_newline? output end @@ -48,16 +48,20 @@ def add_trailing_newline? end def newline_needed(first_line) - first_line ? "".freeze : "\n".freeze + first_line ? "".freeze : newline end def format_row_for_streaming(row, section, first_line) if add_trailing_newline? - section.format(row) + "\n".freeze + section.format(row) + newline else add_newline = newline_needed(first_line) section.format(row).prepend(add_newline) end end + + def newline + @newline ||= @definition.options[:new_line] || "\n".freeze + end end end