mcoin 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 4d8259c823f92b80f9e0124b3443c5e1b74d6120
-  data.tar.gz: 6de15f6269ef50557c690f414bee21dfef1d63fe
+  metadata.gz: 91fde5f4e3e6f808c44d9355719f8c54f1246ef9
+  data.tar.gz: d467d3ce1cca1805f402a55c59f16a28ab230a24
 SHA512:
-  metadata.gz: 47b20bb1bed806bfcd73fe8a5d42c6709f80872cb088e3a23d126dc2dc339d9198fb1b9ff3990ec1b10ad407fb0c54e65037c68eb4bedb0678c591d22deabafd
-  data.tar.gz: eb57c25b083e53e67bb9d1a888e53f47ef56f1bc35038b265ae695d890ef1f767760eb9cb7a6abd41f0f154e8a0702ad543a9a08275d45d8067f0b07740c50d3
+  metadata.gz: 83c9569ebf33899094243524b65f2ca8f5f7793e40b27d3f1b4b295ee34593485b92af2a49f5f35b2edb328ad1fe5c5416ddebf022de77efc0569af116c7a314
+  data.tar.gz: c9f2645d8d1b7126cadcade42809e0a7017334f36a09eea466362acdb72c41ae46581d720e1ed7960c01ef80b7c9b0054bb2693cd404e1cc99dc2b8448779e41

data/.rubocop.yml

@@ -0,0 +1,2 @@
+AllCops:
+  TargetRubyVersion: 2.4.0

data/Gemfile

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 source 'https://rubygems.org'
 
 git_source(:github) { |repo_name| "https://github.com/#{repo_name}" }

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    mcoin (0.1.0)
+    mcoin (0.2.0)
 
 GEM
   remote: https://rubygems.org/

data/README.md

@@ -1,8 +1,6 @@
 # Mcoin
 
