kaboom 0.3.1

data/lib/kaboom/storage/mongodb.rb

@@ -0,0 +1,96 @@
+# coding: utf-8
+
+# Storage adapter that saves data from boom to MongoDB instead of JSON file.
+begin
+  require 'mongo'
+rescue LoadError
+end
+
+module Boom
+  module Storage
+    class Mongodb < Base
+
+      def self.sample_config
+        %(
+          {"backend": "mongodb",
+            "mongodb": {
+              "port": "",
+              "host": ""
+              "database": ""
+              "username": ""
+              "password": ""
+            }
+          })
+      end
+
+      # Public: Initialize MongoDB connection and check dep.
+      #
+      # Returns Mongo connection
+      def mongo
+        @mongo ||= ::Mongo::Connection.new(
+          Boom.config.attributes["mongodb"]["host"],
+          Boom.config.attributes["mongodb"]["port"]
+        ).db(Boom.config.attributes["mongodb"]["database"])
+
+        @mongo.authenticate(
+          Boom.config.attributes['mongodb']['username'],
+          Boom.config.attributes['mongodb']['password']
+        )
+
+        # Return connection
+        @mongo
+      rescue  Exception => exception
+        handle exception, "You don't have the Mongo gem installed yet:\n  gem install mongo"
+      end
+
+      # Public: The MongoDB collection
+      #
+      # Returns the MongoDB collection
+      def collection
+        @collection ||= mongo.collection(Boom.config.attributes["mongodb"]["collection"])
+      end
+
+      # Public: Bootstrap
+      #
+      # Returns
+      def bootstrap
+        collection.insert("boom" => '{"lists": [{}]}') if collection.find_one.nil?
+      end
+
+      # Public: Populates the memory list from MongoDB
+      #
+      # Returns nothing
+      def populate
+        storage = MultiJson.decode(collection.find_one['boom']) || []
+
+        storage['lists'].each do |lists|
+          lists.each do |list_name, items|
+            @lists << list = List.new(list_name)
+
+            items.each do |item|
+              item.each do |name,value|
+                list.add_item(Item.new(name,value))
+              end
+            end
+          end
+        end
+      end
+
+      # Public: Save to MongoDB
+      #
+      # Returns Mongo ID
+      def save
+        doc = collection.find_one()
+        collection.update({"_id" => doc["_id"]}, {'boom' => to_json})
+      end
+
+      # Public: Convert to JSON
+      #
+      # Returns
+      def to_json
+       MultiJson.encode(to_hash)
+      end
+
+    end
+  end
+end

data/lib/kaboom/storage/redis.rb

@@ -0,0 +1,79 @@
+# coding: utf-8
+#
+# Sup, Redis.
+#
+begin
+  require 'digest'
+  require 'redis'
+rescue LoadError
+end
+
+module Boom
+  module Storage
+    class Redis < Base
+      include Boom::Color
+      include Boom::Output
+
+      def self.sample_config
+        %(
+          {"backend": "redis",
+            "redis": {
+              "port": "6379",
+              "host": "<host>"
+            }
+          }
+        )
+      end
+
+      def redis
+        @redis ||= ::Redis.new :host => Boom.config.attributes["redis"]["host"],
+                               :port => Boom.config.attributes["redis"]["port"]
+      rescue  Exception => exception
+        handle exception, "You don't have Redis installed yet:\n  gem install redis"
+      end
+
+      def bootstrap
+      end
+
+      def populate
+        lists = redis.smembers("boom:lists") || []
+
+        lists.each do |sha|
+          list_name  = redis.get("boom:lists:#{sha}:name")
+          @lists << list = List.new(list_name)
+
+          shas = redis.lrange("boom:lists:#{sha}:items",0,-1) || []
+
+          shas.each do |item_sha|
+            name  = redis.get "boom:items:#{item_sha}:name"
+            value = redis.get "boom:items:#{item_sha}:value"
+            list.add_item(Item.new(name, value))
+          end
+        end
+      end
+
+      def clear
+        redis.del "boom:lists"
+        redis.del "boom:items"
+      end
+
+      def save
+        clear
+
+        lists.each do |list|
+          list_sha = Digest::SHA1.hexdigest(list.name)
+          redis.set   "boom:lists:#{list_sha}:name", list.name
+          redis.sadd  "boom:lists", list_sha
+
+          list.items.each do |item|
+            item_sha = Digest::SHA1.hexdigest(item.name)
+            redis.rpush "boom:lists:#{list_sha}:items", item_sha
+            redis.set   "boom:items:#{item_sha}:name", item.name
+            redis.set   "boom:items:#{item_sha}:value", item.value
+          end
+        end
+
+    end
+      end
+  end
+end

