@infrascan/aws-kinesis-scanner/dist/index.js

Infrascan scanner definition for AWS Kinesis

"use strict";
var __create = Object.create;
var __defProp = Object.defineProperty;
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
var __getOwnPropNames = Object.getOwnPropertyNames;
var __getProtoOf = Object.getPrototypeOf;
var __hasOwnProp = Object.prototype.hasOwnProperty;
var __export = (target, all) => {
  for (var name in all)
    __defProp(target, name, { get: all[name], enumerable: true });
};
var __copyProps = (to, from, except, desc) => {
  if (from && typeof from === "object" || typeof from === "function") {
    for (let key of __getOwnPropNames(from))
      if (!__hasOwnProp.call(to, key) && key !== except)
        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
  }
  return to;
};
var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
  // If the importer is in node compatibility mode or this is not an ESM
  // file that has been converted to a CommonJS file using a Babel-
  // compatible transform (i.e. "__esModule" has not been set), then set
  // "default" to the CommonJS "module.exports" for node compatibility.
  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
  mod
));
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);

// src/index.ts
var index_exports = {};
__export(index_exports, {
  default: () => index_default
});
module.exports = __toCommonJS(index_exports);

// src/generated/client.ts
var import_client_kinesis = require("@aws-sdk/client-kinesis");
var import_debug = __toESM(require("debug"));
var clientDebug = (0, import_debug.default)("kinesis:client");
function getClient(credentials, context, retryStrategy) {
  clientDebug("Creating instance with context", context);
  return new import_client_kinesis.KinesisClient({
    credentials,
    region: context.region,
    retryStrategy
  });
}

// src/generated/getters.ts
var import_core = require("@infrascan/core");
var import_client_kinesis2 = require("@aws-sdk/client-kinesis");
var import_debug2 = __toESM(require("debug"));
async function ListStreams(client, stateConnector, context) {
  const getterDebug = (0, import_debug2.default)("kinesis:ListStreams");
  const state = [];
  getterDebug("Fetching state");
  let pagingToken;
  do {
    const preparedParams = {};
    preparedParams.NextToken = pagingToken;
    try {
      const cmd = new import_client_kinesis2.ListStreamsCommand(preparedParams);
      const result = await client.send(cmd);
      state.push({
        _metadata: {
          account: context.account,
          region: context.region,
          timestamp: Date.now()
        },
        _parameters: preparedParams,
        _result: result
      });
      pagingToken = result.NextToken;
      if (pagingToken != null) {
        getterDebug("Found pagination token in response");
      } else {
        getterDebug("No pagination token found in response");
      }
    } catch (err) {
      if (err instanceof import_client_kinesis2.KinesisServiceException) {
        if (err?.$retryable) {
          console.log("Encountered retryable error", err);
        } else {
          console.log("Encountered unretryable error", err);
        }
      } else {
        console.log("Encountered unexpected error", err);
      }
      pagingToken = void 0;
    }
  } while (pagingToken != null);
  getterDebug("Recording state");
  await stateConnector.onServiceScanCompleteCallback(
    context.account,
    context.region,
    "Kinesis",
    "ListStreams",
    state
  );
}
async function DescribeStreamSummary(client, stateConnector, context) {
  const getterDebug = (0, import_debug2.default)("kinesis:DescribeStreamSummary");
  const state = [];
  getterDebug("Fetching state");
  const resolvers = [
    {
      Key: "StreamARN",
      Selector: "Kinesis|ListStreams|[]._result.StreamSummaries[].StreamARN"
    }
  ];
  const parameterQueue = await (0, import_core.resolveFunctionCallParameters)(
    context.account,
    context.region,
    resolvers,
    stateConnector
  );
  for (const parameters of parameterQueue) {
    const preparedParams = parameters;
    try {
      const cmd = new import_client_kinesis2.DescribeStreamSummaryCommand(preparedParams);
      const result = await client.send(cmd);
      state.push({
        _metadata: {
          account: context.account,
          region: context.region,
          timestamp: Date.now()
        },
        _parameters: preparedParams,
        _result: result
      });
    } catch (err) {
      if (err instanceof import_client_kinesis2.KinesisServiceException) {
        if (err?.$retryable) {
          console.log("Encountered retryable error", err);
        } else {
          console.log("Encountered unretryable error", err);
        }
      } else {
        console.log("Encountered unexpected error", err);
      }
    }
  }
  getterDebug("Recording state");
  await stateConnector.onServiceScanCompleteCallback(
    context.account,
    context.region,
    "Kinesis",
    "DescribeStreamSummary",
    state
  );
}
async function ListStreamConsumers(client, stateConnector, context) {
  const getterDebug = (0, import_debug2.default)("kinesis:ListStreamConsumers");
  const state = [];
  getterDebug("Fetching state");
  const resolvers = [
    {
      Key: "StreamARN",
      Selector: "Kinesis|ListStreams|[]._result.StreamSummaries[].StreamARN"
    }
  ];
  const parameterQueue = await (0, import_core.resolveFunctionCallParameters)(
    context.account,
    context.region,
    resolvers,
    stateConnector
  );
  for (const parameters of parameterQueue) {
    let pagingToken;
    do {
      const preparedParams = parameters;
      preparedParams.NextToken = pagingToken;
      try {
        const cmd = new import_client_kinesis2.ListStreamConsumersCommand(preparedParams);
        const result = await client.send(cmd);
        state.push({
          _metadata: {
            account: context.account,
            region: context.region,
            timestamp: Date.now()
          },
          _parameters: preparedParams,
          _result: result
        });
        pagingToken = result.NextToken;
        if (pagingToken != null) {
          getterDebug("Found pagination token in response");
        } else {
          getterDebug("No pagination token found in response");
        }
      } catch (err) {
        if (err instanceof import_client_kinesis2.KinesisServiceException) {
          if (err?.$retryable) {
            console.log("Encountered retryable error", err);
          } else {
            console.log("Encountered unretryable error", err);
          }
        } else {
          console.log("Encountered unexpected error", err);
        }
        pagingToken = void 0;
      }
    } while (pagingToken != null);
  }
  getterDebug("Recording state");
  await stateConnector.onServiceScanCompleteCallback(
    context.account,
    context.region,
    "Kinesis",
    "ListStreamConsumers",
    state
  );
}

