stitch-ui/src/models/PipelineSource.js

import { List, Record } from "immutable";

export default class PipelineSource extends Record({
  output: "",
  stages: List()
}) {
  toInputs() {
    return {
      pipeline: this.stages.map(s => s.toInput()),
      output: this.output
    };
  }
  toRawSource() {
    return {
      pipeline: this.stages.toArray().map(s => s.toRawStage()),
      output: this.output
    };
  }

  describeOutputType() {
    switch (this.output) {
      case "array":
        return "Array";
      case "singleDoc":
        return "Single Document";
      case "boolean":
        return "Boolean";
      default:
        return "(unknown return type)";
    }
  }
}