zold 0.0.4 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: d41d9a57531584dc4084646cd4d211fa55d5e7af
-  data.tar.gz: 50a0a58ed765f7efc239b9dddb870ee89efa6a86
+  metadata.gz: 4af388b06015493f1ef9524f79afae08cdf1c1b6
+  data.tar.gz: 59a0f2cbd9f03218906777eb44008a5deebbcf6a
 SHA512:
-  metadata.gz: 2d4576d6471e6c6bd4589ae2221b8f321291528f6e752d3cddf8ca70df135664d94499b7e5332c06053c87423fbbfdf3392e86c578305bb464b09b944db8b393
-  data.tar.gz: 043d5fdeb31a4afe610074c8426303b3378e8a239087c24ed496c0b40bcdd8bccf01ef3225393527404f1795953aed01099eecfab1d45c53e1e37e4725efa50b
+  metadata.gz: 6b14cca3133a4e94ebf9618d61b65992f2dbbd7b10a81251ee39aa8051f98b35787b305c47e42c1ff7a5bc42d191f644719299d03ac1d1195775b39a457cb372
+  data.tar.gz: d76056868277add9a8cf30134d58e930dd06c9c20dd31a6a80a72c0ffef293e92001b2c95e83351ad471d8e77d2be0b61c6f44df4841b284f485c14034c2eb50

data/.rubocop.yml

@@ -9,9 +9,11 @@ Metrics/CyclomaticComplexity:
   Max: 10
 Metrics/MethodLength:
   Enabled: false
-Style/MultilineMethodCallIndentation:
+Layout/MultilineMethodCallIndentation:
   Enabled: false
 Metrics/AbcSize:
   Enabled: false
 Metrics/BlockLength:
   Max: 50
+Metrics/ClassLength:
+  Max: 200

data/.simplecov

@@ -35,6 +35,6 @@ else
   SimpleCov.start do
     add_filter "/test/"
     add_filter "/features/"
-    minimum_coverage 10
+    minimum_coverage 60
   end
 end

data/README.md

@@ -60,6 +60,7 @@ Or do one of the following:
 
   * `zold init` creates a new wallet (you have to provide PGP keys)
   * `zold pull` pulls a wallet from the network
+  * `zold balance` checks the balance of a wallet
   * `zold send` creates a new transaction
   * `zold push` pushes a wallet to the network
 

data/Rakefile

@@ -59,7 +59,8 @@ RuboCop::RakeTask.new(:rubocop) do |task|
 end
 
 require 'cucumber/rake/task'
-Cucumber::Rake::Task.new(:features) do
+Cucumber::Rake::Task.new(:features) do |t|
+  t.cucumber_opts = 'features --format progress'
   Rake::Cleaner.cleanup_files(['coverage'])
 end
 Cucumber::Rake::Task.new(:'features:html') do |t|

data/assets/wallet.xsd

@@ -38,7 +38,7 @@ SOFTWARE.
     <xs:all>
       <xs:element name="beneficiary" type="wallet" minOccurs="1" maxOccurs="1"/>
       <xs:element name="date" type="xs:dateTime" minOccurs="1" maxOccurs="1"/>
-      <xs:element name="sign" type="xs:string" minOccurs="1" maxOccurs="1"/>
+      <xs:element name="sign" type="xs:string" minOccurs="0" maxOccurs="1"/>
       <xs:element name="amount" type="amount" minOccurs="1" maxOccurs="1"/>
     </xs:all>
     <xs:attribute name="id" use="required">
@@ -51,7 +51,7 @@ SOFTWARE.
   </xs:complexType>
   <xs:complexType name="ledger">
     <xs:sequence>
-      <xs:element name="txn" minOccurs="0" maxOccurs="unbounded"/>
+      <xs:element name="txn" type="txn" minOccurs="0" maxOccurs="unbounded"/>
     </xs:sequence>
   </xs:complexType>
   <xs:element name="wallet">

data/bin/zold

@@ -28,11 +28,16 @@ require 'rainbow'
 require_relative '../lib/zold'
 require_relative '../lib/zold/version'
 require_relative '../lib/zold/wallet'
