adamcooper-ar_mailer 2.0.2

data/lib/smtp_tls.rb

@@ -0,0 +1,105 @@
+# Original code believed public domain from ruby-talk or ruby-core email.
+# Modifications by Kyle Maxwell <kyle@kylemaxwell.com> used under MIT license.
+
+require "openssl"
+require "net/smtp"
+
+# :stopdoc:
+
+class Net::SMTP
+
+  class << self
+    send :remove_method, :start
+  end
+
+  def self.start( address, port = nil,
+                  helo = 'localhost.localdomain',
+                  user = nil, secret = nil, authtype = nil, use_tls = false,
+                  &block) # :yield: smtp
+    new(address, port).start(helo, user, secret, authtype, use_tls, &block)
+  end
+
+  alias tls_old_start start
+
+  def start( helo = 'localhost.localdomain',
+             user = nil, secret = nil, authtype = nil, use_tls = false ) # :yield: smtp
+    start_method = use_tls ? :do_tls_start : :do_start
+    if block_given?
+      begin
+        send start_method, helo, user, secret, authtype
+        return yield(self)
+      ensure
+        do_finish
+      end
+    else
+      send start_method, helo, user, secret, authtype
+      return self
+    end
+  end
+
+  private
+
+  def do_tls_start(helodomain, user, secret, authtype)
+    raise IOError, 'SMTP session already started' if @started
+    check_auth_args user, secret, authtype if user or secret
+
+    sock = timeout(@open_timeout) { TCPSocket.open(@address, @port) }
+    @socket = Net::InternetMessageIO.new(sock)
+    @socket.read_timeout = 60 #@read_timeout
+    @socket.debug_output = STDERR #@debug_output
+
+    check_response(critical { recv_response() })
+    do_helo(helodomain)
+
+    raise 'openssl library not installed' unless defined?(OpenSSL)
+    starttls
+    ssl = OpenSSL::SSL::SSLSocket.new(sock)
+    ssl.sync_close = true
+    ssl.connect
+    @socket = Net::InternetMessageIO.new(ssl)
+    @socket.read_timeout = 60 #@read_timeout
+    @socket.debug_output = STDERR #@debug_output
+    do_helo(helodomain)
+
+    authenticate user, secret, authtype if user
+    @started = true
+  ensure
+    unless @started
+      # authentication failed, cancel connection.
+        @socket.close if not @started and @socket and not @socket.closed?
+      @socket = nil
+    end
+  end
+
+  def do_helo(helodomain)
+     begin
+      if @esmtp
+        ehlo helodomain
+      else
+        helo helodomain
+      end
+    rescue Net::ProtocolError
+      if @esmtp
+        @esmtp = false
+        @error_occured = false
+        retry
+      end
+      raise
+    end
+  end
+
+  def starttls
+    getok('STARTTLS')
+  end
+
+  alias tls_old_quit quit
+
+  def quit
+    begin
+      getok('QUIT')
+    rescue EOFError
+    end
+  end
+
+end unless Net::SMTP.private_method_defined? :do_tls_start or
+           Net::SMTP.method_defined? :tls?

data/share/bsd/ar_sendmail

@@ -0,0 +1,30 @@
+#!/bin/sh
+# PROVIDE: ar_sendmail
+# REQUIRE: DAEMON
+# BEFORE: LOGIN
+# KEYWORD: FreeBSD shutdown
+
+#
+# Add the following lines to /etc/rc.conf to enable ar_sendmail:
+#
+#ar_sendmail_enable="YES"
+
+. /etc/rc.subr
+
+name="ar_sendmail"
+rcvar=`set_rcvar`
+
+command="/usr/local/bin/ar_sendmail"
+command_interpreter="/usr/local/bin/ruby18"
+
+# set defaults
+
+ar_sendmail_rails_env=${ar_sendmail_rails_env:-"production"}
+ar_sendmail_chdir=${ar_sendmail_chdir:-"/"}
+ar_sendmail_enable=${ar_sendmail_enable:-"NO"}
+ar_sendmail_flags=${ar_sendmail_flags:-"-d"}
+
+load_rc_config $name
+export RAILS_ENV=$ar_sendmail_rails_env
+run_rc_command "$1"
+

data/share/linux/ar_sendmail

