1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-05 05:58:23 +00:00
trezor-firmware/crypto/tools/nem_test_vectors.rb

127 lines
2.6 KiB
Ruby
Raw Normal View History

2017-05-21 19:52:05 +00:00
#!/usr/bin/env ruby
require 'highline'
require 'open-uri'
TEMPLATE_NAME = (Pathname.new(__FILE__).sub_ext '.erb').to_s.freeze
@terminal = HighLine.new($stdin, $stderr)
def github_files
require 'octokit'
@github_files ||= Octokit.contents('NemProject/nem-test-vectors')
.select do |file|
file.name.end_with? '.dat'
end
end
def choose_data_file
@terminal.choose do |menu|
github_files.each do |file|
menu.choice(file.name) { file.download_url }
end
menu.prompt = 'Which file? '
menu.index = :none
menu.select_by = :name
end
end
def load_header(line)
line = line.dup
abort 'Header is not a comment' unless line.slice!(0) == '#'
header = line.split(':').each(&:strip!)
header.shift if header.first.empty?
header
2017-05-21 19:52:05 +00:00
end
def parse_field_answer(answer)
if answer.empty?
nil
elsif /^(?:(?<type>\w+) )?(?<identifier>\w+)$/ =~ answer
[identifier, type]
else
raise NotValidQuestionError
end
end
def ask_fields(header)
header.map do |name|
@terminal.ask "Field for `#{name}'? " do |question|
question.answer_type = lambda(&method(:parse_field_answer))
end
end
end
def load_data_line(line)
abort 'Line does not begin with colon' unless line.slice!(0) == ':'
2017-06-11 16:28:07 +00:00
line.strip!
line.chomp!(',')
2017-05-21 19:52:05 +00:00
values = line.split(':').each(&:strip!)
values.pop if values.last.empty?
values
end
def load_data(file, count)
file.each_line.lazy.reject { |line| line.start_with? '#' }
.take(count)
.map(&method(:load_data_line))
.to_a
end
def remove_skipped_fields(fields, data)
data.each do |values|
abort 'Line does not match header' unless values.size == fields.size
values.reject!.each_with_index { |_, index| fields[index].nil? }
end
fields.compact!
end
def format_data_fields(fields, data)
data.each do |values|
fields.each_with_index do |(_, type), index|
values[index] = values[index].dump if type.nil?
end
end
end
def template(source_url, fields, data)
test_name = @terminal.ask('Name for test? ') do |question|
question.validate = /./
end
erb = ERB.new(File.read(TEMPLATE_NAME), nil, '-')
erb.filename = TEMPLATE_NAME
erb.result(binding)
end
download_url = choose_data_file
source_code = open download_url do |file|
line = file.readline
header = load_header(line)
@terminal.say line
count = @terminal.ask('How many vectors to import? ', Integer)
fields = ask_fields(header)
data = load_data(file, count)
remove_skipped_fields(fields, data)
format_data_fields(fields, data)
template(download_url, fields, data)
end
puts source_code