+require_relative '../lib/zold/wallets'
 require_relative '../lib/zold/log'
 require_relative '../lib/zold/key'
 require_relative '../lib/zold/amount'
 require_relative '../lib/zold/commands/create'
 require_relative '../lib/zold/commands/send'
+require_relative '../lib/zold/commands/balance'
+require_relative '../lib/zold/commands/check'
+require_relative '../lib/zold/commands/pull'
+require_relative '../lib/zold/commands/push'
 
 Encoding.default_external = Encoding::UTF_8
 Encoding.default_internal = Encoding::UTF_8
@@ -47,6 +52,10 @@ Available commands:
       Creates a new wallet with a random ID
     #{Rainbow('pull').green} [id...]
       Pulls all local wallets and new ones explicitly required
+    #{Rainbow('check').green} id
+      Checks the validity of the wallet
+    #{Rainbow('balance').green} id
+      Prints the balance of the wallet
     #{Rainbow('send').green} source target amount
       Send ZOLD from one wallet to another
     #{Rainbow('push').green} [id...]
@@ -55,14 +64,17 @@ Available options:"
     o.string '-d', '--dir',
       'The directory where wallets are stored (default: current directory)',
       default: '.'
-    o.string '-p', '--private-key',
+    o.string '--private-key',
       'The location of RSA private key (default: ~/.ssh/id_rsa)',
       default: '~/.ssh/id_rsa'
-    o.string '-u', '--public-key',
+    o.string '--public-key',
       'The location of RSA public key (default: ~/.ssh/id_rsa.pub)',
       default: '~/.ssh/id_rsa.pub'
     o.bool '-h', '--help', 'Show these instructions'
     o.bool '--trace', 'Show full stack trace in case of a problem'
+    o.on '--no-colors', 'Disable colors in the ouput' do
+      Rainbow.enabled = false
+    end
     o.on '-v', '--version', 'Show current version' do
       puts Zold::VERSION
       exit
@@ -78,33 +90,51 @@ Available options:"
 
   command = opts.arguments[0]
 
+  wallets = Zold::Wallets.new(opts['dir'])
+
   case command
   when 'create'
     Zold::Create.new(
-      dir: opts['dir'],
-      pubkey: Zold::Key.new(opts['public-key']),
+      wallets: wallets,
+      pubkey: Zold::Key.new(file: opts['public-key']),
       log: log
     ).run
   when 'send'
+    raise "Payer wallet ID is required" if opts.arguments[1].nil?
+    raise "Recepient wallet ID is required" if opts.arguments[2].nil?
     Zold::Send.new(
-      payer: Zold::Wallet.new(
-        File.join(opts['dir'], "#{opts.arguments[1]}.xml")
-      ),
-      receiver: Zold::Wallet.new(
-        File.join(opts['dir'], "#{opts.arguments[2]}.xml")
-      ),
+      payer: wallets.find(Zold::Id.new(opts.arguments[1])),
+      receiver: wallets.find(Zold::Id.new(opts.arguments[2])),
       amount: Zold::Amount.new(zld: opts.arguments[3].to_f),
-      pvtkey: Zold::Key.new(opts['private-key']),
+      pvtkey: Zold::Key.new(file: opts['private-key']),
+      log: log
+    ).run
+  when 'balance'
+    raise "Wallet ID is required" if opts.arguments[1].nil?
+    Zold::Balance.new(
+      wallet: wallets.find(Zold::Id.new(opts.arguments[1])),
+      log: log
+    ).run
+  when 'check'
+    raise "Wallet ID is required" if opts.arguments[1].nil?
+    Zold::Check.new(
+      wallet: wallets.find(Zold::Id.new(opts.arguments[1])),
+      wallets: wallets,
       log: log
     ).run
   when 'pull'
-    raise 'PULL is not implemented yet'
+    Zold::Pull.new(
+      wallet: wallets.find(Zold::Id.new(opts.arguments[1])),
+      log: log
+    ).run
   when 'push'
-    raise 'PUSH is not implemented yet'
+    Zold::Push.new(
+      wallet: wallets.find(Zold::Id.new(opts.arguments[1])),
+      log: log
+    ).run
   else
     raise "Command '#{command}' is not supported"
   end
