bixby_common 0.2.0

data/lib/bixby_common/exception/bundle_not_found.rb

@@ -0,0 +1,5 @@
+
+module Bixby
+  class BundleNotFound < Exception
+  end
+end

data/lib/bixby_common/exception/command_not_found.rb

@@ -0,0 +1,5 @@
+
+module Bixby
+  class CommandNotFound < Exception
+  end
+end

data/lib/bixby_common/util/crypto_util.rb

@@ -0,0 +1,64 @@
+
+require 'base64'
+require 'openssl'
+
+module Bixby
+  module CryptoUtil
+
+    class << self
+
+      def encrypt(data, key_pem, iv_pem)
+        c = new_cipher()
+        c.encrypt
+        key = c.random_key
+        iv = c.random_iv
+        encrypted = c.update(data) + c.final
+
+        out = []
+        out << w( key_pem.public_encrypt(key) )
+        out << w( iv_pem.private_encrypt(iv) )
+        out << e64(encrypted)
+
+        return out.join("\n")
+      end
+
+      def decrypt(data, key_pem, iv_pem)
+        data = StringIO.new(data, 'rb')
+        key = key_pem.private_decrypt(read_next(data))
+        iv  = iv_pem.public_decrypt(read_next(data))
+
+        c = new_cipher()
+        c.decrypt
+        c.key = key
+        c.iv = iv
+
+        ret = c.update(d64(data.read)) + c.final
+      end
+
+
+      private
+
+      def new_cipher
+        OpenSSL::Cipher.new("AES-256-CBC")
+      end
+
+      def w(s)
+        e64(s).gsub(/\n/, "\\n")
+      end
+
+      def read_next(data)
+        d64(data.readline.gsub(/\\n/, "\n"))
+      end
+
+      def e64(s)
+        Base64.encode64(s)
+      end
+
+      def d64(s)
+        Base64.decode64(s)
+      end
+
+    end
+
+  end # CryptoUtil
+end # Bixby

data/lib/bixby_common/util/hashify.rb

@@ -0,0 +1,16 @@
+
+module Bixby
+
+# Adds to_hash method to an Object
+module Hashify
+
+  # Creates a Hash representation of self
+  #
+  # @return [Hash]
+  def to_hash
+    self.instance_variables.inject({}) { |m,v| m[v[1,v.length].to_sym] = instance_variable_get(v); m }
+  end
+
+end # Hashify
+
+end # Bixby

data/lib/bixby_common/util/http_client.rb

@@ -0,0 +1,70 @@
+
+require 'curb'
+require 'multi_json'
+
+module Bixby
+
+# Utilities for creating HTTP Clients. Just a thin wrapper around curb and JSON
+# for common cases.
+module HttpClient
+
+  # Execute an HTTP GET request to the given URL
+  #
+  # @param [String] url
+  # @return [String] Contents of the response's body
+  def http_get(url)
+    Curl::Easy.http_get(url).body_str
+  end
+
+  # Execute an HTTP GET request (see #http_get) and parse the JSON response
+  #
+  # @param [String] url
+  # @return [Object] Result of calling JSON.parse() on the response body
+  def http_get_json(url)
+    MultiJson.load(http_get(url))
+  end
+
+  # Convert a Hash into a Curl::Postfield Array
+  #
+  # @param [Hash] data
+  def create_post_data(data)
+    if data.kind_of? Hash then
+      data = data.map{ |k,v| Curl::PostField.content(k, v) }
+    end
+    data
+  end
+
+  # Execute an HTTP POST request to the given URL
+  #
+  # @param [String] url
+  # @param [Hash] data  Key/Value pairs to POST
+  # @return [String] Contents of the response's body
+  def http_post(url, data)
+    return Curl::Easy.http_post(url, create_post_data(data)).body_str
+  end
+
+  # Execute an HTTP POST request (see #http_get) and parse the JSON response
+  #
+  # @param [String] url
+  # @param [Hash] data  Key/Value pairs to POST
+  # @return [Object] Result of calling JSON.parse() on the response body
+  def http_post_json(url, data)
+    MultiJson.load(http_post(url, data))
+  end
+
+  # Execute an HTTP post request and save the response body
+  #
+  # @param [String] url
+  # @param [Hash] data  Key/Value pairs to POST
+  # @return [void]
+  def http_post_download(url, data, dest)
+    File.open(dest, "w") do |io|
+      c = Curl::Easy.new(url)
+      c.on_body { |d| io << d; d.length }
+      c.http_post(data)
+    end
+  end
+
+end # HttpClient
+
+end # Bixby

