class ZuoraConnect::Telegraf

def app_name
  return ENV['DEIS_APP'].present? ? ENV['DEIS_APP'] : Rails.application.class.parent_name
end

def connect
  Rails.logger.debug(self.format_metric_log('Telegraf','Need new connection')) if ZuoraConnect.configuration.telegraf_debug
  uri = URI.parse(ZuoraConnect.configuration.telegraf_endpoint)
  self.host = UDPSocket.new.tap do |socket|
    socket.connect uri.host, uri.port
  end
rescue => ex
  self.host = nil
  Rails.logger.warn(self.format_metric_log('Telegraf', 'Failed to connect'))
  Rails.logger.warn(self.format_metric_log('Telegraf', ex.class))
  Rails.logger.warn(self.format_metric_log('Telegraf', ex.message))
end

def format_metric_log(message, dump = nil)
  message_color, dump_color = "1;91", "0;1"
  log_entry = "  \e[#{message_color}m#{message}\e[0m   "
  log_entry << "\e[#{dump_color}m%#{String === dump ? 's' : 'p'}\e[0m" % dump if dump
  log_entry
end

def initialize 
  self.connect
end

def pod_name
  return ENV['HOSTNAME'].present? ? ENV['HOSTNAME'] : 'Unknown'
end

def process_type
  p_type = 'Unknown'
  if ENV['HOSTNAME'] && ENV['DEIS_APP']
    temp = ENV['HOSTNAME'].split(ENV['DEIS_APP'])[1]
    temp = temp.split(/(-[0-9a-zA-Z]{5})$/)[0] # remove the 5 char hash
    p_type = temp[1, temp.rindex("-")-1]
  end
  return p_type
end

def write(direction: 'Unknown', tags: {}, values: {})
  time = Benchmark.measure do |bench|
    # To avoid writing metrics from rspec tests
    if Rails.env.to_sym != :test
      app_instance = Thread.current[:appinstance].present? ? Thread.current[:appinstance].id : 0
      tags = tags.merge({ app_name: app_name, process_type: process_type, app_instance: app_instance, pod_name: pod_name})
      if direction == :inbound
        self.write_udp(series: INBOUND_METRICS_NAME, tags: tags, values: values) if INBOUND_METRICS
      elsif direction == :outbound
        self.write_udp(series: OUTBOUND_METRICS_NAME, tags: tags, values: values) if OUTBOUND_METRICS
      else
        self.write_udp(series: direction, tags: tags, values: values)
      end
    end
  end
  if ZuoraConnect.configuration.telegraf_debug
    Rails.logger.debug(self.format_metric_log('Telegraf', tags.to_s))
    Rails.logger.debug(self.format_metric_log('Telegraf', values.to_s))
    Rails.logger.debug(self.format_metric_log('Telegraf', "Writing '#{direction.capitalize}': #{time.real.round(5)} ms"))
  end
end

def write_udp(series: '', tags: {}, values: {})
  return if !values.present?
  self.host.write InfluxDB::PointValue.new({series: series, tags: tags, values: values}).dump 
rescue => ex
  self.connect
  Rails.logger.warn(self.format_metric_log('Telegraf','Failed to write udp'))
  Rails.logger.warn(self.format_metric_log('Telegraf', ex.class))
  Rails.logger.warn(self.format_metric_log('Telegraf', ex.message))
end