smartguard 0.1.0 → 0.1.1

data/bin/smartguard

@@ -0,0 +1,23 @@
+#!/usr/bin/env ruby
+
+require 'smartguard'
+require 'trollop'
+
+opts = Trollop::options do
+  opt :application, "Application to use", :type => :string
+  opt :path, "Path to application", :type => :string
+  opt :port, "Port to run DRB", :type => :integer, :default => 10000
+end
+
+raise "You should specify application (--application)" if opts[:application].blank?
+raise "You should specify application path (--path)" if opts[:path].blank?
+
+application = Smartguard::Applications.const_get(opts[:application].camelize) rescue nil
+
+raise "Application `#{opts[:application]}` not supported" if application.nil?
+
+application = application.new(opts[:path])
+application.warm_up
+
+DRb.start_service("druby://localhost:#{opts[:port]}", application)
+DRb.thread.join

data/lib/smartguard/application.rb

@@ -0,0 +1,11 @@
+module Smartguard
+  class Application
+    def initialize(path)
+      @base_path     = Pathname.new(File.absolute_path path)
+      @current_path  = @base_path.join('current')
+      @releases_path = @base_path.join('releases')
+      @shared_path   = @base_path.join('shared')
+      @active_path   = File.readlink @current_path
+    end
+  end
+end

data/lib/smartguard/applications/smartkiosk/cronic.rb

@@ -0,0 +1,27 @@
+module Smartguard
+  module Applications
+    class Smartkiosk
+      class Cronic < Smartguard::Process
+        def initialize(path)
+          @path     = path
+          @pid_file = path.join('tmp/pids/cronic.pid')
+          @log_file = path.join('log/cronic.log')
+        end
+
+        def pid
+          File.read(@pid_file).to_i rescue nil
+        end
+
+        def start
+          Logging.logger.info "Starting cronic"
+          run @path, "script/cronic -d -l #{@log_file} -P #{@pid_file} RAILS_ENV=production"
+        end
+
+        def stop
+          Logging.logger.info "Stoping cronic"
+          run @path, "script/cronic -k -P #{@pid_file}"
+        end
+      end
+    end
+  end
+end

data/lib/smartguard/applications/smartkiosk/sidekiq.rb

@@ -0,0 +1,34 @@
+module Smartguard
+  module Applications
+    class Smartkiosk
+      class Sidekiq < Smartguard::Process
+        def initialize(path)
+          @path        = path
+          @config_path = path.join('config/sidekiq.yml')
+          @log_path    = path.join('log/sidekiq_log')
+          @config      = YAML.load_file(@config_path) rescue {}
+        end
+
+        def pid
+          File.read(@path.join @config[:pidfile]).to_i rescue nil
+        end
+
+        def start
+          Logging.logger.info "Starting sidekiq"
+
+          if @config[:pidfile].blank?
+            Logging.logger.warn "Sidekiq's config was not found"
+            return false
+          end
+
+          run @path, "bundle exec sidekiq -e production --config #{@config_path} >> #{@log_path} 2>&1 &"
+        end
+
+        def stop
+          Logging.logger.info "Stoping sidekiq"
+          run @path, "bundle exec sidekiqctl stop #{@config[:pidfile]} 60"
+        end
+      end
+    end
+  end
+end

data/lib/smartguard/applications/smartkiosk/smartware.rb

@@ -0,0 +1,27 @@
+module Smartguard
+  module Applications
+    class Smartkiosk
+      class Smartware < Smartguard::Process
+        def initialize(path)
+          @path      = path
+          @pids_path = path.join('tmp/pids')
+          @logs_path = path.join('log')
+        end
+
+        def pid
+          File.read("#{@pids_path}/smartware.pid").to_i rescue nil
+        end
+
+        def start
+          Logging.logger.info "Starting smartware"
+          run @path, "bundle exec smartware piddir=#{@pids_path} logdir=#{@logs_path} &"
+        end
+
+        def stop
+          Logging.logger.info "Stoping smartware"
+          kill unless pid.blank?
+        end
+      end
+    end
+  end
+end

data/lib/smartguard/applications/smartkiosk/thin.rb

