rmails 0.1.1

data/app/models/certificate_manager.rb

@@ -0,0 +1,46 @@
+class CertificateManager
+
+  attr_reader :interpreter
+
+  def self.save_all_and_restart
+    sm = CertificateManager.new
+    sm.send 'smtpd_cert_gen', {}
+  end
+
+  def initialize
+    @i = @interpreter = AutomateIt.new
+  end
+
+  def smtpd_cert_gen(params)
+    #pass = SecureRandom.base64(25)
+    #@i.edit :file => pass_file = @i.mktemp do
+    #  append pass
+    #end
+    params[:email] ||= 'fandisek@rmails.com' # Property.find_by_key('postmaster').value
+    @i.shell_manager.sh "openssl req -new -newkey rsa:4096 -x509 -days 3650 -nodes -out /etc/ssl/certs/smtpd.pem -keyout /etc/ssl/private/smtpd.pem -subj /C=#{params[:country]}/ST=#{params[:state]}/L=#{params[:locality]}/O=#{params[:org]}/OU=#{params[:org_unit]}/CN=#{params[:name]||Property.find_by_key('myhostname').value}/emailAddress=#{params[:email]}"
+    #@i.shell_manager.rm pass_file
+    #pass
+  end
+
+  def https_cert_gen(params)
+    params[:email] ||= 'fandisek@rmails.com'
+    @i.shell_manager.sh "openssl req -new -newkey rsa:4096 -x509 -days 3650 -nodes -out /etc/ssl/certs/https.pem -keyout /etc/ssl/private/https.pem -subj /C=/ST=/L=/O=/OU=/CN=#{params[:name]||Property.find_by_key('mydomain').value}/emailAddress=#{params[:email]}"
+  end
+
+  def dkim_cert_gen(domain, keyname=domain)
+    @i.shell_manager.sh("opendkim-genkey -r -d #{keyname} -D /etc/ssl/dkim")
+    @i.shell_manager.chown 'opendkim', 'opendkim', "/etc/ssl/dkim/#{keyname}.private"
+
+    key_table = "default._domainkey.#{keyname} #{domain}:default:/etc/ssl/dkim/#{keyname}.private"
+    signing_table = "#{domain} default._domainkey.#{keyname}"
+
+    @i.edit :file => '/etc/opendkim/KeyTable' do
+      append key_table
+    end
+    @i.edit :file => '/etc/opendkim/SigningTable' do
+      append signing_table
+    end
+  end
+
+
+end

data/app/models/property.rb

@@ -0,0 +1,103 @@
+class Property < ActiveRecord::Base
+
+  self.table_name = 'server_configuration'
+
+  attr_reader :collection, :multiple, :translate
+
+  validates_presence_of :key, :value, :service
+  validates_with PropertyValueValidator
+
+  has_paper_trail :ignore => [:new_value], :skip => [:new_value, :template]
+
+  def self.service; APPLICATION ;end
+
+  def self.add(service, key, value, template)
+    p = Property.find_by_key(key)||Property.new
+    p.key = key; p.value = value; p.template = template
+    case service
+      when :postfix
+        p.service = POSTFIX
+      when :dovecot
+        p.service = DOVECOT
+      when :awstats
+        p.service = AWSTATS
+      when :dspam
+        p.service = DSPAM
+      when :nginx
+        p.service = NGINX
+      else
+    end
+    p.save
+  end
+
+  def self.all_new_values
+    self.select([:key, :value, :new_value]).where 'new_value is not null'
+  end
+
+  def self.all_new_values_for_write
+    properties = self.where 'new_value is not null'
+    # template = "template+type", so remove '+type'
+    properties.map {|p| p.template.sub!(/\+.*/, ''); p }
+  end
+
+  def self.all_to_show
+    properties = self.select [:id, :key, :value, :new_value, :template]
+    # template = "template+type", so remove 'template+'
+    properties.map {|p| p.template.sub!(/.*\+/, ''); p }
+  end
+
+  def input_type
+    t = template.to_sym
+    if :yes_no ==  t
+      @collection = [ 'yes', 'no' ]
+      @translate = true
+      return :radio_buttons
+
+    elsif :on_off == t
+      @collection = [ 'on', 'off']
+      @translate = true
+      return :radio_buttons
+
+    elsif [:select_from_enum, :select_from_list].include? t
+      arr = self.value.split(';')
+      @collection = arr.map {|s| s.sub('+','') }
+      @translate = t == :select_from_enum
+      self.value = arr.select {|s| s.first == '+' }.first[1..-1]
+      return :select
+
+    elsif [:multiselect_from_enum, :multiselect_from_list].include? t
+      arr = self.value.split(';')
+      @collection = arr.map {|s| s.sub('+','') }
+      @multiple = true
+      @translate = t == :multiselect_from_enum
+      self.value = arr.map {|s| s[1..-1] if s.first == '+' }.compact
+      return :select
+
+    elsif [:integer, :select].include? t
+      return t
+
+    else
+      return :string
+    end
+  end
+
+  def has_new_value?
+    !self.new_value.nil?
+  end
+
+  def set_value(value)
+    self.update_attribute :new_value, value
+  end
+
+protected
+
+  APPLICATION = 0
+
+  POSTFIX     = 1
+  DOVECOT     = 2
+  NGINX       = 3
+  DSPAM       = 4
+  AMAVIS      = 5
+  AWSTATS     = 6
+
+end