-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/mcoin`. To experiment with that code, run `bin/console` for an interactive prompt.
-
-TODO: Delete this and the text above, and describe your gem
+This is a side-project for me to monitor BTC/ETH. The target is create a helpful command line tool can run as cronjob to import data to InfluxDB and make some analytics.
 
 ## Installation
 
@@ -22,7 +20,32 @@ Or install it yourself as:
 
 ## Usage
 
-TODO: Write usage instructions here
+### Ticker
+
+Add `-m` to specify market name
+
+```
+mcoin ticker -m Bitfinex
+mcoin ticker -m Bitfinex -m Kraken # Multiple
+```
+
+If you want to save it into InfluxDB, add database information.
+
+```
+mcoin ticker -m Bitfinex -d monitor -e http://localhsot:8086
+```
+
+To get more information please use `-h` option.
+
+## Roadmap
+
+* [x] Fetch ticker
+  * [x] Fetch in parallels
+  * [x] Fetch from multiple market
+  * [ ] Fetch multiple coin type via specify `pair`
+* [ ] Support more public API command
+* [ ] Improve printer
+* [ ] Generalize database save interface
 
 ## Development
 
@@ -32,4 +55,4 @@ To install this gem onto your local machine, run `bundle exec rake install`. To
 
 ## Contributing
 
-Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/mcoin.
+Bug reports and pull requests are welcome on GitHub at https://github.com/elct9620/mcoin.

data/Rakefile

@@ -1,2 +1,4 @@
+# frozen_string_literal: true
+
 require 'bundler/gem_tasks'
 task default: :spec

data/bin/console

@@ -1,4 +1,5 @@
 #!/usr/bin/env ruby
+# frozen_string_literal: true
 
 require 'bundler/setup'
 require 'mcoin'

data/exe/mcoin

@@ -1,4 +1,5 @@
 #!/usr/bin/env ruby
+# frozen_string_literal: true
 
 require 'bundler/setup'
 require 'mcoin'

data/lib/mcoin.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require 'mcoin/version'
 
 # :nodoc:
@@ -9,5 +11,6 @@ module Mcoin
   autoload :Market, 'mcoin/market'
   autoload :Data, 'mcoin/data'
   autoload :Printer, 'mcoin/printer'
+  autoload :Parallel, 'mcoin/parallel'
   autoload :InfluxDB, 'mcoin/influx_db'
 end

data/lib/mcoin/command.rb

@@ -1,94 +1,36 @@
+# frozen_string_literal: true
+
 require 'optparse'
 require 'optparse/uri'
 require 'ostruct'
 
 module Mcoin
   # The command line interface
-  class Command < OptionParser
+  module Command
+    autoload :Base, 'mcoin/command/base'
+    autoload :Common, 'mcoin/command/common'
+    autoload :Ticker, 'mcoin/command/ticker'
+
+    autoload :HasMarket, 'mcoin/command/ext/has_market'
+    autoload :HasType, 'mcoin/command/ext/has_type'
+    autoload :Saveable, 'mcoin/command/ext/saveable'
+
     class << self
       def execute
-        new
+        pick.new.parse!.execute
       end
-    end
-
-    attr_reader :option
-
-    def initialize(&block)
-      super
-      prepare_options
-
-      self.banner = '=== Mcoin : Bitcoin Monitor Tools ==='
-      options_for_coin
-      options_for_influxdb
-      common_options
-
-      parse!
-      execute
-    end
-
-    def execute
-      raise OptionParser::MissingArgument, :market if option.market.nil?
-      market.print
-      # TODO: Provide general interface
-      market.save(option) if influxdb?
-    end
-
-    def market
-      @market ||= Market
-                  .pick(option.market)
-                  .new(option.type, option.currency)
-    end
 
-    def influxdb?
-      option.endpoint && option.database
-    end
-
-    def on(name, *args)
-      super(*args, ->(value) { option[name] = value })
-    end
-
-    def separator(content = nil)
-      super ''
-      super content unless content.nil?
-    end
-
-    private
-
-    def prepare_options
-      @option = OpenStruct.new
-      @option.type = :BTC
-      @option.currency = :USD
-    end
-
-    def common_options
-      separator 'Common Options: '
-      on_tail('-h', '--help', 'Show this message') do
-        puts self
-        exit
+      def pick
+        command = ARGV.first&.capitalize
+        return Common unless commands.include?(command&.to_sym)
+        const_get(command)
       end
 
-      on_tail('-v', '--version', 'Show version') do
-        puts "Mcoin #{VERSION}"
-        exit
+      def commands
+        constants.select do |klass|
+          const_get(klass).is_a?(Class)
+        end - %i[Base Common]
       end
     end
-
-    def options_for_influxdb
-      separator 'InfluxDB:'
-      on(:endpoint, '-e', '--endpoint ENDPOINT', URI, 'Database Endpoint')
-      on(:database, '-d', '--database NAME', String, 'Database Name')
-      on(:username, '-u', '--username USERNAME', String, 'Database Username')
-      on(:password, '-p', '--password PASSWORD', String, 'Database Password')
-    end
-
-    def options_for_coin
-      on(:type, '-t', '--type TYPE',
-         Mcoin::TYPES, "Available: #{Mcoin::TYPES.join(', ')}")
-      on(:currency, '-c', '--currency CURRENCY',
-         Mcoin::CURRENCY, "Available: #{Mcoin::CURRENCY.join(', ')}")
-      on(:market, '-m', '--market MARKET',
-         Mcoin::Market.available,
-         "Available: #{Mcoin::Market.available.join(', ')}")
-    end
   end
 end

data/lib/mcoin/command/base.rb

@@ -0,0 +1,70 @@
+# frozen_string_literal: true
+
+module Mcoin
+  module Command
+    # :nodoc:
+    class Base < OptionParser
+      class << self
+        attr_reader :options, :commands
+
+        def description(description = nil)
+          return @description if description.nil?
+          @description = description
+        end
+
+        def option(mode, name, *args)
+          @options ||= []
+          @options.push(Option.new(mode, name, *args))
+        end
+      end
+
+      attr_reader :option
+
+      def initialize
+        super
+
+        prepare
+        self.banner = '=== Mcoin : BTC Monitor Tool ==='
+        self.class.options&.each { |option| option.register(self) }
+      end
+
+      def prepare
+        @option = OpenStruct.new
+      end
+
+      def parse!
+        super
+        self
+      end
+
+      def execute
+        raise NotImplementedError
+      end
+
+      # Option Register
+      class Option
+        def initialize(mode, name, *args)
+          @mode = mode
+          @name = name.to_sym
+          @args = args
+        end
+
+        def register(command)
+          case @mode
+          when :single then register_as_single(command)
+          when :multiple then register_as_multiple(command)
+          end
+        end
+
+        def register_as_single(command)
+          command.on(*@args, ->(value) { command.option[@name] = value })
+        end
+
+        def register_as_multiple(command)
+          command.option[@name] ||= []
+          command.on(*@args, ->(value) { command.option[@name].push(value) })
+        end
+      end
+    end
+  end
+end

data/lib/mcoin/command/common.rb

@@ -0,0 +1,41 @@
+# frozen_string_literal: true
+
+module Mcoin
+  module Command
+    # :nodoc:
+    class Common < Base
+      def initialize
+        super
+
+        separator ''
+        separator 'Commands: '
+        print_commands
+
+        separator ''
+        separator 'Options: '
+        on_tail('-v', '--version', 'Show versions') do
+          puts "Mcoin #{Mcoin::VERSION}"
+          exit
+        end
+      end
+
+      def print_commands
+        Command.commands.each do |command|
+          separator "    #{command.downcase}\t" \
+            "#{Command.const_get(command).description}"
+        end
+      end
+
+      def parse!
+        super
+      rescue OptionParser::InvalidOption
+        puts self
+        exit
+      end
+
+      def execute
+        puts self
+      end
+    end
+  end
+end

data/lib/mcoin/command/ext/has_market.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+module Mcoin
+  module Command
+    # :nodoc:
+    module HasMarket
+      def self.included(base)
+        base.option(:multiple, :market, '-m', '--market MARKET',
+                    Mcoin::Market.available,
+                    "Available: #{Mcoin::Market.available.join(', ')}")
+      end
+    end
+  end
+end

data/lib/mcoin/command/ext/has_type.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module Mcoin
+  module Command
+    # :nodoc:
+    module HasType
+      def self.included(base)
+        base.option(:single, :type, '-t', '--type TYPE',
+                    Mcoin::TYPES, "Available: #{Mcoin::TYPES.join(', ')}")
+        base.option(:single, :currency, '-c', '--currency CURRENCY',
+                    Mcoin::CURRENCY, "Available: #{Mcoin::CURRENCY.join(', ')}")
+      end
+    end
+  end
+end

data/lib/mcoin/command/ext/saveable.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+module Mcoin
+  module Command
+    # :nodoc:
+    module Saveable
+      def self.included(base)
+        base.class_eval do
+          option(:single, :endpoint, '-e',
+                 '--endpoint ENDPOINT', URI, 'Database Endpoint')
+          option(:single, :database, '-d',
+                 '--database NAME', String, 'Database Name')
+          option(:single, :username, '-u',
+                 '--username USERNAME', String, 'Database Username')
+          option(:single, :password, '-p',
+                 '--password PASSWORD', String, 'Database Password')
+        end
+      end
+
+      def save?
+        option.endpoint && option.database
+      end
+
+      def database
+        InfluxDB.new(
+          option.endpoint,
+          option.database,
+          option.username,
+          option.password
+        )
+      end
+    end
+  end
+end

data/lib/mcoin/command/ticker.rb

@@ -0,0 +1,47 @@
+# frozen_string_literal: true
+
+module Mcoin
+  module Command
+    # :nodoc:
+    class Ticker < Base
+      include HasMarket
+      include HasType
+      include Saveable
+
+      attr_reader :option
+
+      description 'Show market information'
+
+      def execute
+        raise OptionParser::MissingArgument, :market if option.market.nil?
+        print
+        save if save?
+      end
+
+      def print
+        tickers = Parallel.map(markets, :fetch).map(&:to_ticker)
+        Printer.new(tickers).print
+      end
+
+      def save
+        database.save(markets.map(&:to_ticker).map(&:to_influx))
+      end
+
+      def markets
+        @markets ||= option.market.map do |name|
+          Market
+            .pick(name)
+            .new(option.type, option.currency)
+        end
+      end
+
+      private
+
+      def prepare
+        super
+        option.type = :BTC
+        option.currency = :USD
+      end
+    end
+  end
+end

data/lib/mcoin/data.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module Mcoin
   # :nodoc:
   module Data

data/lib/mcoin/data/ticker.rb

@@ -1,11 +1,14 @@
+# frozen_string_literal: true
+
 module Mcoin
   module Data
     # :nodoc:
     class Ticker
-      attr_reader :type, :currency
+      attr_reader :market, :type, :currency
       attr_accessor :last, :ask, :bid, :low, :high, :volume
 
-      def initialize(type, currency, data = {})
+      def initialize(market, type, currency, data = {})
+        @market = market
         @type = type
         @currency = currency
         data.each do |key, value|
@@ -14,7 +17,7 @@ module Mcoin
       end
 
       def to_influx(tags = {}, values = {})
-        tags = { type: @type, currency: @currency }.merge(tags)
+        tags = { type: @type, currency: @currency, market: @market }.merge(tags)
         values = {
           last: @last,
           ask: @ask, bid: @bid,

data/lib/mcoin/influx_db.rb

@@ -1,3 +1,6 @@
+# frozen_string_literal: true
+
+require 'pp'
 require 'net/http'
 
 module Mcoin

data/lib/mcoin/market.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module Mcoin
   # :nodoc:
   module Market

data/lib/mcoin/market/base.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require 'pp'
 require 'json'
 require 'net/http'
@@ -11,25 +13,6 @@ module Mcoin
         @currency = currency
       end
 
-      def print
-        fetch
-        puts '=== Result ==='
-        Printer.new(to_ticker).print
-        self
-      end
-
-      def save(option)
-        puts '=== InfluxDB ==='
-        InfluxDB.new(
-          option.endpoint,
-          option.database,
-          option.username,
-          option.password
-        ).save([to_ticker.to_influx(market: name)])
-        puts 'Saved'
-        self
-      end
-
       def name
         self.class.name.split('::').last
       end

data/lib/mcoin/market/bitfinex.rb

@@ -1,13 +1,15 @@
+# frozen_string_literal: true
+
 module Mcoin
   module Market
     # :nodoc:
     class Bitfinex < Base
-      # rubocop:disable Metrics/LineLength
-      ENDPOINT = 'https://api.bitfinex.com/v1/pubticker/%<type>s%<currency>s'.freeze
+      ENDPOINT = 'https://api.bitfinex.com/v1/pubticker/%<type>s%<currency>s'
 
       def to_ticker
         fetch
         Data::Ticker.new(
+          :Bitfinex,
           @type, @currency,
           last: @data['last_price'],
           ask: @data['ask'], bid: @data['bid'],

data/lib/mcoin/market/bitstamp.rb

@@ -1,13 +1,15 @@
+# frozen_string_literal: true
+
 module Mcoin
   module Market
     # :nodoc:
     class Bitstamp < Base
-      # rubocop:disable Metrics/LineLength
-      ENDPOINT = 'https://www.bitstamp.net/api/v2/ticker/%<type>s%<currency>s/'.freeze
+      ENDPOINT = 'https://www.bitstamp.net/api/v2/ticker/%<type>s%<currency>s/'
 
       def to_ticker
         fetch
         Data::Ticker.new(
+          :Bitstamp,
           @type, @currency,
           last: @data['last'],
           ask: @data['ask'], bid: @data['bid'],

data/lib/mcoin/market/kraken.rb

@@ -1,9 +1,11 @@
+# frozen_string_literal: true
+
 module Mcoin
   module Market
     # :nodoc:
     class Kraken < Base
       # rubocop:disable Metrics/LineLength
-      ENDPOINT = 'https://api.kraken.com/0/public/Ticker?pair=%<type>s%<currency>s'.freeze
+      ENDPOINT = 'https://api.kraken.com/0/public/Ticker?pair=%<type>s%<currency>s'
 
       def initialize(type, currency)
         type = swap_btc(type)
@@ -13,6 +15,7 @@ module Mcoin
       def to_ticker
         fetch
         Data::Ticker.new(
+          :Kraken,
           swap_btc(@type), @currency,
           last: @data['c'][0],
           ask: @data['a'][0], bid: @data['b'][0],
@@ -25,6 +28,7 @@ module Mcoin
         super
         return self if @data['result'].nil?
         @data = @data.dig('result', "X#{@type}Z#{@currency}")
+        self
       end
 
       def swap_btc(type)

data/lib/mcoin/parallel.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+module Mcoin
+  # :nodoc:
+  class Parallel
+    class << self
+      def map(array, method)
+        array.map do |item|
+          Thread.new { item.send(method) }
+        end.map(&:join).map(&:value)
+      end
+    end
+  end
+end

data/lib/mcoin/printer.rb

@@ -1,15 +1,37 @@
+# frozen_string_literal: true
+
 module Mcoin
   # :nodoc:
   class Printer
     def initialize(*rows)
       @rows = rows.flatten
+      @outputs = []
+
+      build
     end
 
     def print
-      puts row(columns.map(&:capitalize))
-      puts column_widths.map { |width| '-' * width }.join(' | ')
+      if @rows.empty?
+        puts 'No result found'
+      else
+        puts @outputs
+      end
+    end
+
+    def build
+      return if @rows.empty?
+      build_header
+      build_rows
+    end
+
+    def build_header
+      @outputs << row(columns.map(&:capitalize))
+      @outputs << column_widths.map { |width| '-' * width }.join(' | ')
+    end
+
+    def build_rows
       @rows.each do |row|
-        puts row(columns.map { |column| row.send(column) })
+        @outputs << row(columns.map { |column| row.send(column) })
       end
     end
 
@@ -27,7 +49,8 @@ module Mcoin
     end
 
     def columns
-      %i[currency type last ask bid]
+      # TODO: Load from command
+      %i[market currency type last ask bid]
     end
   end
 end

data/lib/mcoin/version.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module Mcoin
-  VERSION = '0.1.0'.freeze
+  VERSION = '0.2.0'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mcoin
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - 蒼時弦也
@@ -47,6 +47,7 @@ extensions: []
 extra_rdoc_files: []
 files:
 - ".gitignore"
+- ".rubocop.yml"
 - ".travis.yml"
 - Gemfile
 - Gemfile.lock
@@ -57,6 +58,12 @@ files:
 - exe/mcoin
 - lib/mcoin.rb
 - lib/mcoin/command.rb
+- lib/mcoin/command/base.rb
+- lib/mcoin/command/common.rb
+- lib/mcoin/command/ext/has_market.rb
+- lib/mcoin/command/ext/has_type.rb
+- lib/mcoin/command/ext/saveable.rb
+- lib/mcoin/command/ticker.rb
 - lib/mcoin/data.rb
 - lib/mcoin/data/ticker.rb
 - lib/mcoin/influx_db.rb
@@ -65,6 +72,7 @@ files:
 - lib/mcoin/market/bitfinex.rb
 - lib/mcoin/market/bitstamp.rb
 - lib/mcoin/market/kraken.rb
+- lib/mcoin/parallel.rb
 - lib/mcoin/printer.rb
 - lib/mcoin/version.rb
 - mcoin.gemspec