machine-image.ts

import { Construct } from 'constructs';
import { AmazonLinux2022ImageSsmParameter, AmazonLinux2022ImageSsmParameterProps, AmazonLinux2022Kernel } from './amazon-linux-2022';
import { AmazonLinux2023ImageSsmParameter, AmazonLinux2023ImageSsmParameterProps, AmazonLinux2023Kernel } from './amazon-linux-2023';
import { AmazonLinux2ImageSsmParameter, AmazonLinux2ImageSsmParameterProps, AmazonLinux2Kernel } from './amazon-linux2';
import { AmazonLinuxCpuType, AmazonLinuxEdition, AmazonLinuxGeneration, AmazonLinuxStorage, AmazonLinuxVirt, IMachineImage, MachineImageConfig, OperatingSystemType } from './common';
import { lookupImage } from './utils';
import * as ssm from '../../../aws-ssm';
import * as cxschema from '../../../cloud-assembly-schema';
import { ContextProvider, CfnMapping, Aws, Stack, Token } from '../../../core';
import * as cxapi from '../../../cx-api';
import { UserData } from '../user-data';
import { WindowsVersion } from '../windows-versions';

/**
 * Factory functions for standard Amazon Machine Image objects.
 */
export abstract class MachineImage {
  /**
   * A Windows image that is automatically kept up-to-date
   *
   * This Machine Image automatically updates to the latest version on every
   * deployment. Be aware this will cause your instances to be replaced when a
   * new version of the image becomes available. Do not store stateful information
   * on the instance if you are using this image.
   */
  public static latestWindows(version: WindowsVersion, props?: WindowsImageProps): IMachineImage {
    return new WindowsImage(version, props);
  }

  /**
   * An Amazon Linux 2 image that is automatically kept up-to-date
   *
   * This Machine Image automatically updates to the latest version on every
   * deployment. Be aware this will cause your instances to be replaced when a
   * new version of the image becomes available. Do not store stateful information
   * on the instance if you are using this image.
   */
  public static latestAmazonLinux2(props?: AmazonLinux2ImageSsmParameterProps): IMachineImage {
    return new AmazonLinux2ImageSsmParameter({
      cachedInContext: false,
      ...props,
      kernel: AmazonLinux2Kernel.CDK_LATEST,
    });
  }

  /**
   * An Amazon Linux 2022 image that is automatically kept up-to-date
   *
   * This Machine Image automatically updates to the latest version on every
   * deployment. Be aware this will cause your instances to be replaced when a
   * new version of the image becomes available. Do not store stateful information
   * on the instance if you are using this image.
   *
   * @deprecated - use latestAmazonLinux2023() instead
   */
  public static latestAmazonLinux2022(props?: AmazonLinux2022ImageSsmParameterProps): IMachineImage {
    return new AmazonLinux2022ImageSsmParameter({
      cachedInContext: false,
      ...props,
      kernel: AmazonLinux2022Kernel.CDK_LATEST,
    });
  }

  /**
   * An Amazon Linux 2023 image that is automatically kept up-to-date
   *
   * This Machine Image automatically updates to the latest version on every
   * deployment. Be aware this will cause your instances to be replaced when a
   * new version of the image becomes available. Do not store stateful information
   * on the instance if you are using this image.
   */
  public static latestAmazonLinux2023(props?: AmazonLinux2023ImageSsmParameterProps): IMachineImage {
    return new AmazonLinux2023ImageSsmParameter({
      cachedInContext: false,
      ...props,
      kernel: AmazonLinux2023Kernel.CDK_LATEST,
    });
  }

