Rakefile

# frozen_string_literal: true

require 'rspec/core/rake_task'
require 'net/http'
require 'json'

namespace :test do
  desc 'Set up the test environment for functional tests'
  task :functional_setup do
    sh 'go mod vendor'
    sh 'go build -o dalfox .' # Explicitly name the output binary
  end

  desc 'Run the functional tests'
  RSpec::Core::RakeTask.new(functional: :functional_setup) do |t|
    t.pattern = 'spec/functional_tests/**/*_spec.rb'
    t.verbose = true # More output for debugging
  end

  desc 'Run the unit tests'
  task :unit do
    sh 'go test ./...'
  end

  desc 'Run all tests'
  task :all do
    Rake::Task['test:functional'].invoke
    Rake::Task['test:unit'].invoke
  end
end

namespace :docs do
  desc 'Serve the documentation site'
  task :serve do
    within_docs_directory do
      unless system('bundle check')
        puts "Bundler is not installed or dependencies are not met. Please run 'rake docs:install'."
        exit 1
      end
      sh 'bundle exec jekyll s'
    end
  end

  desc 'Install dependencies for the documentation site'
  task :install do
    within_docs_directory do
      sh 'bundle install'
    end
  end

  def within_docs_directory(&block)
    Dir.chdir('docs', &block)
  rescue Errno::ENOENT => e
    puts "Directory 'docs' not found: #{e.message}"
    exit 1
  rescue StandardError => e
    puts "An error occurred: #{e.message}"
    exit 1
  end
end

namespace :assets do
  desc 'Check remote assets'
  task :check do

    def check(endpoint)
      url = URI("https://assets.hahwul.com/#{endpoint}.json")
      response = Net::HTTP.get(url)

      data = JSON.parse(response)
      puts data
    end

    endpoints = [
      'xss-portswigger',
      'xss-payloadbox',
      'wl-params',
      'wl-assetnote-params'
    ]

    endpoints.each do |target|
      check target
    end
  end
end