smartmachine 0.8.0 → 0.9.0

data/lib/smart_machine/grids/elasticsearch.rb

@@ -1,97 +1,78 @@
-# The main SmartMachine Grids Elasticsearch driver
 module SmartMachine
-	class Grids
-		class Elasticsearch < SmartMachine::Base
-			def initialize
-			end
+  class Grids
+    class Elasticsearch < SmartMachine::Base
+      def initialize(name:)
+        config = SmartMachine.config.grids.elasticsearch.dig(name.to_sym)
+        raise "elasticsearch config for #{name} not found." unless config
 
-			def install
-				puts "-----> Installing Elasticsearch"
+        @port = config.dig(:port)
 
-				print "-----> Creating settings for elasticsearch ... "
-				ssh = SmartMachine::SSH.new
-				ssh.run "echo 'vm.max_map_count=262144' | sudo tee /etc/sysctl.d/60-smartmachine-elasticsearch.conf && sudo sysctl --system"
-				puts "done"
+        @name = name.to_s
+        @home_dir = File.expand_path('~')
+      end
 
-				puts "-----> Elasticsearch Installation Complete"
-			end
+      def uper
+        # Creating networks
+        unless system("docker network inspect #{@name}-network", [:out, :err] => File::NULL)
+          print "-----> Creating network #{@name}-network ... "
+          if system("docker network create #{@name}-network", out: File::NULL)
+            puts "done"
+          end
+        end
 
-			def uninstall
-				puts "-----> Uninstalling Elasticsearch"
+        FileUtils.mkdir_p("#{@home_dir}/machine/grids/elasticsearch/#{@name}/data")
+        FileUtils.mkdir_p("#{@home_dir}/machine/grids/elasticsearch/#{@name}/logs")
 
-				print "-----> Removing settings for elasticsearch ... "
-				ssh = SmartMachine::SSH.new
-				# NOTE: sysctl does not reset this setting until restart of system even after sudo sysctl --system is run.
-				ssh.run "test -f /etc/sysctl.d/60-smartmachine-elasticsearch.conf && sudo rm /etc/sysctl.d/60-smartmachine-elasticsearch.conf && sudo sysctl --system"
-				puts "done"
+        # Creating & Starting containers
+        print "-----> Creating container #{@name} ... "
+        command = [
+          "docker create",
+          "--name='#{@name}'",
+          "--label='smartmachine.elasticsearch.name=#{@name}'",
+          "--env discovery.type=single-node",
+          "--env cluster.name=#{@name}-cluster",
+          "--env 'ES_JAVA_OPTS=-Xms512m -Xmx512m -Des.enforce.bootstrap.checks=true'",
+          "--env bootstrap.memory_lock=true",
+          "--ulimit memlock=-1:-1",
+          "--ulimit nofile=65535:65535",
+          "--user `id -u`:`id -g`",
+          "--publish='#{@port}:#{@port}'",
+          "--volume='#{@home_dir}/smartmachine/grids/elasticsearch/#{@name}/data:/usr/share/elasticsearch/data'",
+          "--volume='#{@home_dir}/smartmachine/grids/elasticsearch/#{@name}/logs:/usr/share/elasticsearch/logs'",
+          "--restart='always'",
+          "--network='#{@name}-network'",
+          "elasticsearch:7.4.1"
+        ]
+        if system(command.compact.join(" "), out: File::NULL)
+          puts "done"
+          puts "-----> Starting container #{@name} ... "
+          if system("docker start #{@name}", out: File::NULL)
+            puts "done"
+          else
+            raise "Error: Could not start the created #{@name} container"
+          end
+        else
+          raise "Error: Could not create #{@name} container"
+        end
+      end
 
-				puts "-----> Elasticsearch Uninstallation Complete"
-			end
+      def downer
+        # Stopping & Removing containers - in reverse order
+        print "-----> Stopping container #{@name} ... "
+        if system("docker stop '#{@name}'", out: File::NULL)
+          puts "done"
+          print "-----> Removing container #{@name} ... "
+          if system("docker rm '#{@name}'", out: File::NULL)
+            puts "done"
+          end
+        end
 
