tag.ts

/*
 * Unless explicitly stated otherwise all files in this repository are licensed
 * under the Apache License Version 2.0.
 *
 * This product includes software developed at Datadog (https://www.datadoghq.com/).
 * Copyright 2020-2026 Datadog, Inc.
 */

import { Tags } from "aws-cdk-lib";
import * as lambda from "aws-cdk-lib/aws-lambda";
import * as sfn from "aws-cdk-lib/aws-stepfunctions";
import log from "loglevel";
import { TagKeys, DatadogLambdaProps, DatadogStepFunctionsProps } from "./index";
import { LambdaFunction } from "./interfaces";

const versionJson = require("../version.json");

export function setTags(
  resource: LambdaFunction | sfn.StateMachine,
  props: DatadogLambdaProps | DatadogStepFunctionsProps,
): void {
  log.debug(`Adding datadog tags`);
  const taggable = resource instanceof lambda.SingletonFunction ? resource.permissionsNode.defaultChild! : resource;
  if (props.env) {
    Tags.of(taggable).add(TagKeys.ENV, props.env);
  }
  if (props.service) {
    Tags.of(taggable).add(TagKeys.SERVICE, props.service);
  }
  if (props.version) {
    Tags.of(taggable).add(TagKeys.VERSION, props.version);
  }
  if (props.tags) {
    const tagsArray = props.tags.split(",");
    tagsArray.forEach((tag: string) => {
      const [key, value] = tag.split(":");
      if (key && value) {
        Tags.of(taggable).add(key, value);
      }
    });
  }

  if (resource instanceof sfn.StateMachine) {
    setStepFunctionTags(resource);
  }
}

function setStepFunctionTags(stateMachine: sfn.StateMachine) {
  Tags.of(stateMachine).add(TagKeys.DD_TRACE_ENABLED, "true");
  Tags.of(stateMachine).add(TagKeys.CDK, `v${versionJson.version}`);
}