cerner/cerner_splunk

# frozen_string_literal: true

# Cookbook Name:: cerner_splunk
# Recipe:: _configure_indexes
#
# Configures the indexes.conf file.

indexbag = CernerSplunk.my_cluster_data(node)['indexes']
bag = CernerSplunk::DataBag.load(indexbag, secret: node['splunk']['data_bag_secret'])

unless bag
  Chef::Log.info 'No indexes data bag configured.'
  return
end

config = bag['config'] || {}
flags = bag['flags'] || {}

is_master = node['splunk']['node_type'] == :cluster_master

index_stanzas = config.inject({}) do |result, (stanza, index_config)|
  index_flags = flags[stanza] || {}
  hash = {}.merge! index_config

  stanza_type =
    case stanza
    when 'default'
      :default
    when /^volume:.*/
      :volume
    when /^provider-family:.*/
      :provider_family
    when /^.*:.*/
      :unknown
    else
      :index
    end

  daily_mb = hash.delete('_maxDailyDataSizeMB')
  padding = hash.delete('_dataSizePaddingPercent')
  default_config = config.fetch('default', {})

  s2_enabled_index = hash.key?('remotePath') || default_config.key?('remotePath')
  # _noGenerateTstatsHomePath is false  by default.
  no_gentstat = hash.delete('_noGenerateTstatsHomePath') || default_config['_noGenerateTstatsHomePath'] || s2_enabled_index
  if %i[index default].include?(stanza_type) && daily_mb && ((!s2_enabled_index && !hash.key?('maxTotalDataSizeMB')) || (s2_enabled_index && !hash.key?('maxGlobalDataSizeMB')))
    settings = CernerSplunk.my_cluster_data(node).fetch('settings', {})
    replication_factor = settings['replication_factor'] || 1
    indexer_count = settings['_cerner_splunk_indexer_count'] || 1
    # If the frozen time isn't specified, splunk defaults to 6 years
    frozen_time_in_secs = hash['frozenTimePeriodInSecs'] || default_config['frozenTimePeriodInSecs'] || 188_697_600
    frozen_time_in_days = frozen_time_in_secs / 86_400

    padding ||= default_config['_dataSizePaddingPercent']
    padding = padding.nil? ? 1.1 : 1 + (padding / 100.0)
    # For smartstore enabled indexes only .
    if s2_enabled_index
      hash['maxGlobalDataSizeMB'] = (daily_mb * padding * frozen_time_in_days).to_i
    else
      hash['maxTotalDataSizeMB'] = (daily_mb * padding * frozen_time_in_days * replication_factor).to_i / indexer_count
    end
  end

  if stanza_type == :index
    unless index_flags['noGeneratePaths']
      volume = hash.delete('_volume')
      base_path = volume ? "volume:#{volume}" : '$SPLUNK_DB'
      dir_name = hash.key?('_directory_name') ? hash.delete('_directory_name') : stanza
      hash['coldPath'] = "#{base_path}/#{dir_name}/colddb" unless hash['coldPath']
      hash['homePath'] = "#{base_path}/#{dir_name}/db" unless hash['homePath']
      hash['thawedPath'] = "$SPLUNK_DB/#{dir_name}/thaweddb" unless hash['thawedPath']
      hash['tstatsHomePath'] = "#{base_path}/#{dir_name}/datamodel_summary" if volume && !hash['tstatsHomePath'] && !no_gentstat
    end

    hash['repFactor'] = 'auto' if is_master && !index_flags['noRepFactor'] && !hash['repFactor']
  end

  result[stanza] = hash
  result
end

index_stanzas['_introspection'] = { 'repFactor' => 'auto' } if is_master && index_stanzas['_introspection'].nil?

path = is_master ? 'master-apps/_cluster/indexes.conf' : 'system/indexes.conf'

splunk_template path do
  stanzas index_stanzas
  notifies :touch, 'file[splunk-marker]', :immediately unless is_master
  notifies :run, 'execute[apply-cluster-bundle]' if is_master
end