//===- DiagTool.cpp - Classes for defining diagtool tools -------------------===//

//

// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.

// See https://llvm.org/LICENSE.txt for license information.

// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

//

//===----------------------------------------------------------------------===//

//

// This file implements the boilerplate for defining diagtool tools.

//

//===----------------------------------------------------------------------===//

#include "DiagTool.h"

#include "llvm/ADT/StringMap.h"

#include "llvm/ADT/STLExtras.h"

#include <vector>

using namespace diagtool;

DiagTool::DiagTool(llvm::StringRef toolCmd, llvm::StringRef toolDesc)

    : cmd(std::string(toolCmd)), description(std::string(toolDesc)) {}

DiagTool::~DiagTool() {}

typedef llvm::StringMap<DiagTool *> ToolMap;

static inline ToolMap *getTools(void *v) { return static_cast<ToolMap*>(v); }

DiagTools::DiagTools() : tools(new ToolMap()) {}

DiagTools::~DiagTools() { delete getTools(tools); }

DiagTool *DiagTools::getTool(llvm::StringRef toolCmd) {

  ToolMap::iterator it = getTools(tools)->find(toolCmd);

  return (it == getTools(tools)->end()) ? 
nullptr0
 : it->getValue();

}

void DiagTools::registerTool(DiagTool *tool) {

  (*getTools(tools))[tool->getName()] = tool;

}

void DiagTools::printCommands(llvm::raw_ostream &out) {

  std::vector<llvm::StringRef> toolNames;

  unsigned maxName = 0;

  for (ToolMap::iterator it = getTools(tools)->begin(),

       ei = getTools(tools)->end(); it != ei; ++it) {

    toolNames.push_back(it->getKey());

    unsigned len = it->getKey().size();

    if (len > maxName)

      maxName = len;    

  }

  llvm::sort(toolNames);

  for (std::vector<llvm::StringRef>::iterator it = toolNames.begin(),

       ei = toolNames.end(); it != ei; ++it) {

    out << "  " << (*it);

    unsigned spaces = (maxName + 3) - (it->size());

    for (unsigned i = 0; i < spaces; ++i)

      out << ' ';

    out << getTool(*it)->getDescription() << '\n';    

  }

}

namespace diagtool {

  llvm::ManagedStatic<DiagTools> diagTools;

}