class Growl

# File lib/ruby-growl.rb, line 61
def self.list type
  raise 'you must gem install dnssd' unless Object.const_defined? :DNSSD

  require 'timeout'

  growls = []

  begin
    Timeout.timeout 10 do
      DNSSD.browse! type do |reply|
        next unless reply.flags.add?

        growls << reply

        break unless reply.flags.more_coming?
      end
    end
  rescue Timeout::Error
  end

  hosts = []

  growls.each do |growl|
    DNSSD.resolve! growl do |reply|
      hosts << reply.target
      break
    end
  end

  hosts.uniq
end

# File lib/ruby-growl.rb, line 250
def initialize host, application_name, growl_type = nil
  @host = host
  @application_name = application_name

  @notifications = {}
  @password      = nil

  @growl_type = choose_implementation growl_type
end

# File lib/ruby-growl.rb, line 96
def self.notify options
  message = options[:message]

  unless message then
    puts "Type your message and hit ^D" if $stdin.tty?
    message = $stdin.read
  end

  notify_type = options[:notify_type]

  g = new options[:host], options[:name]
  g.add_notification notify_type, options[:name], options[:icon]
  g.password = options[:password]

  g.notify(notify_type, options[:title], message, options[:priority],
           options[:sticky])
end

# File lib/ruby-growl.rb, line 117
  def self.process_args argv
    require 'optparse'

    options = {
      host:        nil,
      icon:        nil,
      list:        false,
      message:     nil,
      name:        "ruby-growl",
      notify_type: "ruby-growl Notification",
      password:    nil,
      priority:    0,
      sticky:      false,
      title:       "",
    }

    opts = OptionParser.new do |o|
      o.program_name = File.basename $0
      o.version = Growl::VERSION
      o.release = nil

      o.banner = <<-BANNER
Usage: #{o.program_name} -H HOSTNAME [options]

  Where possible, growl is compatible with growlnotify's arguments.
  (Except for -p, use --priority)

Synopsis:
  echo \"message\" | growl -H localhost

  growl -H localhost -m message

      BANNER

      o.separator "Options:"

      o.on("-H", "--host HOSTNAME", "Send notifications to HOSTNAME") do |val|
        options[:host] = val
      end

      o.on("-i", "--icon [ICON]", "Icon url") do |val|
        options[:icon] = URI(val)
      end

      o.on("-n", "--name [NAME]", "Sending application name",
              "(Defaults to \"ruby-growl\")") do |val|
        options[:name] = val
      end

      o.on("-y", "--type [TYPE]", "Notification type",
              "(Defauts to \"Ruby Growl Notification\")") do |val|
        options[:notify_type] = val
      end

      o.on("-t", "--title [TITLE]", "Notification title") do |val|
        options[:title] = val
      end

      o.on("-m", "--message [MESSAGE]",
           "Send this message instead of reading STDIN") do |val|
        options[:message] = val
      end

      # HACK -p -1 raises
      o.on("--priority [PRIORITY]", Integer,
           "Notification priority",
           "Priority can be between -2 and 2") do |val|
        options[:priority] = val
      end

      o.on("-s", "--[no-]sticky", "Make the notification sticky") do |val|
        options[:sticky] = val
      end

      o.on("-P", "--password [PASSWORD]", "Growl UDP Password") do |val|
        options[:password] = val
      end

      o.on("--list", "List growl hosts using dnssd") do |val|
        options[:list] = true
      end
    end

    opts.parse! argv

    abort opts.to_s unless options[:host] or options[:list]

    options
  end

# File lib/ruby-growl.rb, line 210
def self.run argv = ARGV
  options = process_args argv

  if options[:list] then
    begin
      puts 'Growl GNTP hosts:'
      puts list '_gntp._tcp'
      puts
      puts 'Growl UDP hosts:'
      puts list '_growl._tcp'
    rescue => e
      raise unless e.message =~ /gem install dnssd/

      abort "#{e.message} to use --list"
    end
    return
  end

  notify options
end

# File lib/ruby-growl.rb, line 265
def add_notification name, display_name = nil, icon = nil, enabled = true
  @notifications[name] = display_name, icon, enabled
end

# File lib/ruby-growl.rb, line 287
def notify name, title, message, priority = 0, sticky = false, icon = nil
  case @growl_type
  when 'GNTP' then
    notify_gntp name, title, message, priority, sticky, icon
  when 'UDP'  then
    notify_udp name, title, message, priority, sticky
  else
    raise Growl::Error, "bug, unknown growl type #{@growl_type.inspect}"
  end

  self
end