build.rb

# frozen_string_literal: true

# Explicitly load dotenv at the beginning to ensure environment variables are loaded
begin
  require 'dotenv'
  Dotenv.load
  puts 'Loaded environment from .env file'
rescue LoadError
  puts "Warning: dotenv gem not found. Make sure it's installed with 'gem install dotenv'"
end

require_relative 'lib/lottery_factor_tool'
require 'optparse'

# Command-line argument parser
class CommandLineParser

  def self.parse(args)
    options = initialize_default_options
    parser = create_option_parser(options)
    
    parser.parse!(args)
    
    validate_options(options, parser)
    
    options[:database] ||= 'repo.db'
    options
  end
  
  def self.initialize_default_options
    {
      time_range: 30,
      top_display_count: 6,
      force: false,
    }
  end
  
  def self.create_option_parser(options)
    OptionParser.new do |opts|
      opts.banner = 'Usage: ruby build.rb [options]'
      
      define_repository_option(opts, options)
      define_database_option(opts, options)
      define_time_range_option(opts, options)
      define_output_option(opts, options)
      define_force_option(opts, options)
      define_top_display_option(opts, options)
    end
  end
  
  def self.define_repository_option(opts, options)
    opts.on('-r', '--repo REPO',
      "The GitHub repository in 'username/repo' format (required)") do |repo|
      options[:repo] = repo
    end
  end
  
  def self.define_database_option(opts, options)
    opts.on('-d', '--database FILENAME',
      'The SQLite database filename (default: repo.db)') do |filename|
      options[:database] = filename
    end
  end
  
  def self.define_time_range_option(opts, options)
    opts.on('-t', '--time-range DAYS', Integer, 'The time range in days (default: 30)') do |days|
      options[:time_range] = days
    end
  end
  
  def self.define_output_option(opts, options)
    opts.on('-o', '--output FILENAME',
      'The output HTML filename (default: owner-repo-lottery.html)') do |output|
      options[:output] = output
    end
  end
  
  def self.define_force_option(opts, options)
    opts.on('-f', '--force', 'Force fetching new data even if recent data exists') do
      options[:force] = true
    end
  end
  
  def self.define_top_display_option(opts, options)
    opts.on('--top-display COUNT', Integer,
      'Number of top contributors to display individually (default: 6)') do |count|
      options[:top_display_count] = count
    end
  end
  
  def self.validate_options(options, parser)
    return unless options[:repo].nil?

    puts 'Error: Repository (-r) is required.'
    puts parser.help
    exit 1
  end

end

# Main execution
if __FILE__ == $PROGRAM_NAME
  options = CommandLineParser.parse(ARGV)
  tool = LotteryFactorTool.new(options)
  tool.run
end