Rakefile

# frozen_string_literal: true

# Top-level Rakefile which is responsible for running all the other Rakefiles.

TARGET_ARCH = ENV['ARCH']

FOLDERS = if TARGET_ARCH == 'x86'
  %i[storm libraries programs servers storm_tests].freeze
elsif TARGET_ARCH == 'raspberrypi'
  # Nothing else has been ported to this arch yet.
  [:storm].freeze
else
  # Error handling is taken care of by common_settings.rake
  []
end

def command?(name)
  `which #{name}`
  $?.success?
end

verbose false

root = pwd

# Need to set this up using a fully qualified path name, since the Rakefiles in the subfolders won't be able to find the custom
# .rake files otherwise.
Rake.application.options.rakelib = ["#{root}/rakelib"]

desc 'Compiles chaos'
task default: [:create_ramdisk_image] + FOLDERS

desc 'Performs cleanup (removes old .o files and similar)'
task :clean do
  FOLDERS.each do |folder|
    sh "cd #{folder} && #{RAKE_COMMAND} clean"
  end
end

desc 'Creates the initial ramdisk image where the programs and their data is stored'
task :create_ramdisk_image do
  sh "cd servers/block/initial_ramdisk && #{RAKE_COMMAND} create_ramdisk_image"
end

desc 'Compiles and installs chaos'
task install: %i[create_ramdisk_image install_folders iso_image]

task :install_folders do
  FileUtils.rm_rf INSTALL_ROOT

  FOLDERS.each do |folder|
    sh "cd #{folder} && #{RAKE_COMMAND} install"
  end
end

desc 'Builds a bootable ISO image with the kernel, servers and programs.'
task :iso_image do
  FileUtils.mkdir_p "#{INSTALL_ROOT}/boot/grub"
  system "cp grub.cfg #{INSTALL_ROOT}/boot/grub"

  print 'Creating ISO image...'.cyan.bold

  # grub-mkrescue will fail silently if this is not present.
  if !command?('xorriso')
    printf "\n\n"
    raise 'Fatal error: xorriso missing (try something like this: apt-get install xorriso)'
  end

  # Even worse: grub-mkrescue will create a malfunctioning .iso which
  # cannot be booted if this is not present.
  if !File.exist?('/usr/lib/grub/i386-pc/boot_hybrid.img')
    printf "\n\n"
    raise 'Fatal error: grub-pc-bin missing (try something like this: apt-get install grub-pc-bin)'
  end

  sh "grub-mkrescue -o chaos.iso #{INSTALL_ROOT}"
  puts ' done.'
end

desc "Compiles the 'storm' kernel."
task :storm do |folder|
  sh "cd #{folder} && #{RAKE_COMMAND}"
end

desc "Compiles the unit tests for the 'storm' kernel."
task build_storm_tests: :storm do
  sh "cd storm_tests && #{RAKE_COMMAND}"
end

if TARGET_ARCH == 'x86'
  desc 'Runs the unit tests'
  task test: [:build_storm_tests] do
    sh "cd storm_tests && #{RAKE_COMMAND} test"
  end
elsif TARGET_ARCH == 'raspberrypi'
  # No-op since we don't support the tests on this architecture at the moment.
  task :test
end

task libraries: [:storm] do |folder|
  sh "cd #{folder} && #{RAKE_COMMAND}"
end

task :programs do |folder|
  sh "cd #{folder} && rake"
end

task :servers do |folder|
  sh "cd #{folder} && rake"
end

desc 'Builds a Docker image for running chaos'
task :docker_build do
  sh 'docker build -t chaos .'
end