  /**
   * An Amazon Linux image that is automatically kept up-to-date
   *
   * This Machine Image automatically updates to the latest version on every
   * deployment. Be aware this will cause your instances to be replaced when a
   * new version of the image becomes available. Do not store stateful information
   * on the instance if you are using this image.
   *
   * N.B.: "latest" in the name of this function indicates that it always uses the most recent
   * image of a particular generation of Amazon Linux, not that it uses the "latest generation".
   * For backwards compatibility, this function uses Amazon Linux 1 if no generation
   * is specified.
   *
   * Specify the desired generation using the `generation` property:
   *
   * ```ts
   * ec2.MachineImage.latestAmazonLinux({
   *   // Use Amazon Linux 2
   *   generation: ec2.AmazonLinuxGeneration.AMAZON_LINUX_2,
   * })
   * ```
   * @deprecated use MachineImage.latestAmazonLinux2 instead
   */
  public static latestAmazonLinux(props?: AmazonLinuxImageProps): IMachineImage {
    return new AmazonLinuxImage(props);
  }

  /**
   * A Linux image where you specify the AMI ID for every region
   *
   * @param amiMap For every region where you are deploying the stack,
   * specify the AMI ID for that region.
   * @param props Customize the image by supplying additional props
   */
  public static genericLinux(amiMap: Record<string, string>, props?: GenericLinuxImageProps): IMachineImage {
    return new GenericLinuxImage(amiMap, props);
  }

  /**
   * A Windows image where you specify the AMI ID for every region
   *
   * @param amiMap For every region where you are deploying the stack,
   * specify the AMI ID for that region.
   * @param props Customize the image by supplying additional props
   */
  public static genericWindows(amiMap: Record<string, string>, props?: GenericWindowsImageProps): IMachineImage {
    return new GenericWindowsImage(amiMap, props);
  }

  /**
   * An image specified in SSM parameter store that is automatically kept up-to-date
   *
   * This Machine Image automatically updates to the latest version on every
   * deployment. Be aware this will cause your instances to be replaced when a
   * new version of the image becomes available. Do not store stateful information
   * on the instance if you are using this image.
   *
   * @param parameterName The name of SSM parameter containing the AMi id
   * @param os The operating system type of the AMI
   * @param userData optional user data for the given image
   * @deprecated Use `MachineImage.fromSsmParameter()` instead
   */
  public static fromSSMParameter(parameterName: string, os: OperatingSystemType, userData?: UserData): IMachineImage {
    return new GenericSSMParameterImage(parameterName, os, userData);
  }

  /**
   * An image specified in SSM parameter store
   *
   * By default, the SSM parameter is refreshed at every deployment,
   * causing your instances to be replaced whenever a new version of the AMI
   * is released.
   *
   * Pass `{ cachedInContext: true }` to keep the AMI ID stable. If you do, you
   * will have to remember to periodically invalidate the context to refresh
   * to the newest AMI ID.
   */
  public static fromSsmParameter(parameterName: string, options?: SsmParameterImageOptions): IMachineImage {
    return new GenericSsmParameterImage(parameterName, options);
  }

  /**
   * An image specified in SSM parameter store that will be resolved at instance launch time.
   *
   * The AMI ID will be resolved at instance launch time.
   *
   * @param parameterName The name of SSM parameter containing the AMI ID
   * @param options The parameter image options
   *
   * @see https://docs.aws.amazon.com/autoscaling/ec2/userguide/using-systems-manager-parameters.html
   *
   */
  public static resolveSsmParameterAtLaunch(parameterName: string, options?: SsmParameterImageOptions): IMachineImage {
    return new ResolveSsmParameterAtLaunchImage(parameterName, options);
  }

  /**
   * Look up a shared Machine Image using DescribeImages
   *
   * The most recent, available, launchable image matching the given filter
   * criteria will be used. Looking up AMIs may take a long time; specify
   * as many filter criteria as possible to narrow down the search.
   *
   * The AMI selected will be cached in `cdk.context.json` and the same value
   * will be used on future runs. To refresh the AMI lookup, you will have to
   * evict the value from the cache using the `cdk context` command. See
   * https://docs.aws.amazon.com/cdk/latest/guide/context.html for more information.
   *
   * This function can not be used in environment-agnostic stacks.
   */
  public static lookup(props: LookupMachineImageProps): IMachineImage {
    return new LookupMachineImage(props);
  }
}

