mayfly 0.0.1

data/bin/mayfly

@@ -0,0 +1,68 @@
+#!/usr/bin/ruby
+
+$:.push(File.dirname(__FILE__) + "/../lib")
+
+require 'optparse'
+require 'ostruct'
+require 'mayfly'
+
+include Mayfly
+
+options = OpenStruct.new
+
+OptionParser.new do |opts|
+  
+  opts.banner = 'Usage: mayfly [options] file'
+  opts.on("-h", "--help", "Show this message") {|h| options.help = h}  
+  opts.on("-v", "--verbose", "Run verbosely") {|v| options.verbose = v}
+  opts.on("-pnumber", "--port  number", Integer, "Port number, default 7887") {|p| options.port = p}
+  opts.on("-lnumber", "--lives number", Integer, "How long mayfly should live for, how many requests it serves, default 1") {|l| options.lives = l}  
+  opts.on("--[no-]secure", "Turn https on or off, off by default") {|s| options.secure = s}
+  opts.on("--[no-]auth", "Turn authentication on or off, off by default") {|a| options.auth = a}
+  opts.on("--passwd password", String,"If using auth then provide your password as an argument") {|p| options.passwd = p}    
+  opts.on("--version", "Show version") do
+    puts "Mayfly " + Mayfly::VERSION.join('.')
+    exit
+  end  
+    
+  begin
+    opts.parse!
+  rescue OptionParser::MissingArgument, 
+         OptionParser::InvalidOption, 
+         OptionParser::InvalidArgument => e
+    puts e.message + ', use --help for more details'
+    exit 1
+  end
+
+  if (options.help)
+    puts opts
+    exit 1
+  end
+
+  if (ARGV.length != 1)
+    puts "File argument not specified, use --help for more details"
+    exit 1
+  end
+  
+end
+
+file   = ARGV[0]
+port   = options.port  || 7887
+lives  = options.lives || 1
+secure = options.secure
+passwd = options.passwd  ||= Mayfly::Utils::Cli.get_passwd if (options.auth)
+verbose = options.verbose || false
+
+args = [file, port.to_i, lives.to_i, secure, passwd, verbose]
+
+if (ENV['MAYFLY_TESTING']) # Just test the stuff here happens ok
+  system("echo #{args.join(',')}")
+else 
+  begin
+    server = Server.new(*args)
+    server.start
+  rescue FileNotFoundException  => e
+    puts "#{e.message}, shutting down"
+    exit 1  
+  end
+end

data/lib/mayfly.rb

@@ -0,0 +1,96 @@
+require 'webrick'
+require 'webrick/https'
+require 'logger'
+require 'mayfly/file'
+require 'mayfly/servlet'
+require 'mayfly/utils'
+include Mayfly::Utils
+
+module Mayfly
+  
+  VERSION = [0,0,1]
+  
+  class FileNotFoundException < RuntimeError; end
+  
+  class Log < Logger
+    def mayfly_log(msg)
+      (@logdev.nil?) ? puts(msg) : info(msg)
+    end
+  end
+  
+  class Server 
+      
+    attr_accessor :config
+    
+    def initialize(file, port, lives, secure = nil, passwd = nil, verbose = false)
+      
+      local_variables.each {|v| eval("@#{v}=eval(v)")}
+      
+      if (!File.exists?(@file)) 
+        raise FileNotFoundException, "#{@file} does not exist"
+      end
+      
+      logger.mayfly_log("#{file} will be served #{lives} times from: " +
+                        "http#{(secure) ? 's' : ''}://#{local_ip}:#{port}/mayfly/")
+    
+      logger.mayfly_log("Using HTTP Authentication with the username 'mayfly'" +
+                        " and the password you specified") if @passwd
+
+      
+      logger.info("Starting mayfly with the following settings: ")
+      logger.info("file=#{file}")      
+      logger.info("port=#{port}")            
+      logger.info("lives=#{lives}")                  
+      logger.info("secure=#{secure}")                        
+      logger.info("passwd=#{passwd}")
+      
+       @config = {
+          :Port => port,
+          :Logger => logger,
+          :AccessLog => [[$stdout, "%t Request from %h, response code = %s"]],
+        }
+
+        if (secure)
+
+          cert, key = get_cert
+
+          @config.merge!({
+            :SSLEnable => true,
+            :SSLVerifyClient => OpenSSL::SSL::VERIFY_NONE,
+            :SSLCertificate => cert,
+            :SSLPrivateKey => key
+          })
+
+        end
+      
+    end
+    
+    def start
+      
+      server = WEBrick::HTTPServer.new(@config)
+      server.mount('/mayfly', Mayfly::Servlet, @file, @lives, @passwd)
+
+      ['INT', 'TERM'].each { |signal|
+         trap(signal){server.shutdown} 
+      }
+
+      server.start
+      
+    end
+
+    # Generate SSL Cert, and hack stderr to not output some ugly crap from webrick
+    def get_cert
+      old_stderr = $stderr
+      $stderr = StringIO.new
+      cert, key = WEBrick::Utils.create_self_signed_cert(1024, [["C","GB"], ["O","#{local_ip}"], ["CN", "Mayfly"]], "Generated by Ruby/OpenSSL")
+      $stderr = old_stderr
+      return cert, key      
+    end
+    
+    def logger
+      @logger ||= Log.new((@verbose) ? $stdout : nil, WEBrick::Log::INFO)
+    end
+    
+  end
+  
+end