// src/generated/graph.ts
var import_core2 = require("@infrascan/core");
var import_debug3 = __toESM(require("debug"));
var edgesDebug = (0, import_debug3.default)("kinesis:edges");
async function getEdges(stateConnector) {
  edgesDebug("Fetching edges");
  const edges = [];
  edgesDebug("Evaluating Kinesis|ListStreamConsumers|[]");
  const ListStreamConsumersState1 = await (0, import_core2.evaluateSelectorGlobally)(
    "Kinesis|ListStreamConsumers|[]",
    stateConnector
  );
  const ListStreamConsumersEdges1 = ListStreamConsumersState1.flatMap(
    (state) => {
      const source = (0, import_core2.filterState)(state, "_parameters.StreamARN");
      const target = (0, import_core2.filterState)(
        state,
        "_result.Consumers[].{target:ConsumerARN,name:ConsumerName}"
      );
      if (!target || !source) {
        return [];
      }
      if (Array.isArray(target)) {
        return target.map((edgeTarget) => (0, import_core2.formatEdge)(source, edgeTarget));
      }
      return (0, import_core2.formatEdge)(source, target);
    }
  );
  edgesDebug(
    `Evaluated Kinesis|ListStreamConsumers|[]: ${ListStreamConsumersEdges1.length} Edges found`
  );
  edges.push(...ListStreamConsumersEdges1);
  return edges;
}

