vhost_admin 0.0.1

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in vhost_admin.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2012 Hitoshi Kurokawa
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,29 @@
+# VhostAdmin
+
+TODO: Write a gem description
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'vhost_admin'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install vhost_admin
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/bin/vhost_admin

@@ -0,0 +1,5 @@
+#!/usr/bin/env ruby
+
+require 'vhost_admin/runner'
+VhostAdmin::Runner.start
+

data/lib/vhost_admin.rb

@@ -0,0 +1,3 @@
+require "vhost_admin/version"
+require "vhost_admin/base"
+

data/lib/vhost_admin/base.rb

@@ -0,0 +1,244 @@
+require 'vhost_admin/version'
+require 'thor'
+require 'date'
+require 'yaml'
+
+class VhostAdmin::Base
+  CONFIG_FILE = '.vhost_admin.conf'
+
+  def initialize(options={})
+    @config = load_config
+    @crypt = options[:crypt] || false
+  end
+  def load_config
+    unless File.exist?(config_file)
+      create_config
+      puts "configure file: #{config_file} was generated.\nPlease execute after edit it."
+      exit
+    end
+      open(config_file) do |f|
+      YAML.load(f.read)
+    end
+  end
+  def create_config
+    config = {
+      'apache_conf_file'  => '/etc/httpd/conf/virtual/virtualhosts.conf',
+      'home_dir_base'   => '/var/www/vhost',
+      'mail_dir_base' => '/mail',
+      'backup_dir' => '/root/deletetmp',
+      'quota_user'  => 'quota1gb',
+      'encrypt' => 'cleartext'
+    }
+    open(config_file, 'w') do |f|
+      f.write config.to_yaml
+    end
+    File.chmod(0600, config_file)
+  end
+  def config_file
+    File.expand_path(CONFIG_FILE, ENV['HOME'])
+  end
+  def add_domain(domain, password, user=nil)
+    begin
+      unless File.exist?(@config['apache_conf_file'])
+        FileUtils.touch(@config['apache_conf_file'])
+      end
+      if include_apache_domain?(domain)
+        raise "#{domain} is already registered in the apache config file."
+      end
+
+      user = user || domain
+      add_unix_user(user, password)
+      add_apache_domain(domain, user)
+      add_mail_domain(domain, password)
+    rescue => error
+      exit_with_error(error)
+    end
+  end
+  def delete_domain(domain, user=nil)
+    begin
+      user = user || domain
+      backup_domain_data(domain, user)
+      delete_domain_data(domain, user)
+
+      res = apache_test
+      if res
+        puts "Apache test is OK!"
+        exec("apachectl graceful")
+      end
+    rescue => error
+      exit_with_error(error)
+    end
+  end
+
+  private
+
+  def exit_with_error(error)
+    puts "Error: #{error.message}"
+    puts "Exit"
+    exit
+  end
+
+  def delete_domain_data(domain, user)
+    delete_unix_user(user)
+    delete_domain_http_conf(domain)
+    delete_mail_data(domain)
+  end
+  def delete_domain_http_conf(domain)
+    unless include_apache_domain?(domain)
+      raise "#{domain} is not registered in the apache config file."
+    end
+    backup_conf= backup_apache_conf
+
+    flag = false
+    File.open(@config['apache_conf_file'], "w"){|wf|
+      File.open(backup_conf){|f|
+        f.each{|line|
+          if line =~ /# #{domain} :/
+            flag = true
+          end
+          if flag && line =~ /<\/VirtualHost>/
+            flag = false
+            next
+          end
+          wf.write line unless flag
+        }
+      }
+    }
+  end
+  def backup_domain_data(domain, user)
+    backup_web_data(domain, user)
+    backup_mail_data(domain)
+  end
+  def backup_web_data(domain, user)
+    output_value("Home", home_dir(user))
+    output_value("Domain", domain)
+    output_value("User", user)
+    file = "#{domain}_web.tgz"
+    tar_backup(file, home_dir(user))
+  end
+  def backup_mail_data(domain)
+    backup_file = File.join(@config['backup_dir'], domain+'_mail.txt')
+    exec("postfix_admin show #{domain} > #{backup_file}")
+    file = "#{domain}_mail.tgz"
+    tar_backup(file, mail_dir(domain))
+  end
+  def home_dir(user)
+    File.join(@config['home_dir_base'], user)
+  end
+  def mail_dir(domain)
+    File.join(@config['mail_dir_base'], domain)
+  end
+  def tar_backup(file, dir)
+    unless File.exist?(dir)
+      print "Can not find dir: #{dir}\n"
+      return
+    end
+    basename = File.basename(dir)
+    dirname  = File.dirname(dir)
+    tar_file = File.join(@config['backup_dir'], file)
+    res = exec("tar zcf #{tar_file} -C #{dirname} #{basename}")
+    unless res
+      raise("Tar command was failure")
+    end
+  end
+  def delete_dir(dir)
+    FileUtils.rm_rf(dir)
+  end
+  def add_unix_user(user, password)
+    exec("useradd #{user} -m -d #{home_dir(user)} -s /sbin/nologin")
+    exec("echo #{user}:#{password} | /usr/sbin/chpasswd")
+    if @config['quota_user']
+      exec("edquota -p #{@config['quota_user']} #{user}")
+    end
+    FileUtils.chmod(0711, home_dir(user))
+  end
+  def delete_unix_user(user)
+    exec("userdel -r #{user}")
+  end
+  def delete_mail_data(domain)
+    exec("postfix_admin delete_domain #{domain}")
+    delete_dir(mail_dir(domain))
+  end
+  def add_apache_domain(domain, user)
+    backup_apache_conf
+    open(@config['apache_conf_file'], 'a') do |f|
+      f.write(vhost_text(domain, user))
+    end
+
+    res = exec("apachectl -t")
+    if res
+      res_restart = exec("apachectl graceful")
+      unless res_restart
+        raise("Apache restart was failure.")
+      end
+    else
+      raise("Apachec config test was failure.")
+    end
+  end
+  def add_mail_domain(domain, password)
+    set_password =
+      if @crypt
+        md5_crypt(password)
+      else
+        password
+      end
+    exec("postfix_admin add_domain #{domain}")
+    exec("postfix_admin add_admin admin@#{domain} '#{set_password}'")
+    exec("postfix_admin add_admin_domain admin@#{domain} #{domain}")
+  end
+  def md5_crypt(str)
+    salt_set = ('a'..'z').to_a + ('A'..'Z').to_a + ('0'..'9').to_a + ['.', '/']
+    salt = '$1$'+salt_set.sample(8).join+'$'
+    str.crypt(salt)
+  end
+  def vhost_text(domain, user)
+    <<"EOS"
+
+# #{domain} : #{date_time_str} by vhost_admin
+<VirtualHost *:80>
+    ServerAdmin postmaster@#{domain}
+    ServerName www.#{domain}
+    DocumentRoot #{home_dir(user)}/public_html
+    ServerAlias #{domain}
+    SuexecUserGroup #{user} #{user}
+</VirtualHost>
+EOS
+  end
+  # ServerName www.example.com
+  def include_apache_domain?(domain)
+    open(@config['apache_conf_file']) do |f|
+      f.grep(/ServerName\s+www\.#{domain}/).size != 0
+    end
+  end
+  def backup_apache_conf
+    backup_conf = backup_apache_conf_file
+    if File.exist?(@config['apache_conf_file'])
+      FileUtils.copy(@config['apache_conf_file'], backup_apache_conf_file)
+    end
+    backup_conf
+  end
+  def backup_apache_conf_file
+    @config['apache_conf_file'] + backup_suffix
+  end
+  def date_time_str
+    DateTime.now.strftime("%Y-%m-%d %H:%M:%S")
+  end
+  def backup_suffix
+    DateTime.now.strftime(".%Y%m%d%H%M%S")
+  end
+  def exec(cmd, debug=false)
+    if debug
+      output_value("Command(test run)",cmd)
+      true
+    else
+      output_value("Command",cmd)
+      system(cmd)
+    end
+  end
+  def output_value(key, value)
+    print "#{key}\t: #{value}\n"
+  end
+  def apache_test
+    exec("apachectl -t")
+  end
+end

data/lib/vhost_admin/runner.rb

@@ -0,0 +1,16 @@
+require 'vhost_admin/base'
+
+class VhostAdmin::Runner < Thor
+  desc "add_domain", "add a domain"
+  method_options :crypt => :boolean
+  def add_domain(domain, password, user=nil)
+    admin = VhostAdmin::Base.new(options)
+    admin.add_domain(domain, password, user)
+  end
+
+  desc "delete_domain", "delete a domain"
+  def delete_domain(domain, user=nil)
+    admin = VhostAdmin::Base.new
+    admin.delete_domain(domain, user)
+  end
+end

data/lib/vhost_admin/version.rb

@@ -0,0 +1,3 @@
+class VhostAdmin
+  VERSION = "0.0.1"
+end

data/vhost_admin.gemspec

@@ -0,0 +1,22 @@
+# -*- encoding: utf-8 -*-
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'vhost_admin/version'
+
+Gem::Specification.new do |gem|
+  gem.add_dependency 'thor'
+  gem.add_dependency 'postfix_admin'
+
+  gem.name          = "vhost_admin"
+  gem.version       = VhostAdmin::VERSION
+  gem.authors       = ["Hitoshi Kurokawa"]
+  gem.email         = ["hitoshi@nextseed.jp"]
+  gem.description   = %q{Command Line Tools to manage virtual hosts fo Apache and Postfix}
+  gem.summary       = gem.description
+  gem.homepage      = ""
+
+  gem.files         = `git ls-files`.split($/)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.require_paths = ["lib"]
+end

metadata

@@ -0,0 +1,89 @@
+--- !ruby/object:Gem::Specification
+name: vhost_admin
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Hitoshi Kurokawa
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-09-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: postfix_admin
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Command Line Tools to manage virtual hosts fo Apache and Postfix
+email:
+- hitoshi@nextseed.jp
+executables:
+- vhost_admin
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/vhost_admin
+- lib/vhost_admin.rb
+- lib/vhost_admin/base.rb
+- lib/vhost_admin/runner.rb
+- lib/vhost_admin/version.rb
+- vhost_admin.gemspec
+homepage: ''
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: Command Line Tools to manage virtual hosts fo Apache and Postfix
+test_files: []