-			def update
-				self.uninstall
-				self.install
-			end
-
-			def up(*args)
-				args.flatten!
-				exposed = args.empty? ? '' : args.shift
-
-				if SmartMachine::Docker.running?
-					# Creating networks
-					unless system("docker network inspect elasticsearch-network", [:out, :err] => File::NULL)
-						print "-----> Creating network elasticsearch-network ... "
-						if system("docker network create elasticsearch-network", out: File::NULL)
-							puts "done"
-						end
-					end
-
-					# Creating & Starting containers
-					print "-----> Creating container elasticsearch ... "
-					if system("docker create \
-						--name='elasticsearch' \
-						--env discovery.type=single-node \
-						--env cluster.name=elasticsearch-cluster \
-						--env 'ES_JAVA_OPTS=-Xms512m -Xmx512m -Des.enforce.bootstrap.checks=true' \
-						--env bootstrap.memory_lock=true \
-						--ulimit memlock=-1:-1 \
-						--ulimit nofile=65535:65535 \
-						--user `id -u`:`id -g` \
-						#{"--publish='#{SmartMachine.credentials.elasticsearch[:port]}:#{SmartMachine.credentials.elasticsearch[:port]}'" if exposed == '--exposed'} \
-						--volume='#{SmartMachine.config.user_home_path}/.smartmachine/grids/elasticsearch/data:/usr/share/elasticsearch/data' \
-						--volume='#{SmartMachine.config.user_home_path}/.smartmachine/grids/elasticsearch/logs:/usr/share/elasticsearch/logs' \
-						--restart='always' \
-						--network='elasticsearch-network' \
-						elasticsearch:7.4.1", out: File::NULL)
-
-						puts "done"
-						print "-----> Starting container elasticsearch ... "
-						if system("docker start elasticsearch", out: File::NULL)
-							puts "done"
-						end
-					end
-				end
-			end
-	
-			def down
-				if SmartMachine::Docker.running?
-					# Stopping & Removing containers - in reverse order
-					print "-----> Stopping container elasticsearch ... "
-					if system("docker stop 'elasticsearch'", out: File::NULL)
-						puts "done"
-						print "-----> Removing container elasticsearch ... "
-						if system("docker rm 'elasticsearch'", out: File::NULL)
-							puts "done"
-						end
-					end
-
-					# Removing networks
-					print "-----> Removing network elasticsearch-network ... "
-					if system("docker network rm elasticsearch-network", out: File::NULL)
-						puts "done"
-					end
-				end
-			end			
-		end
-	end
-end
+        # Removing networks
+        print "-----> Removing network #{@name}-network ... "
+        if system("docker network rm #{@name}-network", out: File::NULL)
+          puts "done"
+        end
+      end
+    end
+  end
+end

data/lib/smart_machine/grids/minio.rb

@@ -1,77 +1,88 @@
-# The main SmartMachine Grids Minio driver
 module SmartMachine
-	class Grids
-		class Minio < SmartMachine::Base
-			def initialize
-			end
+  class Grids
+    class Minio < SmartMachine::Base
+      def initialize(name:)
+        config = SmartMachine.config.grids.minio.dig(name.to_sym)
+        raise "minio config for #{name} not found." unless config
 
-			def up(*args)
-				args.flatten!
-				exposed = args.empty? ? '' : args.shift
+        @host = config.dig(:host)
+        @access_key = config.dig(:access_key)
+        @secret_key = config.dig(:secret_key)
+        @browser = config.dig(:browser)
+        @worm = config.dig(:worm)
 
-				if SmartMachine::Docker.running?
-					# Creating networks
-					unless system("docker network inspect minio-network", [:out, :err] => File::NULL)
-						print "-----> Creating network minio-network ... "
-						if system("docker network create minio-network", out: File::NULL)
-							puts "done"
-						end
-					end
+        @name = name.to_s
+        @home_dir = File.expand_path('~')
+      end
 
