pickynode 0.0.3 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: a9ed335fbc05dfe5122819dab044c3f087f6855d
-  data.tar.gz: ae580a59583a6c5b355883ee71e76cf8b30f8665
+  metadata.gz: dfe782c9a3c9232314fb733ca5e9a0cc31f8bdd5
+  data.tar.gz: 3044d3f752bdca822b5f0e547f9d478febdad2da
 SHA512:
-  metadata.gz: 1051e7d02b39ee86c2a2cc031d484c5f00ac9ee21c9634822395a10ff03702560ff9d5c76d24a4e8fad5374594920b8b42969dfca2159b5161a28c22a86be952
-  data.tar.gz: 7d215790b39f5b36b0f05cb0e4f4e47bdaaa552e4b975d81a6452e558740f0bdf84cf3818879be03fdc44b2366a3e0e8709df101f1a7faf48161e122f1281ec8
+  metadata.gz: c509c44a20db271c1735296f946e17562f968ce4d52d15cb3720c323b1cbfa90de08acef6bfbf01fc4c7a4b2bc79a356d489d34e991670ecdbcfbd2edd8f8e47
+  data.tar.gz: db50a3333c22a40e18bc7a08f1c2b34d24b92cd5b7d0035a42c6418c794c0cc5e97ffda1d609990a0f304461df8993cc59d2333d93d58de416d0212ee2e49246

data/.gitignore

@@ -0,0 +1,50 @@
+*.gem
+*.rbc
+/.config
+/coverage/
+/InstalledFiles
+/pkg/
+/spec/reports/
+/spec/examples.txt
+/test/tmp/
+/test/version_tmp/
+/tmp/
+
+# Used by dotenv library to load environment variables.
+# .env
+
+## Specific to RubyMotion:
+.dat*
+.repl_history
+build/
+*.bridgesupport
+build-iPhoneOS/
+build-iPhoneSimulator/
+
+## Specific to RubyMotion (use of CocoaPods):
+#
+# We recommend against adding the Pods directory to your .gitignore. However
+# you should judge for yourself, the pros and cons are mentioned at:
+# https://guides.cocoapods.org/using/using-cocoapods.html#should-i-check-the-pods-directory-into-source-control
+#
+# vendor/Pods/
+
+## Documentation cache and generated files:
+/.yardoc/
+/_yardoc/
+/doc/
+/rdoc/
+
+## Environment normalization:
+/.bundle/
+/vendor/bundle
+/lib/bundler/man/
+
+# for a library or gem, you might want to ignore these files since the code is
+# intended to run in multiple environments; otherwise, check them in:
+# Gemfile.lock
+# .ruby-version
+# .ruby-gemset
+
+# unless supporting rvm < 1.11.0 or doing something fancy, ignore this:
+.rvmrc

data/.rubocop.yml

@@ -0,0 +1,4 @@
+Metrics/BlockLength:
+  ExcludedMethods: ['describe', 'context']
+Style/IndentHeredoc:
+  Enabled: false

data/.ruby-gemset

@@ -0,0 +1 @@
+pickynode

data/.ruby-version

@@ -0,0 +1 @@
+2.3.4

data/.travis.yml

@@ -0,0 +1,4 @@
+language: ruby
+rvm:
+  - "2.2.7"
+  - "2.3.4"

data/Gemfile

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+source 'https://rubygems.org'
+ruby '>= 2.0'
+
+gem 'awesome_print', '~> 1.7'
+gem 'rake', '~> 12.0'
+gem 'rspec', '~> 3.6'
+gem 'rubocop', '~> 0.48'
+gem 'trollop', '~> 2.1'

data/Gemfile.lock

@@ -0,0 +1,50 @@
+GEM
+  remote: https://rubygems.org/
+  specs:
+    ast (2.3.0)
+    awesome_print (1.7.0)
+    diff-lcs (1.3)
+    parser (2.4.0.0)
+      ast (~> 2.2)
+    powerpack (0.1.1)
+    rainbow (2.2.2)
+      rake
+    rake (12.0.0)
+    rspec (3.6.0)
+      rspec-core (~> 3.6.0)
+      rspec-expectations (~> 3.6.0)
+      rspec-mocks (~> 3.6.0)
+    rspec-core (3.6.0)
+      rspec-support (~> 3.6.0)
+    rspec-expectations (3.6.0)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.6.0)
+    rspec-mocks (3.6.0)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.6.0)
+    rspec-support (3.6.0)
+    rubocop (0.48.1)
+      parser (>= 2.3.3.1, < 3.0)
+      powerpack (~> 0.1)
+      rainbow (>= 1.99.1, < 3.0)
+      ruby-progressbar (~> 1.7)
+      unicode-display_width (~> 1.0, >= 1.0.1)
+    ruby-progressbar (1.8.1)
+    trollop (2.1.2)
+    unicode-display_width (1.2.1)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  awesome_print (~> 1.7)
+  rake (~> 12.0)
+  rspec (~> 3.6)
+  rubocop (~> 0.48)
+  trollop (~> 2.1)
+
+RUBY VERSION
+   ruby 2.3.4p301
+
+BUNDLED WITH
+   1.14.6

