runnel 0.0.2

data/bin/runnel

@@ -0,0 +1,170 @@
+#!/usr/bin/env ruby
+require 'rubygems'
+require 'yaml'
+
+HELP = <<TXT
+Runnel is a easy way to manage your ssh tunnels. It relies on autossh
+
+Basic Command Line Usage:
+  runnel list_running
+  runnel start <tunnel>
+  runnel start_all
+  runnel kill <tunnel>
+  runnel kill_all
+
+  Configuration is read from '~/.runnel/config'
+  PID's are stored in ~/.runnel/pids/
+
+TXT
+
+TCONF_EXAMPLE = <<YAML
+---
+:socks_proxy:
+  :name: My socks proxy for secure browsing on public WiFi
+  :mport: 44488 #The autossh monitor port
+  :command: -NfD 8080 mysecurebox.net
+:mysql_proxy:
+  :name: mySQL proxy for work
+  :mport: 44490
+  :command: -NfL 3306:localhost:3306 mysqlbox.org
+
+YAML
+
+class Runnel
+  RUNNEL_DIR  = ENV['HOME']+"/.runnel"
+  PIDS_DIR  = "#{RUNNEL_DIR}/pids"
+  PIDS_FILES  = "#{PIDS_DIR}/*"
+  TUNNEL_CONFIG = "#{RUNNEL_DIR}/tunnels.yml"
+
+
+  def self.config
+    @config = YAML.load_file(TUNNEL_CONFIG)
+  end
+
+  def self.setup
+    puts "mkdir -p #{RUNNEL_DIR} #{PIDS_DIR}"
+    `mkdir -p #{RUNNEL_DIR} #{PIDS_DIR}`
+    unless File.exists?(TUNNEL_CONFIG)
+      puts "Populating an example config"
+      File.open(TUNNEL_CONFIG, 'w').puts(TCONF_EXAMPLE)
+    end
+    puts "Now just update #{TUNNEL_CONFIG}"
+  end
+
+  def self.all
+    list = []
+    config.each_pair do |k,v|
+      list << self.new(k, v)
+    end
+    list
+  end
+
+  def self.kill(tid)
+    create_from_tunnel_id(tid).kill
+  end
+
+  def self.start(tid)
+    tunnel = create_from_tunnel_id(tid)
+    tunnel.start
+  end
+
+  def self.start_all
+    self.all.each do |t|
+      t.start unless t.running?
+    end
+  end
+
+  def self.kill_all
+    self.all.each do |t|
+      t.kill if t.running?
+    end
+  end
+
+  def self.create_from_tunnel_id(tid)
+    self.new(tid, config[tid.to_sym])
+  end
+
+  attr_reader :id, :config
+  alias_method :conf, :config
+
+  def initialize(id, config)
+    @id = id
+    @config = config
+  end
+
+  def pp_description
+    txt = "#{id} - #{conf[:name]}"
+    txt += " (Running: #{pid})" if running?
+    running? ? green(txt) : red(txt)
+  end
+
+  def running?
+    if File.exists?(pid_file)
+      if `ps #{pid} | grep autossh`.length == 0
+        File.delete(file)
+        false
+      else
+        true
+      end
+    end
+  end
+
+  def start
+    puts "Starting #{conf[:name]}"
+    puts "AUTOSSH_PIDFILE=#{pid_file} autossh -M #{conf[:mport]} #{conf[:command]}"
+    ENV['AUTOSSH_PIDFILE'] = pid_file
+    `autossh -M #{conf[:mport]} #{conf[:command]}`
+  end
+
+  def kill
+    if running?
+      `kill #{pid}`
+    else
+      puts "Unable to find a running #{conf[:name]}"
+    end
+  end
+
+  private
+
+  def pid_file
+    "#{PIDS_DIR}/#{id}"
+  end
+
+  def pid
+    File.read(pid_file).chomp
+  end
+
+  def red(txt)
+    "\033[31m#{txt}\033[0m"
+  end
+
+  def green(txt)
+    "\033[32m#{txt}\033[0m"
+  end
+
+end
+
+case ARGV[0]
+when "setup"
+  Runnel.setup
+when "kill"
+  Runnel.kill(ARGV[1])
+when "kill_all"
+  Runnel.kill_all
+when "start"
+  Runnel.start(ARGV[1])
+when "start_all"
+  Runnel.start_all
+when "help"
+  puts HELP
+else # List tunnels
+  if File.exists?(Runnel::TUNNEL_CONFIG)
+    Runnel.all.each do |r|
+      puts r.pp_description
+    end
+  else
+    puts "Looks like this is your first time, lets set everything up."
+    Runnel.setup
+  end
+end
+

metadata

@@ -0,0 +1,67 @@
+--- !ruby/object:Gem::Specification 
+name: runnel
+version: !ruby/object:Gem::Version 
+  hash: 27
+  prerelease: 
+  segments: 
+  - 0
+  - 0
+  - 2
+  version: 0.0.2
+platform: ruby
+authors: 
+- Mark Percival
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-03-19 00:00:00 -07:00
+default_executable: runnel
+dependencies: []
+
+description: 
+email: mark@mpercival.com
+executables: 
+- runnel
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- bin/runnel
+has_rdoc: true
+homepage: https://github.com/mdp/runnel
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.5.2
+signing_key: 
+specification_version: 3
+summary: An autossh tunnel manager, written in Ruby
+test_files: []
+