-					# Creating & Starting containers
-					print "-----> Creating container minio ... "
-					if system("docker create \
-						--name='minio' \
-						--env VIRTUAL_HOST=#{SmartMachine.credentials.minio[:name]}.#{SmartMachine.config.apps_domain} \
-						--env LETSENCRYPT_HOST=#{SmartMachine.credentials.minio[:name]}.#{SmartMachine.config.apps_domain} \
-						--env LETSENCRYPT_EMAIL=#{SmartMachine.config.sysadmin_email} \
-						--env LETSENCRYPT_TEST=false \
-						--env MINIO_ACCESS_KEY=#{SmartMachine.credentials.minio[:access_key]} \
-						--env MINIO_SECRET_KEY=#{SmartMachine.credentials.minio[:secret_key]} \
-						--env MINIO_BROWSER=#{SmartMachine.credentials.minio[:browser]} \
-						--env MINIO_WORM=#{SmartMachine.credentials.minio[:worm]} \
-						--user `id -u`:`id -g` \
-						--volume='#{SmartMachine.config.user_home_path}/.smartmachine/grids/minio/data:/data' \
-						--restart='always' \
-						--network='minio-network' \
-						minio/minio:RELEASE.2020-02-27T00-23-05Z server /data", out: File::NULL)
+      def uper
+        # Creating networks
+        unless system("docker network inspect #{@name}-network", [:out, :err] => File::NULL)
+          print "-----> Creating network #{@name}-network ... "
+          if system("docker network create #{@name}-network", out: File::NULL)
+            puts "done"
+          end
+        end
 
-						# The alias is necessary to support internal network requests directed to minio container using public url
-						system("docker network connect --alias #{SmartMachine.credentials.minio[:name]}.#{SmartMachine.config.apps_domain} minio-network nginx")
-						system("docker network connect nginx-network minio")
+        FileUtils.mkdir_p("#{@home_dir}/machine/grids/minio/#{@name}/data")
 
-						puts "done"
-						print "-----> Starting container minio ... "
-						if system("docker start minio", out: File::NULL)
-							puts "done"
-						end
-					end
-				end
-			end
+        # Creating & Starting containers
+        print "-----> Creating container #{@name} ... "
+        command = [
+          "docker create",
+          "--name='#{@name}'",
+          "--env VIRTUAL_HOST=#{@host}",
+          "--env LETSENCRYPT_HOST=#{@host}",
+          "--env LETSENCRYPT_EMAIL=#{SmartMachine.config.sysadmin_email}",
+          "--env LETSENCRYPT_TEST=false",
+          "--env MINIO_ACCESS_KEY=#{@access_key}",
+          "--env MINIO_SECRET_KEY=#{@secret_key}",
+          "--env MINIO_BROWSER=#{@browser}",
+          "--env MINIO_WORM=#{@worm}",
+          "--user `id -u`:`id -g`",
+          "--volume='#{@home_dir}/smartmachine/grids/minio/#{@name}/data:/data'",
+          "--restart='always'",
+          "--network='#{@name}-network'",
+          "minio/minio:RELEASE.2020-02-27T00-23-05Z server /data"
+        ]
+        if system(command.compact.join(" "), out: File::NULL)
+          # The alias is necessary to support internal network requests directed to minio container using public url
+          system("docker network connect --alias #{@host} #{@name}-network nginx")
+          system("docker network connect nginx-network #{@name}")
 
-			def down
-				if SmartMachine::Docker.running?
-					# Disconnecting networks
-					system("docker network disconnect nginx-network minio")
-					system("docker network disconnect minio-network nginx")
+          puts "done"
+          puts "-----> Starting container #{@name} ... "
+          if system("docker start #{@name}", out: File::NULL)
+            puts "done"
+          else
+            raise "Error: Could not start the created #{@name} container"
+          end
+        else
+          raise "Error: Could not create #{@name} container"
+        end
+      end
 
-					# Stopping & Removing containers - in reverse order
-					print "-----> Stopping container minio ... "
-					if system("docker stop 'minio'", out: File::NULL)
-						puts "done"
-						print "-----> Removing container minio ... "
-						if system("docker rm 'minio'", out: File::NULL)
-							puts "done"
-						end
-					end
+      def downer
+        # Disconnecting networks
+        system("docker network disconnect nginx-network #{@name}")
+        system("docker network disconnect #{@name}-network nginx")
 
-					# Removing networks
-					print "-----> Removing network minio-network ... "
-					if system("docker network rm minio-network", out: File::NULL)
-						puts "done"
-					end
-				end
-			end
-		end
-	end
-end
+        # Stopping & Removing containers - in reverse order
+        print "-----> Stopping container #{@name} ... "
+        if system("docker stop '#{@name}'", out: File::NULL)
+          puts "done"
+          print "-----> Removing container #{@name} ... "
+          if system("docker rm '#{@name}'", out: File::NULL)
+            puts "done"
+          end
+        end
+
+        # Removing networks
+        print "-----> Removing network #{@name}-network ... "
+        if system("docker network rm #{@name}-network", out: File::NULL)
+          puts "done"
+        end
+      end
+    end
+  end
+end