# frozen_string_literal: true

require "logger"
require "yologga/version"
require "yologga/gzip"
require "yologga/logs_lifetime"

pattern = File.join(File.dirname(__FILE__), "yologga", "patch", "*.rb")
Dir[pattern].each { |file| require_relative(file) }

module Yologga
  class << self
    attr_reader :gzip, :logs_lifetime

    def add_hourly_rotation_period_support!
      if hourly_mixin
        Logger::LogDevice.prepend(hourly_mixin)
      else
        warn unsupported_warning
      end
    end

    def default_rotation_period=(period)
      raise unsupported_warning if period == "hourly" && !hourly_mixin

      mixin = Module.new do
        define_method(:initialize) do |logdev, shift_age = period, *args|
          super(logdev, shift_age, *args)
        end
      end

      Logger.prepend(mixin)
    end

    #
    # :call-seq:
    #   gzip=(enabled)
    #
    # === Args
    #
    # +enabled+:: Boolean true of false
    #
    # === Examples
    #
    # Yologga.gzip = false
    # Yologga.gzip = true
    #
    # === Return
    #
    # supplied value or raises an Error
    #
    def gzip=(enabled)
      raise unable_to_find_gzip if enabled && !system("gzip --help &> /dev/null")
      @gzip = enabled
    end

    #
    # :call-seq:
    #   logs_lifetime=(logs_lifetime)
    #
    # === Args
    #
    # +logs_lifetime+:: ActiveSupport::Duration representing logs lifetime
    #
    # === Examples
    #
    # Yologga.logs_lifetime = 9.hours
    # Yologga.logs_lifetime = 30.days
    #
    # === Return
    #
    # supplied value
    #
    def logs_lifetime=(logs_lifetime)
      @logs_lifetime = logs_lifetime
    end

    private

    def hourly_mixin
      return @mixin if defined?(@mixin)

      mixin_name = "Yologga::Patch::Ruby_#{RUBY_VERSION.split('.').first(2).join('_')}"
      @mixin = begin
        Module.const_get(mixin_name)
      rescue NameError
        nil
      end
    end

    def unable_to_find_gzip
      "Unable to find gzip in PATH environment variable."
    end

    def unsupported_warning
      "Hourly log rotation period is not supported for Ruby #{RUBY_VERSION}"
    end
  end
end

Yologga.add_hourly_rotation_period_support!