Update tests to ensure referer is working

[rails.git] / script / deliver-message

diff --git a/script/deliver-message b/script/deliver-message
index 46f9642e7f60ea5bd9d2b7b16d6d5f7100149356..2c440c3ec60144b00d6cb571dae9ec8e3569df4e 100755 (executable)
--- a/script/deliver-message
+++ b/script/deliver-message
@@ -6,7 +6,7 @@ if recipient = ARGV[0].match(/^c-(\d+)-(\d+)-(.*)$/)
   comment = DiaryComment.find(recipient[1])
   digest = comment.digest
   date = comment.created_at
-  from = User.find(recipient[2])
+  from = comment.diary_entry.subscribers.find(recipient[2])
   to = comment.user
   token = recipient[3]
 elsif recipient = ARGV[0].match(/^m-(\d+)-(.*)$/)
@@ -23,9 +23,11 @@ end
 exit 0 unless token == digest[0, 6]
 exit 0 if date < 1.month.ago
 
-message.update(:message_read => true) if message
+message&.update(:message_read => true)
 
-mail = Mail.new(STDIN.readlines.join)
+mail = Mail.new(STDIN.read
+                     .encode(:universal_newline => true)
+                     .encode(:crlf_newline => true))
 
 message = Message.from_mail(mail, from, to)
 message.save!