Менеджер паролей

#include <iostream>
#include <string>
#include <random>
#include "cxxopts.hpp"
#include "sqlite3.h"
#include <conio.h>
#include <locale>
#include "sqlite3.h"
 
const char set_lowercase[] = "abcdefghijklmnopqrstuvwxyz";
const char set_uppercase[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZ";
const char set_digits[] = "0123456789";
const char set_special[] = "!@#$%^&*_";
const char *sqlite3_errmsg (sqlite3*);
 
class Database {
public:
    sqlite3 *db;
    int openStatus, execStatus;
 
    Database() {
        openStatus = sqlite3_open("passwords.db", &db);
        if (openStatus != SQLITE_OK)
            std::cout << "Database Error: " << sqlite3_errmsg(db) << std::endl;
        else {
            std::string sql = "CREATE TABLE IF NOT EXISTS passwords(ID INTEGER PRIMARY KEY AUTOINCREMENT, PASSWORD TEXT NOT NULL);";
            execStatus = sqlite3_exec(db, sql.c_str(), 0, 0, 0);
            if (execStatus != SQLITE_OK)
                std::cout << "Create Table Error!" << std::endl;
        }
    }
 
    int savePassword(std::string password) {
        std::string sql = "INSERT INTO passwords (PASSWORD) VALUES ('" + password + "')";
        execStatus = sqlite3_exec(db, sql.c_str(), 0, 0, 0);
        if (execStatus != SQLITE_OK)
            return -1;
        return 0;
    }
 
    ~Database() {
        sqlite3_close(db);
    }
};
 
class PasswordGenerator {
    std::random_device rd{};
    std::mt19937 rng{ rd() };
public:
    std::string generate(size_t length, bool flag_lowercase, bool flag_uppercase, bool flag_digits, bool flag_special) {
        std::string set_result, password;
        if (!flag_lowercase && !flag_uppercase && !flag_digits && !flag_special)
            return "";
        if (flag_lowercase)
            set_result += set_lowercase;
        if (flag_uppercase)
            set_result += set_uppercase;
        if (flag_digits)
            set_result += set_digits;
        if (flag_special)
            set_result += set_special;
        std::uniform_int_distribution<int> dist(0, set_result.length() - 1);
        for (size_t i = 0; i < length; i++) {
            password += set_result[dist(rng)];
        }
        return password;
    }
};
 
int main(int argc, char** argv) {
    setlocale(LC_ALL, "rus");
    cxxopts::Options options("passwordgen", "Password Generator v0.0.0.1");
    options.add_options()
        ("l,lowercase", "Включить строчные буквы")
        ("u,uppercase", "Включить заглавные буквы")
        ("d,digits", "Включить цифры")
        ("s,special", "Включить специальные символы !@#$%^&*_")
        ("n,length", "Длина пароля", cxxopts::value<size_t>()->default_value("8"))
        ("c,count", "Сколько паролей сгенерировать", cxxopts::value<size_t>()->default_value("1"))
        ("h,help", "Использование печати")
        ;
 
    auto args = options.parse(argc, argv);
 
    if (args.count("h")) {
        std::cout << options.help() << std::endl;
        exit(0);
    }
 
    bool f1, f2, f3, f4;
    if (args.count("l") == 0 && args.count("u") == 0
        && args.count("d") == 0 && args.count("s") == 0) {
        f1 = true; f2 = true; f3 = true; f4 = true;
    }
    else {
        f1 = args.count("l");
        f2 = args.count("u");
        f3 = args.count("d");
        f4 = args.count("s");
    }
 
    Database db;
    PasswordGenerator pg;
    for (size_t i = 0; i < args["c"].as<size_t>(); i++) {
        std::string password = pg.generate(args["n"].as<size_t>(), f1, f2, f3, f4);
        std::cout << password << std::endl;
        db.savePassword(password);
    }
 
    char action;
    action = _getch();
    return 0;
 
}