@@ -0,0 +1,26 @@
+module Smartguard
+  module Applications
+    class Smartkiosk
+      class Thin < Smartguard::Process
+        def initialize(path)
+          @path     = path
+          @pid_file = path.join('tmp/pids/thin.pid')
+        end
+
+        def pid
+          File.read(@pid_file).to_i rescue nil
+        end
+
+        def start
+          Logging.logger.info "Starting thin"
+          run @path, "thin -d -e production start"
+        end
+
+        def stop
+          Logging.logger.info "Stoping thin"
+          run(@path, "thin -d -e production stop") unless pid.blank?
+        end
+      end
+    end
+  end
+end

data/lib/smartguard/applications/smartkiosk.rb

@@ -0,0 +1,132 @@
+module Smartguard
+  module Applications
+    class Smartkiosk < Smartguard::Application
+      def initialize(*args)
+        super
+        @head_path = @base_path.join('head')
+      end
+
+      def services
+        [:sidekiq, :smartware, :cronic, :thin]
+      end
+
+      def status
+        data = {}
+
+        services.each do |s|
+          service = send(s)
+
+          data[s] = [service.pid, service.active?]
+        end
+
+        data
+      end
+
+      def warm_up
+        Logging.logger.info 'Warming up'
+
+        services.each do |s|
+          service = send(s)
+
+          if !service.active?
+            if !service.start
+              stop_services
+              puts "Could not be started: #{s}"
+              exit
+            end
+          else
+            Logging.logger.info "#{s} is already active: #{service.pid}"
+          end
+        end
+      end
+
+      def restart(path=nil)
+        stop_services(path)
+        start_services(path)
+      end
+
+      def restart_asyn(path=nil)
+        Thread.new{ self.restart(path) }
+        true
+      end
+
+      def switch_release(release)
+        release = release.is_a?(Symbol) ? @releases_path.join(release.to_s)
+                                        : @base_path.join(release)
+
+        raise "Release doesn't exist: #{release.to_s}" unless File.exist? release
+
+        FileUtils.cd(release) do
+          Logging.logger.info "Installing release gems"
+          `bundle install`
+
+          Logging.logger.info "Migrating database"
+          `bundle exec rake db:migrate --trace  RAILS_ENV=production`
+
+          Logging.logger.info "Compiling assets"
+          `bundle exec rake assets:precompile --trace RAILS_ENV=production`
+        end
+
+        self.stop_services
+
+        Logging.logger.info "Switching symlink from `#{@active_path}` to `#{release}`"
+        File.delete @current_path if File.symlink? @current_path
+        FileUtils.ln_s(release, @current_path)
+
+        @active_path = release if self.start_services do
+          Logging.logger.warn "New release `#{release}` did not start!"
+          self.stop_services
+
+          Logging.logger.info "Switching symlink back to `#{@active_path}`"
+          File.delete @current_path
+          FileUtils.ln_s @active_path, @current_path
+
+          self.start_services
+        end
+      end
+
+      def switch_release_async(release)
+        Thread.new{ switch_release(release) }
+        true
+      end
+
+      def start_services(path=nil, &block)
+        services.each do |s|
+          if !send(s, path).start && block_given?
+            Logging.logger.debug "Startup of #{s} failed: running safety block"
+            yield 
+            return false
+          end
+        end
+      end
+
+      def stop_services(path=nil)
+        services.each do |s|
+          send(s, path).stop
+        end
+      end
+
+    protected
+
+      def sidekiq(path=nil)
+        Sidekiq.new wrap_path(path)
+      end
+
+      def smartware(path=nil)
+        Smartware.new wrap_path(path)
+      end
+
+      def cronic(path=nil)
+        Cronic.new wrap_path(path)
+      end
+
+      def thin(path=nil)
+        Thin.new wrap_path(path)
+      end
+
+      def wrap_path(path)
+        path.blank? ? @current_path : Pathname.new(path)
+      end
+    end
+  end
+end

data/lib/smartguard/client.rb

