smartmachine 0.8.0 → 0.9.0

data/lib/smart_machine/grids/nginx.rb

@@ -1,137 +1,178 @@
-# The main SmartMachine Grids Nginx driver
+require 'yaml'
+require "base64"
+require 'bcrypt'
+
 module SmartMachine
-	class Grids
-		class Nginx < SmartMachine::Base
-			def initialize
-			end
-	
-			def up(*args)
-				args.flatten!
-				exposed = args.empty? ? '' : args.shift
-
-				if SmartMachine::Docker.running?
-					# Creating volumes
-					print "-----> Creating volume nginx-confd ... "
-					if system("docker volume create nginx-confd", out: File::NULL)
-						puts "done"
-					end
-
-					print "-----> Creating volume nginx-vhost ... "
-					if system("docker volume create nginx-vhost", out: File::NULL)
-						puts "done"
-					end
-
-					print "-----> Creating volume nginx-shtml ... "
-					if system("docker volume create nginx-shtml", out: File::NULL)
-						puts "done"
-					end
-
-					# Creating networks
-					unless system("docker network inspect nginx-network", [:out, :err] => File::NULL)
-						print "-----> Creating network nginx-network ... "
-						if system("docker network create nginx-network", out: File::NULL)
-							puts "done"
-						end
-					end
-
-					# Creating & Starting containers
-					print "-----> Creating container nginx ... "
-					if system("docker create \
-						--name='nginx' \
-						#{"--publish='80:80' --publish='443:443'" if exposed == '--exposed'} \
-						--volume='nginx-confd:/etc/nginx/conf.d/' \
-						--volume='nginx-vhost:/etc/nginx/vhost.d/' \
-						--volume='nginx-shtml:/usr/share/nginx/html' \
-						--volume='#{SmartMachine.config.user_home_path}/.smartmachine/grids/nginx/certificates:/etc/nginx/certs' \
-						--volume='#{SmartMachine.config.user_home_path}/.smartmachine/grids/nginx/fastcgi.conf:/etc/nginx/fastcgi.conf:ro' \
-						--volume='#{SmartMachine.config.user_home_path}/.smartmachine/grids/nginx/htpasswd:/etc/nginx/htpasswd:ro' \
-						--restart='always' \
-						--network='nginx-network' \
-						nginx:alpine", out: File::NULL)
-
-						puts "done"
-						print "-----> Starting container nginx ... "
-						if system("docker start nginx", out: File::NULL)
-							puts "done"
-						end
-					end
-
-					print "-----> Creating container nginx-gen ... "
-					if system("docker create \
-						--name='nginx-gen' \
-						--volumes-from nginx \
-						--volume='#{SmartMachine.config.user_home_path}/.smartmachine/grids/nginx/nginx.tmpl:/etc/docker-gen/templates/nginx.tmpl:ro' \
-						--volume='/var/run/docker.sock:/tmp/docker.sock:ro' \
-						--restart='always' \
-						--network='nginx-network' \
-						jwilder/docker-gen \
-						-notify-sighup nginx -watch /etc/docker-gen/templates/nginx.tmpl /etc/nginx/conf.d/default.conf", out: File::NULL)
-
-						puts "done"
-						print "-----> Starting container nginx-gen ... "
-						if system("docker start nginx-gen", out: File::NULL)
-							puts "done"
-						end
-					end
-
-					print "-----> Creating container nginx-letsencrypt ... "
-					if system("docker create \
-						--name='nginx-letsencrypt' \
-						--env NGINX_PROXY_CONTAINER=nginx \
-						--env NGINX_DOCKER_GEN_CONTAINER=nginx-gen \
-						--env DEFAULT_EMAIL=#{SmartMachine.config.sysadmin_email} \
-						--volumes-from nginx \
-						--volume='/var/run/docker.sock:/var/run/docker.sock:ro' \
-						--restart='always' \
-						--network='nginx-network' \
-						jrcs/letsencrypt-nginx-proxy-companion", out: File::NULL)
-
-						puts "done"
-						print "-----> Starting container nginx-letsencrypt ... "
-						if system("docker start nginx-letsencrypt", out: File::NULL)
-							puts "done"
-						end
-					end
-				end
-			end
-
-			def down
-				if SmartMachine::Docker.running?
-					# Stopping & Removing containers - in reverse order
-					print "-----> Stopping container nginx-letsencrypt ... "
-					if system("docker stop 'nginx-letsencrypt'", out: File::NULL)
-						puts "done"
-						print "-----> Removing container nginx-letsencrypt ... "
-						if system("docker rm 'nginx-letsencrypt'", out: File::NULL)
-							puts "done"
-						end
-					end
-
-					print "-----> Stopping container nginx-gen ... "
-					if system("docker stop 'nginx-gen'", out: File::NULL)
-						puts "done"
-						print "-----> Removing container nginx-gen ... "
-						if system("docker rm 'nginx-gen'", out: File::NULL)
-							puts "done"
-						end
-					end
-
-					print "-----> Stopping container nginx ... "
-					if system("docker stop 'nginx'", out: File::NULL)
-						puts "done"
-						print "-----> Removing container nginx ... "
-						if system("docker rm 'nginx'", out: File::NULL)
-							puts "done"
-						end
-					end
-
-					# Removing networks
-					print "-----> Removing network nginx-network ... "
-					if system("docker network rm nginx-network", out: File::NULL)
-						puts "done"
-					end
-				end
-			end
-		end
-	end
+  class Grids
+    class Nginx < SmartMachine::Base
+      def initialize
+        @home_dir = File.expand_path('~')
+      end
+
+      def uper
+        # Creating volumes
+        puts "-----> Creating volume nginx-confd ... "
+        if system("docker volume create nginx-confd", out: File::NULL)
+          puts "done"
+        end
+
+        puts "-----> Creating volume nginx-vhost ... "
+        if system("docker volume create nginx-vhost", out: File::NULL)
+          puts "done"
+        end
+
+        puts "-----> Creating volume nginx-shtml ... "
+        if system("docker volume create nginx-shtml", out: File::NULL)
+          puts "done"
+        end
+
+        # Creating networks
+        unless system("docker network inspect nginx-network", [:out, :err] => File::NULL)
+          puts "-----> Creating network nginx-network ... "
+          if system("docker network create nginx-network", out: File::NULL)
+            puts "done"
+          end
+        end
+
+        # Creating & Starting containers
+        puts "-----> Creating container nginx ... "
+        command = [
+          "docker create",
+          "--name='nginx'",
+          "--publish='80:80' --publish='443:443'",
+          "--volume='nginx-confd:/etc/nginx/conf.d/'",
+          "--volume='nginx-vhost:/etc/nginx/vhost.d/'",
+          "--volume='nginx-shtml:/usr/share/nginx/html'",
+          "--volume='#{@home_dir}/smartmachine/grids/nginx/certificates:/etc/nginx/certs'",
+          "--volume='#{@home_dir}/smartmachine/grids/nginx/fastcgi.conf:/etc/nginx/fastcgi.conf:ro'",
+          "--volume='#{@home_dir}/smartmachine/grids/nginx/htpasswd:/etc/nginx/htpasswd:ro'",
+          "--restart='always'",
+          "--network='nginx-network'",
+          "nginx:alpine"
+        ]
+        if system(command.compact.join(" "), out: File::NULL)
+          puts "done"
+          puts "-----> Starting container nginx ... "
+          if system("docker start nginx", out: File::NULL)
+            puts "done"
+          else
+            raise "Error: Could not start the created nginx container."
+          end
+        else
+          raise "Error: Could not create nginx container."
+        end
+
+        puts "-----> Creating container nginx-gen ... "
+        command = [
+          "docker create",
+          "--name='nginx-gen'",
+          "--volumes-from nginx",
+          "--volume='#{@home_dir}/smartmachine/grids/nginx/nginx.tmpl:/etc/docker-gen/templates/nginx.tmpl:ro'",
+          "--volume='/var/run/docker.sock:/tmp/docker.sock:ro'",
+          "--restart='always'",
+          "--network='nginx-network'",
+          "jwilder/docker-gen",
+          "-notify-sighup nginx -watch /etc/docker-gen/templates/nginx.tmpl /etc/nginx/conf.d/default.conf"
+        ]
+        if system(command.compact.join(" "), out: File::NULL)
+          puts "done"
+          puts "-----> Starting container nginx-gen ... "
+          if system("docker start nginx-gen", out: File::NULL)
+            puts "done"
+          else
+            raise "Error: Could not start the created nginx-gen container."
+          end
+        else
+          raise "Error: Could not create nginx-gen container."
+        end
+
+        puts "-----> Creating container nginx-letsencrypt ... "
+        command = [
+          "docker create",
+          "--name='nginx-letsencrypt'",
+          "--env NGINX_PROXY_CONTAINER=nginx",
+          "--env NGINX_DOCKER_GEN_CONTAINER=nginx-gen",
+          "--env DEFAULT_EMAIL=#{SmartMachine.config.sysadmin_email}",
+          "--volumes-from nginx",
+          "--volume='/var/run/docker.sock:/var/run/docker.sock:ro'",
+          "--restart='always'",
+          "--network='nginx-network'",
+          "jrcs/letsencrypt-nginx-proxy-companion"
+        ]
+        if system(command.compact.join(" "), out: File::NULL)
+          puts "done"
+          puts "-----> Starting container nginx-letsencrypt ... "
+          if system("docker start nginx-letsencrypt", out: File::NULL)
+            puts "done"
+          else
+            raise "Error: Could not start the created nginx-letsencrypt container."
+          end
+        else
+          raise "Error: Could not create nginx-letsencrypt container."
+        end
+      end
+
+      # Stopping & Removing containers - in reverse order
+      def downer
+        puts "-----> Stopping container nginx-letsencrypt ... "
+        if system("docker stop 'nginx-letsencrypt'", out: File::NULL)
+          puts "done"
+          puts "-----> Removing container nginx-letsencrypt ... "
+          if system("docker rm 'nginx-letsencrypt'", out: File::NULL)
+            puts "done"
+          end
+        end
+
+        puts "-----> Stopping container nginx-gen ... "
+        if system("docker stop 'nginx-gen'", out: File::NULL)
+          puts "done"
+          puts "-----> Removing container nginx-gen ... "
+          if system("docker rm 'nginx-gen'", out: File::NULL)
+            puts "done"
+          end
+        end
+
+        puts "-----> Stopping container nginx ... "
+        if system("docker stop 'nginx'", out: File::NULL)
+          puts "done"
+          puts "-----> Removing container nginx ... "
+          if system("docker rm 'nginx'", out: File::NULL)
+            puts "done"
+          end
+        end
+
+        # Removing networks
+        puts "-----> Removing network nginx-network ... "
+        if system("docker network rm nginx-network", out: File::NULL)
+          puts "done"
+        end
+      end
+
+      def create_htpasswd_files
+        htpasswd_dir = "#{Dir.pwd}/grids/nginx/htpasswd"
+
+        # Remove existing htpasswd_dir and recreate it.
+        FileUtils.rm_r htpasswd_dir if Dir.exist?(htpasswd_dir)
+        FileUtils.mkdir htpasswd_dir
+        FileUtils.touch "#{htpasswd_dir}/.keep"
+
+        # Add hostfiles to htpasswd_dir
+        get_users_from_file.each do |domain_name, users|
+          next unless users
+
+          file_data = ""
+          users.each do |user, password|
+            file_data += "#{user}:#{BCrypt::Password.create(password)}\n"
+          end
+          File.open("#{Dir.pwd}/grids/nginx/htpasswd/#{domain_name}", "w") { |file| file.write(file_data) }
+        end
+      end
+
+      private
+
+      def get_users_from_file
+        YAML.load_file("#{Dir.pwd}/config/users.yml") || Hash.new
+      end
+    end
+  end
 end