/**
 * Select the image based on a given SSM parameter at deployment time of the CloudFormation Stack.
 *
 * This Machine Image automatically updates to the latest version on every
 * deployment. Be aware this will cause your instances to be replaced when a
 * new version of the image becomes available. Do not store stateful information
 * on the instance if you are using this image.
 *
 * The AMI ID is selected using the values published to the SSM parameter store.
 */
export class GenericSSMParameterImage implements IMachineImage {
  // FIXME: this class ought to be `@deprecated` and removed from v2, but that
  // is causing build failure right now. Ref: https://github.com/aws/jsii/issues/3025
  // @-deprecated Use `MachineImage.fromSsmParameter()` instead

  /**
   * Name of the SSM parameter we're looking up
   */
  public readonly parameterName: string;

  constructor(parameterName: string, private readonly os: OperatingSystemType, private readonly userData?: UserData) {
    this.parameterName = parameterName;
  }

  /**
   * Return the image to use in the given context
   */
  public getImage(scope: Construct): MachineImageConfig {
    const ami = ssm.StringParameter.valueForTypedStringParameterV2(scope, this.parameterName, ssm.ParameterValueType.AWS_EC2_IMAGE_ID);
    return {
      imageId: ami,
      osType: this.os,
      userData: this.userData ?? (this.os === OperatingSystemType.WINDOWS ? UserData.forWindows() : UserData.forLinux()),
    };
  }
}

/**
 * Select the image based on a given SSM parameter at instance launch time.
 *
 * This Machine Image comes with an imageId as `resolve:ssm:parameter-name` or `resolve:ssm:parameter-name:version` format
 * as described in the document:
 *
 * @see https://docs.aws.amazon.com/autoscaling/ec2/userguide/using-systems-manager-parameters.html
 *
 * The AMI ID would be selected at instance launch time.
 */
export class ResolveSsmParameterAtLaunchImage implements IMachineImage {
  /**
   * Name of the SSM parameter we're looking up
   */
  public readonly parameterName: string;

  constructor(parameterName: string, private readonly props: SsmParameterImageOptions = {}) {
    this.parameterName = parameterName;
  }

  /**
   * Return the image to use in the given context
   */
  public getImage(_: Construct): MachineImageConfig {
    const versionString = this.props.parameterVersion ? `:${this.props.parameterVersion}` : '';
    const osType = this.props.os ?? OperatingSystemType.LINUX;
    return {
      imageId: `resolve:ssm:${this.parameterName}${versionString}`,
      osType,
      userData: this.props.userData ?? (osType === OperatingSystemType.WINDOWS ? UserData.forWindows() : UserData.forLinux()),
    };
  }
}

/**
 * Properties for GenericSsmParameterImage
 */
export interface SsmParameterImageOptions {
  /**
   * Operating system
   *
   * @default OperatingSystemType.LINUX
   */
  readonly os?: OperatingSystemType;

  /**
   * Custom UserData
   *
   * @default - UserData appropriate for the OS
   */
  readonly userData?: UserData;

  /**
   * Whether the AMI ID is cached to be stable between deployments
   *
   * By default, the newest image is used on each deployment. This will cause
   * instances to be replaced whenever a new version is released, and may cause
   * downtime if there aren't enough running instances in the AutoScalingGroup
   * to reschedule the tasks on.
   *
   * If set to true, the AMI ID will be cached in `cdk.context.json` and the
   * same value will be used on future runs. Your instances will not be replaced
   * but your AMI version will grow old over time. To refresh the AMI lookup,
   * you will have to evict the value from the cache using the `cdk context`
   * command. See https://docs.aws.amazon.com/cdk/latest/guide/context.html for
   * more information.
   *
   * Can not be set to `true` in environment-agnostic stacks.
   *
   * @default false
   */
  readonly cachedInContext?: boolean;

  /**
   * The version of the SSM parameter.
   *
   * @default no version specified.
   */
  readonly parameterVersion?: string;
}

