sqs-cli 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: b8eaec729349100a3d6b9da3e0fe95411c46655e
+  data.tar.gz: 945de266458be8d6118ce7ebf1b91f8b452e3900
+SHA512:
+  metadata.gz: ddd1066f3ba9a3b2c6448fa1e938dd72c12c2845fd9d0c8748013ab98a6a65c314aa514bb78be778da9fa7a5fd389a0ecc69d2a120bf44577e92bacf94f8636d
+  data.tar.gz: e7202d7e88c511d638bb70e338d105dbd9cf922ef11fbd2fce074c77927110dcf139dca4edbc17f3399500c6d4bb99974b395fd5d31786661853178e6ca43a7e

data/.gitignore

@@ -0,0 +1,9 @@
+/.bundle/
+/.yardoc
+/Gemfile.lock
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in sqs-cli.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2016 Steven Occhipinti
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

data/README.md

@@ -0,0 +1,41 @@
+# Sqs::Cli
+
+Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/sqs/cli`. To experiment with that code, run `bin/console` for an interactive prompt.
+
+TODO: Delete this and the text above, and describe your gem
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'sqs-cli'
+```
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install sqs-cli
+
+## Usage
+
+TODO: Write usage instructions here
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake test` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+
+To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/sqs-cli.
+
+
+## License
+
+The gem is available as open source under the terms of the [MIT License](http://opensource.org/licenses/MIT).
+

data/Rakefile

@@ -0,0 +1,10 @@
+require "bundler/gem_tasks"
+require "rake/testtask"
+
+Rake::TestTask.new(:test) do |t|
+  t.libs << "test"
+  t.libs << "lib"
+  t.test_files = FileList['test/**/*_test.rb']
+end
+
+task :default => :test

data/bin/sqs-cli

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+require_relative "../lib/sqs_cli"
+
+SqsCli.run

data/lib/sqs_cli.rb

@@ -0,0 +1,59 @@
+require "inquirer"
+require_relative "sqs_cli/sqs"
+require_relative "sqs_cli/cli"
+require_relative "sqs_cli/ticker"
+require_relative "sqs_cli/data_file"
+
+module SqsCli
+  extend self
+
+  def run
+    all_queue_urls = Cli.wait_with_message("Fetching queues...") do
+      SQS.all_queues
+    end
+
+    source = Cli.list_options "Source queue or file",
+      all_queue_urls,
+      include_file: true
+    source_is_queue = source[:selected_item]
+
+    destination = Cli.list_options "Destination queue or file (appends)",
+      all_queue_urls,
+      include_file: source_is_queue,
+      include_stdout: source_is_queue
+
+    if source_is_queue
+      delete_when_done = Ask.confirm "Delete message when processed?"
+    end
+
+
+    ticker = Ticker.new
+
+    if source_url = source[:selected_item]
+      SQS.read_message_batches(source_url) do |batch|
+        batch.each { ticker.increment }
+
+        if destination_url = destination[:selected_item]
+          SQS.send_message_batch(destination_url, batch)
+        elsif destination_file = destination[:filename]
+          DataFile.new(destination_file).write_batch(batch)
+        elsif stream = destination[:stream]
+          batch.each { |message| stream.puts message.body }
+        end
+
+        SQS.delete_message_batch(source_url, batch) if delete_when_done
+      end
+
+    elsif source_file = source[:filename]
+      DataFile.new(source_file).read_batches do |batch|
+        batch.each { ticker.increment }
+
+        if destination_url = destination[:selected_item]
+          SQS.send_message_batch(destination_url, batch)
+        end
+      end
+    end
+
+    puts "\nProcessed #{ticker.count} messages in #{ticker.uptime} seconds"
+  end
+end

data/lib/sqs_cli/cli.rb

@@ -0,0 +1,26 @@
+require "inquirer"
+
+module Cli
+  extend self
+
+  def self.wait_with_message(message)
+    print message
+    result = yield if block_given?
+    print "\r\e[K"
+    result
+  end
+
+  def self.list_options(message, items, opts={})
+    items += ["File..."] if opts[:include_file]
+    items += ["STDOUT"] if opts[:include_stdout]
+
+    index = Ask.list message, items
+    if opts[:include_file] && items[index] == "File..."
+      { filename: (Ask.input "Filename", default: "sqs-cli.b64") }
+    elsif opts[:include_stdout] && items[index] == "STDOUT"
+      { stream: $stdout }
+    else
+      { selected_item: items[index] }
+    end
+  end
+end

data/lib/sqs_cli/data_file.rb

@@ -0,0 +1,29 @@
+require 'ostruct'
+require 'base64'
+
+class DataFile
+  def initialize(filename)
+    @filename = filename
+  end
+
+  def read_batches
+    count = 0
+    File.readlines(@filename).each_slice(10).each do |batch|
+      yield (batch.map { |message|
+        count += 1
+        OpenStruct.new(
+          message_id: count.to_s,
+          body: Base64.strict_decode64(message.strip)
+        )
+      })
+    end
+  end
+
+  def write_batch(batch)
+    File.open(@filename, "a") do |f|
+      batch.each do |message|
+        f.puts Base64.strict_encode64(message.body.strip)
+      end
+    end
+  end
+end

data/lib/sqs_cli/sqs.rb

@@ -0,0 +1,41 @@
+require "aws-sdk"
+
+module SQS
+  extend self
+
+  def all_queues
+    sqs.list_queues.queue_urls
+  end
+
+  def read_message_batches(queue_url, &block)
+    loop do
+      resp = sqs.receive_message(
+        queue_url: queue_url,
+        max_number_of_messages: 10,
+        visibility_timeout: 10
+      )
+      break if resp.messages.empty?
+      block.call resp.messages
+    end
+  end
+
+  def send_message_batch(queue_url, batch)
+    entries = batch.map do |msg|
+      { id: msg.message_id, message_body: msg.body }
+    end
+    sqs.send_message_batch(queue_url: queue_url, entries: entries)
+  end
+
+  def delete_message_batch(queue_url, batch)
+    entries = batch.map do |m|
+      { id: m.message_id, receipt_handle: m.receipt_handle }
+    end
+    sqs.delete_message_batch(queue_url: queue_url, entries: entries)
+  end
+
+  private
+
+  def self.sqs
+    @sqs ||= Aws::SQS::Client.new
+  end
+end

data/lib/sqs_cli/ticker.rb

@@ -0,0 +1,19 @@
+class Ticker
+  attr_reader :count
+
+  def initialize(count=0, opts={})
+    @count = count
+    @silent = opts.fetch(:silent, false)
+    @start_time = Time.now
+  end
+
+  def increment
+    print "." unless @silent
+    @count += 1
+  end
+
+  def uptime
+    @end_time = Time.now
+    (@end_time - @start_time).round(2)
+  end
+end

data/lib/sqs_cli/version.rb

@@ -0,0 +1,3 @@
+module SqsCli
+  VERSION = "0.1.0"
+end

data/sqs-cli.gemspec

@@ -0,0 +1,28 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'sqs_cli/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "sqs-cli"
+  spec.version       = SqsCli::VERSION
+  spec.authors       = ["Steven Occhipinti"]
+  spec.email         = ["dev@stevenocchipinti.com"]
+
+  spec.summary       = "A simple CLI to move messages between SQS queues"
+  spec.homepage      = "https://github.com/stevenocchipinti/sqscli"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0").reject do |f|
+    f.match(%r{^(test|spec|features)/})
+  end
+  spec.bindir        = "bin"
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_runtime_dependency "aws-sdk"
+  spec.add_runtime_dependency "inquirer"
+  spec.add_development_dependency "bundler", "~> 1.13"
+  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_development_dependency "minitest", "~> 5.0"
+end

metadata

@@ -0,0 +1,129 @@
+--- !ruby/object:Gem::Specification
+name: sqs-cli
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Steven Occhipinti
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2017-06-30 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: aws-sdk
+  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: inquirer
+  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: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.13'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.13'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+description: 
+email:
+- dev@stevenocchipinti.com
+executables:
+- sqs-cli
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/sqs-cli
+- lib/sqs_cli.rb
+- lib/sqs_cli/cli.rb
+- lib/sqs_cli/data_file.rb
+- lib/sqs_cli/sqs.rb
+- lib/sqs_cli/ticker.rb
+- lib/sqs_cli/version.rb
+- sqs-cli.gemspec
+homepage: https://github.com/stevenocchipinti/sqscli
+licenses:
+- MIT
+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.6.11
+signing_key: 
+specification_version: 4
+summary: A simple CLI to move messages between SQS queues
+test_files: []