Contents

require 'json'
require 'redis'
require 'notifyor/growl'
require 'notifyor/util/formatter'
require 'net/ssh/gateway'
module Notifyor
  module Remote
    class Connection

      def initialize
        @ssh_host = ENV['ssh_host'] || 'localhost'
        @ssh_port = ENV['ssh_port'] || '22'
        @ssh_user = ENV['ssh_user']
        @tunnel_port = ENV['ssh_tunnel_port'] || '2000'
        @redis_port = ENV['ssh_redis_port'] || '6379'
        @redis_channel = ENV['channel'] || 'notifyor'
        @ssh_gateway = nil
        @redis_tunnel_connection = nil
      end

      def build_tunnel
        unless ['127.0.0.1', 'localhost'].include? @ssh_host
          @ssh_gateway = Net::SSH::Gateway.new(@ssh_host, @ssh_user, port: @ssh_port)
          @ssh_gateway.open('127.0.0.1', @redis_port, @tunnel_port)
        end
      end

      def build_redis_tunnel_connection
        redis_port = (['127.0.0.1', 'localhost'].include? @ssh_host) ? @redis_port : @tunnel_port
        @redis_tunnel_connection = Redis.new(port: redis_port)
      end

      def subscribe_to_redis
        @redis_tunnel_connection.subscribe(@redis_channel) do |on|
          on.message do |channel, msg|
            STDERR.write "INFO - Message received on channel: #{channel} \n"
            growl_message(msg)
          end
        end
      end

      def growl_message(message)
        ::Notifyor::Growl.create_growl("Notifyor", message) unless Notifyor::Util::Formatter.squish!(message).empty?
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
notifyor-0.8.1	lib/notifyor/remote/connection.rb
notifyor-0.8.0	lib/notifyor/remote/connection.rb