Contents

require 'thread'
require 'socket'
require 'securerandom'
require 'fileutils'
require 'nrser'

using NRSER

module QB; end
module QB::Util; end

module QB::Util::STDIO
  SOCKET_DIR = Pathname.new('/').join 'tmp', 'qb-stdio'
  
  # STDIO as a service exposed on a UNIX socket so that modules can stream
  # their output to it, which is in turn printed to the console `qb` is running
  # in.
  class Service
    def initialize name, dest
      @name = name
      @dest = dest
      @thread = nil
      @server = nil
      @socket = nil
      @env_key = "QB_STDIO_#{ name.upcase }"
      
      unless SOCKET_DIR.exist?
        FileUtils.mkdir SOCKET_DIR
      end
      
      @path = SOCKET_DIR.join "#{ name }.#{ SecureRandom.uuid }.sock"
      
      @debug_header = "#{ name }@#{ @path.to_s }"
    end
    
    def debug *args
      QB.debug "#{ @debug_header }", *args
    end
    
    def open!
      debug "opening..."
      
      # make sure env var is not already set (basically just prevents you from
      # accidentally opening two instances with the same name)
      if ENV.key? @env_key
        raise <<-END.squish
          env already contains key #{ @env_key } with value #{ ENV[@env_key] }
        END
      end
      
      @thread = Thread.new do
        debug "thread started."
        
        @server = UNIXServer.new @path.to_s
        @socket = @server.accept
        
        while (line = @socket.gets) do
          @dest.puts line
        end
      end
      
      # set the env key so children can find the socket path
      ENV[@env_key] = @path.to_s
      debug "set env var #{ @env_key }=#{ ENV[@env_key] }"
      
      debug "service open."
    end # open
    
    def close!
      # clean up.
      # 
      # TODO not sure how correct this is...
      # 
      debug "closing..."
      
      @thread.kill
      @socket.close
      @socket = nil
      @server.close
      @server = nil
      FileUtils.rm @path
      
      debug "closed."
    end
  end # Service
end # QB::Util::STDIO

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
qb-0.1.37	lib/qb/util/stdio.rb