-
 rescue StandardError => ex
   log.error(ex.message)
   puts ex.backtrace if opts['trace']

data/features/cli.feature

@@ -11,3 +11,7 @@ Feature: Command Line Processing
     When I run bin/zold with "--version"
     Then Exit code is zero
 
+  Scenario: Wallet can be created
+    When I run bin/zold with "create --private-key id_rsa --public-key id_rsa.pub"
+    Then Exit code is zero
+

data/features/step_definitions/steps.rb

@@ -26,6 +26,8 @@ require_relative '../../lib/zold'
 Before do
   @cwd = Dir.pwd
   @dir = Dir.mktmpdir('test')
+  FileUtils.copy('fixtures/id_rsa', @dir)
+  FileUtils.copy('fixtures/id_rsa.pub', @dir)
   FileUtils.mkdir_p(@dir) unless File.exist?(@dir)
   Dir.chdir(@dir)
 end

data/lib/zold/amount.rb

@@ -18,6 +18,8 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 # SOFTWARE.
 
+require 'rainbow'
+
 # The amount.
 # Author:: Yegor Bugayenko (yegor256@gmail.com)
 # Copyright:: Copyright (c) 2018 Zerocracy, Inc.
@@ -29,8 +31,11 @@ module Zold
       raise 'You can\'t specify both coints and zld' if !coins.nil? && !zld.nil?
       @coins = coins unless coins.nil?
       @coins = (zld * 2**24).to_i unless zld.nil?
+      raise "Integer is required: #{@coins.class}" unless @coins.is_a?(Integer)
     end
 
+    ZERO = Amount.new(coins: 0)
+
     def to_i
       @coins
     end
@@ -40,13 +45,22 @@ module Zold
     end
 
     def to_s
-      "#{to_zld}ZLD"
+      text = "#{to_zld}ZLD"
+      if negative?
+        Rainbow(text).red
+      else
+        Rainbow(text).green
+      end
     end
 
     def ==(other)
       @coins == other.to_i
     end
 
+    def +(other)
+      Amount.new(coins: @coins + other.to_i)
+    end
+
     def zero?
       @coins.zero?
     end

data/lib/zold/commands/balance.rb

@@ -0,0 +1,41 @@
+# Copyright (c) 2018 Zerocracy, Inc.
+#
+# 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 NONINFINGEMENT. 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.
+
+require_relative '../log.rb'
+
+# BALANCE command.
+# Author:: Yegor Bugayenko (yegor256@gmail.com)
+# Copyright:: Copyright (c) 2018 Zerocracy, Inc.
+# License:: MIT
+module Zold
+  # Balance checking command
+  class Balance
+    def initialize(wallet:, log: Log::Quiet.new)
+      @wallet = wallet
+      @log = log
+    end
+
+    def run
+      balance = @wallet.balance
+      @log.info("The balance of #{@wallet} is #{balance}")
+      balance
+    end
+  end
+end

data/lib/zold/commands/check.rb

@@ -0,0 +1,58 @@
+# Copyright (c) 2018 Zerocracy, Inc.
+#
+# 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 NONINFINGEMENT. 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.
+
+require_relative '../log.rb'
+require_relative '../id.rb'
+require_relative 'pull.rb'
+
+# CHECK command.
+# Author:: Yegor Bugayenko (yegor256@gmail.com)
+# Copyright:: Copyright (c) 2018 Zerocracy, Inc.
+# License:: MIT
+module Zold
+  # Wallet checking command
+  class Check
+    def initialize(wallet:, wallets:, log: Log::Quiet.new)
+      @wallet = wallet
+      @wallets = wallets
+      @log = log
+    end
+
+    def run
+      clean = true
+      @wallet.income do |t|
+        bnf = Pull.new(
+          wallet: @wallets.find(Id.new(t[:beneficiary])),
+          log: @log
+        ).run
+        clean = bnf.check(t[:id], t[:amount], @wallet.id)
+        next if clean
+        @log.error("Txn ##{t[:id]} for #{t[:amount]} is absent at #{bnf.id}")
+        break
+      end
+      if clean
+        @log.info("The #{@wallet} is clean")
+      else
+        @log.error("The #{@wallet} is compromised")
+      end
+      clean
+    end
+  end
+end