data/test/examples/config_json.json

@@ -0,0 +1,3 @@
+{
+  backend: 'Json'
+}

data/test/examples/test_json.json

@@ -0,0 +1,3 @@
+{
+  "backend": "Json"
+}

data/test/examples/urls.json

@@ -0,0 +1 @@
+{"lists":[{"urls":[{"github":"https://github.com"},{"blog":"http://zachholman.com"}]}]}

data/test/helper.rb

@@ -0,0 +1,25 @@
+# coding: utf-8
+
+require 'test/unit'
+
+begin
+  require 'rubygems'
+  require 'redgreen'
+rescue LoadError
+end
+
+require 'mocha'
+
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+
+require 'kaboom'
+
+def boom_json(name)
+  root = File.expand_path(File.dirname(__FILE__))
+  Boom::Storage::Json.any_instance.stubs(:save).returns(true)
+  Boom::Storage::Json.any_instance.stubs(:json_file).
+    returns("#{root}/examples/#{name}.json")
+  Boom.use_remote false
+  Boom.stubs(:storage).returns(Boom::Storage::Json.new)
+end

data/test/output_interceptor.rb

@@ -0,0 +1,28 @@
+module Boom
+  module Output
+
+    def capture_output
+      @output = ''
+    end
+
+    def captured_output
+      @output
+    end
+
+    def output(s)
+      @output << s
+    end
+
+    extend self
+  end
+end
+
+# Intercept STDOUT and collect it
+class Boom::Command
+
+
+  def self.save!
+    # no-op
+  end
+
+end

data/test/test_color.rb

@@ -0,0 +1,30 @@
+require 'helper'
+
+class TestColor < Test::Unit::TestCase
+
+  def test_colorize
+    assert_equal "\e[35mBoom!\e[0m",
+      Boom::Color.colorize("Boom!", :magenta)
+  end
+
+  def test_magenta
+    assert_equal "\e[35mMagenta!\e[0m",
+      Boom::Color.magenta("Magenta!")
+  end
+
+  def test_red
+    assert_equal "\e[31mRed!\e[0m",
+      Boom::Color.red("Red!")
+  end
+
+  def test_yellow
+    assert_equal "\e[33mYellow!\e[0m",
+      Boom::Color.yellow("Yellow!")
+  end
+
+  def test_cyan
+    assert_equal "\e[36mCyan!\e[0m",
+      Boom::Color.cyan("Cyan!")
+  end
+
+end

data/test/test_command.rb

