packages/core/src/shared/sam/cli/samCliRemoteTestEvent.ts

/*!
 * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 * SPDX-License-Identifier: Apache-2.0
 */

import * as vscode from 'vscode'
import { logAndThrowIfUnexpectedExitCode, SamCliProcessInvoker } from './samCliInvokerUtils'
import path from 'path'
import * as os from 'os'
import { getSpawnEnv } from '../../env/resolveEnv'
import { getLogger } from '../../../shared/logger/logger'

export const TestEventsOperation = {
    List: 'list',
    Get: 'get',
    Put: 'put',
}
export interface SamCliRemoteTestEventsParameters {
    operation: string
    functionArn?: string
    name?: string
    region?: string
    eventSample?: string
    projectRoot?: vscode.Uri
    stackName?: string
    logicalId?: string
    force?: boolean
}

export async function runSamCliRemoteTestEvents(
    remoteTestEventsParameters: SamCliRemoteTestEventsParameters,
    invoker: SamCliProcessInvoker
): Promise<string> {
    const args = ['remote', 'test-event', remoteTestEventsParameters.operation]
    if (remoteTestEventsParameters.stackName === '' && remoteTestEventsParameters.functionArn === '') {
        getLogger().info('No remote test events found. This stack is not deployed.')
        return 'No remote test events found. This stack is not deployed.'
    }
    if (remoteTestEventsParameters.functionArn) {
        args.push(remoteTestEventsParameters.functionArn)
    } else if (remoteTestEventsParameters.stackName && remoteTestEventsParameters.logicalId) {
        args.push('--stack-name', remoteTestEventsParameters.stackName, remoteTestEventsParameters.logicalId)
    }

    if (remoteTestEventsParameters.region) {
        args.push('--region', remoteTestEventsParameters.region)
    }

    if (remoteTestEventsParameters.name) {
        args.push('--name', remoteTestEventsParameters.name)
    }

    if (remoteTestEventsParameters.operation === TestEventsOperation.Put && remoteTestEventsParameters.eventSample) {
        const tempFileUri = vscode.Uri.file(path.join(os.tmpdir(), 'event-sample.json'))
        const encoder = new TextEncoder()
        await vscode.workspace.fs.writeFile(tempFileUri, encoder.encode(remoteTestEventsParameters.eventSample))
        args.push('--file', tempFileUri.fsPath)

        // Add --force flag when updating existing events
        if (remoteTestEventsParameters.force) {
            args.push('--force')
        }
    }

    const childProcessResult = await invoker.invoke({
        arguments: args,
        spawnOptions: {
            env: await getSpawnEnv(process.env),
            cwd: remoteTestEventsParameters.projectRoot?.fsPath,
        },
    })

    if (childProcessResult.stderr && childProcessResult.stderr.startsWith('Error: No events found for function')) {
        return ''
    }
    logAndThrowIfUnexpectedExitCode(childProcessResult, 0)

    return childProcessResult.stdout
}