data/lib/zold/commands/create.rb

@@ -29,18 +29,17 @@ require_relative '../id.rb'
 module Zold
   # Create command
   class Create
-    def initialize(dir:, pubkey:, log: Log::Quiet.new)
-      @dir = dir
+    def initialize(wallets:, pubkey:, log: Log::Quiet.new)
+      @wallets = wallets
       @pubkey = pubkey
       @log = log
     end
 
     def run
       id = Id.new
-      file = File.join(@dir, "#{id}.xml")
-      wallet = Wallet.new(file)
+      wallet = @wallets.find(id)
       wallet.init(id, @pubkey)
-      @log.info("#{wallet} created at #{file}")
+      @log.info("#{wallet} created at #{@wallets}")
       wallet
     end
   end

data/lib/zold/commands/pull.rb

@@ -0,0 +1,41 @@
+# Copyright (c) 2018 Zerocracy, Inc.
+#
+# 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 NONINFINGEMENT. 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.
+
+require_relative '../log.rb'
+
+# PULL command.
+# Author:: Yegor Bugayenko (yegor256@gmail.com)
+# Copyright:: Copyright (c) 2018 Zerocracy, Inc.
+# License:: MIT
+module Zold
+  # Wallet pulling command
+  class Pull
+    def initialize(wallet:, log: Log::Quiet.new)
+      @wallet = wallet
+      @log = log
+    end
+
+    def run
+      raise 'PULL doesn\'t work and the wallet is absent' unless @wallet.exists?
+      @log.info("The #{@wallet} is here")
+      @wallet
+    end
+  end
+end

data/lib/zold/commands/push.rb

@@ -0,0 +1,40 @@
+# Copyright (c) 2018 Zerocracy, Inc.
+#
+# 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 NONINFINGEMENT. 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.
+
+require_relative '../log.rb'
+
+# PUSH command.
+# Author:: Yegor Bugayenko (yegor256@gmail.com)
+# Copyright:: Copyright (c) 2018 Zerocracy, Inc.
+# License:: MIT
+module Zold
+  # Wallet pushing command
+  class Push
+    def initialize(wallet:, log: Log::Quiet.new)
+      @wallet = wallet
+      @log = log
+    end
+
+    def run
+      raise 'The wallet is absent' unless @wallet.exists?
+      raise 'PUSH is not implemented yet'
+    end
+  end
+end

data/lib/zold/commands/send.rb

@@ -37,8 +37,10 @@ module Zold
 
     def run
       raise "The amount can't be negative: #{@amount}" if @amount.negative?
-      @receiver.add(@payer.sub(@amount, @receiver.id, @pvtkey))
+      txn = @payer.sub(@amount, @receiver.id, @pvtkey)
+      @receiver.add(txn)
       @log.info("#{@amount} sent from #{@payer} to #{@receiver}")
+      txn[:id]
     end
   end
 end

data/lib/zold/id.rb

@@ -25,8 +25,8 @@
 module Zold
   # Id of the wallet
   class Id
-    def initialize
-      @id = rand(2**32..2**64 - 1)
+    def initialize(id = nil)
+      @id = id.nil? ? rand(2**32..2**64 - 1) : Integer("0x#{id}", 16)
     end
 
     def to_s

data/lib/zold/key.rb

@@ -28,23 +28,31 @@ require 'base64'
 module Zold
   # A key
   class Key
-    def initialize(file)
-      @file = File.expand_path(file)
+    def initialize(file: nil, text: nil)
+      unless file.nil?
+        path = File.expand_path(file)
+        raise "Can't find RSA key at #{file} (#{path})" unless File.exist?(path)
+        @body = File.read(path)
+      end
+      @body = text unless text.nil?
     end
 
     def to_s
       rsa.to_s.strip
     end
 
