lib/copy/server.rb in copy-0.1.0 vs lib/copy/server.rb in copy-0.1.1

@@ -55,15 +55,19 @@
           # Side-step the output buffer so we can capture the block, but not output it.
           @_out_buf, old_buffer = '', @_out_buf
           content = yield
           @_out_buf = old_buffer
           
+          # Remove newlines from the beginning
+          content.gsub!(/^\n/, '')
           # Get the first line from captured text.
           first_line = content.split("\n").first
           # Determine how much white space it has in front.
           white_space = first_line.match(/^(\s)*/)[0]
           # Remove that same amount of white space from the beginning of every line.
           content.gsub!(Regexp.new("^#{white_space}"), '')
+          # Remove all whitespace from the end, but keep the newline if present.
+          content.gsub!(/\s*$/, '') << "\n" if content =~ /\n/
           
           # Save the content so it can be edited.
           Copy::Storage.set(name, content) if Copy::Storage.connected?
         end
         
\ No newline at end of file