Class: Zold::Fetch

Inherits:
Object
  • Object
show all
Defined in:
lib/zold/commands/fetch.rb

Overview

FETCH pulling command

Instance Method Summary collapse

Constructor Details

#initialize(wallets:, remotes:, copies:, log: Log::Quiet.new) ⇒ Fetch

Returns a new instance of Fetch.



39
40
41
42
43
44
 
# File 'lib/zold/commands/fetch.rb', line 39

def initialize(wallets:, remotes:, copies:, log: Log::Quiet.new)
  @wallets = wallets
  @remotes = remotes
  @copies = copies
  @log = log
end

Instance Method Details

#fetch(id, cps, opts) ⇒ Object 



65
66
67
68
69
70
71
72
 
# File 'lib/zold/commands/fetch.rb', line 65

def fetch(id, cps, opts)
  total = 0
  @remotes.iterate(@log) do |r|
    fetch_one(id, r, cps, opts)
    total += 1
  end
  @log.debug("#{total} copies of #{id} fetched, there are #{cps.all.count} available locally")
end

#fetch_one(id, r, cps, opts) ⇒ Object 



74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
 
# File 'lib/zold/commands/fetch.rb', line 74

def fetch_one(id, r, cps, opts)
  start = Time.now
  if opts['ignore-node'].include?(r.to_s)
    @log.info("#{r} ignored because of --ignore-node")
    return false
  end
  res = r.http("/wallet/#{id}").get
  raise "Wallet #{id} not found" if res.code == '404'
  r.assert_code(200, res)
  json = JSON.parse(res.body)
  score = Score.parse_json(json['score'])
  r.assert_valid_score(score)
  raise "Score is too weak #{score.strength}" if score.strength < Score::STRENGTH && !opts['ignore-score-weakness']
  cps.add(json['body'], score.host, score.port, score.value)
  @log.info("#{r} returned #{json['body'].length}b/#{Rainbow(score.value).green} \
of #{id} (#{json['version']}) in #{(Time.now - start).round(2)}s")
end

#run(args = []) ⇒ Object 



46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
 
# File 'lib/zold/commands/fetch.rb', line 46

def run(args = [])
  opts = Slop.parse(args, help: true, suppress_errors: true) do |o|
    o.banner = "Usage: zold fetch [ID...] [options]
Available options:"
    o.bool '--ignore-score-weakness',
      'Don\'t complain when their score is too weak',
      default: false
    o.array '--ignore-node',
      'Ignore this node and don\'t fetch from it',
      default: []
    o.bool '--help', 'Print instructions'
  end
  mine = Args.new(opts, @log).take || return
  mine = @wallets.all if mine.empty?
  mine.each do |id|
    fetch(id, Copies.new(File.join(@copies, id)), opts)
  end
end