# Adapted from https://github.com/eric/metriks-graphite/blob/master/lib/metriks/reporter/graphite.rb

require 'metriks'

module Dyndnsd
  class TextfileReporter
    attr_reader :file

    def initialize(file, options = {})
      @file = file

      @prefix = options[:prefix]

      @registry  = options[:registry] || Metriks::Registry.default
      @interval  = options[:interval] || 60
      @on_error  = options[:on_error] || proc { |ex| }
    end

    def start
      @thread ||= Thread.new do
        loop do
          sleep @interval

          Thread.new do
            begin
              write
            rescue StandardError => e
              @on_error[e] rescue nil
            end
          end
        end
      end
    end

    def stop
      @thread&.kill
      @thread = nil
    end

    def restart
      stop
      start
    end

    def write
      File.open(@file, 'w') do |f|
        @registry.each do |name, metric|
          case metric
          when Metriks::Meter
            write_metric f, name, metric, [
              :count, :one_minute_rate, :five_minute_rate,
              :fifteen_minute_rate, :mean_rate
            ]
          when Metriks::Counter
            write_metric f, name, metric, [
              :count
            ]
          when Metriks::UtilizationTimer
            write_metric f, name, metric, [
              :count, :one_minute_rate, :five_minute_rate,
              :fifteen_minute_rate, :mean_rate,
              :min, :max, :mean, :stddev,
              :one_minute_utilization, :five_minute_utilization,
              :fifteen_minute_utilization, :mean_utilization
            ], [
              :median, :get_95th_percentile
            ]
          when Metriks::Timer
            write_metric f, name, metric, [
              :count, :one_minute_rate, :five_minute_rate,
              :fifteen_minute_rate, :mean_rate,
              :min, :max, :mean, :stddev
            ], [
              :median, :get_95th_percentile
            ]
          when Metriks::Histogram
            write_metric f, name, metric, [
              :count, :min, :max, :mean, :stddev
            ], [
              :median, :get_95th_percentile
            ]
          end
        end
      end
    end

    def write_metric(file, base_name, metric, keys, snapshot_keys = [])
      time = Time.now.to_i

      base_name = base_name.to_s.gsub(/ +/, '_')
      base_name = "#{@prefix}.#{base_name}" if @prefix

      keys.flatten.each do |key|
        name = key.to_s.gsub(/^get_/, '')
        value = metric.send(key)
        file.write("#{base_name}.#{name} #{value} #{time}\n")
      end

      unless snapshot_keys.empty?
        snapshot = metric.snapshot
        snapshot_keys.flatten.each do |key|
          name = key.to_s.gsub(/^get_/, '')
          value = snapshot.send(key)
          file.write("#{base_name}.#{name} #{value} #{time}\n")
        end
      end
    end
  end
end