data/app/models/property/awstats.rb

@@ -0,0 +1,15 @@
+class Property::Awstats < Property
+
+  def self.service; AWSTATS ;end
+
+  default_scope where(:service => self.service)
+
+
+protected
+
+  def self.etc_awstats
+    '/etc/awstats'
+  end
+
+
+end

data/app/models/property/dovecot.rb

@@ -0,0 +1,61 @@
+class Property::Dovecot < Property
+
+  def self.service; DOVECOT ;end
+
+  default_scope where(:service => self.service)
+
+  def self.etc_dovecot
+    '/etc/dovecot'
+  end
+
+  def self.value_template(a, locals)
+    a.edit :file => "#{etc_dovecot}/dovecot.conf", :backup => false do
+      locals.each do |key, value|
+        replace /#{key}\s*=\s*.*/, "#{key} = #{value}"
+      end
+    end
+  end
+
+  def self.yes_no_template(a, values)
+    a.edit  :file => "#{etc_dovecot}/dovecot.conf", :backup => false do
+      values.each do |key, value|
+        replace /^#{key}\s*=\s*.*$/, "#{key} = #{value}"
+      end
+    end
+  end
+
+  def self.commented_value_template(a, values)
+    a.edit :file => "#{etc_dovecot}/dovecot.conf", :backup => false do
+      values.each do |key, value|
+        param_name = key.split('__').last
+        rxp = /(\s*#\s#{key}\s+!!.*!!\s*$\s*#{param_name}\s*=\s*)(\d+)/
+        manipulate do |text|
+          text =~ rxp
+          text.gsub rxp, "#{$1}#{value}"
+        end
+      end
+    end
+  end
+
+  def self.storage_template(a, values)
+    a.edit :file => "#{etc_dovecot}/dovecot.conf", :backup => false do
+      values.each do |key, value|
+        value += '%' if value =~ /%$/
+        manipulate do |text|
+          case key.to_sym
+            when :storage_size
+              text =~ rxp = /(\*:storage\s*=\s*).*$/
+            when :storage_trash
+              text =~ rxp = /(Trash:storage\s*=\s*).*$/
+            when :storage_spam
+              text =~ rxp = /(Spam:storage\s*=\s*).*$/
+            else
+          end
+          text.gsub rxp, "#{$1}#{value}"
+        end
+      end
+    end
+  end
+
+
+end

data/app/models/property/dspam.rb

@@ -0,0 +1,45 @@
+class Property::Dspam< Property
+
+  def self.service; DSPAM ;end
+
+  default_scope where(:service => self.service)
+
+
+  def self.on_off_pref_template(a, locals)
+    a.edit :file => "#{etc_dspam}/dspam.conf" do
+      locals.each do |key, value|
+        value = value == true or value == 'on' ? 'on' : 'off'
+        delete /^Preference\s+"#{key}/
+        append "Preference\t\t\"#{key}=#{value}\""
+      end
+    end
+  end
+
+  def self.preference_template(a, locals)
+    a.edit :file => "#{etc_dspam}/dspam.conf" do
+      locals.each do |key, value|
+        delete /^Preference\s+"#{key}/
+        append "Preference\t\t\"#{key}=#{value}\""
+      end
+    end
+  end
+
+  def self.enum_template(a, locals)
+    a.edit :file => "#{etc_dspam}/dspam.conf" do
+      locals.each do |key, value|
+        value = value.split(';').select {|s| sfirst=='+' }. first[1..-1]
+        delete /^Preference\s+"#{key}/
+        append "Preference\t\t\"#{key}=#{value}\""
+      end
+    end
+  end
+
+
+protected
+
+  def self.etc_dspam
+    '/etc/dspam'
+  end
+
+
+end

data/app/models/property/nginx.rb

@@ -0,0 +1,47 @@
+class Property::Nginx < Property
+
+  def self.service; NGINX ;end
+
+  default_scope where(:service => self.service)
+
+  def self.render_thin_site_template(a, site, key_value_string)
+    # example key_value_string= 'hostnames=rmails.com www.rmails.com;a=b;c=d'
+    locals = Hash[ key_value_string.split(';').map  {|it|  it.split('=', 2)  } ]
+    site_path = "#{etc_nginx}/#{sites_available}/#{site}"
+    render(
+      :file   => "#{dist}nginx/rmails.erb",
+      :to     => site_path,
+      :mode   => 0664,
+      :locals => locals
+    )
+  end
+
+  def site_power(a, site, run)
+    site_enabled = "#{etc_nginx}/#{sites_enabled}/#{site}"
+    if run
+      unless File.file? site_enabled
+        a.shell_manager.ln_s  "#{etc_nginx}/#{sites_available}/#{site}",
+                              site_enabled
+      end
+    else
+      a.shell_manager.rm site_enabled
+    end
+  end
+
+protected
+
+  def self.etc_nginx
+    '/etc/nginx'
+  end
+
+  def self.sites_enabled
+    'sites-enabled'
+  end
+
+  def self.sites_available
+    'sites-available'
+  end
+
+
+
+end

data/app/models/property/postfix.rb

@@ -0,0 +1,64 @@
+class Property::Postfix < Property
+
+  def self.service; POSTFIX ;end
+
+  default_scope where(:service => self.service)
+
+  def self.main_value_template(a, locals)
+    a.edit :file => "#{etc_postfix}/main.cf", :backup => false do
+      locals.each do |key, value|
+        replace /^#{key}\s*=\s*.*$/, "#{key}=#{value}"
+      end
+    end
+  end
+
+  # reject_rbl_client bl.spamcop.net
+  # reject_rbl_client zen.spamhaus.org  ...
+  # values = {'bl.spamcop.net' => true, 'zen.spamhaus.org' => false}
+  def self.rbl_client_template(a, values)
+    a.edit :file => "#{etc_postfix}/main.cf", :backup => false do
+      values.each do |key, value|
+        value ? uncomment(key) : comment(key)
+      end
+    end
+  end
+
+  def self.dspam_power(a, run)
+    a.edit :file => "#{etc_postfix}/master.cf", :backup => false do
+      if run
+        uncomment 'filter=lmtp:unix:/var/run/dspam'
+        uncomment 'localhost:7711 inet'
+      else
+        comment 'filter=lmtp:unix:/var/run/dspam'
+        comment 'localhost:7711 inet'
+      end
+    end
+  end
+
+  def self.dkim_power(a, run)
+    a.edit :file => "#{etc_postfix}/main.cf", :backup => false do
+      if run
+        uncomment 'smtpd_milters = inet:127.0.0.1:8891'
+        uncomment 'non_smtpd_milters = inet:127.0.0.1:8891'
+      else
+        comment 'smtpd_milters = inet:127.0.0.1:8891'
+        comment 'non_smtpd_milters = inet:127.0.0.1:8891'
+      end
+    end
+    a.edit :file => "/etc/default/opendkim", :backup => false do
+      if run
+        uncomment 'SOCKET="inet:8891@localhost"'
+      else
+        comment 'SOCKET="inet:8891@localhost"'
+      end
+    end
+  end
+
+protected
+
+  def self.etc_postfix
+    '/etc/postfix'
+  end
+
+
+end

data/app/models/property_value_validator.rb

@@ -0,0 +1,41 @@
+class PropertyValueValidator < ActiveModel::Validator
+
+  def validate(record)
+    return true unless record.new_value
+    template, rxp = record.template.split('+')
+    rxp = template unless rxp
+    if validation = Validations[rxp]
+      if record.new_value ~! validation
+        record.errors[:base] << I18n.t('validation.format')
+        return false
+      end
+    end
+    true
+  end
+
+private
+
+  Validations = {
+      'on_off' => /\A(on)|(off)\Z/i,
+
+      'yes_no' => /\A(yes)|(no)\Z/i,
+
+      'mega_size' => /\A\d+[bBkKmMgG]?\Z/i,
+
+      'mail_name' => /\A[a-z0-9](\s|[a-z0-9])*\Z/i,
+
+      'domain' => /\A[a-z0-9]+([\-\.]{1}[a-z0-9]+)*\.[a-z]{2,6}\Z/i,
+
+      'param_string' => /\A(\w|\d|\s|\$|@|%|_|-|\.|:)*\Z/i,
+
+      'string' => /\A[a-zA-Z0-9_-]*\Z/i,
+
+      'number' => /\A(\+|-)?\d+\Z/i,
+
+      'percent' => /\A\d+%\Z/i,
+
+      'select_from_enum' => /\A(\+?\w+;?)*\Z/i
+  }
+
+
+end

data/app/models/system_manager.rb

@@ -0,0 +1,88 @@
+class SystemManager
+
+  attr_reader :interpreter
+
+  SERVICES = ['rmails', 'postfix', 'dovecot', 'nginx', 'dspam', 'amavisd']
+
+  def self.change_configuration_and_restart
+    sm = ServiceManager.new
+    sm.write_new_values_for Property::Dovecot
+    sm.write_new_values_for Property::Postfix
+    sm.write_new_values_for Property::Dspam
+    sm.smart_restart_all
+  end
+
+  def self.run(service, goal)
+    sm = ServiceManager.new
+    sm.send "#{service}_switch", goal
+  end
+
+  def restart_all
+    SERVICES[1..-1].each do |s|
+      @interpreter.service_manager.restart s
+    end
+  end
+
+  def smart_restart_all
+    @services_to_restart.each do |klass|
+      @interpreter.service_manager.restart SERVICES[klass.service]
+    end
+    @services_to_restart = []
+  end
+
+  def initialize
+    @i = @interpreter = AutomateIt.new
+    @services_to_restart = []
+  end
+
+  def write_new_values_for(properties_of_service)
+    # find properties to write
+    new_values = properties_of_service.all_new_values_for_write
+    if new_values.any?
+      @services_to_restart << properties_of_service
+    else
+      return true
+    end
+    # update database in transaction
+    properties_of_service.transaction do
+      new_values.each do |property|
+        property.value = property.new_value
+        property.new_value = nil
+        property.save
+      end
+    end
+    # write properties with template
+    new_values.group_by(&:template).each do |template, properties|
+      locals = {}
+      properties.each do |property|
+        locals[property.key] = property.value
+      end
+      properties_of_service.send "#{template}_template", @interpreter, locals
+    end
+  end
+
+  def dspam_switch(goal)
+    Property::Postfix.dspam_power @interpreter, goal
+    action = goal ? 'restart' : 'stop'
+    @i.service_manager.send('restart', "postfix") # postfix will only restart
+    @i.service_manager.send(action, "dspam")
+  end
+
+
+  def amavis_switch(goal)
+
+  end
+
+  def dkim_switch(goal)
+    Property::Postfix.dkim_power @interpreter, goal
+    action = goal ? 'restart' : 'stop'
+    @i.service_manager.send('restart', "postfix")
+    @i.service_manager.send(action, "opendkim")
+  end
+
+  def postfix_switch(goal)
+    action = goal ? 'restart' : 'stop'
+    @i.service_manager.send(action, "postfix")
+  end
+
+end