@@ -0,0 +1,227 @@
+# coding: utf-8
+
+require 'helper'
+require 'output_interceptor'
+
+class TestCommand < Test::Unit::TestCase
+
+  def setup
+    boom_json :urls
+  end
+
+  def command(cmd)
+    cmd = cmd.split(' ') if cmd
+    Boom::Command.capture_output
+    Boom::Command.execute(*cmd)
+    output = Boom::Command.captured_output
+    output.gsub(/\e\[\d\d?m/, '')
+  end
+
+  def test_use_remote
+    response = command('remote the_office fun awesome')
+
+    assert_match /a new list called the_office/, response
+    assert_match /fun in the_office is awesome/, response
+  end
+
+  def test_remote_checks_for_acceptable_config
+    response = command('remote the_office fun awesome')
+
+    assert_match /a new list called the_office/, response
+    assert_match /fun in the_office is awesome/, response
+  end
+
+  def test_overview_for_empty
+    storage = Boom::Storage
+    storage.stubs(:lists).returns([])
+    Boom::Command.stubs(:storage).returns(storage)
+    assert_match /have anything yet!/, command(nil)
+  end
+
+  def test_overview
+    assert_equal '  urls (2)', command(nil)
+  end
+
+  def test_list_detail
+    assert_match /github/, command('urls')
+  end
+
+  def test_list_all
+    cmd = command('all')
+    assert_match /urls/,    cmd
+    assert_match /github/,  cmd
+  end
+
+  def test_list_creation
+    assert_match /a new list called newlist/, command('newlist')
+  end
+
+  def test_list_creation_with_item
+    assert_match /a new list called newlist.* item in newlist/, command('newlist item blah')
+  end
+
+  def test_list_creation_with_item_stdin
+    STDIN.stubs(:read).returns('blah')
+    STDIN.stubs(:stat)
+    STDIN.stat.stubs(:size).returns(4)
+
+    assert_match /a new list called newlist.* item in newlist is blah/, command('newlist item')
+  end
+
+  def test_item_access
+    IO.stubs(:popen)
+    assert_match /copied https:\/\/github\.com to your clipboard/,
+      command('github')
+  end
+
+  def test_item_access_scoped_by_list
+    IO.stubs(:popen)
+    assert_match /copied https:\/\/github\.com to your clipboard/,
+      command('urls github')
+  end
+
+  def test_item_open_item
+    Boom::Platform.stubs(:system).returns('')
+    assert_match /opened https:\/\/github\.com for you/, command('open github')
+  end
+
+  def test_item_open_specific_item
+    Boom::Platform.stubs(:system).returns('')
+    assert_match /opened https:\/\/github\.com for you/, command('open urls github')
+  end
+
+  def test_item_open_lists
+    Boom::Platform.stubs(:system).returns('')
+    assert_match /opened all of urls for you/, command('open urls')
+  end
+
+  def test_item_creation
+    assert_match /twitter in urls/,
+      command('urls twitter http://twitter.com/holman')
+  end
+
+  def test_item_creation_long_value
+    assert_match /is tanqueray hendricks bombay/,
+      command('urls gins tanqueray hendricks bombay')
+  end
+
+  def test_list_deletion_no
+    STDIN.stubs(:gets).returns('n')
+    assert_match /Just kidding then/, command('urls delete')
+  end
+
+  def test_list_deletion_yes
+    STDIN.stubs(:gets).returns('y')
+    assert_match /Deleted all your urls/, command('urls delete')
+  end
+
+  def test_item_deletion
+    assert_match /github is gone forever/, command('urls github delete')
+  end
+
+  def test_edit
+    Boom::Platform.stubs(:system).returns('')
+    assert_match 'Make your edits', command('edit')
+  end
+
+  def test_help
+    assert_match 'boom help', command('help')
+    assert_match 'boom help', command('-h')
+    assert_match 'boom help', command('--help')
+  end
+
+  def test_noop_options
+    assert_match 'boom help', command('--anything')
+    assert_match 'boom help', command('-d')
+  end
+
+  def test_nonexistent_item_access_scoped_by_list
+    assert_match /twitter not found in urls/, command('urls twitter')
+  end
+
+  def test_echo_item
+    assert_match /https:\/\/github\.com/, command('echo github')
+  end
+
+  def test_echo_item_shorthand
+    assert_match /https:\/\/github\.com/, command('e github')
+  end
+
+  def test_echo_item_does_not_exist
+    assert_match /wrong not found/, command('echo wrong')
+  end
+
+  def test_echo_list_item
+    assert_match /https:\/\/github\.com/, command('echo urls github')
+  end
+
+  def test_echo_list_item_does_not_exist
+    assert_match /wrong not found in urls/, command('echo urls wrong')
+  end
+
+  def test_show_storage
+    Boom::Config.any_instance.stubs(:attributes).returns('backend' => 'json')
+    assert_match /You're currently using json/, command('storage')
+  end
+
+  def test_nonexistant_storage_switch
+    Boom::Config.any_instance.stubs(:save).returns(true)
+    assert_match /couldn't find that storage engine/, command('switch dkdkdk')
+  end
+
+  def test_storage_switch
+    Boom::Config.any_instance.stubs(:save).returns(true)
+    assert_match /We've switched you over to redis/, command('switch redis')
+  end
+
+  def test_version_switch
+    assert_match /#{Boom::VERSION}/, command('-v')
+  end
+
+  def test_version_long
+    assert_match /#{Boom::VERSION}/, command('--version')
+  end
+
+  def test_stdin_pipes
+    stub = Object.new
+    stub.stubs(:stat).returns([1,2])
+    stub.stubs(:read).returns("http://twitter.com")
+    Boom::Command.stubs(:stdin).returns(stub)
+    assert_match /twitter in urls/, command('urls twitter')
+  end
+
+  def test_random
+    Boom::Platform.stubs(:system).returns('')
+    assert_match /opened .+ for you/, command('random')
+  end
+
+  def test_random_from_list
+    Boom::Platform.stubs(:system).returns('')
+    assert_match /(github|zachholman)/, command('random urls')
+  end
+
+  def test_random_list_not_exist
+    Boom::Platform.stubs(:system).returns('')
+    assert_match /couldn't find that list\./, command('random 39jc02jlskjbbac9')
+  end
+
+  def test_delete_item_list_not_exist
+    assert_match /couldn't find that list\./, command('urlz github delete')
+  end
+
+  def test_delete_item_wrong_list
+    command('urlz twitter https://twitter.com/')
+    assert_match /github not found in urlz/, command('urlz github delete')
+  end
+
+  def test_delete_item_different_name
+    command('foo bar baz')
+    assert_match /bar is gone forever/, command('foo bar delete')
+  end
+
+  def test_delete_item_same_name
+    command('duck duck goose')
+    assert_match /duck is gone forever/, command('duck duck delete')
+  end
+
+end