cli.cpp

#include "cli.hpp"
#include "crypto.hpp"
#include "fileops.hpp"
#include "logger.hpp"
#include <iostream>
#include <string>
#include <vector>
#include <filesystem>
#include <termios.h>
#include <unistd.h>

namespace sflk {
static std::string prompt_hidden(const std::string& label) {
    std::cerr << label;
    termios oldt; tcgetattr(STDIN_FILENO, &oldt); termios newt = oldt; newt.c_lflag &= ~ECHO; tcsetattr(STDIN_FILENO, TCSANOW, &newt);
    std::string s; std::getline(std::cin, s);
    tcsetattr(STDIN_FILENO, TCSANOW, &oldt); std::cerr << "\n";
    return s;
}

static void usage() {
    std::cout << "sflk - Secure File Locker\n\n"
              << "Usage:\n"
              << "  sflk encrypt <file> [--iter N] [--out file.sflk]\n"
              << "  sflk decrypt <file.sflk> [--out restored_name]\n"
              << "  sflk verify  <file.sflk>\n"
              << "  sflk list    <file.sflk>\n"
              << "  sflk delete-secure <file>\n"
              << "\n";
}

int run_cli(int argc, char** argv) {
    if (argc < 2) { usage(); return 1; }
    std::string cmd = argv[1];

    try {
        if (cmd == "encrypt" && argc >= 3) {
            std::string in = argv[2]; std::string out = in + ".sflk"; uint32_t iters=200000;
            for (int i=3;i<argc;i++) {
                std::string a=argv[i];
                if (a=="--iter" && i+1<argc) { iters=std::stoul(argv[++i]); }
                else if (a=="--out" && i+1<argc) { out=argv[++i]; }
            }
            std::string p1 = prompt_hidden("Password: ");
            std::string p2 = prompt_hidden("Confirm  : ");
            if (p1!=p2) { std::cerr << "Passwords do not match\n"; return 2; }
            encrypt_file(in, out, p1, iters);
            std::cout << "Encrypted -> " << out << "\n";
            return 0;
        } else if (cmd == "decrypt" && argc >= 3) {
            std::string in = argv[2]; std::string out;
            for (int i=3;i<argc;i++) if (std::string(argv[i])=="--out" && i+1<argc) out=argv[++i];
            std::string p = prompt_hidden("Password: ");
            std::string restored = decrypt_file(in, out, p);
            std::cout << "Restored -> " << restored << "\n";
            return 0;
        } else if (cmd == "verify" && argc >= 3) {
            std::string in = argv[2]; std::string p = prompt_hidden("Password: ");
            verify_file(in, p);
            auto m = read_header(in);
            std::cout << "OK: size=" << m.origSize << ", name=\"" << m.origName << "\"\n";
            return 0;
        } else if (cmd == "list" && argc >= 3) {
            auto m = read_header(argv[2]);
            std::cout << "Iterations: " << m.iterations << "\n"
                      << "SaltLen  : " << m.salt.size() << "\n"
                      << "IVLen    : " << m.iv.size() << "\n"
                      << "Name     : " << m.origName << "\n"
                      << "Size     : " << m.origSize << " bytes\n";
            return 0;
        } else if (cmd == "delete-secure" && argc >= 3) {
            secure_delete(argv[2]);
            std::cout << "Securely deleted: " << argv[2] << "\n";
            return 0;
        } else {
            usage();
            return 1;
        }
    } catch (const std::exception& e) {
        log_error(std::string("fatal: ")+ e.what());
        std::cerr << "Error: " << e.what() << "\n";
        return 3;
    }
}
}