Move script files to lib

Author: cadamini

distribution.rb lib/distribution.rbdistribution.rb renamed to lib/distribution.rb

@@ -0,0 +1,27 @@
+require 'date'
+
+class ExportDateRange
+  class << self
+    def for(year)
+      [start_date(year), end_date(year)]
+    end
+
+    private
+
+    def start_date(year)
+      Date.new(year, 1, 1)
+    end
+
+    def end_date(year)
+      if this_year?(year)
+        Date.new(year, Date.today.month, Date.today.day - 1)
+      else
+        Date.new(year, 12, 31)
+      end
+    end
+
+    def this_year?(year)
+      year == Date.today.year
+    end
+  end
+end

lib/export_date_range.rb

@@ -0,0 +1,34 @@
+require 'holidays'
+
+class HolidayClient
+  attr_reader :startdate, :enddate
+
+  def initialize(startdate, enddate, country_code)
+    @startdate = startdate
+    @enddate = enddate
+    @country_code = country_code
+  end
+
+  def holidays
+    holidays = {}
+    holidays_between.each do |holiday|
+      holidays[holiday[:name]] = holiday[:date]
+    end
+    holidays
+  end
+
+  private
+
+  def holiday_regions
+    Holidays.available_regions.select do |region|
+      region.to_s.start_with?(@country_code)
+    end
+  end
+
+  def holidays_between
+    Holidays.cache_between(
+      startdate, startdate.next_year(1) - 1,
+      *holiday_regions, :informal, :observed
+    ).values.flatten
+  end
+end

lib/holiday_client.rb

@@ -0,0 +1,85 @@
+require_relative 'holiday_client'
+require_relative 'distribution'
+require_relative 'interval_value_randomizer'
+
+class OpeningHours
+  def initialize(hours_array)
+    @hours_array = hours_array
+  end
+
+  def open
+    @hours_array[0]
+  end
+
+  def close
+    @hours_array[1]
+  end
+end
+
+class OutputGenerator
+  attr_reader :start_date, :end_date, :region, :queue_name
+
+  def initialize(start_date:, end_date:, region:, queue_name:, opening_hours:)
+    @region = region
+    @start_date = start_date
+    @end_date = end_date
+    @queue_name = queue_name
+    @opening_hours = OpeningHours.new(opening_hours)
+  end
+
+  def run
+    puts header
+
+    (start_date..end_date).each do |date|
+      output_csv_lines_for(date)
+    end
+  end
+
+  def output_csv_lines_for(date)
+    if holiday_dates.include?(date)
+      generate_output(
+        date, rand(12_000..13_500),
+        distribution.special_day, distribution.special_week
+      )
+    else
+      generate_output(date, rand(13_500..14_500),
+                      distribution.normal_day, distribution.normal_week)
+    end
+  end
+
+  def holiday_dates
+    HolidayClient.new(
+      start_date, end_date, region
+    ).holidays.map { |h| h[1] }
+  end
+
+  private
+
+  def build_time_string(interval)
+    time_in_minutes = interval * 15
+    hours = time_in_minutes / 60
+    minutes = time_in_minutes * 60 / 60 % 60
+    time = Time.new(2001, 1, 1, hours, minutes, 0)
+    time.strftime('%H:%M')
+  end
+
+  def header
+    'Queue;Date;Time;CallsOffered;CallsHandled;AHT'
+  end
+
+  def generate_output(date, weekly_volume,
+                      daily_distribution, weekly_destribution)
+    daily_volume = weekly_destribution[date.wday] * weekly_volume
+    daily_distribution.each_with_index do |value, interval|
+      random = IntervalValueRandomizer.new(daily_volume, value)
+      time_string = build_time_string(interval)
+      puts "#{queue_name};#{date.strftime('%d.%m.%Y')};#{time_string};" \
+           "#{random.calls};#{random.handled_calls};#{random.aht}"
+    end
+  end
+
+  def distribution
+    Distribution.new(open_time: @opening_hours.open,
+                     close_time: @opening_hours.close)
+  end
+end

interval_value_randomizer.rb lib/interval_value_randomizer.rbinterval_value_randomizer.rb renamed to lib/interval_value_randomizer.rb

@@ -1,6 +1,6 @@
-require_relative 'holiday_client'
-require_relative 'distribution'
-require_relative 'interval_value_randomizer'
+require_relative 'lib/holiday_client'
+require_relative 'lib/distribution'
+require_relative 'lib/interval_value_randomizer'
 
 class OpeningHours
   def initialize(hours_array)

lib/output_generator.rb

@@ -1,5 +1,5 @@
-require_relative 'export_date_range'
-require_relative 'output_generator'
+require_relative 'lib/export_date_range'
+require_relative 'lib/output_generator'
 
 # generate file for importer using > operator
 # ruby random_csv.rb > filename.csv