// src/graph.ts
var import_core3 = require("@infrascan/core");
var KinesisStreamEntity = {
  version: "0.1.1",
  debugLabel: "kinesis",
  provider: "aws",
  command: "DescribeStreamSummary",
  category: "kinesis",
  subcategory: "stream",
  nodeType: "kinesis-stream",
  selector: "Kinesis|DescribeStreamSummary|[]",
  getState(state, context) {
    return (0, import_core3.evaluateSelector)(
      context.account,
      context.region,
      KinesisStreamEntity.selector,
      state
    );
  },
  translate(val) {
    if (val._result.StreamDescriptionSummary == null) {
      return [];
    }
    const enrichedDescription = Object.assign(
      val._result.StreamDescriptionSummary,
      {
        $metadata: val._metadata,
        $parameters: val._parameters
      }
    );
    return [enrichedDescription];
  },
  components: {
    $metadata(val) {
      return {
        version: KinesisStreamEntity.version,
        timestamp: val.$metadata.timestamp
      };
    },
    $graph(val) {
      return {
        id: val.StreamARN,
        label: val.StreamName,
        nodeClass: "visual",
        nodeType: KinesisStreamEntity.nodeType,
        parent: `${val.$metadata.account}-${val.$metadata.region}`
      };
    },
    $source(val) {
      return {
        command: KinesisStreamEntity.command,
        parameters: val.$parameters
      };
    },
    tenant(val) {
      return {
        tenantId: val.$metadata.account,
        provider: KinesisStreamEntity.provider,
        partition: val.$metadata.partition
      };
    },
    location(val) {
      return {
        code: val.$metadata.region
      };
    },
    resource(val) {
      return {
        id: val.StreamARN,
        name: val.StreamName,
        category: KinesisStreamEntity.category,
        subcategory: KinesisStreamEntity.subcategory
      };
    },
    audit(val) {
      return {
        createdAt: val.StreamCreationTimestamp
      };
    },
    kinesis(val) {
      return {
        stream: {
          mode: val.StreamModeDetails?.StreamMode != null ? (0, import_core3.toLowerCase)(val.StreamModeDetails?.StreamMode) : void 0,
          status: val.StreamStatus != null ? (0, import_core3.toLowerCase)(val.StreamStatus) : void 0
        }
      };
    },
    encryption(val) {
      return {
        keyId: val.KeyId
      };
    }
  }
};
var KinesisConsumerEntity = {
  version: "0.1.1",
  debugLabel: "kinesis",
  provider: "aws",
  command: "ListStreamConsumers",
  category: "kinesis",
  subcategory: "consumer",
  nodeType: "kinesis-consumer",
  selector: "Kinesis|ListStreams|[]",
  getState(state, context) {
    return (0, import_core3.evaluateSelector)(
      context.account,
      context.region,
      KinesisConsumerEntity.selector,
      state
    );
  },
  translate(val) {
    return (val._result.Consumers ?? []).map(
      (stream) => Object.assign(stream, {
        $metadata: val._metadata,
        $parameters: val._parameters
      })
    );
  },
  components: {
    $metadata(val) {
      return {
        version: KinesisConsumerEntity.version,
        timestamp: val.$metadata.timestamp
      };
    },
    $graph(val) {
      return {
        id: val.ConsumerARN,
        label: val.ConsumerName,
        nodeClass: "informational",
        nodeType: KinesisConsumerEntity.nodeType,
        parent: `${val.$metadata.account}-${val.$metadata.region}`
      };
    },
    $source(val) {
      return {
        command: KinesisConsumerEntity.command,
        parameters: val.$parameters
      };
    },
    tenant(val) {
      return {
        tenantId: val.$metadata.account,
        provider: KinesisConsumerEntity.provider,
        partition: val.$metadata.partition
      };
    },
    location(val) {
      return {
        code: val.$metadata.region
      };
    },
    resource(val) {
      return {
        id: val.ConsumerARN,
        name: val.ConsumerName,
        category: KinesisConsumerEntity.category,
        subcategory: KinesisConsumerEntity.subcategory
      };
    },
    audit(val) {
      return {
        createdAt: val.ConsumerCreationTimestamp
      };
    },
    kinesis(val) {
      return {
        consumer: {
          status: val.ConsumerStatus != null ? (0, import_core3.toLowerCase)(val.ConsumerStatus) : void 0
        }
      };
    }
  }
};

// src/index.ts
var KinesisScanner = {
  provider: "aws",
  service: "kinesis",
  key: "Kinesis",
  getClient,
  callPerRegion: true,
  getters: [ListStreams, DescribeStreamSummary, ListStreamConsumers],
  getEdges,
  entities: [KinesisConsumerEntity, KinesisStreamEntity]
};
var index_default = KinesisScanner;