Merge branch 'update_puppet' of github.com:colab-community/colab into update_puppet

Sergio Oliveira
2 parents fd68bc4f 24b89caf
Showing 2 changed files with 101 additions and 54 deletions Show diff stats
Vagrantfile
fabfile.py
 # -*- mode: ruby -*-
 # vi: set ft=ruby :
+# CHOOSE THE DISTRO FOR COLAB VM (set the distro variable):
+# - ubuntu_precise
+# - ubuntu_trusty
+# - centos_65
+
+distro = "centos_65"
+
 # Vagrantfile API/syntax version. Don't touch unless you know what you're doing!
 VAGRANTFILE_API_VERSION = "2"
@@ -10,24 +17,43 @@ Vagrant.configure(VAGRANTFILE_API_VERSION) do |config|
   # please see the online documentation at vagrantup.com.
   # Every Vagrant virtual environment requires a box to build off of.
-  config.vm.box = "trusty64"
-  # The url from where the 'config.vm.box' box will be fetched if it
-  # doesn't already exist on the user's system.
-  config.vm.box_url = "https://cloud-images.ubuntu.com/vagrant/trusty/current/trusty-server-cloudimg-amd64-vagrant-disk1.box"
+  config.vm.define "colab" do |colab|
+
+    colab.vm.box = distro
+
+    if distro == "ubuntu_precise"
+      puts "Installing Ubuntu Precise VM!"
+      colab.vm.box_url = "https://cloud-images.ubuntu.com/vagrant/precise/current/precise-server-cloudimg-amd64-vagrant-disk1.box"
+    elsif distro == "ubuntu_trusty"
+      puts "Installing Ubuntu Trusty VM!"
+      colab.vm.box_url = "https://cloud-images.ubuntu.com/vagrant/trusty/current/trusty-server-cloudimg-amd64-vagrant-disk1.box"
+    elsif distro == "centos_65"
+      puts "Installing CentOS 6.5 VM!"
+      colab.vm.box_url = "https://github.com/2creatives/vagrant-centos/releases/download/v6.5.3/centos65-x86_64-20140116.box"
+    end
+
+    colab.vm.network "forwarded_port", guest: 80, host: 80
+    colab.vm.network "private_network", ip: "192.168.33.10"
+    colab.vm.provider "virtualbox" do |vb|
+      # Use VBoxManage to customize the VM. For example to change memory:
+      vb.customize ["modifyvm", :id, "--memory", "1024"]
+    end
+  end
+
+  # Disable automatic box update checking. If you disable this, then
+  # boxes will only be checked for updates when the user runs
+  # `vagrant box outdated`. This is not recommended.
+  # config.vm.box_check_update = false
   # Create a forwarded port mapping which allows access to a specific port
   # within the machine from a port on the host machine. In the example below,
   # accessing "localhost:8080" will access port 80 on the guest machine.
-  config.vm.network :forwarded_port, guest: 80, host: 8080
-  config.vm.network :forwarded_port, guest: 8000, host: 8000
-  config.vm.network :forwarded_port, guest: 5280, host: 5280
-  config.vm.network :forwarded_port, guest: 8080, host: 8081
-  config.vm.network :forwarded_port, guest: 8983, host: 8983
+  # config.vm.network "forwarded_port", guest: 80, host: 8080
   # Create a private network, which allows host-only access to the machine
   # using a specific IP.
-  config.vm.network "private_network", ip: "192.168.33.10"
+  # config.vm.network "private_network", ip: "192.168.33.10"
   # Create a public network, which generally matched to bridged network.
   # Bridged networks make the machine appear as another physical device on
@@ -43,40 +69,42 @@ Vagrant.configure(VAGRANTFILE_API_VERSION) do |config|
   # the path on the guest to mount the folder. And the optional third
   # argument is a set of non-required options.
   # config.vm.synced_folder "../data", "/vagrant_data"
-  config.vm.synced_folder ".", "/vagrant", type: "nfs"
   # Provider-specific configuration so you can fine-tune various
   # backing providers for Vagrant. These expose provider-specific options.
   # Example for VirtualBox:
   #
