robit 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 02d1b158b2bc49358abeacbeedc00010bc121c48
+  data.tar.gz: 23507548cd768c1f65617c2f7833737b902f0997
+SHA512:
+  metadata.gz: 7fa9e47a7bba52f516857e402ed05b15fe51882adcfc4bec3732f16ec5e79a60ce9838e8c1f02230f38b23e6b4475b35adee882fe5a98684a8741a5ae9b84a51
+  data.tar.gz: b57810f74891d858fe399bc7432c9b773a2aebb4e71995db2d84d96c8084166b4eeb40f63c74b9b1db1c7f7a935c17035c93b252dc78d0b6db640d4215138e98

data/.gitignore

@@ -0,0 +1,15 @@
+*~
+*.db
+*.gem
+*.log
+*.out
+*.pid
+*.swp
+.DS_Store
+.robit
+.robut_store
+.vagrant
+.yardoc
+Gemfile.lock
+doc
+pkg

data/Gemfile

@@ -0,0 +1,17 @@
+source 'https://rubygems.org'
+
+gemspec
+
+gem 'robut', git: 'git://git/inf/robut.git'
+
+group :development do
+  gem 'pry'
+  gem 'rake'
+  gem 'yard'
+  gem 'version'
+  gem 'rubygems-tasks'
+end
+
+group :test do
+  gem 'minitest'
+end

data/Rakefile

@@ -0,0 +1,33 @@
+require 'rubygems'
+require 'bundler'
+require 'rake'
+
+
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.test_files = FileList['test/test*.rb']
+  test.verbose = true
+end
+
+task :default => :test
+
+
+require 'yard'
+YARD::Rake::YardocTask.new do |t|
+  t.files = %w[ --readme Readme.md lib/**/*.rb - VERSION ]
+end
+
+
+require 'rubygems/tasks'
+Gem::Tasks.new({
+  push: false,
+  sign: {}
+}) do |tasks|
+  tasks.console.command = 'pry'
+end
+Gem::Tasks::Sign::Checksum.new sha2: true
+
+
+require 'rake/version_task'
+Rake::VersionTask.new

data/Readme.md

@@ -0,0 +1,3 @@
+# Robit
+
+See `start.sh` and `test.sh` for examples.

data/VERSION

@@ -0,0 +1 @@
+0.0.1

data/bin/robit

@@ -0,0 +1,95 @@
+#!/usr/bin/env ruby
+require 'robut'
+require 'robut/storage/hash_store'
+require 'robut/storage/yaml_store'
+require 'robut/plugin/calc'
+require 'robut/plugin/pick'
+require 'robut/plugin/quips'
+require 'robut/plugin/lunch'
+require 'robut/plugin/later'
+require 'robut/plugin/help'
+require 'robut/plugin/ping'
+require 'robut/plugin/echo'
+require 'robut/plugin/alias'
+require 'robut_quiz'
+require 'ostruct'
+require 'logger'
+require 'trollop'
+require 'hipchat'
+
+require_relative '../lib/robit'
+
+
+Opts = Trollop::options do
+  version Robit::VERSION
+  banner Robit::ART + "\n\n" + <<-EOS.gsub(/^    /, '')
+    #{Robit::SUMMARY}
+
+    Usage: robit [<options>]
+
+    Options:
+  EOS
+  opt :jid,   "Jabber ID",             type: :string, required: true, short: 'j'
+  opt :pass,  "Jabber password",       type: :string, required: true, short: 'p'
+  opt :nick,  "HipChat @mention name", type: :string, required: true, short: 'n'
+  opt :name,  "HipChat full name",     type: :string, required: true, short: 'N'
+  opt :rooms, "Hipchat rooms",         type: :string, required: true, short: 'r'
+  opt :api_key, "HipChat v1 API key",  type: :string, required: true, short: 'k'
+end
+
+short_room_names = Opts[:rooms].split(',')
+
+
+Robut::Connection.configure do |config|
+  config.jid          = Opts[:jid]
+  config.password     = Opts[:pass]
+  config.mention_name = Opts[:nick]
+  config.nick         = Opts[:name]
+  config.rooms        = Opts[:rooms].split(',')
+  Robut::Storage::YamlStore.file = ".robut_store"
+  config.store        = Robut::Storage::YamlStore
+  config.logger       = Logger.new(STDOUT)
+
+  config.nick_db = Daybreak::DB.new '.nick.db', default: []
+
+  config.hipchat = HipChat::Client.new Opts[:api_key]
+
+  config.hipchat_rooms = []
+  hipchat_rooms = config.hipchat.rooms
+
+  raise 'No rooms!' if hipchat_rooms.length.zero?
+
+  hipchat_rooms.each_with_index do |r, i|
+    next unless short_room_names.include? r.xmpp_jid
+    puts 'room_name: %s, room: %s' % [
+      r.name.downcase_and_snakify, r.inspect
+    ]
+    config.hipchat_rooms << {
+      'xmpp_jid' => r.xmpp_jid,
+      'name' => r.name,
+      'raw' => r
+    }
+    short_room_names.delete r.xmpp_jid
+    sleep 3 # So we don't blow the API rate limit
+  end
+end
+
+raise short_room_names.inspect unless short_room_names.empty?
+
+
+Robut::Plugin.plugins = [
+  Robut::Plugin::Alias,
+  Robut::Plugin::Quiz,
+  Robut::Plugin::Ping,
+  Robut::Plugin::Echo,
+  Robut::Plugin::Calc,
+  Robut::Plugin::Pick,
+  Robut::Plugin::Quips,
+  Robut::Plugin::Lunch,
+  Robut::Plugin::Later,
+  Robut::Plugin::Help,
+  Robut::Plugin::AliasNick
+]
+
+Robut::Web.set :connection, Robut::Connection.new.connect
+loop { sleep 1 }