/**
 * Select the image based on a given SSM parameter at deployment time of the CloudFormation Stack.
 *
 * This Machine Image automatically updates to the latest version on every
 * deployment. Be aware this will cause your instances to be replaced when a
 * new version of the image becomes available. Do not store stateful information
 * on the instance if you are using this image.
 *
 * The AMI ID is selected using the values published to the SSM parameter store.
 */
class GenericSsmParameterImage implements IMachineImage {
  constructor(private readonly parameterName: string, private readonly props: SsmParameterImageOptions = {}) {
  }

  /**
   * Return the image to use in the given context
   */
  public getImage(scope: Construct): MachineImageConfig {
    const imageId = lookupImage(scope, this.props.cachedInContext, this.parameterName);

    const osType = this.props.os ?? OperatingSystemType.LINUX;
    return {
      imageId,
      osType,
      userData: this.props.userData ?? (osType === OperatingSystemType.WINDOWS ? UserData.forWindows() : UserData.forLinux()),
    };
  }
}

/**
 * Configuration options for WindowsImage
 */
export interface WindowsImageProps {
  /**
   * Initial user data
   *
   * @default - Empty UserData for Windows machines
   */
  readonly userData?: UserData;
}

/**
 * Select the latest version of the indicated Windows version
 *
 * This Machine Image automatically updates to the latest version on every
 * deployment. Be aware this will cause your instances to be replaced when a
 * new version of the image becomes available. Do not store stateful information
 * on the instance if you are using this image.
 *
 * The AMI ID is selected using the values published to the SSM parameter store.
 *
 * https://aws.amazon.com/blogs/mt/query-for-the-latest-windows-ami-using-systems-manager-parameter-store/
 */
export class WindowsImage extends GenericSSMParameterImage {
  private static DEPRECATED_VERSION_NAME_MAP: Partial<Record<WindowsVersion, WindowsVersion>> = {
    [WindowsVersion.WINDOWS_SERVER_2016_GERMAL_FULL_BASE]: WindowsVersion.WINDOWS_SERVER_2016_GERMAN_FULL_BASE,
    [WindowsVersion.WINDOWS_SERVER_2012_R2_SP1_PORTUGESE_BRAZIL_64BIT_CORE]: WindowsVersion.WINDOWS_SERVER_2012_R2_SP1_PORTUGUESE_BRAZIL_64BIT_CORE,
    [WindowsVersion.WINDOWS_SERVER_2016_PORTUGESE_PORTUGAL_FULL_BASE]: WindowsVersion.WINDOWS_SERVER_2016_PORTUGUESE_PORTUGAL_FULL_BASE,
    [WindowsVersion.WINDOWS_SERVER_2012_R2_RTM_PORTUGESE_BRAZIL_64BIT_BASE]: WindowsVersion.WINDOWS_SERVER_2012_R2_RTM_PORTUGUESE_BRAZIL_64BIT_BASE,
    [WindowsVersion.WINDOWS_SERVER_2012_R2_RTM_PORTUGESE_PORTUGAL_64BIT_BASE]:
      WindowsVersion.WINDOWS_SERVER_2012_R2_RTM_PORTUGUESE_PORTUGAL_64BIT_BASE,
    [WindowsVersion.WINDOWS_SERVER_2016_PORTUGESE_BRAZIL_FULL_BASE]: WindowsVersion.WINDOWS_SERVER_2016_PORTUGUESE_BRAZIL_FULL_BASE,
    [WindowsVersion.WINDOWS_SERVER_2012_SP2_PORTUGESE_BRAZIL_64BIT_BASE]: WindowsVersion.WINDOWS_SERVER_2012_SP2_PORTUGUESE_BRAZIL_64BIT_BASE,
    [WindowsVersion.WINDOWS_SERVER_2012_RTM_PORTUGESE_BRAZIL_64BIT_BASE]: WindowsVersion.WINDOWS_SERVER_2012_RTM_PORTUGUESE_BRAZIL_64BIT_BASE,
    [WindowsVersion.WINDOWS_SERVER_2008_R2_SP1_PORTUGESE_BRAZIL_64BIT_BASE]: WindowsVersion.WINDOWS_SERVER_2008_R2_SP1_PORTUGUESE_BRAZIL_64BIT_BASE,
    [WindowsVersion.WINDOWS_SERVER_2008_SP2_PORTUGESE_BRAZIL_32BIT_BASE]: WindowsVersion.WINDOWS_SERVER_2008_SP2_PORTUGUESE_BRAZIL_32BIT_BASE,
    [WindowsVersion.WINDOWS_SERVER_2012_RTM_PORTUGESE_PORTUGAL_64BIT_BASE]: WindowsVersion.WINDOWS_SERVER_2012_RTM_PORTUGUESE_PORTUGAL_64BIT_BASE,
    [WindowsVersion.WINDOWS_SERVER_2019_PORTUGESE_BRAZIL_FULL_BASE]: WindowsVersion.WINDOWS_SERVER_2019_PORTUGUESE_BRAZIL_FULL_BASE,
    [WindowsVersion.WINDOWS_SERVER_2019_PORTUGESE_PORTUGAL_FULL_BASE]: WindowsVersion.WINDOWS_SERVER_2019_PORTUGUESE_PORTUGAL_FULL_BASE,
  }
  constructor(version: WindowsVersion, props: WindowsImageProps = {}) {
    const nonDeprecatedVersionName = WindowsImage.DEPRECATED_VERSION_NAME_MAP[version] ?? version;
    super('/aws/service/ami-windows-latest/' + nonDeprecatedVersionName, OperatingSystemType.WINDOWS, props.userData);
  }
}

