const opentelemetry = require('@opentelemetry/sdk-node')
const {
  getNodeAutoInstrumentations,
} = require('@opentelemetry/auto-instrumentations-node')
const {
  diag,
  DiagConsoleLogger,
  DiagLogLevel,
  trace,
} = require('@opentelemetry/api')
const { JaegerExporter } = require('@opentelemetry/exporter-jaeger')
const { Resource } = require('@opentelemetry/resources')
const {
  SemanticResourceAttributes,
} = require('@opentelemetry/semantic-conventions')
const GCP = require('@google-cloud/opentelemetry-cloud-trace-exporter')

let tracer

function tracingEnabled() {
  return process.env.GCP_OPENTELEMETRY || process.env.JAEGER_OPENTELEMETRY
}

function initialize(appName) {
  diag.setLogger(new DiagConsoleLogger(), DiagLogLevel.INFO)

  const resource = new Resource({
    [SemanticResourceAttributes.SERVICE_NAME]: appName,
    [SemanticResourceAttributes.SERVICE_NAMESPACE]: 'Overleaf',
    'host.type': 'VM',
  })

  let exporter
  if (process.env.GCP_OPENTELEMETRY) {
    exporter = new GCP.TraceExporter()
  } else if (process.env.JAEGER_OPENTELEMETRY) {
    exporter = new JaegerExporter({ host: process.env.JAEGER_HOST || 'jaeger' })
  } else {
    return
  }

  const sdk = new opentelemetry.NodeSDK({
    traceExporter: exporter,
    logger: console,

    instrumentations: [getNodeAutoInstrumentations()],
    resource,
  })

  tracer = trace.getTracer(appName)
  sdk.start()
}

function getTracer() {
  return tracer
}

module.exports = { initialize, getTracer, tracingEnabled }