data/lib/mayfly/file.rb

@@ -0,0 +1,19 @@
+module Mayfly
+  
+  class File < File
+
+    @close_callback
+    
+    def initialize(file, close_callback = nil)
+      super file
+      @close_callback = close_callback      
+    end
+    
+    def close
+      super
+      @close_callback.call if @close_callback != nil
+    end
+    
+  end
+  
+end

data/lib/mayfly/servlet.rb

@@ -0,0 +1,69 @@
+module Mayfly 
+  
+  require 'rubygems'
+  require 'thread'
+  require 'webrick/httputils'
+  require 'mayfly/utils'    
+  include Mayfly::Utils
+  
+  class Servlet < WEBrick::HTTPServlet::AbstractServlet
+  
+    @@instance = nil
+    @@instance_creation_mutex = Mutex.new
+    
+    def self.get_instance(config, *options)
+      @@instance_creation_mutex.synchronize {
+        @@instance = @@instance || self.new(config, *options)
+      }
+    end
+
+    def initialize(config, file, max, passwd = nil)
+      super
+      @file = file
+      @max = max
+      @count = 1
+      @count_mutex = Mutex.new
+      @passwd = passwd
+    end
+  
+    def do_GET(req, resp)
+
+      if (@count > @max)
+        resp.status = 401
+      end
+      
+      if (@passwd)
+        WEBrick::HTTPAuth.basic_auth(req, resp, 'mayfly') do |user, pass|
+            user == 'mayfly' && pass == @passwd
+        end
+      end
+            
+      st = File::stat(@file)
+      resp['content-type'] = WEBrick::HTTPUtils::mime_type(@file, WEBrick::HTTPUtils::DefaultMimeTypes) 
+      resp['content-length'] = st.size
+      resp['content-disposition'] = "attachment; filename=\"#{File.basename(@file)}\""
+      resp['last-modified'] = st.mtime.httpdate
+      resp['Cache-Control'] = 'no-cache, must-revalidate'
+      
+      close_callback = nil 
+      
+      @count_mutex.synchronize {
+        @count += 1
+        close_callback = Proc.new do
+          growl("mayfly served #{@file} to #{req.peeraddr[2]}")            
+          if (@count > @max)
+            @server.shutdown if (@count >= @max) 
+            growl("mayfly has died, it served #{@file} #{@max} times")            
+          end
+        end
+      }
+      
+      if (@count <= @max + 1)
+        resp.body = File.new(@file, close_callback)
+      end
+      
+    end
+    
+  end
+
+end

data/lib/mayfly/utils.rb

@@ -0,0 +1,51 @@
+module Mayfly
+  module Utils
+
+    require 'rubygems'
+    gem 'growl', '=1.0.3'
+    require 'growl'
+    include Growl
+    
+    require 'socket'
+    
+    def growl(msg)
+      if Growl.installed?
+        notify_info(msg, {:title => 'mayfly'}) 
+      end
+    end
+
+    # http://coderrr.wordpress.com/2008/05/28/get-your-local-ip-address/
+    def local_ip
+          
+      orig, Socket.do_not_reverse_lookup = Socket.do_not_reverse_lookup, true  # turn off reverse DNS resolution temporarily
+
+      UDPSocket.open do |s|
+        s.connect '64.233.187.99', 1
+        s.addr.last
+      end
+      
+    ensure
+      Socket.do_not_reverse_lookup = orig
+    end
+    module_function :local_ip
+    
+    # Some helpers for the cli  
+    class Cli
+      
+      gem 'highline', '=1.5.1'
+      require 'highline/import'
+            
+      def self.get_passwd  
+        passwd = ask('Password, leave blank for none: ') {|q| q.echo = false}
+        return nil if passwd == ''
+        if (passwd != ask('Confirm password: ') {|q| q.echo = false})
+          puts 'Passwords do not match, exiting'
+          exit 1
+        end
+        passwd
+      end          
+             
+    end
+    
+  end
+end

metadata

@@ -0,0 +1,77 @@
+--- !ruby/object:Gem::Specification 
+name: mayfly
+version: !ruby/object:Gem::Version 
+  version: 0.0.1
+platform: ruby
+authors: 
+- Matt Haynes
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-10-18 00:00:00 +01:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: growl
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        version: 1.0.3
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: highline
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        version: 1.5.1
+    version: 
+description: "mayfly is a tiny HTTP server with a very short lifespan, existing only to serve a single file for a predefined number of times, it then quietly shuffles off this mortal coil. "
+email: matt@matthaynes.net
+executables: 
+- mayfly
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/mayfly.rb
+- lib/mayfly/file.rb
+- lib/mayfly/servlet.rb
+- lib/mayfly/utils.rb
+has_rdoc: true
+homepage: http://github.com/matth/mayfly/tree/master
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: A tiny HTTP server with a very short lifespan
+test_files: []
+