testmycode/tmc-server

# frozen_string_literal: true

require 'fileutils'

# If you want to change the TMC Langs Rust version, see app/background_tasks/rust_langs_downloader_task.rb

module RustLangsCliExecutor
  def self.prepare_submission(clone_path, output_path, submission_path, extra_params = {}, config = {})
    command = "#{self.langs_executable_path}/current prepare-submission --clone-path #{clone_path} --output-path #{output_path} --submission-path #{submission_path}"
    command = command + " --top-level-dir-name #{config[:toplevel_dir_name]}" if !!config[:toplevel_dir_name]
    command = command + " --stub-zip-path #{config[:tests_from_stub]}" if !!config[:tests_from_stub]
    command = command + " --output-format #{config[:format]}" if !!config[:format]
    extra_params.each do |k, v|
      command = command + " --tmc-param #{k}=#{v}" unless v.nil?
    end

    command_output = `#{command}`
    Rails.logger.info(command_output)

    result = self.process_prepare_submission_command_output(command_output)

    if result['result'] == 'error'
      Rails.logger.error('Preparing submission failed: ' + result.to_s)
    end

    result
  end

  def self.refresh(course_template_refresh_task)
    course_template = course_template_refresh_task.course_template
    command = "#{self.langs_executable_path}/current refresh-course"\
    " --cache-path #{course_template.cache_path}"\
    " --cache-root #{Course.cache_root}"\
    " --course-name #{course_template.name}"\
    " --git-branch #{course_template.git_branch}"\
    " --source-url #{course_template.source_url}"

    Rails.logger.info(command)
    @course_refresh = course_template_refresh_task

    Open3.popen2(command) do |stdin, stdout, status_thread|
      Timeout.timeout(600, Timeout::Error, 'Refresh process took more than 10 minutes...') do
        stdout.each_line do |line|
          Rails.logger.info("Rust Refresh output \n#{line}")
          data = parse_as_JSON(line)
          return unless data

          @parsed_data = self.process_refresh_command_output(data)
          if data['output-kind'] == 'status-update'
            ActionCable.server.broadcast("CourseTemplateRefreshChannel-#{course_template.id}", @parsed_data) unless @parsed_data[:percent_done] == 1.0
            @course_refresh.percent_done = @parsed_data[:percent_done]
            @course_refresh.create_phase(@parsed_data[:message], @parsed_data[:time])
          end
        end
      end
    rescue Timeout::Error => e
      Rails.logger.info("Rust Refresh took too long, killing process. #{e}")
      status_thread.kill
      raise e
    end
    @course_refresh.percent_done = 0.95
    @course_refresh.save!
    @parsed_data
  end

  private
    def self.parse_as_JSON(output)
      JSON.parse output
    rescue StandardError => e
      Rails.logger.info("Could not parse output line. #{e}")
    end

    def self.langs_executable_path
      Pathname.new("#{::Rails.root}/vendor/tmc-langs-rust").relative_path_from(Pathname.new(Dir.pwd)).to_s
    end

    def self.process_prepare_submission_command_output(command_output)
      output_lines = command_output.split("\n")
      valid_lines = output_lines.map do |line|
        JSON.parse line
      rescue
        nil
      end.select { |line| !!line }

      last_line = valid_lines.last

      if last_line['status'] == 'crashed'
        Rails.logger.info('TMC-langs-rust crashed')
        Rails.logger.info(last_line)
        raise 'TMC-langs-rust crashed: ' + last_line
      end

      last_line
    end

    def self.process_refresh_command_output(command_output)
      data = command_output['data']
      if command_output['status'] == 'crashed'
        raise "TMC-langs-rust crashed: \n#{data}"
      elsif command_output['output-kind'] == 'status-update'
        {
          message: command_output['message'],
          percent_done: command_output['percent-done'],
          time: command_output['time'],
        }
      elsif command_output['status'] == 'finished'
        if command_output['result'] == 'error'
          raise data['output-data']['trace'].join("\n")
        elsif command_output['result'] == 'executed-command'
          data
        end
      end
    end
end