@@ -0,0 +1,78 @@
+#!/usr/bin/env ruby
+#
+# ar_sendmail Startup script for ar_mailer by Adam Meehan
+#
+# chkconfig: - 85 15
+# description: ar_sendmail manages sending emails for Rails apps. 
+#
+require 'yaml'
+
+# Config file app mailers
+config_file = '/etc/ar_sendmail.conf'
+
+begin
+  config = YAML::load(IO.read(config_file)) || {}
+  if config.empty? || (config.has_key?('defaults') && config.size == 1)
+    puts "No mailers defined. Exiting."
+    exit
+  end
+rescue Errno::ENOENT
+  puts "Config file not found at '#{config_file}'!"
+  exit
+end
+
+default_options = {'pidfile' => './log/ar_sendmail.pid'}.merge(config.delete('defaults') || {})
+
+command, app_name = *ARGV
+
+def start(app, options)
+  switches = ""
+  options.each {|k, v| switches << " --#{k} #{v}"}
+  STDOUT.write "Starting mailer for #{app} in #{options['environment']} mode ... "
+  status = system("ar_sendmail -d #{switches}") ? "started" : "failed"
+  puts status
+end
+
+def stop(app, options)
+  pid_file = File.expand_path(options['pidfile'], options['chdir'])
+  if File.exist? pid_file
+    begin
+      pid = open(pid_file).read.to_i
+      STDOUT.write "Stopping mailer for #{app}... "
+      Process.kill('TERM', pid)
+      puts "stopped"
+    rescue Errno::ESRCH
+      puts "Mailer process does not exist. Is not running."
+    end
+  else
+    puts "Skipping mailer for #{app}, no pid file."
+  end
+end
+
+def restart(app, options)
+  puts "Restarting mailer for #{app} ..."
+  stop app, options  
+  start app, options
+end
+
+def command_error(msg)
+  puts msg
+  exit
+end
+
+if ['start', 'stop', 'restart'].include?(command)
+  apps = config
+  if app_name
+    command_error "No such app defined in ar_sendmail config" unless config.include?(app_name)
+    app_options = config[app_name]
+    command_error "Must specify chdir for app in ar_sendmail config" if app_options['chdir'].nil?    
+    apps = {app_name => app_options}
+  end
+
+  apps.each do |app, options|     
+    options = default_options.merge(options)
+    send(command, app, options)
+  end
+else
+  command_error "Usage: ar_sendmail {start|stop|restart} [optional app_name]"
+end

data/share/linux/ar_sendmail.conf

@@ -0,0 +1,30 @@
+# This is a configuration file for ar_sendmail daemon init.d script
+#
+# Define the settings for each app which has a mailer that you want start 
+# automatically on startup.
+#
+# You can define an optional defaults section which will apply to all 
+# applications unless the setting is specified under the applications specific 
+# config.
+#
+# Settings not specified in either the defaults or app config section will
+# be implied by the gem binary defaults. The option names are the same as the 
+# long format binary option switches. Run 'ar_sendmail -h' to see option switches
+# and default values.
+#
+# Copy this file to /etc/ar_sendmail.conf and it will be read by the init.d
+# script.
+#
+
+## Demo app config
+#
+#defaults:
+#  batch-size: 10
+#  max-age: 0
+#  delay: 60
+#
+#app_name:
+#  chdir: /var/www/apps/app_name
+#  environment: production
+#  pidfile: ./log/ar_sendmail.pid
+

data/test/action_mailer.rb