-  config.vm.provider "virtualbox" do |vb|
+  # config.vm.provider "virtualbox" do |vb|
   #   # Don't boot with headless mode
   #   vb.gui = true
   #
   #   # Use VBoxManage to customize the VM. For example to change memory:
-      vb.customize ["modifyvm", :id, "--memory", "1024"]
-  end
+  #   vb.customize ["modifyvm", :id, "--memory", "1024"]
+  # end
   #
   # View the documentation for the provider you're using for more
   # information on available options.
+  # Enable provisioning with CFEngine. CFEngine Community packages are
+  # automatically installed. For example, configure the host as a
+  # policy server and optionally a policy file to run:
+  #
+  # config.vm.provision "cfengine" do |cf|
+  #   cf.am_policy_hub = true
+  #   # cf.run_file = "motd.cf"
+  # end
+  #
+  # You can also configure and bootstrap a client to an existing
+  # policy server:
+  #
+  # config.vm.provision "cfengine" do |cf|
+  #   cf.policy_server_address = "10.0.2.15"
+  # end
+
   # Enable provisioning with Puppet stand alone.  Puppet manifests
   # are contained in a directory path relative to this Vagrantfile.
   # You will need to create the manifests directory and a manifest in
-  # the file base.pp in the manifests_path directory.
-  #
-  # An example Puppet manifest to provision the message of the day:
-  #
-  # # group { "puppet":
-  # #   ensure => "present",
-  # # }
-  # #
-  # # File { owner => 0, group => 0, mode => 0644 }
-  # #
-  # # file { '/etc/motd':
-  # #   content => "Welcome to your Vagrant-built virtual machine!
-  # #               Managed by Puppet.\n"
-  # # }
+  # the file default.pp in the manifests_path directory.
   #
   # config.vm.provision "puppet" do |puppet|
   #   puppet.manifests_path = "manifests"
@@ -95,7 +123,7 @@ Vagrant.configure(VAGRANTFILE_API_VERSION) do |config|
   #   chef.add_role "web"
   #
   #   # You may also specify custom JSON attributes:
-  #   chef.json = { :mysql_password => "foo" }
+  #   chef.json = { mysql_password: "foo" }
   # end
   # Enable provisioning with chef server, specifying the chef server URL,
@@ -9,6 +9,19 @@ from fabric.api import env, run, sudo, local
 from fabric.contrib.files import exists
 from fabric.context_managers import prefix, cd, settings, shell_env
+LINUX_DISTRO = ''
+DEBIAN_FAMILY = ['debian', 'ubuntu']
+REDHAT_FAMILY = ['centos', 'fedora']
+
+DISTRO_CMD = {'debian': ('apt-get', {
+				'install': '-y',
+				'update': '-y',
+			}), 
+	      'redhat': ('yum', {
+				'install': '-y',
+				'update': '-y',
+	                })
+             }
 APP_USER = APP_NAME = VENV_NAME = 'colab'
 REPO_URL = 'git@github.com:colab-community/colab.git'
@@ -34,6 +47,19 @@ WORKON_ENV = &#39;{} &amp;&amp; workon {}&#39;.format(SOURCE_VENV, VENV_NAME)
 MANAGE_PATH = os.path.join(REPO_PATH, 'src')
 SETTINGS_PATH = os.path.join(MANAGE_PATH, APP_NAME)
+def get_distro_family():
+    linux_name = run('python -c "import platform; print platform.dist()[0]"').lower()
+    if linux_name in DEBIAN_FAMILY:
+	return 'debian'
+    elif linux_name in REDHAT_FAMILY:
+	return 'redhat'
+    else :
+	error(colors.red('Distribuiton `{}` not supported.'.format(linux_name)))
+	exit(1)
+
+def cmd(family, command, args = ''):
+    pkgmanager, commands = DISTRO_CMD[family]
+    return '{} {} {} {}'.format(pkgmanager, command, commands[command], args)
 @task
 def environment(name=DEFAULT_ENVIRONMENT):
@@ -53,12 +79,10 @@ def environment(name=DEFAULT_ENVIRONMENT):
     env.update(environments[name])
     env.environment = name
