import { ElasticClient } from '../ElasticClient'
import nodeIndex from '../Definitions/nodeIndex'
import dateProperty from '../Properties/dateProperty'

const assertNodeIndex = async (elastic: ElasticClient):Promise<void> => {
  console.info('Setting node pipeline')
  await elastic.ingest.putPipeline({
    id: 'node',
    description: 'Default node pipeline',
    processors: [
      {
        // @ts-ignore
        geoip: {
          ignore_missing: true,
          field: 'serverIps',
          properties: [
            'location',
            'continent_name',
            'country_name',
            'country_iso_code',
            'region_iso_code',
            'region_name',
            'city_name'
          ],
          target_field: 'geoip'
        }
      },
      {
        grok: {
          ignore_missing: true,
          field: 'softwareVersion',
          patterns: ['%{VERSION:standardizedSoftwareVersion}'],
          pattern_definitions: {
            VERSION: '(?:[0-9]+\\.?[0-9]+\\.?[0-9]+)'
          }
        }
      }
    ]
  })

  console.info('Checking node index')
  const exists = await elastic.indices.exists({
    index: nodeIndex
  })
  if (exists) {
    console.info('Node index exists')
  } else {
    console.info('Creating node index')
    await elastic.indices.create({
      index: nodeIndex,
      settings: {
        default_pipeline: 'node'
      }
    })
  }

  console.info('Setting node mapping')
  await elastic.indices.putMapping({
    index: nodeIndex,
    properties: {
      name: { type: 'text' },
      domain: { type: 'text' },
      softwareName: { type: 'keyword' },
      softwareVersion: { type: 'text' },
      standardizedSoftwareVersion: { type: 'keyword' },
      totalUserCount: { type: 'integer' },
      monthActiveUserCount: { type: 'integer' },
      halfYearActiveUserCount: { type: 'integer' },
      statusesCount: { type: 'integer' },
      foundAt: dateProperty,
      refreshedAt: dateProperty,
      refreshAttemptedAt: dateProperty,
      openRegistrations: { type: 'boolean' },
      serverIps: { type: 'ip' },
      discoveredByDomain: { type: 'text' },
      geoip: {
        type: 'nested',
        properties: {
          location: { type: 'geo_point' },
          continent_name: { type: 'keyword' },
          country_name: { type: 'keyword' },
          country_iso_code: { type: 'keyword' },
          region_name: { type: 'keyword' },
          region_iso_code: { type: 'keyword' },
          city_name: { type: 'keyword' }
        }
      },
      accountFeedCount: { type: 'integer' },
      channelFeedCount: { type: 'integer' }
    }
  })
  await elastic.indices.refresh({ index: nodeIndex })
}

export default assertNodeIndex