@@ -0,0 +1,197 @@
+require 'net/smtp'
+require 'smtp_tls' unless Net::SMTP.instance_methods.include?("enable_starttls_auto")
+require 'time'
+
+class Net::SMTP
+
+  @reset_called = 0
+
+  @deliveries = []
+
+  @send_message_block = nil
+
+  @start_block = nil
+
+  class << self
+
+    attr_reader :deliveries
+    attr_reader :send_message_block
+    attr_accessor :reset_called
+
+    send :remove_method, :start
+
+  end
+
+  def self.on_send_message(&block)
+    @send_message_block = block
+  end
+
+  def self.on_start(&block)
+    if block_given?
+      @start_block = block
+    else
+      @start_block
+    end
+  end
+
+  def self.clear_on_start
+    @start_block = nil
+  end
+
+  def self.reset
+    deliveries.clear
+    on_start
+    on_send_message
+    @reset_called = 0
+  end
+
+  def start(*args)
+    self.class.on_start.call if self.class.on_start
+    yield self
+  end
+
+  alias test_old_reset reset if instance_methods.include? 'reset'
+
+  def reset
+    self.class.reset_called += 1
+  end
+
+  alias test_old_send_message send_message
+
+  def send_message(mail, to, from)
+    return self.class.send_message_block.call(mail, to, from) unless
+      self.class.send_message_block.nil?
+    self.class.deliveries << [mail, to, from]
+    return "queued"
+  end
+
+end
+
+##
+# Stub for ActionMailer::Base
+
+module ActionMailer; end
+
+class ActionMailer::Base
+
+  @server_settings = {}
+
+  class << self
+    attr_accessor :delivery_method
+  end
+
+  def self.logger
+    o = Object.new
+    def o.info(arg) end
+    return o
+  end
+
+  def self.method_missing(meth, *args)
+    meth.to_s =~ /deliver_(.*)/
+    super unless $1
+    new($1, *args).deliver!
+  end
+
+  def self.reset
+    server_settings.clear
+  end
+
+  def self.server_settings
+    @server_settings
+  end
+
+  def initialize(meth = nil)
+    send meth if meth
+  end
+
+  def deliver!
+    perform_delivery_activerecord @mail
+  end
+
+end
+
+##
+# Stub for an ActiveRecord model
+
+class Email
+
+  START = Time.parse 'Thu Aug 10 2006 11:19:48'
+
+  attr_accessor :from, :to, :mail, :last_send_attempt, :created_on, :id
+
+  @records = []
+  @id = 0
+
+  class << self; attr_accessor :records, :id; end
+
+  def self.create(record)
+    record = new record[:from], record[:to], record[:mail],
+                 record[:last_send_attempt]
+    records << record
+    return record
+  end
+
+  def self.destroy_all(conditions)
+    timeout = conditions.last
+    found = []
+
+    records.each do |record|
+      next if record.last_send_attempt == 0
+      next if record.created_on == 0
+      next unless record.created_on < timeout
+      record.destroy
+      found << record
+    end
+
+    found
+  end
+
+  def self.find(_, conditions = nil)
+    return records if conditions.nil?
+    now = Time.now.to_i - 300
+    return records.select do |r|
+      r.last_send_attempt < now
+    end
+  end
+
+  def self.reset
+    @id = 0
+    records.clear
+  end
+
+  def initialize(from, to, mail, last_send_attempt = nil)
+    @from = from
+    @to = to
+    @mail = mail
+    @id = self.class.id += 1
+    @created_on = START + @id
+    @last_send_attempt = last_send_attempt || 0
+  end
+
+  def destroy
+    self.class.records.delete self
+    self.freeze
+  end
+
+  def ==(other)
+    other.id == id
+  end
+
+  def save
+  end
+
+end
+
+Mail = Email
+
+class String
+  def classify
+    self
+  end
+
+  def tableize
+    self.downcase
+  end
+
+end
+

data/test/test_armailer.rb

@@ -0,0 +1,51 @@
+require 'test/unit'
+require 'action_mailer'
+require 'action_mailer/ar_mailer'
+
+##
+# Pretend mailer
+
+class Mailer < ActionMailer::Base
+  self.delivery_method = :activerecord
+
+  def mail
+    @mail = Object.new
+    def @mail.encoded() 'email' end
+    def @mail.from() ['nobody@example.com'] end
+    def @mail.destinations() %w[user1@example.com user2@example.com] end
+  end
+
+end
+
+class TestARMailer < Test::Unit::TestCase
+
+  def setup
+    Mailer.email_class = Email
+
+    Email.records.clear
+    Mail.records.clear
+  end
+
+  def test_self_email_class_equals
+    Mailer.email_class = Mail
+
+    Mailer.deliver_mail
+
+    assert_equal 2, Mail.records.length
+  end
+
+  def test_perform_delivery_activerecord
+    Mailer.deliver_mail
+
+    assert_equal 2, Email.records.length
+
+    record = Email.records.first
+    assert_equal 'email', record.mail
+    assert_equal 'user1@example.com', record.to
+    assert_equal 'nobody@example.com', record.from
+
+    assert_equal 'user2@example.com', Email.records.last.to
+  end
+
+end
+