cheese 0.0.1

data/lib/cheese-setup.rb

@@ -0,0 +1,144 @@
+#!/usr/bin/env ruby
+# cheese-setup.rb
+#
+# Configure cheese-tools defaults and misc information
+# This is a simple run-once script which needs LOTS of refactoring and cleanup
+#
+
+require 'rubygems'
+require 'highline/import'
+require 'fileutils'
+require 'yaml'
+
+def draw_line(count=20)
+  @line = '-' * count
+  say "<%= color(@line, :horizontal_line) %>"
+end
+
+def draw_box
+  draw_line 50
+  yield
+  draw_line 50
+end
+
+CHEESE_CONF = "/etc/cheese/cheese.conf"
+SUPPORTED_SCMS = { 
+  :svn =>   { :name => 'Subversion', :binary => 'svnserve' } }
+SUPPORTED_DBS =  { 
+  :psql =>  { :name => 'Postgresql', :binary => 'psql' },
+  :mysql => { :name => 'MySQL', :binary => 'mysql' } }
+SILENCER = ">/dev/null 2>/dev/null"
+begin
+  FileUtils.mkdir '/etc/cheese' unless File.exists? '/etc/cheese'
+  FileUtils.touch CHEESE_CONF
+  
+  preferences = {}
+  
+  # Create a color scheme, naming color patterns with symbol names.
+  ft = HighLine::ColorScheme.new do |cs|
+          cs[:headline]         = [ :bold, :yellow ]
+          cs[:horizontal_line]  = [ :bold, :white ]
+          cs[:important]        = [ :bold, :red ]
+       end
+
+  # Assign that color scheme to HighLine...
+  HighLine.color_scheme = ft
+
+  say "<%= color('Welcome to the cheese setup utility', :headline) %>"
+  draw_line 37
+  
+  say "Answer each of the following questions and then you'll be"
+  say "ready to start using cheese:"
+
+  # SCM
+  choose do |menu|
+    SUPPORTED_SCMS.each do |scm, details|
+      menu.choice(scm, details[:name]) do |command, extras|
+        preferences[:scm] = { :name => details[:name], :binary => details[:binary]}
+        say "What would you like the default #{details[:name]} username to be?"
+        preferences[:scm_user] = ask("=> ", String)
+        say "and what password would you like?"
+        preferences[:scm_pass] = ask("=> ", String){|prompt| prompt.echo = "*" }
+      end
+    end
+    menu.choice(:skip, "You don't want to use one") do |command, details|
+      preferences[:scm] = :skip
+    end
+    menu.choice(:quit, "Exit setup") { exit }
+  end
+  
+  # Database
+  say "What database engine are you using?"
+  choose do |menu|
+    SUPPORTED_DBS.each do |db, details|
+      menu.choice(db, details[:name]) do |command, extras|
+        preferences[:database_type] = { :name => details[:name], :binary => details[:binary] }
+      end
+    end
+    menu.choice(:skip, "You don't want to use one") do |command, details|
+      preferences[:database_type] = :skip
+    end
+    menu.choice(:quit, "Exit setup") { exit }
+  end
+  
+  # Rails stack combination
+  say "What Ruby on Rails stack are you using?"
+  choose do |menu|
+    menu.choice(:nginx, "Nginx with Mongrel") do |command, details|
+      preferences[:stack] = :nginx
+    end
+    menu.choice(:skip, "You don't want to use one") do |command, details|
+      preferences[:scm] = :skip
+    end
+    menu.choice(:quit, "Exit setup") { exit }
+  end
+  
+  draw_box do
+    say "<%= color('You need a normal account for safety reasons, running as', :bold) %>"
+    say "<%= color('root is dangerous, so we need to do that', :bold) %>"
+  end
+  
+  user = ask("What username would you like for logging into the system?", String)
+  unless user.empty?
+    preferences[:user] = user
+    %x{ useradd -c "Cheesey User" -d /home/#{preferences[:user].chomp} --create-home #{preferences[:user].chomp} #{SILENCER} }
+    say "and what password would you like to use with that?"
+    %x{ passwd #{preferences[:user]} }
+    %x{ addgroup scm #{SILENCER} }
+    %x{ usermod -g scm #{preferences[:user]} }
+  else
+    say "Skipping user creation (I assume you must have done this already then)"
+  end
+  
+  say "Saving and setting up those selected options"
+  
+  scm = %x{ which #{ preferences[:scm][:binary] } }
+  xinetd_service = <<-EOF
+    service #{ preferences[:scm][:binary] }
+    {
+      socket_type = stream
+      protocol = tcp
+      user = scm
+      wait = no
+      disable = no
+      server = #{ scm.chomp }
+      server_args = -i --root=/var/src
+      port = 3690
+    }
+    EOF
+  File.open("/etc/xinetd.d/#{preferences[:scm][:binary]}", 'w+') do |file|
+    file.puts xinetd_service
+  end
+  %x{ /etc/init.d/xinetd restart #{SILENCER} }
+  
+  File.open(CHEESE_CONF, "w+", 0640) {|f| YAML.dump(preferences, f)}
+  
+  say "Done"
+  
+rescue Errno::EPERM
+  puts "This script must be run with root privileges"
+  exit
+rescue Errno::EACCES
+  puts "This script must be run with root privileges"
+  exit
+end

data/lib/cheese/version.rb

@@ -0,0 +1,9 @@
+module Cheese #:nodoc:
+  module VERSION #:nodoc:
+    MAJOR = 0
+    MINOR = 0
+    TINY  = 1
+
+    STRING = [MAJOR, MINOR, TINY].join('.')
+  end
+end

data/lib/controller.rb

@@ -0,0 +1,152 @@
+# controller.rb
+#
+# Validate the options given and delegate jobs to the other classes
+#
+
+begin
+  require 'rubygems'
+rescue LoadError
+  # no rubygems to load, so we fail silently
+end
+
+require 'yaml'
+require 'process'
+require 'verbose'
+require 'database/mysql'
+require 'database/postgresql'
+require 'scm/subversion'
+require 'web/domain'
+require 'web/mongrel'
+require 'web/nginx'
+require 'web/proxy'
+require 'web/virtual_host'
+
+module Cheese
+  class Controller
+    
+    CHEESE_CONF = "/etc/cheese/cheese.conf"
+    NGINX_CONF  = '/etc/nginx/nginx.conf'
+    
+    def initialize(options={})
+      # Check cheese --setup has been run
+      unless File.exists?(CHEESE_CONF)
+        puts "Please run cheese --setup before using any cheese actions."
+        exit
+      end
+      # Need a better way of doing this, so it works in all files
+      Cheese::Verbose.dry_run = options.delete(:dry_run)
+      Cheese::Verbose.loud = options.delete(:verbose)
+      @preferences = YAML.load(File.open(CHEESE_CONF))
+      @options = options
+      run_actions
+    end
+
+private
+    
+    # Run all actions necessary
+    def run_actions
+      # List vhosts
+      if ( @options[:actions].include?(:web_server) || @options[:actions].include?(:list_vhosts))
+        @nginx = Cheese::Nginx::Config.new(NGINX_CONF)
+      end
+      
+      if @options[:actions].include? :list_vhosts
+        Cheese::Verbose.log_task("listing vhosts in nginx.conf") do
+          begin
+            @nginx.domains.each_with_index {|domain, i| puts "#{i}. #{domain.vhost.domain} - #{domain.proxy.ports.size} threads" }
+          rescue Exception => e
+            puts "Error listing vhosts:"
+            puts e.message
+            puts "exiting"
+            exit
+          end
+        end
+      end
+
+      # Nginx
+      if @options[:actions].include? :web_server
+        begin
+          Cheese::Verbose.log_task("back up nginx.conf") do
+            FileUtils.cp(NGINX_CONF, NGINX_CONF + ".old") if File.exists?(NGINX_CONF)
+          end
+        rescue Errno::EPERM
+          puts "This script must be run with root privileges"
+          exit
+        rescue Errno::EACCES
+          puts "This script must be run with root privileges"
+          exit
+        end
+        
+        case @options[:remove]
+        when false
+          Cheese::Verbose.log_task("create nginx vhost (#{@options[:name]})") do
+            @added_domain = @nginx.add @options
+          end
+        when true
+          Cheese::Verbose.log_task("remove nginx vhost (#{@options[:name]})") do 
+            @removed_domain = @nginx.remove @options
+          end
+        end
+        
+        @nginx.save
+        @nginx.restart
+      end
+      
+      # Subversion
+      if @options[:actions].include? :scm
+        if @options[:remove]
+          Cheese::Verbose.log_task("remove subversion repository (#{@options[:name]})") do
+            svn = Cheese::Subversion::Repository.remove @options[:name]
+          end
+        else
+          Cheese::Verbose.log_task("add subversion repository (#{@options[:name]})") do
+            svn = Cheese::Subversion::Repository.create @options[:name]
+          end
+          Cheese::Verbose.log_task("set the default permissions on the repository") do
+            user, pass = @preferences[:scm_user].chomp, @preferences[:scm_pass].chomp
+            Cheese::Subversion::Repository.set_permissions( :name => @options[:name],
+                                                            :access => {:anon => :none, :auth => :write},
+                                                            :users => {:user => user, :pass => pass})
+          end
+        end
+      end
+
+      # Mongrel cluster file
+      if @options[:actions].include? :app_server
+        if @options[:remove]
+          Cheese::Verbose.log_task("remove the mongrel_cluster file") do
+            Cheese::Mongrel.remove(@removed_domain)
+          end
+        else
+          Cheese::Verbose.log_task("create the mongrel_cluster file") do
+            Cheese::Mongrel.create(@added_domain.host, @added_domain.proxy.ports)
+          end
+        end
+      end
+      
+      # Database
+      if @options[:actions].include? :database
+        if @options[:remove]
+          Cheese::Verbose.log_task("drop a database") do
+            Cheese::Verbose.log_task(" requiring lib/#{@options[:database_type]}")
+            require "database/#{@options[:database_type]}"
+            Cheese::Verbose.log_task(" creating class #{@options[:database_type].to_s.capitalize}")
+            db_klass = Cheese.const_get(@options[:database_type].to_s.capitalize.intern)
+            Cheese::Verbose.log_task(" executing remove command on #{@options[:name]}")
+            db_klass.remove(@options[:name])
+          end
+        else
+          Cheese::Verbose.log_task("create a database") do
+            Cheese::Verbose.log_task(" requiring lib/#{@options[:database_type]}")
+            require "database/#{@options[:database_type]}"
+            Cheese::Verbose.log_task(" creating class #{@options[:database_type].to_s.capitalize}")
+            db_klass = Cheese.const_get(@options[:database_type].to_s.capitalize.intern)
+            Cheese::Verbose.log_task(" executing create command")
+            db_klass.create(@options[:name])
+          end
+        end
+      end
+    end
+    
+  end
+end

data/lib/database/mysql.rb

@@ -0,0 +1,50 @@
+# mysql.rb
+#
+# Create and remove mysql databases
+#
+
+require 'highline/import'
+require 'tempfile'
+
+module Cheese
+  
+  class Mysql
+    
+    # create a user and new db
+    def self.create(name)
+      puts "What is the root password for MySQL?"
+      pass = ask("=> "){|prompt| prompt.echo = "*" }.chomp
+      while true
+        puts "What is the database password you want for #{name.gsub(".", "_")}?"
+        dbpass1 = ask("=> "){|prompt| prompt.echo = "*" }.chomp
+        puts "Confirm that"
+        dbpass2 = ask("=> "){|prompt| prompt.echo = "*" }.chomp
+        if dbpass1.chomp == dbpass2.chomp
+          break
+        else
+          puts "Those passwords didn't match, try again:"
+        end
+      end
+      puts "adding db and user"
+      tmpfile = Tempfile.new("mysql-create")
+      tmpfile.puts "CREATE DATABASE #{name.gsub(".", "_")};"
+      tmpfile.puts "GRANT SELECT,INSERT,UPDATE,DELETE,CREATE,DROP ON *.* TO '#{name.gsub(".", "_")}'@'localhost' IDENTIFIED BY '#{dbpass1}';"
+      tmpfile.close
+      %x{ mysql -u root -p#{pass} mysql < #{tmpfile.path} }
+    end
+    
+    # remove a user and new db
+    def self.remove(name)
+      puts "What is the root password for MySQL?"
+      pass = ask("=> "){|prompt| prompt.echo = "*" }
+      
+      tmpfile = Tempfile.new("mysql-drop")
+      tmpfile.puts "DROP DATABASE #{name.gsub(".", "_")};"
+      tmpfile.puts "DROP USER '#{name.gsub(".", "_")}'@'localhost';"
+      tmpfile.close
+      %x{ mysql -u root -p#{pass} < #{tmpfile.path} }
+    end
+    
+  end
+  
+end

data/lib/database/postgresql.rb

@@ -0,0 +1,29 @@
+# postgresql.rb
+#
+# Create and remove postgresql databases
+#
+
+module Cheese
+  
+  class Postgresql
+    SILENCER = ">/dev/null 2>/dev/null"
+    
+    # create a user and new db
+    def self.create(name)
+      while
+        puts "We need to set a database password for #{name.gsub(".", "_")}."
+      end
+      
+      %x{ su -c "createuser -S -D -R #{name.gsub(".", "_")} -W" postgres }
+      %x{ su -c "createdb #{name.gsub(".", "_")}" postgres #{SILENCER} }
+    end
+    
+    # remove a user and new db
+    def self.remove(name)
+      %x{ su -c "dropuser #{name.gsub(".", "_")}" postgres #{SILENCER} }
+      %x{ su -c "dropdb #{name.gsub(".", "_")}" postgres #{SILENCER} }
+    end
+    
+  end
+  
+end

data/lib/process.rb

@@ -0,0 +1,17 @@
+# process.rb
+#
+# Run a process as a different user
+# Thanks to the "Ruby Cookbook" for this one
+
+module Process
+  def as_uid(uid=0)
+    old_euid, old_uid = Process.euid, Process.uid
+    Process.euid, Process.uid = uid, uid
+    begin
+      yield
+    ensure
+      Process.euid, Process.uid = old_euid, old_uid
+    end
+  end
+  module_function(:as_uid)
+end

data/lib/scm/subversion.rb

@@ -0,0 +1,83 @@
+# subversion.rb
+#
+# Acts as a subversion repository
+# This could be done with the svn ruby bindings, but it's simple
+# stuff that for now will be fine using system
+
+require 'fileutils'
+
+module Cheese
+  module Subversion
+    class Repository
+      DEFAULT_ACCESS = { :anon => :none, :auth => :write }
+      SILENCER = ">/dev/null 2>/dev/null"
+      def self.create(name)
+        # Create the repo
+        %x{ svnadmin create /var/src/#{name} #{SILENCER} }
+        %x{ chown svn:svn /var/src/#{name} #{SILENCER} }
+        %x{ chmod -R 774 /var/src/#{name} #{SILENCER} }
+      
+        folder = self.create_tmp_structure
+        self.import(name, folder)
+      end
+      
+      # Create the default file structure
+      def self.create_tmp_structure
+        FileUtils.mkdir_p(  File.join(Dir.tmpdir, "/svn_structure/branches")) unless File.exists? File.join(Dir.tmpdir, "/svn_structure/branches")
+        FileUtils.mkdir(    File.join(Dir.tmpdir, "/svn_structure/trunk"))    unless File.exists? File.join(Dir.tmpdir, "/svn_structure/trunk")
+        FileUtils.mkdir(    File.join(Dir.tmpdir, "/svn_structure/tags"))     unless File.exists? File.join(Dir.tmpdir, "/svn_structure/tags")
+        File.join(Dir.tmpdir, "/svn_structure")
+      end
+      
+      # Set the permissions on a repository, defaults are anon-access: none and auth-access: write.
+      # These options can be changed with options
+      #
+      # ==== Options
+      # +options+
+      #   +access+: A Hash containing :anon and :auth which can be set to either :none, :read or :write
+      #   +users+:  An Array or Hash containing user/pass combinations 
+      #             e.g. { :name => "Jamie", :password => "Chees3y" }
+      def self.set_permissions(options={})
+        access = options.delete(:access) || DEFAULT_ACCESS
+        name = options.delete(:name)
+        File.open("/var/src/#{name}/conf/svnserve.conf", "w+") do |file|
+          file.puts "# Generated on #{Time.now.to_s}"
+          file.puts "[general]"
+          file.puts "anon-access = #{access[:anon].to_s}"
+          file.puts "auth-access = #{access[:auth].to_s}"
+          file.puts "password-db = passwd"
+        end
+        File.open("/var/src/#{name}/conf/passwd", "w+") do |file|
+          file.puts "[users]"
+          if options[:users].is_a?Array
+            options[:users].each do |user, pass|
+              file.puts "#{user} = #{pass}"
+            end
+          elsif options[:users].is_a?Hash
+            file.puts "#{options[:users][:user]} = #{options[:users][:pass]}"
+          end
+        end
+      end
+      
+      def self.remove(name)
+        FileUtils.rm_rf("/var/src/#{name}")
+      end
+      
+      def self.import(name, files=[], extra_path="")
+        files.each do |file|
+          %x{ svn import #{file} file:///var/src/#{name}#{extra_path} -m "import of #{file}" }
+        end
+      end
+      
+      def self.remove_file(name, files=[])
+        if files.is_a?Array
+          files.each do |file|
+            %x{ svn delete #{file} file:///var/src/#{name} -m "deleted #{file}" #{SILENCER} } if File.exists? file
+          end
+        else
+          %x{ svn delete #{files} file:///var/src/#{name} -m "deleted #{files} #{SILENCER}" } if File.exists? files
+        end
+      end
+    end
+  end
+end