data/lib/bixby_common/util/jsonify.rb

@@ -0,0 +1,27 @@
+
+require 'multi_json'
+
+module Bixby
+module Jsonify
+
+  include Hashify
+
+  def to_json
+    MultiJson.dump(self.to_hash)
+  end
+
+  module ClassMethods
+    def from_json(json)
+      json = MultiJson.load(json) if json.kind_of? String
+      obj = self.allocate
+      json.each{ |k,v| obj.send("#{k}=".to_sym, v) }
+      obj
+    end
+  end
+
+  def self.included(receiver)
+    receiver.extend(ClassMethods)
+  end
+
+end # Jsonify
+end # Bixby

data/test/helper.rb

@@ -0,0 +1,74 @@
+require 'rubygems'
+require 'bundler'
+begin
+  Bundler.setup(:default, :development)
+rescue Bundler::BundlerError => e
+  $stderr.puts e.message
+  $stderr.puts "Run `bundle install` to install missing gems"
+  exit e.status_code
+end
+require 'minitest/unit'
+require 'turn'
+require 'turn/reporter'
+require 'turn/reporters/outline_reporter'
+
+Turn.config.framework = :minitest
+Turn.config.format = :outline
+
+module Turn
+  class OutlineReporter < Reporter
+    def start_test(test)
+      @stdout = StringIO.new
+      @stderr = StringIO.new
+
+      name = naturalized_name(test)
+
+      io.print "    %-57s" % name
+
+      @stdout.rewind
+      @stderr.rewind
+
+      $stdout = @stdout
+      $stderr = @stderr unless $DEBUG
+    end
+  end
+end
+
+begin
+  require 'simplecov'
+  SimpleCov.start do
+    add_filter "/test/"
+  end
+rescue Exception => ex
+end
+
+
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+require 'bixby_common'
+
+class MiniTest::Unit::TestCase
+
+  # minitest assert_throws doesn't seem to work properly
+  def assert_throws(clazz, msg = nil, &block)
+    begin
+      yield
+    rescue Exception => ex
+      puts "#{ex.class}: #{ex.message}"
+      puts ex.backtrace.join("\n")
+      if clazz.to_s == ex.class.name then
+        if msg.nil?
+          return
+        elsif msg == ex.message then
+          return
+        end
+      end
+    end
+    flunk("Expected #{mu_pp(clazz)} to have been thrown")
+  end
+
+end
+
+Dir.glob(File.dirname(__FILE__) + "/../lib/**/*.rb").each{ |f| require f }
+
+MiniTest::Unit.autorun

data/test/support/test_bundle/bin/cat

@@ -0,0 +1,2 @@
+#!/bin/bash
+cat -

data/test/support/test_bundle/bin/echo

@@ -0,0 +1,2 @@
+#!/bin/sh
+echo hi

data/test/support/test_bundle/digest

@@ -0,0 +1,17 @@
+{
+  "digest": "2429629015110c29f8fae8ca97e0e494410a28b981653c0e094cfe4a7567f1b7",
+  "files": [
+    {
+      "file": "bin/cat",
+      "digest": "6d14ab63802f3ef51bdc16db1cfeb6df93167fe7a2aa09a6c737293d42c118b2"
+    },
+    {
+      "file": "bin/echo",
+      "digest": "299001868fb8c02fd431c336c6d058f5558c5dff5b5af5e6fe04b870a6a9cbba"
+    },
+    {
+      "file": "manifest.json",
+      "digest": "e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855"
+    }
+  ]
+}

data/test/test_bixby_common.rb

