import core = require('@aws-cdk/core');
import sfn  = require('@aws-cdk/aws-stepfunctions');

export interface ParameterizedPassProps extends sfn.PassProps {
  parameters: { [name: string]: any };
}

export class ParameterizedPass extends sfn.Pass {
  private parameterizedProps: ParameterizedPassProps;

  constructor(parentConstruct: core.Construct, id: string, parameterizedProps: ParameterizedPassProps) {
    super(parentConstruct, id, parameterizedProps);
    this.parameterizedProps = parameterizedProps;
  }

  public toStateJson(): object {
    const stateJson = {
      ...super.toStateJson(),
      Parameters: this.parameterizedProps.parameters
    };
    return stateJson;
  }
}

const passSfnTask = new ParameterizedPass(
  this,
  'pass',
  {
    parameters: {
      'x.$': '$.foo.a',
      'y.$': '$.foo.b'
    },
    resultPath: '$.bar'
  }
);