Add interactive black-belt prompt to C++ CLI

Author: mearvk

bin/blackbelt.cpp

@@ -20,16 +20,59 @@ static std::string shellQuote(const std::string& s) {
     return r + "'";
 }
 
+static int runEngine(const std::string& input) {
+    char tmp[] = "/tmp/blackbelt-cpp-XXXXXX";
+    int fd = mkstemp(tmp);
+    if (fd < 0) { perror("mkstemp"); return 2; }
+    FILE* fp = fdopen(fd, "wb");
+    if (!fp) { perror("fdopen"); close(fd); unlink(tmp); return 2; }
+    if (fwrite(input.data(), 1, input.size(), fp) != input.size() || fputc('\n', fp) == EOF) {
+        perror("write"); fclose(fp); unlink(tmp); return 2;
+    }
+    fclose(fp);
+
+    const char* env = std::getenv("BBEA_ENGINE_CMD");
+    std::string engine = (env && *env) ? env : "bin/blackbelt-engine.sh";
+    std::string command = "cat " + shellQuote(tmp) + " | " + shellQuote(engine);
+    int status = std::system(command.c_str());
+    unlink(tmp);
+    if (status == -1) return 3;
+    return WIFEXITED(status) ? WEXITSTATUS(status) : 3;
+}
+
+static int interactive() {
+    std::cout << "Black Belt Ethical Auditor\n"
+              << "Enter a JSON audit object, or type 'help' or 'quit'.\n";
+    std::string line;
+    for (;;) {
+        std::cout << "black-belt> " << std::flush;
+        if (!std::getline(std::cin, line)) break;
+        if (line.empty()) continue;
+        if (line == "quit" || line == "exit") break;
+        if (line == "help") {
+            std::cout << "Paste a complete BBEA JSON audit object at the prompt.\n"
+                      << "Commands: help, quit, exit\n";
+            continue;
+        }
+        int rc = runEngine(line);
+        if (rc != 0) std::cerr << "Audit failed (exit " << rc << ").\n";
+    }
+    std::cout << '\n';
+    return 0;
+}
+
 int main(int argc, char** argv) {
     std::string file;
     for (int i = 1; i < argc; ++i) {
         std::string a = argv[i];
         if (a == "--help" || a == "-h") { usage(argv[0]); return 0; }
-        if (a == "--version") { std::cout << "blackbelt-cpp 1.0 (BBEA CLI v2 transport)\n"; return 0; }
+        if (a == "--version") { std::cout << "blackbelt-cpp 1.1 (BBEA CLI v2 transport)\n"; return 0; }
         if (a == "--file" && i + 1 < argc) { file = argv[++i]; continue; }
         std::cerr << "Unknown argument: " << a << '\n'; usage(argv[0]); return 2;
     }
 
+    if (file.empty() && isatty(STDIN_FILENO)) return interactive();
+
     std::string input;
     if (file.empty()) {
         input.assign(std::istreambuf_iterator<char>(std::cin), std::istreambuf_iterator<char>());
@@ -41,20 +84,5 @@ int main(int argc, char** argv) {
     if (input.find_first_not_of(" \t\r\n") == std::string::npos) {
         std::cerr << "No JSON audit input received.\n"; return 2;
     }
-
-    char tmp[] = "/tmp/blackbelt-cpp-XXXXXX";
-    int fd = mkstemp(tmp);
-    if (fd < 0) { perror("mkstemp"); return 2; }
-    FILE* fp = fdopen(fd, "wb");
-    if (!fp) { perror("fdopen"); close(fd); unlink(tmp); return 2; }
-    fwrite(input.data(), 1, input.size(), fp);
-    fclose(fp);
-
-    const char* env = std::getenv("BBEA_ENGINE_CMD");
-    std::string engine = (env && *env) ? env : "bin/blackbelt-engine.sh";
-    std::string command = "cat " + shellQuote(tmp) + " | " + shellQuote(engine);
-    int status = std::system(command.c_str());
-    unlink(tmp);
-    if (status == -1) return 3;
-    return WIFEXITED(status) ? WEXITSTATUS(status) : 3;
+    return runEngine(input);
 }