/**
 * Amazon Linux image properties
 */
export interface AmazonLinuxImageProps {
  /**
   * What generation of Amazon Linux to use
   *
   * @default AmazonLinux
   */
  readonly generation?: AmazonLinuxGeneration;

  /**
   * What edition of Amazon Linux to use
   *
   * @default Standard
   */
  readonly edition?: AmazonLinuxEdition;

  /**
   * What kernel version of Amazon Linux to use
   *
   * @default -
   */
  readonly kernel?: AmazonLinuxKernel;

  /**
   * What storage backed image to use
   *
   * @default GeneralPurpose
   */
  readonly storage?: AmazonLinuxStorage;

  /**
   * Initial user data
   *
   * @default - Empty UserData for Linux machines
   */
  readonly userData?: UserData;

  /**
   * CPU Type
   *
   * @default X86_64
   */
  readonly cpuType?: AmazonLinuxCpuType;

  /**
   * Virtualization type
   *
   * @default HVM
   */
  readonly virtualization?: AmazonLinuxVirt;

  /**
   * Whether the AMI ID is cached to be stable between deployments
   *
   * By default, the newest image is used on each deployment. This will cause
   * instances to be replaced whenever a new version is released, and may cause
   * downtime if there aren't enough running instances in the AutoScalingGroup
   * to reschedule the tasks on.
   *
   * If set to true, the AMI ID will be cached in `cdk.context.json` and the
   * same value will be used on future runs. Your instances will not be replaced
   * but your AMI version will grow old over time. To refresh the AMI lookup,
   * you will have to evict the value from the cache using the `cdk context`
   * command. See https://docs.aws.amazon.com/cdk/latest/guide/context.html for
   * more information.
   *
   * Can not be set to `true` in environment-agnostic stacks.
   *
   * @default false
   */
  readonly cachedInContext?: boolean;
}

/**
 * Selects the latest version of Amazon Linux
 *
 * This Machine Image automatically updates to the latest version on every
 * deployment. Be aware this will cause your instances to be replaced when a
 * new version of the image becomes available. Do not store stateful information
 * on the instance if you are using this image.
 *
 * The AMI ID is selected using the values published to the SSM parameter store.
 */
