osquery-1/osquery/config/parsers/kafka_topics.cpp

/*
 *  Copyright (c) 2014-present, Facebook, Inc.
 *  All rights reserved.
 *
 *  This source code is licensed under the BSD-style license found in the
 *  LICENSE file in the root directory of this source tree. An additional grant
 *  of patent rights can be found in the PATENTS file in the same directory.
 *
 */

#include <iostream>

#include <osquery/config.h>

#include "osquery/config/parsers/kafka_topics.h"

namespace osquery {

/// Root key to retrieve Kafka topic configurations.
const std::string kKafkaTopicParserRootKey("kafka_topics");

std::vector<std::string> KafkaTopicsConfigParserPlugin::keys() const {
  return {kKafkaTopicParserRootKey};
}

Status KafkaTopicsConfigParserPlugin::setUp() {
  data_.put_child(kKafkaTopicParserRootKey, boost::property_tree::ptree());
  return Status(0, "OK");
}

Status KafkaTopicsConfigParserPlugin::update(const std::string& source,
                                             const ParserConfig& config) {
  if (config.count(kKafkaTopicParserRootKey) > 0) {
    data_ = boost::property_tree::ptree();
    data_.put_child(kKafkaTopicParserRootKey,
                    config.at(kKafkaTopicParserRootKey));
  }

  return Status(0, "OK");
}

REGISTER_INTERNAL(KafkaTopicsConfigParserPlugin,
                  "config_parser",
                  "kafka_topics");
} // namespace osquery