[SYCLomatic] DPCT/C2S bash auto-complete

clang/lib/DPCT/AutoComplete.cpp

@@ -0,0 +1,166 @@
+//===--------------- AutoComplete.cpp -------------------------------------===//
+//
+// 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
+//
+//===----------------------------------------------------------------------===//
+
+#include "AutoComplete.h"
+#include "Error.h"
+#include "Utility.h"
+#include "llvm/Support/raw_ostream.h"
+#include <algorithm>
+
+namespace clang {
+namespace dpct {
+
+static std::map<std::string, std::set<std::string>> DPCTOptionInfoMap = {
+  // To avoid make llvm library depends on this file, hard code 2 options here.
+  {"help", {}},
+  {"version", {}},
+#define DPCT_OPTIONS_IN_LLVM_SUPPORT
+#define DPCT_OPTIONS_IN_CLANG_TOOLING
+#define DPCT_OPTIONS_IN_CLANG_DPCT
+
+#define DPCT_OPT_TYPE(...) 0
+#define DPCT_OPT_ENUM(NAME, ...) NAME
+#define DPCT_OPTION_VALUES(...)                                                \
+  { __VA_ARGS__ }
+#define DPCT_NON_ENUM_OPTION(OPT_TYPE, OPT_VAR, OPTION_NAME, ...)              \
+  {OPTION_NAME, {}},
+#define DPCT_ENUM_OPTION(OPT_TYPE, OPT_VAR, OPTION_NAME, OPTION_VALUES, ...)   \
+  {OPTION_NAME, OPTION_VALUES},
+
+#include "clang/DPCT/DPCTOptions.inc"
+
+#undef DPCT_ENUM_OPTION
+#undef DPCT_NON_ENUM_OPTION
+#undef DPCT_OPT_ENUM
+#undef DPCT_OPT_TYPE
+
+#undef DPCT_OPTIONS_IN_CLANG_DPCT
+#undef DPCT_OPTIONS_IN_CLANG_TOOLING
+#undef DPCT_OPTIONS_IN_LLVM_SUPPORT
+};
+
+void AutoCompletePrinter::process() {
+  llvm::StringRef InputRef(Input);
+  LastSharpPos = InputRef.find_last_of('#');
+  if (LastSharpPos == (InputRef.size() - 1)) {
+    // No char after last '#', it means user press tab after a space,
+    // return empty line to show files/folders
+    printAndExit();
+  }
+
+  bool SuggestEnumValue =
+      (InputRef.find("=", LastSharpPos) != llvm::StringRef::npos) ||
+      (LastSharpPos >= 1 && InputRef[LastSharpPos - 1] == '=');
+  if (!SuggestEnumValue) {
+    // suggest option names
+    llvm::StringRef OptionPrefixRef = InputRef.substr(LastSharpPos + 1);
+
+    // Case0: "" : return empty.
+    // Case1: "-" : suggest all options. All candidates have "--" prefix except "-p".
+    // Case2: "--" : suggest all options except -p. All candidates have "--" prefix.
+    // Case3: "-abc" : suggest options have abc prefix. All candidates have "-" prefix.
+    // Case4: "--abc" : suggest options have abc prefix. All candidates have "--" prefix.
+
+    if (OptionPrefixRef.size() == 0) {
+      // Case0
+      printAndExit();
+    } else if (OptionPrefixRef.size() == 1 && OptionPrefixRef[0] == '-') {
+      // Case1
+      addSuggestions(OptionSet, false);
+      printAndExit();
+    } else if (OptionPrefixRef.size() == 2 && OptionPrefixRef[0] == '-' &&
+        OptionPrefixRef[1] == '-') {
+      // Case2
+      addSuggestions(OptionSet, true);
+      printAndExit();
+    } else if (OptionPrefixRef.size() >= 2 && OptionPrefixRef[0] == '-' &&
+        OptionPrefixRef[1] != '-') {
+      // Case3
+      addSuggestions(OptionSet, OptionPrefixRef.substr(1), "-");
+      printAndExit();
+    } else if (OptionPrefixRef.size() >= 3 && OptionPrefixRef[0] == '-' &&
+        OptionPrefixRef[1] == '-') {
+      // Case4
+      addSuggestions(OptionSet, OptionPrefixRef.substr(2), "--");
+      printAndExit();
+    }
+    printAndExit();
+  }
+
+  // suggest enum values
+  // E.g., "--foobar=abc,def,g"
+  // CurOptionNameRef is "foobar="
+  // EnumPrefixRef is "g"
+  // ResultPrefixRef is "abc,def,"
+  llvm::StringRef CurOptionNameRef;
+  llvm::StringRef EnumPrefixRef;
+  llvm::StringRef ResultPrefixRef;
+
+  auto EqualPos = InputRef.find("=", LastSharpPos);
+  if (EqualPos != llvm::StringRef::npos) {
+    // InputRef: #--foo=
+    auto OptionNameStartPos = InputRef.find_first_not_of('-', LastSharpPos + 1);
+    CurOptionNameRef = InputRef.substr(OptionNameStartPos,
+                                       InputRef.size() - OptionNameStartPos);
+    addSuggestions(OptionEnumsMap[CurOptionNameRef.str()], "", "");
+    printAndExit();
+  }
+  // InputRef: #--foo=#...
+  auto SharpPosBeforeLastSharp =
+      InputRef.substr(0, LastSharpPos).find_last_of('#');
+  auto OptionNameStartPos =
+      InputRef.find_first_not_of('-', SharpPosBeforeLastSharp + 1);
+  CurOptionNameRef =
+      InputRef.substr(OptionNameStartPos, LastSharpPos - OptionNameStartPos);
+  auto StrAfterLastSharp = InputRef.substr(LastSharpPos + 1);
+  auto LastCommaPosAfterLastSharp = StrAfterLastSharp.find_last_of(',');
+  if (LastCommaPosAfterLastSharp != llvm::StringRef::npos) {
+    // InputRef: #--foo=#abc,def,g
+    EnumPrefixRef = StrAfterLastSharp.substr(LastCommaPosAfterLastSharp + 1);
+    ResultPrefixRef =
+        StrAfterLastSharp.substr(0, LastCommaPosAfterLastSharp + 1);
+    addSuggestions(OptionEnumsMap[CurOptionNameRef.str()], EnumPrefixRef, ResultPrefixRef);
+    printAndExit();
+  } else {
+    // InputRef: #--foo=#ab
+    EnumPrefixRef = StrAfterLastSharp;
+    addSuggestions(OptionEnumsMap[CurOptionNameRef.str()], EnumPrefixRef, "");
+    printAndExit();
+  }
+}
+
+void AutoCompletePrinter::operator=(std::string RawInput) {
+  for (const auto &I : DPCTOptionInfoMap) {
+    if (I.second.empty()) {
+      OptionSet.insert(I.first);
+    } else {
+      const std::string Key = I.first + "=";
+      OptionSet.insert(Key);
+      for (const auto &Enum : I.second) {
+        OptionEnumsMap[Key].insert(Enum);
+      }
+    }
+  }
+  Input = "#" + RawInput;
+  process();
+}
+
+void AutoCompletePrinter::printAndExit() {
+  if (Suggestions.empty()) {
+    llvm::outs() << "\n";
+    dpctExit(MigrationSucceeded);
+  }
+
+  for (const auto &Opt : Suggestions) {
+    llvm::outs() << Opt << "\n";
+  }
+  dpctExit(MigrationSucceeded);
+}
+
+} // namespace dpct
+}

