AFDU.cpp

#include <iostream>
#include <cstdlib>
#include <string>
#include <unistd.h> 

void format_disk(const std::string& device, const std::string& fs_type) {
    if (getuid() != 0) {
        std::cerr << "❌ Root required, restart with sudo\n";
        exit(1);
    }
    
    std::cout << "⚠️  WARNING all data on " << device 
              << " will be deleted!\n";
    std::cout << "Continue? (yes/no): ";
    
    std::string answer;
    std::cin >> answer;
    
    if (answer != "yes") {
        std::cout << "Canceled.\n";
        return;
    }
    
    std::string command = "mkfs." + fs_type + " " + device;
    
    std::cout << "Formatting...\n";
    
    int result = system(command.c_str());
    
    if (result == 0) {
        std::cout << "✅ Done!\n";
    } else {
        std::cerr << "❌ Format error\n";
    }
}

int main(int argc, char* argv[]) {
    if (argc != 3) {
        std::cout << "Using: " << argv[0] 
                  << " <disk> <fs_type>\n";
        std::cout << "Example: " << argv[0] 
                  << " /dev/sdb1 ext4\n";
        return 1;
    }
    
    std::string device = argv[1];
    std::string fs_type = argv[2];
    int unmount;
    std::cout << "do you really want to unmout disk?\n 1.Yes 2.No\n";
    std::cin >> unmount;
    if((unmount = 1)){
        int unmount = std::system("umount "), device;
        }else{
            std::cerr << "unmount error, you should unmount disk before formatting\n";
        }
    format_disk(device, fs_type);

    
    
    
    return 0;
}