-environment()
-
-
-def aptget_install(pkg):
-    sudo('DEBIAN_FRONTEND=noninteractive apt-get install -y -q {}'.format(pkg))
+def package_install(pkg): 
+    family = get_distro_family()
+    sudo(cmd(family, 'install', pkg ))
 def install_requirements():
     with cd(REPO_PATH), prefix(WORKON_ENV):
@@ -72,14 +96,12 @@ def install_requirements():
         else:
             run('pip install -r requirements.txt')
-
 def mkvirtualenv():
     if not exists('~/.virtualenvs/' + VENV_NAME):
         with prefix(SOURCE_VENV):
             run('mkvirtualenv ' + VENV_NAME)
             return True
-
 def manage(command):
     django_settings = env.get('django_settings')
     env_vars = {}
@@ -90,21 +112,17 @@ def manage(command):
         with cd(MANAGE_PATH), prefix(WORKON_ENV):
             run('python manage.py {}'.format(command))
-
 def syncdb():
     manage('syncdb')
-
 def migrate():
     manage('migrate')
-
 def collectstatic():
     sudo('mkdir -p /usr/share/nginx/{}'.format(APP_NAME))
     sudo('chown {} /usr/share/nginx/{}'.format(env.user, APP_NAME))
     manage('collectstatic --noinput')
-
 def create_local_settings():
     with cd(SETTINGS_PATH), settings(user=env.superuser):
         env_local_settings = 'local_settings-{}.py'.format(env.environment)
@@ -113,7 +131,6 @@ def create_local_settings():
             run('ln -s {} {}'.format(env_local_settings, 'local_settings.py'))
             run('chown {} local_settings.py'.format(env.user))
-
 def update_code():
     if env.is_vagrant:
         if not exists(REPO_PATH):
@@ -126,24 +143,27 @@ def update_code():
         with cd(REPO_PATH):
             run('git pull')
-
 @task
 def bootstrap():
     """Bootstrap machine to run fabric tasks"""
-
     with settings(user=env.superuser):
-
-        if not exists('/usr/bin/git'):
-            aptget_install('git')
+	family = get_distro_family()
+	sudo(cmd(family, 'update'))
+        
+	if not exists('/usr/bin/git'):
+            package_install('git-core')
         if env.is_vagrant:
-            groups = 'sudo,vagrant'
+            groups = ['sudo' ,'vagrant']
             local('chmod -fR g+w {}'.format(PROJECT_PATH))
         else:
-            groups = 'sudo'
+            groups = ['sudo']
+
+	for group in groups:
+  	    sudo('groupadd -f {}'.format(group))
+
+        sudo('useradd {} -G {} -m -s /bin/bash'.format(APP_USER, ','.join(groups)))
-        sudo('useradd {} -G {} -m -s /bin/bash'.format(APP_USER, groups),
-             quiet=True)
         ssh_dir = '/home/{0}/.ssh/'.format(APP_USER)
         if not exists(ssh_dir):
             sudo('mkdir -p {0}'.format(ssh_dir))
@@ -162,13 +182,11 @@ def bootstrap():
             sudo('chmod 440 {}'.format(tmp_file))
             sudo('mv {} {}'.format(tmp_file, sudoers_file))
-
 @task
 def provision():
     """Run puppet"""
     update_code()
-
     puppet_path = os.path.join(REPO_PATH, 'puppet/')
     modules_path = os.path.join(puppet_path, 'modules')
     puppet_modules = '{}:/etc/puppet/modules'.format(modules_path)
@@ -187,7 +205,6 @@ def provision():
     sudo('puppet apply --modulepath={} {}'.format(puppet_modules, cmd))
-
 @task
 def ssh_keygen():
     """Create SSH credentials"""
@@ -201,7 +218,6 @@ def ssh_keygen():
     print('')
     print('Add the key above to your github repository deploy keys')
-
 @task
 def deploy(noprovision=False):
     """Deploy and run the new code (master branch)"""
@@ -222,3 +238,6 @@ def deploy(noprovision=False):
     migrate()
     sudo('supervisorctl start all')
+
+# Main
+environment()