-    def encrypt(text)
-      Base64.encode64(rsa.private_encrypt(text))
+    def sign(text)
+      Base64.encode64(rsa.sign(OpenSSL::Digest::SHA256.new, text)).delete("\n")
+    end
+
+    def verify(signature, text)
+      rsa.verify(OpenSSL::Digest::SHA256.new, Base64.decode64(signature), text)
     end
 
     private
 
     def rsa
-      raise "Can't find RSA key at #{@file}" unless File.exist?(@file)
-      text = File.read(@file).strip
+      text = @body.strip
       unless text.start_with?('-----BEGIN')
         text = OpenSSHKeyConverter.decode_pubkey(text.split[1])
       end

data/lib/zold/version.rb

@@ -23,5 +23,5 @@
 # Copyright:: Copyright (c) 2018 Zerocracy, Inc.
 # License:: MIT
 module Zold
-  VERSION = '0.0.4'.freeze
+  VERSION = '0.0.5'.freeze
 end

data/lib/zold/wallet.rb

@@ -36,7 +36,12 @@ module Zold
       id
     end
 
+    def exists?
+      File.exist?(@file)
+    end
+
     def init(id, pubkey)
+      raise "File '#{@file}' already exists" if File.exist?(@file)
       File.write(
         @file,
         valid(
@@ -65,16 +70,22 @@ module Zold
     end
 
     def sub(amount, target, pvtkey)
-      txn = 1
-      date = Time.now.iso8601
       xml = load
+      txn = 1
+      unless xml.xpath('/wallet/ledger[txn]').empty?
+        txn = xml.xpath('/wallet/ledger/txn/@id')
+          .map(&:to_s)
+          .map(&:to_i)
+          .max + 1
+      end
+      date = Time.now
       t = xml.xpath('/wallet/ledger')[0].add_child('<txn/>')[0]
       t['id'] = txn
-      t.add_child('<date/>')[0].content = date
+      t.add_child('<date/>')[0].content = date.iso8601
       t.add_child('<amount/>')[0].content = -amount.to_i
       t.add_child('<beneficiary/>')[0].content = target
-      t.add_child('<sign/>')[0].content = pvtkey.encrypt(
-        "#{id} #{date} #{amount.to_i} #{target}"
+      t.add_child('<sign/>')[0].content = pvtkey.sign(
+        "#{txn} #{amount.to_i} #{target}"
       )
       save(xml)
       { id: txn, date: date, amount: amount, beneficiary: id }
@@ -84,15 +95,39 @@ module Zold
       xml = load
       t = xml.xpath('/wallet/ledger')[0].add_child('<txn/>')[0]
       t['id'] = "/#{txn[:id]}"
-      t.add_child('<date/>')[0].content = txn[:date]
+      t.add_child('<date/>')[0].content = txn[:date].iso8601
       t.add_child('<amount/>')[0].content = txn[:amount].to_i
       t.add_child('<beneficiary/>')[0].content = txn[:beneficiary]
       save(xml).to_s
     end
 
+    def check(id, amount, beneficiary)
+      xml = load
+      txn = xml.xpath("/wallet/ledger/txn[@id='#{id}']")[0]
+      Amount.new(coins: txn.xpath('amount/text()')[0].to_s.to_i).mul(-1) ==
+        amount &&
+        txn.xpath('beneficiary/text()')[0].to_s == beneficiary &&
+        Key.new(text: xml.xpath('/wallet/pkey/text()')[0].to_s).verify(
+          txn.xpath('sign/text()')[0].to_s,
+          "#{id} #{amount.to_i} #{beneficiary}"
+        )
+    end
+
+    def income
+      load.xpath('/wallet/ledger/txn[amount > 0]').each do |txn|
+        hash = {
+          id: txn['id'][1..-1].to_i,
+          beneficiary: Id.new(txn.xpath('beneficiary/text()')[0].to_s),
+          amount: Amount.new(coins: txn.xpath('amount/text()')[0].to_s.to_i)
+        }
+        yield hash
+      end
+    end
+
     private
 
     def load
+      raise "File '#{@file}' is absent" unless File.exist?(@file)
       valid(Nokogiri::XML(File.read(@file)))
     end
 
@@ -101,7 +136,14 @@ module Zold
     end
 
     def valid(xml)
-      xsd = Nokogiri::XML::Schema(File.open('assets/wallet.xsd'))
+      xsd = Nokogiri::XML::Schema(
+        File.open(
+          File.join(
+            File.dirname(__FILE__),
+            '../../assets/wallet.xsd'
+          )
+        )
+      )
       errors = xsd.validate(xml)
       unless errors.empty?
         errors.each do |error|

data/lib/zold/wallets.rb

@@ -0,0 +1,42 @@
+# Copyright (c) 2018 Zerocracy, Inc.
+#
+# 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 NONINFINGEMENT. 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.
+
+require_relative 'wallet.rb'
+
+# The local collection of wallets.
+# Author:: Yegor Bugayenko (yegor256@gmail.com)
+# Copyright:: Copyright (c) 2018 Zerocracy, Inc.
+# License:: MIT
+module Zold
+  # Collection of local wallets
+  class Wallets
+    def initialize(dir)
+      @dir = dir
+    end
+
+    def to_s
+      @dir
+    end
+
+    def find(id)
+      Zold::Wallet.new(File.join(@dir, "z-#{id}.xml"))
+    end
+  end
+end

data/test/commands/test_balance.rb

@@ -0,0 +1,44 @@
+# Copyright (c) 2018 Zerocracy, Inc.
+#
+# 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 NONINFINGEMENT. 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.
+
+require 'minitest/autorun'
+require 'tmpdir'
+require_relative '../../lib/zold/wallet.rb'
+require_relative '../../lib/zold/key.rb'
+require_relative '../../lib/zold/id.rb'
+require_relative '../../lib/zold/commands/balance.rb'
+
+# BALANCE test.
+# Author:: Yegor Bugayenko (yegor256@gmail.com)
+# Copyright:: Copyright (c) 2018 Zerocracy, Inc.
+# License:: MIT
+class TestBalance < Minitest::Test
+  def test_checks_wallet_balance
+    Dir.mktmpdir 'test' do |dir|
+      id = Zold::Id.new
+      wallet = Zold::Wallets.new(dir).find(id)
+      wallet.init(Zold::Id.new, Zold::Key.new(file: 'fixtures/id_rsa.pub'))
+      balance = Zold::Balance.new(
+        wallet: wallet
+      ).run
+      assert balance == Zold::Amount::ZERO
+    end
+  end
+end

data/test/commands/test_create.rb

@@ -20,7 +20,7 @@
 
 require 'minitest/autorun'
 require 'tmpdir'
-require_relative '../../lib/zold/wallet.rb'
+require_relative '../../lib/zold/wallets.rb'
 require_relative '../../lib/zold/key.rb'
 require_relative '../../lib/zold/commands/create.rb'
 
@@ -32,12 +32,12 @@ class TestCreate < Minitest::Test
   def test_creates_wallet
     Dir.mktmpdir 'test' do |dir|
       wallet = Zold::Create.new(
-        dir: dir,
-        pubkey: Zold::Key.new('fixtures/id_rsa.pub')
+        wallets: Zold::Wallets.new(dir),
+        pubkey: Zold::Key.new(file: 'fixtures/id_rsa.pub')
       ).run
       assert wallet.balance.zero?
       assert(
-        File.exist?(File.join(dir, "#{wallet.id}.xml")),
+        File.exist?(File.join(dir, "z-#{wallet.id}.xml")),
         "Wallet file not found: #{wallet.id}.xml"
       )
     end

data/test/commands/test_send.rb

@@ -33,15 +33,15 @@ class TestSend < Minitest::Test
   def test_sends_from_wallet_to_wallet
     Dir.mktmpdir 'test' do |dir|
       source = Zold::Wallet.new(File.join(dir, 'source.xml'))
-      source.init(Zold::Id.new, Zold::Key.new('fixtures/id_rsa.pub'))
+      source.init(Zold::Id.new, Zold::Key.new(file: 'fixtures/id_rsa.pub'))
       target = Zold::Wallet.new(File.join(dir, 'target.xml'))
-      target.init(Zold::Id.new, Zold::Key.new('fixtures/id_rsa.pub'))
+      target.init(Zold::Id.new, Zold::Key.new(file: 'fixtures/id_rsa.pub'))
       amount = Zold::Amount.new(zld: 14.95)
       Zold::Send.new(
         payer: source,
         receiver: target,
         amount: amount,
-        pvtkey: Zold::Key.new('fixtures/id_rsa')
+        pvtkey: Zold::Key.new(file: 'fixtures/id_rsa')
       ).run
       assert source.balance == amount.mul(-1)
       assert target.balance == amount

data/test/test_amount.rb

@@ -30,7 +30,7 @@ class TestAmount < Minitest::Test
   def test_parses_zld
     amount = Zold::Amount.new(zld: 14.95)
     assert(
-      amount.to_s == '14.95ZLD',
+      amount.to_s.include?('14.95ZLD'),
       "#{amount} is not equal to '14.95ZLD'"
     )
   end
@@ -38,7 +38,7 @@ class TestAmount < Minitest::Test
   def test_parses_coins
     amount = Zold::Amount.new(coins: 900_000_000)
     assert(
-      amount.to_s == '53.64ZLD',
+      amount.to_s.include?('53.64ZLD'),
       "#{amount} is not equal to '53.64ZLD'"
     )
   end

data/test/test_id.rb

@@ -41,4 +41,10 @@ class TestId < Minitest::Test
       assert id.to_s == before
     end
   end
+
+  def test_parses_id
+    hex = 'ff01889402fe0954'
+    id = Zold::Id.new(hex)
+    assert(id.to_s == hex, "#{id} is not equal to #{hex}")
+  end
 end

data/test/test_key.rb

@@ -28,14 +28,31 @@ require_relative '../lib/zold/key.rb'
 # License:: MIT
 class TestKey < Minitest::Test
   def test_reads_public_rsa
-    key = Zold::Key.new('fixtures/id_rsa.pub')
+    key = Zold::Key.new(file: 'fixtures/id_rsa.pub')
     assert key.to_s.start_with?("-----BEGIN PUBLIC KEY-----\nMIICI")
     assert key.to_s.end_with?("EAAQ==\n-----END PUBLIC KEY-----")
   end
 
   def test_reads_private_rsa
-    key = Zold::Key.new('fixtures/id_rsa')
+    key = Zold::Key.new(file: 'fixtures/id_rsa')
     assert key.to_s.start_with?("-----BEGIN RSA PRIVATE KEY-----\nMIIJJ")
     assert key.to_s.end_with?("Sg==\n-----END RSA PRIVATE KEY-----")
   end
+
+  def test_signs_and_verifies
+    pub = Zold::Key.new(file: 'fixtures/id_rsa.pub')
+    pvt = Zold::Key.new(file: 'fixtures/id_rsa')
+    text = 'How are you, my friend?'
+    signature = pvt.sign(text)
+    assert pub.verify(signature, text)
+  end
+
+  def test_signs_and_verifies_with_random_key
+    key = OpenSSL::PKey::RSA.new(2048)
+    pub = Zold::Key.new(text: key.public_key.to_s)
+    pvt = Zold::Key.new(text: key.to_s)
+    text = 'How are you doing, dude?'
+    signature = pvt.sign(text)
+    assert pub.verify(signature, text)
+  end
 end

data/test/test_wallet.rb

@@ -24,6 +24,7 @@ require_relative '../lib/zold/key.rb'
 require_relative '../lib/zold/id.rb'
 require_relative '../lib/zold/wallet.rb'
 require_relative '../lib/zold/amount.rb'
+require_relative '../lib/zold/commands/send.rb'
 
 # Wallet test.
 # Author:: Yegor Bugayenko (yegor256@gmail.com)
@@ -32,28 +33,79 @@ require_relative '../lib/zold/amount.rb'
 class TestWallet < Minitest::Test
   def test_adds_transaction
     Dir.mktmpdir 'test' do |dir|
-      file = File.join(dir, 'source.xml')
-      wallet = Zold::Wallet.new(file)
-      wallet.init(Zold::Id.new, Zold::Key.new('fixtures/id_rsa.pub'))
+      wallet = wallet(dir)
       amount = Zold::Amount.new(zld: 39.99)
-      wallet.sub(amount, 100, Zold::Key.new('fixtures/id_rsa'))
+      key = Zold::Key.new(file: 'fixtures/id_rsa')
+      wallet.sub(amount, Zold::Id.new, key)
+      wallet.sub(amount, Zold::Id.new, key)
       assert(
-        wallet.balance == amount.mul(-1),
-        "#{wallet.balance} is not equal to #{amount.mul(-1)}"
+        wallet.balance == amount.mul(-2),
+        "#{wallet.balance} is not equal to #{amount.mul(-2)}"
       )
     end
   end
 
   def test_initializes_it
     Dir.mktmpdir 'test' do |dir|
-      file = File.join(dir, 'source.xml')
-      wallet = Zold::Wallet.new(file)
-      id = Zold::Id.new.to_s
-      wallet.init(id, Zold::Key.new('fixtures/id_rsa.pub'))
+      pkey = Zold::Key.new(file: 'fixtures/id_rsa.pub')
+      Dir.chdir(dir) do
+        file = File.join(dir, 'source.xml')
+        wallet = Zold::Wallet.new(file)
+        id = Zold::Id.new.to_s
+        wallet.init(id, pkey)
+        assert(
+          wallet.id == id,
+          "#{wallet.id} is not equal to #{id}"
+        )
+      end
+    end
+  end
+
+  def test_iterates_income_transactions
+    Dir.mktmpdir 'test' do |dir|
+      wallet = wallet(dir)
+      wallet.add(
+        id: 1,
+        date: Time.now, amount: Zold::Amount.new(zld: 39.99),
+        beneficiary: Zold::Id.new
+      )
+      wallet.add(
+        id: 2,
+        date: Time.now, amount: Zold::Amount.new(zld: 14.95),
+        beneficiary: Zold::Id.new
+      )
+      sum = Zold::Amount::ZERO
+      wallet.income do |t|
+        sum += t[:amount]
+      end
       assert(
-        wallet.id == id,
-        "#{wallet.id} is not equal to #{id}"
+        sum == Zold::Amount.new(coins: 921_740_246),
+        "#{sum} is not equal to #{Zold::Amount.new(zld: 54.94)}"
       )
     end
   end
+
+  def test_checks_transaction
+    Dir.mktmpdir 'test' do |dir|
+      payer = wallet(dir)
+      receiver = wallet(dir)
+      amount = Zold::Amount.new(zld: 14.95)
+      txn = Zold::Send.new(
+        payer: payer, receiver: receiver,
+        amount: amount,
+        pvtkey: Zold::Key.new(file: 'fixtures/id_rsa')
+      ).run
+      assert payer.check(txn, amount, receiver.id)
+    end
+  end
+
+  private
+
+  def wallet(dir)
+    id = Zold::Id.new
+    file = File.join(dir, "#{id}.xml")
+    wallet = Zold::Wallet.new(file)
+    wallet.init(id, Zold::Key.new(file: 'fixtures/id_rsa.pub'))
+    wallet
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: zold
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.5
 platform: ruby
 authors:
 - Yegor Bugayenko
@@ -212,13 +212,19 @@ files:
 - fixtures/id_rsa.pub
 - lib/zold.rb
 - lib/zold/amount.rb
+- lib/zold/commands/balance.rb
+- lib/zold/commands/check.rb
 - lib/zold/commands/create.rb
+- lib/zold/commands/pull.rb
+- lib/zold/commands/push.rb
 - lib/zold/commands/send.rb
 - lib/zold/id.rb
 - lib/zold/key.rb
 - lib/zold/log.rb
 - lib/zold/version.rb
 - lib/zold/wallet.rb
+- lib/zold/wallets.rb
+- test/commands/test_balance.rb
 - test/commands/test_create.rb
 - test/commands/test_send.rb
 - test/test__helper.rb
@@ -258,6 +264,7 @@ test_files:
 - features/gem_package.feature
 - features/step_definitions/steps.rb
 - features/support/env.rb
+- test/commands/test_balance.rb
 - test/commands/test_create.rb
 - test/commands/test_send.rb
 - test/test__helper.rb