data/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2017 Josh Ellithorpe
+
+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,61 @@
+[![Gem Version](https://badge.fury.io/rb/pickynode.svg)](https://badge.fury.io/rb/pickynode) [![Build Status](https://travis-ci.org/zquestz/pickynode.svg)](https://travis-ci.org/zquestz/pickynode) [![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+# pickynode
+
+Some people are picky about the bitcoin nodes they connect to.
+
+### Requirements:
+
+You need a working full node on your machine. The `bitcoin-cli` command should be functional.
+
+### Installation:
+
+```
+gem install pickynode
+```
+
+### Usage:
+
+Display list of currently connected nodes:
+
+```
+pickynode
+```
+
+Add node type:
+```
+pickynode --add=USER_AGENT_FILTER
+```
+
+Ban node type:
+```
+pickynode --ban=USER_AGENT_FILTER
+```
+
+Connect to node type:
+```
+pickynode --connect=USER_AGENT_FILTER
+```
+
+Disconnect from node type:
+
+```
+pickynode --disconnect=USER_AGENT_FILTER
+```
+
+### Help:
+
+```
+pickynode v0.1.0
+Options:
+  -a, --add=<s>           Add node type
+  -c, --connect=<s>       Connect to node type
+  -b, --ban=<s>           Ban node type
+  -d, --debug             Debug mode
+  -o, --output            Output commands
+  -i, --disconnect=<s>    Disconnect from node type
+  -v, --version           Print version and exit
+  -h, --help              Show this message
+
+```
+
+The --add and --connect commands pull data from Bitnodes.

data/Rakefile

@@ -0,0 +1,9 @@
+#!/usr/bin/env rake
+# frozen_string_literal: true
+
+require File.join('bundler', 'gem_tasks')
+require File.join('rspec', 'core', 'rake_task')
+
+RSpec::Core::RakeTask.new(:spec)
+
+task default: :spec

data/bin/pickynode

@@ -1,86 +1,16 @@
 #!/usr/bin/env ruby
-require 'awesome_print'
-require 'json'
-require 'net/http'
-require 'pp'
-require 'trollop'
-require 'uri'
-
-VERSION = "0.0.3"
-
-opts = Trollop::options do
-  version "pickynode v#{VERSION}"
-  opt :add, "Add node type", :type => :string
-  opt :connect, "Connect to node type", :type => :string
-  opt :ban, "Ban node type", :type => :string
-  opt :debug, "Debug mode"
-  opt :output, "Output commands"
-  opt :disconnect, "Disconnect from node type", :type => :string
-end
-
-def run(cmd, opts={})
-  puts "Running #{cmd}" if opts[:output] || opts[:debug]
-  `#{cmd}` unless opts[:debug]
-end
-
-def get_addr_types
-  nodes = `bitcoin-cli getpeerinfo`
-  parsed_nodes = JSON.parse(nodes)
-  parsed_nodes.map { |n| [n["addr"], n["subver"]] }.to_h
-end
-
-def get_bitnode_addr_types
-  u = URI.parse("https://bitnodes.21.co/api/v1/snapshots/latest/")
-  nodelist = Net::HTTP.get(u)
-  parsed_nodelist = JSON.parse(nodelist)
-  parsed_nodelist['nodes'].map { |k,v| [k, v[1]] }.to_h
-end
-
-if opts[:add] || opts[:connect]
-  bitnode_addr_types = get_bitnode_addr_types
-
-  if filter = opts[:add]
-    bitnode_addr_types.each do |k,v|
-      if v.include?(filter)
-        u = URI.parse("https://#{k}")
-        run(%^bitcoin-cli addnode "#{u.host}" "add"^, opts)
-      end
-    end
-  end
-
-  if filter = opts[:connect]
-    bitnode_addr_types.each do |k,v|
-      if v.include?(filter)
-        u = URI.parse("https://#{k}")
-        run(%^bitcoin-cli addnode "#{u.host}" "onetry"^, opts)
-      end
-    end
-  end
-end
-
-if opts[:ban] || opts[:disconnect]
-  addr_types = get_addr_types
-
-  if filter = opts[:ban]
-    addr_types.each do |k,v|
-      if v.include?(filter)
-        u = URI.parse("https://#{k}")
-        run(%^bitcoin-cli setban "#{u.host}" "add"^, opts)
-      end
-    end
-  end
-
-  if filter = opts[:disconnect]
-    addr_types.each do |k,v|
-      if v.include?(filter)
-        run(%^bitcoin-cli disconnectnode "#{k}"^, opts)
-      end
-    end
-  end
-end
-
-unless opts.values.select { |v| v }.empty?
-  exit(0)
+# frozen_string_literal: true
+
+require 'pickynode'
+
+opts = Trollop.options do
+  version "pickynode v#{Pickynode::VERSION}"
+  opt :add, 'Add node type', type: :string
+  opt :connect, 'Connect to node type', type: :string
+  opt :ban, 'Ban node type', type: :string
+  opt :debug, 'Debug mode'
+  opt :output, 'Output commands'
+  opt :disconnect, 'Disconnect from node type', type: :string
 end
 
-ap get_addr_types
+Pickynode.new(opts).run

data/lib/pickynode.rb

@@ -0,0 +1,105 @@
+# frozen_string_literal: true
+
+require 'awesome_print'
+require 'json'
+require 'net/http'
+require 'trollop'
+require 'uri'
+
+# Allows you to easily add/ban/connect/disconnect nodes
+# based on User Agent.
+class Pickynode
+  VERSION = '0.1.0'
+
+  def initialize(opts = {})
+    @opts = opts
+  end
+
+  def add(filter)
+    return unless filter
+    bitnode_addr_types.each do |k, v|
+      run_cmd(%(bitcoin-cli addnode "#{k}" "add")) if v.include?(filter)
+    end
+  end
+
+  def ban(filter)
+    return unless filter
+    addr_types.each do |k, v|
+      if v.include?(filter)
+        u = URI.parse("https://#{k}")
+        run_cmd(%(bitcoin-cli setban "#{u.host}" "add"))
+      end
+    end
+  end
+
+  def connect(filter)
+    return unless filter
+    bitnode_addr_types.each do |k, v|
+      run_cmd(%(bitcoin-cli addnode "#{k}" "onetry")) if v.include?(filter)
+    end
+  end
+
+  def disconnect(filter)
+    return unless filter
+    addr_types.each do |k, v|
+      run_cmd(%(bitcoin-cli disconnectnode "#{k}")) if v.include?(filter)
+    end
+  end
+
+  def display
+    ap addr_types
+  end
+
+  def run
+    add(@opts[:add])
+    connect(@opts[:connect])
+
+    ban(@opts[:ban])
+    disconnect(@opts[:disconnect])
+
+    return unless @opts.values.select { |v| v }.empty?
+
+    display
+  end
+
+  def clear_cache
+    @addr_types = nil
+    @bitnode_addr_types = nil
+  end
+
+  private
+
+  def run_cmd(cmd)
+    puts "Running #{cmd}" if @opts[:output] || @opts[:debug]
+    `#{cmd}` unless @opts[:debug]
+  end
+
+  def addr_types
+    return @addr_types if @addr_types
+    nodes = getpeerinfo
+    parsed_nodes = JSON.parse(nodes)
+    @addr_types = parsed_nodes.map do |n|
+      [n['addr'], n['subver']]
+    end.to_h
+  rescue JSON::ParserError
+    {}
+  end
+
+  def bitnode_addr_types
+    return @bitnode_addr_types if @bitnode_addr_types
+    parsed_nodelist = JSON.parse(bitnodes_snapshot)
+    @bitnode_addr_types = parsed_nodelist['nodes'].map do |k, v|
+      [k, v[1]]
+    end.to_h
+  rescue JSON::ParserError
+    {}
+  end
+
+  def bitnodes_snapshot
+    Net::HTTP.get(URI.parse('https://bitnodes.21.co/api/v1/snapshots/latest/'))
+  end
+
+  def getpeerinfo
+    `bitcoin-cli getpeerinfo`
+  end
+end

data/pickynode.gemspec

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+require File.expand_path(File.join('..', 'lib', 'pickynode'), __FILE__)
+
+Gem::Specification.new do |s|
+  s.name        = 'pickynode'
+  s.version     = Pickynode::VERSION
+  s.date        = '2017-05-19'
+  s.summary     = 'Manage connections to your bitcoin node'
+  s.description = 'Some people are picky about the \
+bitcoin nodes they connect to.'
+  s.authors     = ['Josh Ellithorpe']
+  s.email       = 'quest@mac.com'
+  s.homepage    = 'http://github.com/zquestz/pickynode'
+  s.license     = 'MIT'
+  s.executables << 'pickynode'
+  s.files         = `git ls-files`.split("\n")
+  s.require_paths = ['lib']
+  s.required_ruby_version = '>= 2.0'
+
+  s.add_dependency 'awesome_print', '~> 1.7'
+  s.add_dependency 'trollop', '~> 2.1'
+
+  s.add_development_dependency 'rspec', '~> 3.6'
+  s.add_development_dependency 'rake', '~> 12.0'
+  s.add_development_dependency 'rubocop', '~> 0.48'
+end

data/spec/mocks.rb

@@ -0,0 +1,68 @@
+# frozen_string_literal: true
+
+BITNODES_SNAPSHOT = <<-HEREDOC
+{
+  "timestamp": 1495414383,
+  "total_nodes": 7055,
+  "latest_height": 467520,
+  "nodes": {
+    "88.99.199.87:8333": [80002, "/BitcoinUnlimited:1.0.2(EB16; AD12)/", 1494608364, 5, 467520, "static.87.199.99.88.clients.your-server.de", null, "DE", 51.2993, 9.491, "Europe/Berlin", "AS24940", "Hetzner Online GmbH"],
+    "[2a01:e34:ee3a:5730:21f:5bff:fec5:e356]:8333": [70015, "/Satoshi:0.14.0/", 1495378733, 13, 467520, "2a01:e34:ee3a:5730:21f:5bff:fec5:e356", null, "FR", 46.0, 2.0, "Europe/Paris", "AS12322", "Free SAS"]
+  }
+}
+HEREDOC
+
+PEER_INFO = <<-HEREDOC
+[
+  {
+    "id": 2,
+    "addr": "131.114.88.218:33422",
+    "addrlocal": "67.188.11.253:8333",
+    "services": "0000000000000000",
+    "relaytxes": true,
+    "lastsend": 1495412829,
+    "lastrecv": 1495412825,
+    "bytessent": 5200917,
+    "bytesrecv": 2044140,
+    "conntime": 1495393261,
+    "timeoffset": -24,
+    "pingtime": 0.369602,
+    "minping": 0.210565,
+    "version": 70002,
+    "subver": "/FirstClient/",
+    "inbound": true,
+    "startingheight": 435862,
+    "banscore": 0,
+    "synced_headers": 467515,
+    "synced_blocks": -1,
+    "inflight": [
+    ],
+    "whitelisted": false
+  },
+  {
+    "id": 3,
+    "addr": "[2a01:e34:ee3a:5730:21f:5bff:fec5:e356]:8333",
+    "addrlocal": "[2a01:e34:ee3a:5730:21f:5bff:fec5:e356]:8333",
+    "services": "0000000000000005",
+    "relaytxes": true,
+    "lastsend": 1495412829,
+    "lastrecv": 1495412830,
+    "bytessent": 10235040,
+    "bytesrecv": 2729345,
+    "conntime": 1495393262,
+    "timeoffset": -11,
+    "pingtime": 0.275245,
+    "minping": 0.07081899999999999,
+    "version": 80002,
+    "subver": "/SecondClient/",
+    "inbound": false,
+    "startingheight": 467483,
+    "banscore": 0,
+    "synced_headers": 467515,
+    "synced_blocks": 467515,
+    "inflight": [
+    ],
+    "whitelisted": false
+  }
+]
+HEREDOC

data/spec/pickynode_spec.rb

@@ -0,0 +1,202 @@
+# frozen_string_literal: true
+
+require_relative 'spec_helper'
+require_relative 'mocks'
+
+describe Pickynode do
+  # Debug mode makes sure we don't execute real commands.
+  let(:opts) do
+    { debug: true }
+  end
+
+  # IPv6 addresses are long, easier to refer by let.
+  let(:ipv6_ip) { '[2a01:e34:ee3a:5730:21f:5bff:fec5:e356]:8333' }
+  let(:ipv6_ip_no_port) { '[2a01:e34:ee3a:5730:21f:5bff:fec5:e356]' }
+
+  # The currently connected nodes for the specs.
+  let(:node_hash) do
+    { '131.114.88.218:33422' => '/FirstClient/',
+      ipv6_ip => '/SecondClient/' }
+  end
+
+  subject { Pickynode.new(opts) }
+
+  describe '.add' do
+    it 'should add nodes based on user agent' do
+      expect(subject).to receive(:bitnodes_snapshot).once
+        .and_return(BITNODES_SNAPSHOT)
+      expect(subject).to receive(:run_cmd)
+        .with(%(bitcoin-cli addnode "#{ipv6_ip}" "add"))
+      subject.add('Satoshi')
+      expect(subject).to receive(:run_cmd)
+        .with('bitcoin-cli addnode "88.99.199.87:8333" "add"')
+      subject.add('Unlimited')
+    end
+
+    it 'should return if the filter is falsy' do
+      expect(subject).to_not receive(:bitnodes_snapshot)
+      expect(subject).to_not receive(:run_cmd)
+      subject.add(false)
+      subject.add(nil)
+    end
+  end
+
+  describe '.ban' do
+    it 'should ban nodes based on user agent' do
+      expect(subject).to receive(:getpeerinfo).once
+        .and_return(PEER_INFO)
+      expect(URI).to receive(:parse).with('https://131.114.88.218:33422')
+        .and_call_original
+      expect(subject).to receive(:run_cmd)
+        .with('bitcoin-cli setban "131.114.88.218" "add"')
+      subject.ban('FirstClient')
+      expect(URI).to receive(:parse).with("https://#{ipv6_ip}")
+        .and_call_original
+      expect(subject).to receive(:run_cmd)
+        .with(%(bitcoin-cli setban "#{ipv6_ip_no_port}" "add"))
+      subject.ban('SecondClient')
+    end
+
+    it 'should return if the filter is falsy' do
+      expect(subject).to_not receive(:getpeerinfo)
+      expect(URI).to_not receive(:parse)
+      expect(subject).to_not receive(:run_cmd)
+      subject.ban(false)
+      subject.ban(nil)
+    end
+  end
+
+  describe '.connect' do
+    it 'should connect to nodes based on user agent' do
+      expect(subject).to receive(:bitnodes_snapshot).once
+        .and_return(BITNODES_SNAPSHOT)
+      expect(subject).to receive(:run_cmd)
+        .with('bitcoin-cli addnode "88.99.199.87:8333" "onetry"')
+      subject.connect('Unlimited')
+      expect(subject).to receive(:run_cmd)
+        .with(%(bitcoin-cli addnode "#{ipv6_ip}" "onetry"))
+      subject.connect('Satoshi')
+    end
+
+    it 'should return if the filter is falsy' do
+      expect(subject).to_not receive(:bitnodes_snapshot)
+      expect(subject).to_not receive(:run_cmd)
+      subject.connect(false)
+      subject.connect(nil)
+    end
+  end
+
+  describe '.disconnect' do
+    it 'should disconnect nodes based on user agent' do
+      expect(subject).to receive(:getpeerinfo).once
+        .and_return(PEER_INFO)
+      expect(subject).to receive(:run_cmd)
+        .with(%(bitcoin-cli disconnectnode "#{ipv6_ip}"))
+      subject.disconnect('SecondClient')
+      expect(subject).to receive(:run_cmd)
+        .with('bitcoin-cli disconnectnode "131.114.88.218:33422"')
+      subject.disconnect('FirstClient')
+    end
+
+    it 'should return if the filter is falsy' do
+      expect(subject).to_not receive(:getpeerinfo)
+      expect(subject).to_not receive(:run_cmd)
+      subject.disconnect(false)
+      subject.disconnect(nil)
+    end
+  end
+
+  describe '.display' do
+    it 'should display connected nodes' do
+      expect(subject).to receive(:getpeerinfo).and_return(PEER_INFO)
+      expect(subject).to receive(:ap).with(node_hash).and_return(node_hash)
+      expect(subject.display).to eq(node_hash)
+    end
+  end
+
+  describe '.run' do
+    context 'with opts' do
+      let(:opts) do
+        {
+          debug: true,
+          add: 'Wanted',
+          connect: 'Now',
+          ban: 'Nefarious',
+          disconnect: 'Fools'
+        }
+      end
+
+      it 'should call add, connect, ban and disconnect' do
+        expect(subject).to receive(:bitnodes_snapshot).once
+          .and_return(BITNODES_SNAPSHOT)
+        expect(subject).to receive(:getpeerinfo).and_return(PEER_INFO)
+        expect(subject).to receive(:add).with(opts[:add])
+          .and_call_original
+        expect(subject).to receive(:connect).with(opts[:connect])
+          .and_call_original
+        expect(subject).to receive(:ban).with(opts[:ban])
+          .and_call_original
+        expect(subject).to receive(:disconnect).with(opts[:disconnect])
+          .and_call_original
+        expect(subject).to_not receive(:display)
+        subject.run
+      end
+    end
+
+    context 'without opts' do
+      let(:opts) do
+        {}
+      end
+
+      it 'should call display' do
+        expect(subject).to receive(:getpeerinfo).once
+          .and_return(PEER_INFO)
+        expect(subject).to receive(:ap).with(node_hash).and_return(node_hash)
+        expect(subject).to receive(:display).and_call_original
+        subject.run
+      end
+    end
+  end
+
+  describe 'clear_cache' do
+    it 'should clear the bitnodes cache' do
+      expect(subject).to receive(:bitnodes_snapshot).once
+        .and_return(BITNODES_SNAPSHOT)
+      expect(subject).to receive(:run_cmd)
+        .with(%(bitcoin-cli addnode "#{ipv6_ip}" "add"))
+      expect(subject).to receive(:run_cmd)
+        .with('bitcoin-cli addnode "88.99.199.87:8333" "add"')
+      subject.add('Satoshi')
+      subject.add('Unlimited')
+      subject.clear_cache
+      expect(subject).to receive(:bitnodes_snapshot).once
+        .and_return(BITNODES_SNAPSHOT)
+      expect(subject).to receive(:run_cmd)
+        .with(%(bitcoin-cli addnode "#{ipv6_ip}" "add"))
+      subject.add('Satoshi')
+    end
+
+    it 'should clear the getpeerinfo cache' do
+      expect(subject).to receive(:getpeerinfo).once
+        .and_return(PEER_INFO)
+      expect(URI).to receive(:parse).with('https://131.114.88.218:33422')
+        .and_call_original
+      expect(URI).to receive(:parse).with(%(https://#{ipv6_ip}))
+        .and_call_original
+      expect(subject).to receive(:run_cmd)
+        .with('bitcoin-cli setban "131.114.88.218" "add"')
+      expect(subject).to receive(:run_cmd)
+        .with(%(bitcoin-cli setban "#{ipv6_ip_no_port}" "add"))
+      subject.ban('FirstClient')
+      subject.ban('SecondClient')
+      subject.clear_cache
+      expect(subject).to receive(:getpeerinfo).once
+        .and_return(PEER_INFO)
+      expect(URI).to receive(:parse).with('https://131.114.88.218:33422')
+        .and_call_original
+      expect(subject).to receive(:run_cmd)
+        .with('bitcoin-cli setban "131.114.88.218" "add"')
+      subject.ban('FirstClient')
+    end
+  end
+end

data/spec/rubocop_spec.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+require_relative 'spec_helper'
+
+describe 'Rubocop' do
+  it 'should pass with no offenses detected' do
+    expect(`rubocop`).to include('no offenses detected')
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+require File.join('bundler', 'setup')
+require 'rspec'
+require 'pickynode'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pickynode
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.1.0
 platform: ruby
 authors:
 - Josh Ellithorpe
@@ -38,14 +38,74 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '2.1'
-description: Some people are picky about the bitcoin nodes they connect to.
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.6'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.6'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '12.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '12.0'
+- !ruby/object:Gem::Dependency
+  name: rubocop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.48'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.48'
+description: |-
+  Some people are picky about the \
+  bitcoin nodes they connect to.
 email: quest@mac.com
 executables:
 - pickynode
 extensions: []
 extra_rdoc_files: []
 files:
+- ".gitignore"
+- ".rubocop.yml"
+- ".ruby-gemset"
+- ".ruby-version"
+- ".travis.yml"
+- Gemfile
+- Gemfile.lock
+- LICENSE
+- README.md
+- Rakefile
 - bin/pickynode
+- lib/pickynode.rb
+- pickynode.gemspec
+- spec/mocks.rb
+- spec/pickynode_spec.rb
+- spec/rubocop_spec.rb
+- spec/spec_helper.rb
 homepage: http://github.com/zquestz/pickynode
 licenses:
 - MIT
@@ -58,7 +118,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '0'
+      version: '2.0'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="