clang/lib/DPCT/AutoComplete.h

@@ -0,0 +1,56 @@
+//===--------------- AutoComplete.h ---------------------------------------===//
+//
+// 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
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef DPCT_AUTOCOMPLETE_H
+#define DPCT_AUTOCOMPLETE_H
+
+#include "llvm/ADT/StringRef.h"
+#include <string>
+#include <vector>
+#include <tuple>
+#include <map>
+#include <set>
+
+namespace clang {
+namespace dpct {
+class AutoCompletePrinter {
+  std::set<std::string> Suggestions;
+  std::set<std::string> OptionSet;
+  std::map<std::string, std::set<std::string>> OptionEnumsMap;
+  std::string Input;
+  void printAndExit();
+  void process();
+  std::string::size_type LastSharpPos;
+  void addSuggestions(const std::set<std::string>& Set,
+                      const llvm::StringRef Prefix,
+                      const llvm::StringRef ResultPrefix) {
+    for (const auto &Item : Set) {
+      llvm::StringRef ItemRef(Item);
+      if (ItemRef.startswith(Prefix)) {
+        Suggestions.insert(ResultPrefix.str() + Item);
+      }
+    }
+  }
+  void addSuggestions(const std::set<std::string>& Set,
+                      bool MustBeginWithDashDash) {
+    for (const auto &Opt : OptionSet) {
+      if (Opt.size() == 1 && !MustBeginWithDashDash) {
+        Suggestions.insert("-" + Opt);
+      } else {
+        Suggestions.insert("--" + Opt);
+      }
+    }
+  }
+public:
+  AutoCompletePrinter() {}
+  void operator=(std::string RawInput);
+};
+}
+}
+
+#endif