preview_add 0.0.1

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Load dependencies from preview.gemspec
+gemspec

data/README.md

@@ -0,0 +1,26 @@
+preview_add
+===========
+
+Add preview sites.
+
+Add preview sites for the specified host. Creates the DNS, Vhost, htpasswd file and checks the site out from revision control
+
+Installation
+------------
+
+    $ gem install preview_add
+
+preview_add requires a config file in /etc/preview_add, the structure should look like:
+
+    ---
+    :preview_domain: flickerbox.com
+    :svn_repos: https://repos.com
+    :locations:
+      :htpasswd: /path/to/htpasswd/directory
+      :vhosts: /path/to/vhosts/directory
+      :sites_available: /path/to/sites_available/directory
+    :zerigo:
+      :user: zerigo_user
+      :api_key: zerigo_api_key
+      :zone_id: zone_id
+      :cname: the cname to the preview domain

data/Rakefile

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

data/bin/preview_add

@@ -0,0 +1,79 @@
+#!/usr/bin/env ruby
+#/ Usage: preview_add <options> <host>...
+#/
+#/ The host should be the host of the live site, not the preview site.
+#/
+#/ Add preview sites for the specified host. Creates the DNS, Vhost, htpasswd
+#/ file and checks the site out from revision control
+#/ 
+#/ Options
+#/   -f, --htpasswd   specify the htpasswd file to use
+#/   -u, --username   username for the htpasswd file
+#/   -p, --password   password for the htpasswd file
+#/   -h, --help       show this help message
+require 'preview_add'
+require 'optparse'
+
+if ENV['USER'] != 'root'
+  puts "You must run preview_add as root"
+  exit 1
+end
+
+def usage
+  puts File.readlines(__FILE__).
+    grep(/^#\/.*/).
+    map { |line| line.chomp[3..-1] }.
+    join("\n")
+end
+
+options = {}
+
+optparse = OptionParser.new do |opts| 
+  
+  opts.on( '-f', '--htpasswd [file]') do |arg|
+    options[:htpasswd] = arg
+  end
+  
+  opts.on( '-u', '--username [username]') do |arg|
+    options[:username] = arg
+  end
+  
+  opts.on( '-p', '--password [password]') do |arg|
+    options[:password] = arg
+  end
+
+  opts.on('-v', '--version') do
+    puts PreviewAdd::VERSION
+    exit 0
+  end
+
+  opts.on('-h', '--help', 'Display help info') do
+    usage
+    exit 0
+  end
+end
+
+begin
+  optparse.parse!
+  
+  # The url has no flag and will be in ARGV[0] if it's set
+  options[:host] = ARGV[0] || nil
+
+  # Reset ARGV for gets later
+  ARGV.clear
+  
+  mandatory = [:host]
+  missing = mandatory.select{ |param| options[param].nil? }
+  unless missing.empty?
+    puts "Missing option#{'s' if missing.length > 1}: #{missing.join(', ')}"
+    usage
+    exit 1
+  end
+rescue OptionParser::InvalidOption, OptionParser::MissingArgument
+  puts $!.to_s
+  usage
+  exit 1
+end
+
+preview = PreviewAdd::PreviewAdd.new(options)
+preview.add_site

data/lib/preview_add/config.rb

@@ -0,0 +1,20 @@
+require 'yaml'
+
+module PreviewAdd
+  
+  class Config
+    
+    FILE = "/etc/preview_add"
+    
+    def self.config
+      if File.exist?(FILE)
+        YAML.load_file(FILE)
+      else
+        puts "Please create a config file in /etc/preview_add"
+        exit 1
+      end
+    end
+
+  end
+
+end

data/lib/preview_add/dns.rb

@@ -0,0 +1,54 @@
+require 'rubygems'
+require 'zerigo_dns'
+
+module PreviewAdd
+  class Dns
+    
+    def initialize(options, host)
+      Zerigo::DNS::Base.user     = options[:zerigo][:user]
+      Zerigo::DNS::Base.password = options[:zerigo][:api_key]
+      Zerigo::DNS::Base.format   = :xml
+      
+      @zone_id = options[:zerigo][:zone_id]
+      @host    = host
+      @cname   = options[:preview_domain]
+    end
+    
+    def add_cname
+      vals = {
+        :hostname  => @host,
+        :host_type => 'CNAME',
+        :data      => @cname,
+        :zone_id   => @zone_id
+      }
+      
+      begin
+        Zerigo::DNS::Host.create(vals)
+      rescue Exception => e
+        if e.to_s.include? 'wrong number of arguments (2 for 1)'
+          puts "Successfully created #{@host}.#{@cname}"
+        else
+          puts e
+        end
+      end
+      
+    end
+    
+    def record_exists?(hostname)
+      begin
+        hosts = Zerigo::DNS::Host.find(:all, :params=>{:zone_id => @zone_id})
+        hosts.any? { |hash| hash.hostname == hostname }
+      rescue Exception => e
+        puts e
+        exit 1
+      end
+    end
+    
+    def create_if_not_exists
+      unless record_exists? @host
+        add_cname
+      end
+    end
+    
+  end
+end

data/lib/preview_add/names.rb

@@ -0,0 +1,18 @@
+module PreviewAdd
+  class Names
+    
+    attr_reader :document_root, :domain, :server_name, :host, :base_name
+    
+    def initialize(host)
+      @host = host.dup
+      @host.gsub!(/^(https?:\/\/)?(www\.)?/, '')
+      
+      parts          = @host.split('.')
+      @document_root = parts.reverse.join('.')
+      @domain        = parts[-2, 2].reverse.join('.')
+      @server_name   = parts[0..-2].join('.')
+      @base_name     = parts[-2]
+    end
+    
+  end
+end

data/lib/preview_add/svn.rb

@@ -0,0 +1,13 @@
+module PreviewAdd
+  class Svn
+    
+    def self.co(config, names)
+      url      = "#{config[:svn_repos]}/#{names.document_root}/trunk/"
+      doc_root = "#{config[:locations][:vhosts]}/#{names.document_root}/"
+      system "svn co #{url} #{doc_root}"
+      system "chown -R webmaster:webmaster #{doc_root}"
+    end
+    
+  end
+end
+    

data/lib/preview_add/version.rb

@@ -0,0 +1,3 @@
+module PreviewAdd
+  VERSION = "0.0.1"
+end

data/lib/preview_add/vhost.erb

@@ -0,0 +1,10 @@
+<VirtualHost *:80>
+	ServerName <%= @server_name %>
+	DocumentRoot <%= @document_root %>
+	<Directory <%= @document_root %>>
+		AuthName "Preview Website"
+		AuthType Basic
+		AuthUserFile <%= @auth_user_file %>
+		Require valid-user
+	</Directory>
+</VirtualHost>

data/lib/preview_add/vhost.rb

@@ -0,0 +1,86 @@
+require 'erb'
+require 'fileutils'
+
+module PreviewAdd
+  class Vhost
+    
+    include FileUtils
+    
+    attr_accessor :vhost
+    
+    def initialize(config, names)
+      @host = config[:host]
+      
+      @config = config
+      @names = names
+      
+      @auth_name      = 'TODO'
+      @auth_user_file = auth_user_file
+      @document_root  = File.join(@config[:locations][:vhosts], @names.document_root)
+      @server_name    = "#{@names.server_name}.#{@config[:preview_domain]}"
+    end
+    
+    def auth_user_file
+      htpasswd_dir = @config[:locations][:htpasswd];
+      extension = '.htpasswd'
+      auth_user_file = File.join(htpasswd_dir, "#{@names.domain + extension}")
+
+      unless @config[:htpasswd].nil?
+        if @config[:htpasswd].end_with?(extension)
+          auth_user_file = File.join(htpasswd_dir, "#{@config[:htpasswd]}")
+        else
+          auth_user_file = File.join(htpasswd_dir, "#{@config[:htpasswd] + extension}")
+        end
+      end
+
+      auth_user_file
+    end
+
+    def generate_vhost
+      template = File.join(File.dirname(__FILE__), 'vhost.erb')
+      vhost = ERB.new(File.read(template))
+      vhost.result(binding)
+    end
+    
+    def htpasswd(username, password)
+      `htpasswd -bn #{username} #{password}`
+    end
+    
+    def create!
+      # Create the document root
+      unless File.exist? @document_root
+        mkdir(@document_root)
+      end
+      
+      # Create the vhost
+      File.open(File.join(@config[:locations][:sites_available], @names.document_root), 'w') { 
+        |f| f.write(generate_vhost) 
+      }
+      
+      # Create the htpasswd file if it doesn't exist
+      unless File.exists? @auth_user_file
+        username = @config[:username] || @names.base_name
+        password = @config[:password] || nil
+        
+        if password.nil?
+          puts "No htpasswd file exists."
+          puts "Username: #{username}"
+          print "Password: "
+          password = gets.to_s.strip
+        end
+        
+        File.open(@auth_user_file, 'w') { |f| f.write(htpasswd(username, password)) }
+      end
+      
+      unless @config[:mode] == 'test'
+        # Enable the site
+        `a2ensite #{@names.document_root}`
+
+        # reload apache
+        `service apache2 reload`
+      end
+      
+    end
+    
+  end
+end

data/lib/preview_add.rb

@@ -0,0 +1,45 @@
+require "preview_add/version"
+require "preview_add/config"
+require "preview_add/names"
+require "preview_add/dns"
+require "preview_add/vhost"
+require "preview_add/svn"
+
+module PreviewAdd
+  class PreviewAdd
+    
+    attr_reader :config, :names
+    
+    def initialize(opts = {})
+      @config = Config.config.merge(opts)
+      @names = Names.new(@config[:host])
+    end
+    
+    def add_dns
+      dns = Dns.new(@config, @names.server_name)
+      dns.create_if_not_exists
+    end
+    
+    def add_vhost
+      vhost = Vhost.new(@config, @names)
+      vhost.create!
+    end
+    
+    def checkout_site
+      Svn.co(@config, @names)
+    end
+    
+    def add_site
+      # Add DNS Entry
+      add_dns
+      
+      # Add Vhost
+      add_vhost
+      
+      # Checkout site
+      checkout_site
+      
+    end
+    
+  end
+end

data/preview_add.gemspec

@@ -0,0 +1,22 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "preview_add/version"
+
+Gem::Specification.new do |s|
+  s.name        = "preview_add"
+  s.version     = PreviewAdd::VERSION
+  s.authors     = ["Ben Ubois"]
+  s.email       = ["ben@benubois.com"]
+  s.homepage    = "https://github.com/flickerbox/preview"
+  s.summary     = %q{Add preview sites}
+  s.description = %q{Add preview sites for the specified host. Creates the DNS, Vhost, htpasswd file and checks the site out from revision control}
+
+  s.rubyforge_project = "preview"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+  
+  s.add_dependency "zerigo_dns"
+end

metadata

@@ -0,0 +1,94 @@
+--- !ruby/object:Gem::Specification 
+name: preview_add
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: 
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Ben Ubois
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-09-18 00:00:00 -07:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: zerigo_dns
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+description: Add preview sites for the specified host. Creates the DNS, Vhost, htpasswd file and checks the site out from revision control
+email: 
+- ben@benubois.com
+executables: 
+- preview_add
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Gemfile
+- README.md
+- Rakefile
+- bin/preview_add
+- lib/preview_add.rb
+- lib/preview_add/config.rb
+- lib/preview_add/dns.rb
+- lib/preview_add/names.rb
+- lib/preview_add/svn.rb
+- lib/preview_add/version.rb
+- lib/preview_add/vhost.erb
+- lib/preview_add/vhost.rb
+- preview_add.gemspec
+has_rdoc: true
+homepage: https://github.com/flickerbox/preview
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: preview
+rubygems_version: 1.6.2
+signing_key: 
+specification_version: 3
+summary: Add preview sites
+test_files: []
+