/*
Copyright 2023-2024 SolarWinds Worldwide, LLC.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

/*
Every instrumentation config patch should live in its own file in this
directory.

It should export a single `patch` function of type
`Patch<InstrumentationConfig>` where `InstrumentationConfig` is the config type
specific to the instrumentation being patched.
*/

import { type TextMapPropagator } from "@opentelemetry/api"
import { type InstrumentationConfig } from "@opentelemetry/instrumentation"

import { type SwConfiguration } from "../config"
import * as awsLambda from "./aws-lambda"
import * as awsSdk from "./aws-sdk"
import * as bunyan from "./bunyan"
import * as fs from "./fs"
import * as http from "./http"
import * as mysql2 from "./mysql2"
import * as pg from "./pg"
import * as pino from "./pino"
import * as winston from "./winston"

export interface SwPatchesConfiguration extends SwConfiguration {
  responsePropagator: TextMapPropagator<unknown>
}

export type Patch<Config extends InstrumentationConfig> = (
  config: Config,
  options: SwPatchesConfiguration,
) => Config

export const RESOURCE_SERVICE_NAME = "resource.service.name"

const patches = {
  "aws-lambda": awsLambda,
  "aws-sdk": awsSdk,
  bunyan,
  fs,
  http,
  mysql2,
  pg,
  pino,
  winston,
} as const
type Patches = typeof patches
type PatchableConfigs = {
  [Module in keyof Patches as `@opentelemetry/instrumentation-${Module}`]?: Patches[Module] extends {
    patch: Patch<infer Config>
  }
    ? Config
    : never
}

export function patch(
  configs: PatchableConfigs,
  options: SwPatchesConfiguration,
): PatchableConfigs {
  const patched = { ...configs }
  for (const [name, { patch }] of Object.entries(patches)) {
    const prefixed = `@opentelemetry/instrumentation-${name}` as const
    patched[prefixed] = (patch as Patch<InstrumentationConfig>)(
      configs[prefixed] ?? {},
      options,
    )
  }
  return patched
}