restartable 0.1.2

checksums.yaml

@@ -0,0 +1,15 @@
+---
+!binary "U0hBMQ==":
+  metadata.gz: !binary |-
+    ZjBlMWMyMTgxZjllMGY4NTNkZGMzNTcxODY2ZjlmYTU5ZjA3MGNmMQ==
+  data.tar.gz: !binary |-
+    ZWE5Nzc2YjgxYTY2MjBjMDA4OGE1ZDcxZTVkNzk2Y2VlMDVmN2Q5NA==
+!binary "U0hBNTEy":
+  metadata.gz: !binary |-
+    NDBkMmUxOTczOTQ5N2JlZDJjYzc1ZjMyOTc1YjgxYjNkMWMyMmEyMmJjODVl
+    OTlhMzA5ZTFiNmM5MWVjMjUwYzFiNDk1ZjA1ZTJiMGU0ZDQ3ZjI2ZWFlNWQx
+    MWYxNjRlZjZhN2Q3ZDU3YTliYjZjNjBjMTQ1YWEwMTNiMTcwYzQ=
+  data.tar.gz: !binary |-
+    NDU2NTA2M2M3N2M1YjBjZWE2ZWE5MzY5NzQxNjViYWU2ZjM1MDU3NmM2N2Ez
+    MGRiZDE1YTE2MmM1N2MwYmQ0NTMwMTdjZTI0ZmVmNWY4NWFhNjU5NDM0ZGEz
+    YTJmYjlhYWRmOGM4MDliMTQzN2Y5MGJiOTg0NDQzMmIyY2E4YWQ=

data/.gitignore

@@ -0,0 +1,12 @@
+/pkg/
+/*.gem
+
+/doc/
+/rdoc/
+/.yardoc/
+/coverage/
+
+Makefile
+*.o
+*.bundle
+/tmp/

data/LICENSE.txt

@@ -0,0 +1,20 @@
+Copyright (c) 2013 Ivan Kuchin
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.markdown

@@ -0,0 +1,7 @@
+# restartable
+
+Run code, Ctrl-C to restart, two Ctrl-C to stop.
+
+## Copyright
+
+Copyright (c) 2012-2013 Ivan Kuchin. See LICENSE.txt for details.

data/TODO

@@ -0,0 +1,2 @@
+run block on restart
+bin option to run command on restart

data/bin/restartable

@@ -0,0 +1,37 @@
+#!/usr/bin/env ruby
+# encoding: UTF-8
+
+require 'optparse'
+require 'restartable'
+
+options = {}
+option_parser = OptionParser.new do |op|
+  op.banner = <<-TEXT
+#{op.program_name} v#{Restartable.version}
+
+Usege:
+  #{op.program_name} [options] command [arg…]
+
+  TEXT
+
+  op.on_tail('-h', '--help', 'Show full help') do
+    puts option_parser.help
+    exit
+  end
+
+  op.on_tail('--version', 'Show version') do
+    puts Restartable.version
+    exit
+  end
+end
+
+begin
+  option_parser.order!
+  raise OptionParser::ParseError, 'No command to run' if ARGV.empty?
+rescue OptionParser::ParseError => e
+  abort "#{e.to_s}\n\n#{option_parser.help}"
+end
+
+Restartable.new(options) do
+  system *ARGV
+end

data/lib/restartable.rb

@@ -0,0 +1,83 @@
+# encoding: UTF-8
+
+require 'sys/proctable'
+require 'colored'
+
+class Restartable
+  def self.version
+    Gem.loaded_specs['restartable'].version.to_s rescue 'DEV'
+  end
+
+  WAIT_SIGNALS = [[1, 'INT'], [1, 'INT'], [1, 'INT'], [1, 'TERM'], [5, 'KILL']]
+
+  def initialize(options, &block)
+    @options, @block = options, block
+    @mutex = Mutex.new
+    synced_trap('INT'){ interrupt! }
+    synced_trap('TERM'){ no_restart!; interrupt! }
+    cycle
+  end
+
+  def interrupt!
+    unless @interrupted
+      @interrupted = true
+      Thread.list.each do |thread|
+        unless Thread.current == thread
+          thread.raise SignalException.new('INT')
+        end
+      end
+    else
+      no_restart!
+    end
+  end
+
+  def no_restart!
+    @stop = true
+    puts 'Don\'t restart!'.red.bold
+  end
+
+  def cycle
+    until @stop
+      @interrupted = false
+      puts '^C to restart, double ^C to stop'.green
+      begin
+        @block.call
+        sleep # wait ^C even if block finishes
+      rescue SignalException
+        unless children.empty?
+          puts 'Killing children…'.yellow.bold
+          ripper = Thread.new do
+            WAIT_SIGNALS.each do |time, signal|
+              sleep time
+              puts "…SIG#{signal}…".yellow
+              children.each do |child|
+                Process.kill(signal, child.pid)
+              end
+            end
+          end
+          Process.waitall
+          ripper.terminate
+        end
+        unless @stop
+          puts 'Waiting ^C 0.5 second than restart…'.yellow.bold
+          sleep 0.5
+        end
+      end
+    end
+  end
+
+private
+
+  def synced_trap(signal, &block)
+    Signal.trap(signal) do
+      Thread.new do
+        @mutex.synchronize(&block)
+      end
+    end
+  end
+
+  def children
+    pid = Process.pid
+    Sys::ProcTable.ps.select{ |pe| pid == pe.ppid }
+  end
+end

data/restartable.gemspec

@@ -0,0 +1,20 @@
+# encoding: UTF-8
+
+Gem::Specification.new do |s|
+  s.name        = 'restartable'
+  s.version     = '0.1.2'
+  s.summary     = %q{Run code, Ctrl-C to restart, once more Ctrl-C to stop}
+  s.homepage    = "http://github.com/toy/#{s.name}"
+  s.authors     = ['Ivan Kuchin']
+  s.license     = 'MIT'
+
+  s.rubyforge_project = s.name
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = %w[lib]
+
+  s.add_dependency 'colored', '~> 1.2'
+  s.add_dependency 'sys-proctable', '~> 0.9.3'
+end

metadata

@@ -0,0 +1,79 @@
+--- !ruby/object:Gem::Specification
+name: restartable
+version: !ruby/object:Gem::Version
+  version: 0.1.2
+platform: ruby
+authors:
+- Ivan Kuchin
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-04-01 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: colored
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.2'
+- !ruby/object:Gem::Dependency
+  name: sys-proctable
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.9.3
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.9.3
+description: 
+email: 
+executables:
+- restartable
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- LICENSE.txt
+- README.markdown
+- TODO
+- bin/restartable
+- lib/restartable.rb
+- restartable.gemspec
+homepage: http://github.com/toy/restartable
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: restartable
+rubygems_version: 2.0.3
+signing_key: 
+specification_version: 4
+summary: Run code, Ctrl-C to restart, once more Ctrl-C to stop
+test_files: []