data/lib/robit.rb

@@ -0,0 +1,3 @@
+require_relative 'robit/metadata'
+require_relative 'robit/plugins/status'
+require_relative 'robit/plugins/alias_nick'

data/lib/robit/metadata.rb

@@ -0,0 +1,28 @@
+module Robit
+
+  # We use a VERSION file to tie into our build pipeline
+  VERSION  = File.read(File.join(File.dirname(__FILE__), '..', '..', 'VERSION')).strip
+
+  # We don't really do all that much, be humble
+  SUMMARY  = 'Like Robut, but better and pronounced correctly'
+
+  # Your benevolent dictator for life
+  AUTHOR   = 'Sean Clemmer'
+
+  # Turn here to strangle your dictator
+  EMAIL    = 'sclemmer@bluejeans.com'
+
+  # Every project deserves its own ASCII art
+  ART      = <<-'EOART' % VERSION
+
+     _  .-')              .-. .-')            .-') _
+    ( \( -O )             \  ( OO )          (  OO) )
+     ,------.  .-'),-----. ;-----.\   ,-.-') /     '._
+     |   /`. '( OO'  .-.  '| .-.  |   |  |OO)|'--...__)
+     |  /  | |/   |  | |  || '-' /_)  |  |  \'--.  .--'
+     |  |_.' |\_) |  |\|  || .-. `.   |  |(_/   |  |
+     |  .  '.'  \ |  | |  || |  \  | ,|  |_.'   |  |
+     |  |\  \    `'  '-'  '| '--'  /(_|  |      |  |
+     `--' '--'     `-----' `------'   `--'      `--'  v%s
+  EOART
+end

data/lib/robit/plugins/alias_nick.rb

@@ -0,0 +1,190 @@
+require 'hipchat'
+require 'daybreak'
+require 'shellwords'
+
+class String
+  # Like a downcase, de-camelcaser:
+  # test => test
+  # Test => test
+  # OpsOnly => ops_only
+  def downcase_and_snakify
+    self.gsub(/\s+/,'').gsub(/([A-Z])/, '_\1').delete('@').sub(/^_/, '').downcase
+  end
+end
+
+
+
+class Robut::Plugin::AliasNick
+  include Robut::Plugin
+
+
+  def usage
+    '#alias <nick> <rooms> - When mentioning <nick>, <rooms> will be notified'
+  end
+
+
+  def handle time, sender_nick, message
+    if message =~ /^#aliases\s+@??(.+)/
+      handle_aliases "@#{$1}"
+    elsif message =~ /^#alias\s+@??(.+?)\s+(@.*+)/
+      handle_alias "@#{$1}", $2.split(' ')
+    elsif sent_to_me?(message)
+      # nop
+    else
+      mentions = message.scan /@\S+/
+      handle_notification sender_nick, mentions, message
+    end
+  end
+
+
+private
+
+  def handle_aliases nick
+    puts 'handle_aliases nick=%s' % nick.inspect
+    rooms = db[nick]
+    if rooms.empty?
+      reply "Couldn't find any aliases."
+    else
+      reply rooms.join(', ')
+    end
+  end
+
+
+  def handle_alias nick, rooms
+    puts 'handle_alias nick=%s rooms=%s' % [ nick.inspect, rooms.inspect ]
+    db.lock { db[nick] = rooms }
+    db.flush
+    reply affirmation
+  end
+
+
+  def handle_notification from, mentions, message
+    puts 'handle_notification from=%s mentions=%s message=%s' % [
+      from.inspect, mentions.inspect, message.inspect
+    ]
+
+    tos = mentions.select { |m| db.keys.include? m }
+
+
+    tos.each do |to|
+
+      db[to].each do |name|
+        rroom, hroom = nil, nil
+
+        if name =~ /^@/
+          rroom = robut_room name
+          raise if rroom.nil?
+          hroom = hipchat_room rroom.name
+          raise if hroom.nil?
+        end
+
+        subject = '%s mentioned "%s" in "%s"' % [
+          from, name, current_room
+        ]
+
+        notification = "<i>%s</i> writes in <b>%s</b>: <i>%s</i>" % [
+          from, current_room, highlight(message, to)
+        ]
+
+        if name =~ /^@/
+          handle_hipchat_notification notification, hroom
+        else
+          handle_email_notification subject, notification, name
+        end
+      end
+
+      reply "Notified #{to_sentence db[to]}."
+    end
+  end
+
+
+  def handle_hipchat_notification notification, hroom
+    puts 'handle_hipchat_notification notification=%s hroom=%s' % [
+      notification.inspect, hroom.inspect
+    ]
+
+    hipchat[hroom['name']].send \
+      reply_to.connection.config.nick, notification, \
+      message_format: :html, notify: true
+  end
+
+
+  def handle_email_notification subject, notification, email_alias
+    puts 'handle_email_notification subject=%s notification=%s email_alias=%s' % [
+      subject.inspect, notification.inspect, email_alias.inspect
+    ]
+
+    send_email = 'echo %s | mail -a %s -a %s -s %s %s' % [
+      Shellwords::escape(notification),
+      Shellwords::escape('From: noreply@bluejeansnet.com'),
+      Shellwords::escape('Content-Type: text/html'),
+      Shellwords::escape(subject),
+      "#{email_alias}@bluejeansnet.com"
+    ]
+
+    system send_email
+  end
+
+
+################################################################################
+
+
+  def to_sentence array
+    return if array.empty?
+    return array.first.to_s if array.length == 1
+    array = array.dup
+    array[-1] = "and #{array[-1]}"
+    array.join(', ').sub(', and', ' and')
+  end
+
+
+  def hipchat
+    reply_to.connection.config.hipchat
+  end
+
+
+  def hipchat_rooms
+    reply_to.connection.config.hipchat_rooms
+  end
+
+
+  def db
+    reply_to.connection.config.nick_db
+  end
+
+
+  def affirmation
+    [ 'Aight, then',
+      'Sure thing',
+      'No problemo',
+      'You got it, bud'
+    ].sample(1).shift
+  end
+
+
+  def robut_room name
+    robut_room = reply_to.connection.rooms.select do |r|
+      r.name =~ /^\d+_#{name.downcase_and_snakify}@/
+    end.shift
+  end
+
+
+  def hipchat_room name
+    hipchat_room = hipchat_rooms.select do |r|
+      r['xmpp_jid'] == name
+    end.shift
+  end
+
+
+  def current_room
+    hipchat_room = hipchat_rooms.select do |r|
+      r['xmpp_jid'] == reply_to.name
+    end.shift['name']
+  end
+
+
+  def highlight message, nick
+    message.gsub nick, "<b>#{nick}</b>"
+  end
+
+end

data/lib/robit/plugins/status.rb

@@ -0,0 +1,75 @@
+require 'open-uri'
+require 'nokogiri'
+require 'json'
+
+def partitions
+  endpoint = 'http://floorplan.bluejeansnet.com:8080/partitionmagick/'
+  html = Nokogiri::HTML open(endpoint) { |f| f.read }
+  partitions = {}
+
+  html.css('.partition').each do |partition|
+    name = partition.css('.name').text
+    next if name.empty?
+    partitions[name] = {
+      name:        name,
+      environment: partition.css('.environment').text,
+      release:     partition.css('.release').text,
+      color:       partition.css('.release')[0]['class'].sub(/release +/, ''),
+      branch:      partition.css('.branch').text
+    }
+  end
+
+  return partitions
+end
+
+
+class Robut::Plugin::Status
+  include Robut::Plugin
+
+  def usage
+    [
+      "#{at_nick} partition - Return the configuration for all partitions",
+      "#{at_nick} partition <name> - Return the configuration for the given partition",
+      "#{at_nick} color - Return the plugin color for all paritions",
+      "#{at_nick} color <name> - Return the plugin color for the given partition"
+    ].join("\n")
+  end
+
+  def handle_partition time, sender, message
+    pname = words(message)[1]
+    if pname.nil? or pname.empty?
+      partitions.each do |_, partition|
+        reply "%{name} %{environment} - %{release} %{branch}" % partition
+      end
+    else
+      begin
+        reply "%{name} %{environment} - %{release} %{branch}" % partitions[pname.downcase]
+      rescue
+        reply "Sorry, I don't know of any partition named '#{pname}'"
+      end
+    end
+  end
+
+  def handle_color time, sender, message
+    pname = words(message)[1]
+    if pname.nil? || pname.empty?
+      partitions.each do |_, partition|
+        reply "%{name} %{environment} - %{color}" % partition
+      end
+    else
+      begin
+        reply "%{name} %{environment} - %{color}" % partitions[pname.downcase]
+      rescue
+        reply "Sorry, I don't know of any partition named '#{pname}'"
+      end
+    end
+  end
+
+  def handle time, sender, message
+    if sent_to_me?(message) && words(message)[0] =~ /^partitions?$/i
+      handle_partition time, sender, message
+    elsif sent_to_me?(message) && words(message)[0] =~ /^colors?$/i
+      handle_color time, sender, message
+    end
+  end
+end

data/robit.gemspec

@@ -0,0 +1,29 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path(File.join('..', 'lib'), __FILE__)
+require 'robit/metadata'
+
+Gem::Specification.new do |s|
+  s.name        = 'robit'
+  s.version     = Robit::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.author      = Robit::AUTHOR
+  s.email       = Robit::EMAIL
+  s.summary     = Robit::SUMMARY
+  s.description = Robit::SUMMARY + '.'
+
+  s.add_runtime_dependency 'twss-classifier', '~> 0'
+  s.add_runtime_dependency 'calc', '~> 1'
+  s.add_runtime_dependency 'nokogiri', '~> 1.5'
+  s.add_runtime_dependency 'robut-quiz', '~> 0'
+  s.add_runtime_dependency 'rubypress', '~> 1'
+  s.add_runtime_dependency 'robut', '~> 0'
+  s.add_runtime_dependency 'trollop', '~> 2'
+  s.add_runtime_dependency 'awesome_print', '~> 1'
+  s.add_runtime_dependency 'daybreak', '~> 0'
+  s.add_runtime_dependency 'hipchat', '~> 1'
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- test/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map { |f| File::basename(f) }
+  s.require_paths = %w[ lib ]
+end

data/start.sh

@@ -0,0 +1,20 @@
+#!/usr/bin/env sh
+set -x
+set -e
+rm -rf nohup.out nohup.pid
+
+rooms=9455_test@conf.hipchat.com
+for r in ops_only all_engr lunch_room qoe \
+         a2m project_easyfit non_functional_test_team \
+         sawmill_feedback indigo_feedback media_team \
+         support_and_ops bitcouncil hangouts_project ; do
+  rooms="9455_$r@conf.hipchat.com,$rooms"
+done
+
+bundle exec bin/robit \
+  -j 9455_468553@chat.hipchat.com \
+  -p "$1"                         \
+  -n 'robit'                      \
+  -N 'Robit (bot)'                \
+  -k "$2"                         \
+  -r "$rooms"

data/test.sh

@@ -0,0 +1,8 @@
+#!/usr/bin/env sh
+bundle exec bin/robit \
+  -j 9455_468553@chat.hipchat.com \
+  -p "$1"                         \
+  -n 'robit'                      \
+  -N 'Robit (bot)'                \
+  -k "$2"                         \
+  -r 9455_test@conf.hipchat.com,9455_ops_only@conf.hipchat.com

metadata

@@ -0,0 +1,197 @@
+--- !ruby/object:Gem::Specification
+name: robit
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Sean Clemmer
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-12-23 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: twss-classifier
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: calc
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1'
+- !ruby/object:Gem::Dependency
+  name: nokogiri
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.5'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.5'
+- !ruby/object:Gem::Dependency
+  name: robut-quiz
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rubypress
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1'
+- !ruby/object:Gem::Dependency
+  name: robut
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: trollop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2'
+- !ruby/object:Gem::Dependency
+  name: awesome_print
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1'
+- !ruby/object:Gem::Dependency
+  name: daybreak
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: hipchat
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1'
+description: Like Robut, but better and pronounced correctly.
+email: sclemmer@bluejeans.com
+executables:
+- robit
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- Gemfile.lock
+- Rakefile
+- Readme.md
+- VERSION
+- bin/robit
+- lib/robit.rb
+- lib/robit/metadata.rb
+- lib/robit/plugins/alias_nick.rb
+- lib/robit/plugins/status.rb
+- robit.gemspec
+- start.sh
+- test.sh
+homepage: 
+licenses: []
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.2.2
+signing_key: 
+specification_version: 4
+summary: Like Robut, but better and pronounced correctly
+test_files: []