rust-raspberrypi-OS-tutorials/utils/devtool.rb
2022-04-25 23:43:16 +02:00

348 lines
8.2 KiB
Ruby
Executable File

#!/usr/bin/env ruby
# frozen_string_literal: true
# SPDX-License-Identifier: MIT OR Apache-2.0
#
# Copyright (c) 2020-2022 Andre Richter <andre.o.richter@gmail.com>
require 'rubygems'
require 'bundler/setup'
require 'colorize'
require 'fileutils'
require_relative 'devtool/copyright'
# Actions for tutorial folders.
class TutorialCrate
attr_reader :folder
def initialize(folder)
@folder = folder
end
def tutorial?
/[0-9]/.match?(@folder[0])
end
def clean
puts 'Cleaning '.light_blue + @folder
# No output needed.
Dir.chdir(@folder) { `make clean` }
end
def update
puts "\n\n"
puts 'Updating '.light_blue + @folder
Dir.chdir(@folder) { system('cargo update') }
end
def clippy(bsp)
puts "\n\n"
puts "Clippy #{@folder} - BSP: #{bsp}".light_blue
Dir.chdir(@folder) { exit(1) unless system("BSP=#{bsp} make clippy") }
end
def fmt_cargo_rust(args)
Dir.chdir(@folder) { exit(1) unless system("cargo fmt #{args}") }
end
def make(bsp)
puts "\n\n"
puts "Make #{@folder} - BSP: #{bsp}".light_blue
Dir.chdir(@folder) { exit(1) unless system("BSP=#{bsp} make") }
end
def test(bsp)
return unless boot_test?
puts "\n\n"
puts "Test #{@folder} - BSP: #{bsp}".light_blue
Dir.chdir(@folder) { exit(1) unless system("BSP=#{bsp} make test") }
end
def test_boot(bsp)
return unless boot_test?
puts "\n\n"
puts "Test Boot #{@folder} - BSP: #{bsp}".light_blue
Dir.chdir(@folder) { exit(1) unless system("BSP=#{bsp} make test_boot") }
end
def test_unit(bsp)
return unless unit_integration_tests?
puts "\n\n"
puts "Test Unit #{@folder} - BSP: #{bsp}".light_blue
Dir.chdir(@folder) { exit(1) unless system("BSP=#{bsp} make test_unit") }
end
def test_integration(bsp)
return unless unit_integration_tests?
puts "\n\n"
puts "Test Integration #{@folder} - BSP: #{bsp}".light_blue
Dir.chdir(@folder) { exit(1) unless system("BSP=#{bsp} make test_integration") }
end
private
def boot_test?
Dir.exist?("#{@folder}/kernel/tests")
end
def unit_integration_tests?
!Dir.glob("#{@folder}/kernel/tests/00_*.rs").empty?
end
end
# Forks commands to all applicable receivers.
class DevTool
def initialize
@user_has_supplied_crates = false
@bsp = bsp_from_env || SUPPORTED_BSPS.first
cl = user_supplied_crate_list || Dir['*/Cargo.toml'].sort
@crates = cl.map { |c| TutorialCrate.new(c.delete_suffix('/Cargo.toml')) }
end
def clean
@crates.each(&:clean)
end
def update
@crates.each(&:update)
end
def clippy(bsp = nil)
bsp ||= @bsp
@crates.each { |c| c.clippy(bsp) }
end
def diff
tuts = tutorials.map(&:folder)
padding = tuts.map(&:length).max
tuts[0..-2].each_with_index do |original, i|
update = tuts[i + 1]
diff_pair(original, update, padding)
end
end
def fmt
fmt_cargo_rust(check: false)
puts
fmt_prettier(check: false)
end
def fmt_check
fmt_cargo_rust(check: true)
puts
fmt_prettier(check: true)
end
def make(bsp = nil)
bsp ||= @bsp
@crates.each { |c| c.make(bsp) }
end
def make_xtra
return if @user_has_supplied_crates
puts "\n\n"
puts 'Make Xtra stuff'.light_blue
system('cd *_uart_chainloader && bash update.sh')
system('cd X1_JTAG_boot && bash update.sh')
end
def test(bsp = nil)
bsp ||= @bsp
@crates.each { |c| c.test(bsp) }
end
def test_boot(bsp = nil)
bsp ||= @bsp
@crates.each { |c| c.test_boot(bsp) }
end
def test_unit(bsp = nil)
bsp ||= @bsp
@crates.each { |c| c.test_unit(bsp) }
end
def test_integration(bsp = nil)
bsp ||= @bsp
@crates.each { |c| c.test_integration(bsp) }
end
def copyright
exit(1) unless copyright_check_files(copyright_source_files)
end
def misspell
puts 'Misspell'.light_blue
translations = ['README.CN.md', 'README.ES.md']
files = tracked_files.reject { |f| translations.include?(File.basename(f)) }
files = files.join(' ')
exit(1) unless system(".vendor/misspell -error #{files}")
end
def rubocop
puts 'Rubocop'.light_blue
exit(1) unless system('bundle exec rubocop')
end
def ready_for_publish_no_rust
clean
fmt
rubocop
copyright
diff
misspell
clean
end
def ready_for_publish
ready_for_publish_no_rust
make_xtra
clippy('rpi4')
clippy('rpi3')
test_boot('rpi3')
test_unit('rpi3')
test_integration('rpi3')
clean
end
private
SUPPORTED_BSPS = %w[rpi3 rpi4].freeze
def bsp_from_env
bsp = ENV.fetch('BSP', nil)
return bsp if SUPPORTED_BSPS.include?(bsp)
nil
end
def fmt_cargo_rust(check: false)
args = '-- --check' if check
@crates.each do |c|
print 'Rust cargo fmt '.light_blue
print "#{args} ".light_blue unless args.nil?
puts c.folder
Process.fork { c.fmt_cargo_rust(args) }
end
Process.waitall
end
def fmt_prettier(check: false)
args = if check
'--check'
else
'--write'
end
args += if @user_has_supplied_crates
" #{@crates.map(&:folder).join(' ')}"
else
' .'
end
puts 'Prettier:'.light_blue
exit(1) unless system("./node_modules/.bin/prettier #{args}")
end
def user_supplied_crate_list
folders = ARGV.drop(1)
return nil if folders.empty?
crates = folders.map { |d| "#{d}/Cargo.toml" }.sort
crates.each do |c|
unless File.exist?(c)
puts "Crate not found: #{c}"
exit(1)
end
end
@user_has_supplied_crates = true
crates
end
def tutorials
@crates.select(&:tutorial?)
end
def tracked_files
crate_list = @crates.map(&:folder).join(' ') if @user_has_supplied_crates
`git ls-files #{crate_list}`.split("\n") # crates_list == nil means all files
end
def diff_pair(original, update, padding)
# Only diff adjacent tutorials. This checks the numbers of the tutorial folders.
return unless original[0..1].to_i + 1 == update[0..1].to_i
# Skip for tutorial 11. Due to the change to virtual manifest, the diff is rather
# unreadable.
if original[0..1].to_i == 11
puts 'Skipping '.light_yellow +
"#{original}: Too noisy due to change to virtual manifest"
return
end
puts 'Diffing '.light_blue + original.ljust(padding) + " -> #{update}"
system("bash utils/diff_tut_folders.bash #{original} #{update}")
end
def copyright_source_files
extensions = ['.S', '.rs', '.rb']
# NOTE: The selection result is the return value of the function.
tracked_files.select do |f|
next unless File.exist?(f)
next if f.include?('build.rs')
next if f.include?('boot_test_string.rb')
f.include?('Makefile') ||
f.include?('Dockerfile') ||
extensions.include?(File.extname(f))
end
end
end
##--------------------------------------------------------------------------------------------------
## Execution starts here
##--------------------------------------------------------------------------------------------------
tool = DevTool.new
cmd = ARGV[0]
commands = tool.public_methods(false).sort
if commands.include?(cmd&.to_sym)
tool.public_send(cmd)
else
puts "Usage: ./#{__FILE__.split('/').last} COMMAND [optional list of folders]"
puts
puts 'Commands:'
commands.each { |m| puts " #{m}" }
exit(1)
end