Invoking MWAA DAG from JavaScript

invoke_mwaa_dag_from_js.js

const axios = require('axios');
const { MWAAClient, CreateCliTokenCommand } = require('@aws-sdk/client-mwaa');

const client = new MWAAClient({ region: 'AWS_REGION' });

const environmentName = 'MWAA_ENV_NAME'
const dagName = 'DAG_NAME';
const payload = {
  }

const getAirflowCliToken = async (environmentName) => {
  const command = new CreateCliTokenCommand({ Name: environmentName });
  const { CliToken, WebServerHostname } = await client.send(command);

  return { token: CliToken, host: WebServerHostname };
};


const triggerDag = async (environmentName, dagName, payload) => {
    const { token, host } = await getAirflowCliToken(environmentName);
    const url = `https://${host}/aws_mwaa/cli`;
    const formattedPayload = JSON.stringify(payload);
    const data = `dags trigger -c '${formattedPayload}' ${dagName}`;
    const options = {
      headers: {
        Authorization: `Bearer ${token}`,
        'Content-Type': 'text/plain'
      }
    };
  
    await axios.post(url, data, options);
  };

triggerDag(environmentName, dagName, payload);