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
73
74
75
76
 
# File 'lib/zold/commands/fetch.rb', line 65

def fetch(id, cps, opts)
  total = 0
  @remotes.all.each do |r|
    done = fetch_one(id, r, cps, opts)
    if done
      total += 1
    else
      @remotes.error(r[:host], r[:port])
    end
  end
  @log.debug("#{total} copies fetched, there are #{cps.all.count} available locally")
end

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



78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
 
# File 'lib/zold/commands/fetch.rb', line 78

def fetch_one(id, r, cps, opts)
  address = "#{r[:host]}:#{r[:port]}".downcase
  if opts['ignore-node'].include?(address)
    @log.info("#{address} ignored because of --ignore-node")
    return false
  end
  uri = URI("#{r[:home]}wallet/#{id}")
  res = Http.new(uri).get
  if res.code == '404'
    @log.info("#{address} wallet #{Rainbow('not found').red}")
    return false
  end
  unless res.code == '200'
    @log.error("#{address} #{Rainbow(res.code).red}/#{res.message} at #{uri}")
    return false
  end
  json = JSON.parse(res.body)
  score = Score.parse_json(json['score'])
  unless score.valid?
    @log.error("#{address}: invalid score: #{score}")
    return false
  end
  if score.expired?
    @log.error("#{address}: score expired: #{score}")
    return false
  end
  if score.strength < Score::STRENGTH && !opts['ignore-score-weakness']
    @log.error("#{address} score is too weak (#{score.strength}<#{Score::STRENGTH}): #{score}")
    return false
  end
  cps.add(json['body'], score.host, score.port, score.value)
  @log.info("#{address} #{json['body'].length}b/#{Rainbow(score.value).green} (#{json['version']})")
  true
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