@@ -0,0 +1,18 @@
+require 'helper'
+
+module Bixby
+module Test
+
+class TestBixbyCommon < MiniTest::Unit::TestCase
+
+  def test_autoloading
+    assert_equal(JsonRequest, JsonRequest.new(nil, nil).class)
+    assert_equal(BundleNotFound, BundleNotFound.new.class)
+    assert_equal(CommandNotFound, CommandNotFound.new.class)
+    assert_equal(CommandSpec, CommandSpec.new.class)
+  end
+
+end
+
+end # Test
+end # Bixby

data/test/test_command_spec.rb

@@ -0,0 +1,92 @@
+
+
+require 'helper'
+
+module Bixby
+module Test
+
+class TestCommandSpec < MiniTest::Unit::TestCase
+
+  def setup
+    BundleRepository.path = File.expand_path(File.dirname(__FILE__))
+    h = { :repo => "support", :bundle => "test_bundle", 'command' => "echo", :foobar => "baz" }
+    @c = CommandSpec.new(h)
+  end
+
+
+  def test_init_with_hash
+
+    assert(@c)
+    assert_equal("support", @c.repo)
+    assert_equal("test_bundle", @c.bundle)
+    assert_equal("echo", @c.command)
+
+  end
+
+  def test_to_hash
+
+    assert_equal("support", @c.to_hash[:repo])
+    assert_equal("test_bundle", @c.to_hash[:bundle])
+    assert_equal("echo", @c.to_hash[:command])
+
+  end
+
+  def test_validate
+    @c.validate
+  end
+
+  def test_validate_failures
+    assert_throws(BundleNotFound) do
+      CommandSpec.new(:repo => "support", :bundle => "foobar").validate
+    end
+    assert_throws(CommandNotFound) do
+      CommandSpec.new(:repo => "support", :bundle => "test_bundle", :command => "foobar").validate
+    end
+  end
+
+  def test_execute
+    (status, stdout, stderr) = @c.execute
+    assert(status.success?)
+    assert_equal("hi\n", stdout)
+    assert_equal("", stderr)
+  end
+
+  def test_execute_stdin
+    @c.command = "cat"
+    @c.stdin = "hi"
+    (status, stdout, stderr) = @c.execute
+    assert(status.success?)
+    assert_equal("hi", stdout)
+    assert_equal("", stderr)
+  end
+
+  def test_digest
+    assert_equal "2429629015110c29f8fae8ca97e0e494410a28b981653c0e094cfe4a7567f1b7", @c.digest
+  end
+
+  def test_digest_no_err
+    c = CommandSpec.new({ :repo => "support", :bundle => "test_bundle", 'command' => "echofoo" })
+  end
+
+  def test_update_digest
+    expected = MultiJson.load(File.read(BundleRepository.path + "/support/test_bundle/digest"))
+
+    t = "/tmp/foobar_test_repo"
+    d = "#{t}/support/test_bundle/digest"
+    `mkdir -p #{t}`
+    `cp -a #{BundleRepository.path}/support #{t}/`
+    `rm #{d}`
+    BundleRepository.path = t
+
+    refute File.exist? d
+    @c.update_digest
+    assert File.exist? d
+
+    assert_equal MultiJson.dump(expected), MultiJson.dump(MultiJson.load(File.read(d)))
+    `rm -rf #{t}`
+  end
+
+end # TestCommandSpec
+
+end # Test
+end # Bixby

data/test/test_jsonify.rb

@@ -0,0 +1,36 @@
+
+require 'helper'
+
+module Bixby
+module Test
+
+class TestJsonify < MiniTest::Unit::TestCase
+
+  def test_to_json
+    j = JFoo.new
+    j.bar = "baz"
+    json = j.to_json
+    assert json
+    assert_equal '{"bar":"baz"}', json
+  end
+
+  def test_from_json
+    j1 = JFoo.new
+    j1.bar = "baz"
+
+    j2 = JFoo.from_json(j1.to_json)
+    assert_equal "baz", j2.bar
+
+    j3 = JFoo.from_json('{"bar":"baz"}')
+    assert_equal "baz", j3.bar
+  end
+
+end # Jsonify
+
+class JFoo
+  include Jsonify
+  attr_accessor :bar
+end
+
+end # Test
+end # Bixby