@@ -0,0 +1,12 @@
+module Smartguard
+  class Client
+    def initialize(port=10000)
+      DRb.start_service
+      @instance = DRbObject.new_with_uri("druby://localhost:#{port}")
+    end
+
+    def method_missing(method, *args, &block)
+      @instance.send method, *args, &block
+    end
+  end
+end

data/lib/smartguard/logging.rb

@@ -0,0 +1,32 @@
+require 'logger'
+
+module Smartguard
+  module Logging
+
+    def self.logdir=(dir)
+      @logdir = dir
+    end
+
+    def self.logger
+      @logger ||= begin
+        log = Logger.new(STDOUT)
+        log.level = Logger::DEBUG
+        log.formatter = Logger::Formatter.new
+        log
+      end
+    end
+
+    def self.logger=(val)
+      @logger = (val ? val : Logger.new('/dev/null'))
+    end
+
+    def self.logfile=(val)
+      @logfile = val
+    end
+
+    def self.logfile
+      @logfile
+    end
+
+  end
+end

data/lib/smartguard/process.rb

@@ -0,0 +1,25 @@
+module Smartguard
+  class Process
+    include DRb::DRbUndumped
+
+    def run(path, command)
+      result = false
+
+      Bundler.with_clean_env do
+        FileUtils.cd(path) do
+          result = Kernel.system command
+        end
+      end
+
+      result
+    end
+
+    def kill
+      run @path, "kill -9 #{pid}"
+    end
+
+    def active?
+      !!::Process.getpgid(pid) rescue false
+    end
+  end
+end

data/lib/smartguard/version.rb

@@ -1,3 +1,3 @@
 module Smartguard
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

data/smartguard.gemspec

@@ -10,7 +10,7 @@ Gem::Specification.new do |gem|
   gem.email         = ["boris@roundlake.ru"]
   gem.description   = %q{Smartguard is the Smartkiosk services control daemon}
   gem.summary       = gem.description
-  gem.homepage      = ""
+  gem.homepage      = "https://github.com/roundlake/smartguard"
 
   gem.files         = `git ls-files`.split($/)
   gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: smartguard
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
   prerelease: 
 platform: ruby
 authors:
@@ -13,7 +13,7 @@ date: 2012-12-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: i18n
-  requirement: &70194555401080 !ruby/object:Gem::Requirement
+  requirement: &70236313873140 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,10 +21,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70194555401080
+  version_requirements: *70236313873140
 - !ruby/object:Gem::Dependency
   name: active_support
-  requirement: &70194555400180 !ruby/object:Gem::Requirement
+  requirement: &70236313872100 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -32,10 +32,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70194555400180
+  version_requirements: *70236313872100
 - !ruby/object:Gem::Dependency
   name: trollop
-  requirement: &70194555399460 !ruby/object:Gem::Requirement
+  requirement: &70236313871280 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -43,10 +43,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *70194555399460
+  version_requirements: *70236313871280
 - !ruby/object:Gem::Dependency
   name: pry
-  requirement: &70194555398800 !ruby/object:Gem::Requirement
+  requirement: &70236313870400 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -54,11 +54,12 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *70194555398800
+  version_requirements: *70236313870400
 description: Smartguard is the Smartkiosk services control daemon
 email:
 - boris@roundlake.ru
-executables: []
+executables:
+- smartguard
 extensions: []
 extra_rdoc_files: []
 files:
@@ -67,10 +68,20 @@ files:
 - LICENSE.txt
 - README.md
 - Rakefile
+- bin/smartguard
 - lib/smartguard.rb
+- lib/smartguard/application.rb
+- lib/smartguard/applications/smartkiosk.rb
+- lib/smartguard/applications/smartkiosk/cronic.rb
+- lib/smartguard/applications/smartkiosk/sidekiq.rb
+- lib/smartguard/applications/smartkiosk/smartware.rb
+- lib/smartguard/applications/smartkiosk/thin.rb
+- lib/smartguard/client.rb
+- lib/smartguard/logging.rb
+- lib/smartguard/process.rb
 - lib/smartguard/version.rb
 - smartguard.gemspec
-homepage: ''
+homepage: https://github.com/roundlake/smartguard
 licenses: []
 post_install_message: 
 rdoc_options: []