data/lib/smart_machine/grids/prereceiver/Dockerfile

@@ -1,7 +1,7 @@
 ARG SMARTMACHINE_VERSION
 
-FROM smartmachine:$SMARTMACHINE_VERSION
-LABEL maintainer="Timeboard <hello@timeboard.me>"
+FROM smartmachine/smartengine:$SMARTMACHINE_VERSION
+LABEL maintainer="Gaurav Goel <gaurav@timeboard.me>"
 
 # FCGI Essentials
 # --- IMPORTANT NOTE: This is custom built fcgiwrap package for alpine linux to account for NO_BUFFERING option.

data/lib/smart_machine/grids/prereceiver/pre-receive

@@ -0,0 +1,17 @@
+#!/bin/sh
+
+# oldrev, newrev, refname are a feature of the way in which Git executes the pre-receive hook.
+# See https://www.kernel.org/pub/software/scm/git/docs/githooks.html
+while read oldrev newrev refname; do
+    if [ $(git rev-parse --is-bare-repository) = true ]; then
+	REPOSITORY_BASENAME=$(basename "$PWD")
+	REPOSITORY_BASENAME=${REPOSITORY_BASENAME%.git}
+    else
+	REPOSITORY_BASENAME=$(basename $(readlink -nf "$PWD"/..))
+    fi
+    script --quiet --return --flush --command "smartmachine grid prereceiver prereceive $REPOSITORY_BASENAME $REMOTE_USER $oldrev $newrev $refname" /home/$(whoami)/machine/apps/containers/$REPOSITORY_BASENAME/typescript
+    if [ "$?" -eq 10 ]; then
+	exit 0
+    fi
+    exit 1
+done