export class AmazonLinuxImage extends GenericSSMParameterImage {
  /**
   * Return the SSM parameter name that will contain the Amazon Linux image with the given attributes
   */
  public static ssmParameterName(props: AmazonLinuxImageProps = {}) {
    const generation = (props && props.generation) || AmazonLinuxGeneration.AMAZON_LINUX;
    const edition = (props && props.edition) || AmazonLinuxEdition.STANDARD;
    const cpu = (props && props.cpuType) || AmazonLinuxCpuType.X86_64;
    let kernel = (props && props.kernel) || undefined;
    let virtualization: AmazonLinuxVirt | undefined;
    let storage: AmazonLinuxStorage | undefined;

    if (generation === AmazonLinuxGeneration.AMAZON_LINUX_2022) {
      kernel = AmazonLinuxKernel.KERNEL5_X;
      if (props && props.storage) {
        throw new Error('Storage parameter does not exist in smm parameter name for Amazon Linux 2022.');
      }
      if (props && props.virtualization) {
        throw new Error('Virtualization parameter does not exist in smm parameter name for Amazon Linux 2022.');
      }
    } else {
      virtualization = (props && props.virtualization) || AmazonLinuxVirt.HVM;
      storage = (props && props.storage) || AmazonLinuxStorage.GENERAL_PURPOSE;
    }

    const parts: Array<string|undefined> = [
      generation,
      'ami',
      edition !== AmazonLinuxEdition.STANDARD ? edition : undefined,
      kernel,
      virtualization,
      cpu,
      storage,
    ].filter(x => x !== undefined); // Get rid of undefineds

    return '/aws/service/ami-amazon-linux-latest/' + parts.join('-');
  }

  private readonly cachedInContext: boolean;

  constructor(private readonly props: AmazonLinuxImageProps = {}) {
    super(AmazonLinuxImage.ssmParameterName(props), OperatingSystemType.LINUX, props.userData);

    this.cachedInContext = props.cachedInContext ?? false;
  }

  /**
   * Return the image to use in the given context
   */
  public getImage(scope: Construct): MachineImageConfig {
    const imageId = lookupImage(scope, this.cachedInContext, this.parameterName);

    const osType = OperatingSystemType.LINUX;
    return {
      imageId,
      osType,
      userData: this.props.userData ?? UserData.forLinux(),
    };
  }
}

/**
 * Amazon Linux Kernel
 */
export enum AmazonLinuxKernel {
  /**
   * Standard edition
   */
  KERNEL5_X = 'kernel-5.10',
}

/**
 * Configuration options for GenericLinuxImage
 */
export interface GenericLinuxImageProps {
  /**
   * Initial user data
   *
   * @default - Empty UserData for Linux machines
   */
  readonly userData?: UserData;
}

/**
 * Configuration options for GenericWindowsImage
 */
export interface GenericWindowsImageProps {
  /**
   * Initial user data
   *
   * @default - Empty UserData for Windows machines
   */
  readonly userData?: UserData;
}

/**
 * Construct a Linux machine image from an AMI map
 *
 * Linux images IDs are not published to SSM parameter store yet, so you'll have to
 * manually specify an AMI map.
 */
export class GenericLinuxImage implements IMachineImage {
  constructor(private readonly amiMap: { [region: string]: string }, private readonly props: GenericLinuxImageProps = {}) {
  }

  public getImage(scope: Construct): MachineImageConfig {
    const userData = this.props.userData ?? UserData.forLinux();
    const osType = OperatingSystemType.LINUX;
    const region = Stack.of(scope).region;
    if (Token.isUnresolved(region)) {
      const mapping: { [k1: string]: { [k2: string]: any } } = {};
      for (const [rgn, ami] of Object.entries(this.amiMap)) {
        mapping[rgn] = { ami };
      }
      const amiMap = new CfnMapping(scope, 'AmiMap', { mapping });
      return {
        imageId: amiMap.findInMap(Aws.REGION, 'ami'),
        userData,
        osType,
      };
    }
    const imageId = region !== 'test-region' ? this.amiMap[region] : 'ami-12345';
    if (!imageId) {
      throw new Error(`Unable to find AMI in AMI map: no AMI specified for region '${region}'`);
    }
    return {
      imageId,
      userData,
      osType,
    };
  }
}

/**
 * Construct a Windows machine image from an AMI map
 *
 * Allows you to create a generic Windows EC2 , manually specify an AMI map.
 */
export class GenericWindowsImage implements IMachineImage {
  constructor(private readonly amiMap: {[region: string]: string}, private readonly props: GenericWindowsImageProps = {}) {
  }

  public getImage(scope: Construct): MachineImageConfig {
    const userData = this.props.userData ?? UserData.forWindows();
    const osType = OperatingSystemType.WINDOWS;
    const region = Stack.of(scope).region;
    if (Token.isUnresolved(region)) {
      const mapping: { [k1: string]: { [k2: string]: any } } = {};
      for (const [rgn, ami] of Object.entries(this.amiMap)) {
        mapping[rgn] = { ami };
      }
      const amiMap = new CfnMapping(scope, 'AmiMap', { mapping });
      return {
        imageId: amiMap.findInMap(Aws.REGION, 'ami'),
        userData,
        osType,
      };
    }
    const imageId = region !== 'test-region' ? this.amiMap[region] : 'ami-12345';
    if (!imageId) {
      throw new Error(`Unable to find AMI in AMI map: no AMI specified for region '${region}'`);
    }
    return {
      imageId,
      userData,
      osType,
    };
  }
}

/**
 * A machine image whose AMI ID will be searched using DescribeImages.
 *
 * The most recent, available, launchable image matching the given filter
 * criteria will be used. Looking up AMIs may take a long time; specify
 * as many filter criteria as possible to narrow down the search.
 *
 * The AMI selected will be cached in `cdk.context.json` and the same value
 * will be used on future runs. To refresh the AMI lookup, you will have to
 * evict the value from the cache using the `cdk context` command. See
 * https://docs.aws.amazon.com/cdk/latest/guide/context.html for more information.
 */
export class LookupMachineImage implements IMachineImage {
  constructor(private readonly props: LookupMachineImageProps) {
  }

  public getImage(scope: Construct): MachineImageConfig {
    // Need to know 'windows' or not before doing the query to return the right
    // osType for the dummy value, so might as well add it to the filter.
    const filters: Record<string, string[] | undefined> = {
      'name': [this.props.name],
      'state': ['available'],
      'image-type': ['machine'],
      'platform': this.props.windows ? ['windows'] : undefined,
    };
    Object.assign(filters, this.props.filters);

    const value = ContextProvider.getValue(scope, {
      provider: cxschema.ContextProvider.AMI_PROVIDER,
      props: {
        owners: this.props.owners,
        filters,
      } as cxschema.AmiContextQuery,
      dummyValue: 'ami-1234',
    }).value as cxapi.AmiContextResponse;

    if (typeof value !== 'string') {
      throw new Error(`Response to AMI lookup invalid, got: ${value}`);
    }

    const osType = this.props.windows ? OperatingSystemType.WINDOWS : OperatingSystemType.LINUX;

    return {
      imageId: value,
      osType,
      userData: this.props.userData ?? UserData.forOperatingSystem(osType),
    };
  }
}

/**
 * Properties for looking up an image
 */
export interface LookupMachineImageProps {
  /**
   * Name of the image (may contain wildcards)
   */
  readonly name: string;

  /**
   * Owner account IDs or aliases
   *
   * @default - All owners
   */
  readonly owners?: string[];

  /**
   * Additional filters on the AMI
   *
   * @see https://docs.aws.amazon.com/AWSEC2/latest/APIReference/API_DescribeImages.html
   * @default - No additional filters
   */
  readonly filters?: {[key: string]: string[]};

  /**
   * Look for Windows images
   *
   * @default false
   */
  readonly windows?: boolean;

  /**
   * Custom userdata for this image
   *
   * @default - Empty user data appropriate for the platform type
   */
  readonly userData?: UserData;
}