Course-Design-C--/userAuth.cpp

#include "userAuth.h"
#include "tools.h"
#include <cstdlib>

const char *CREATE_USER_TABLE = "CREATE TABLE IF NOT EXISTS user_data ("
                                "id INTEGER PRIMARY KEY AUTOINCREMENT,"
                                "username TEXT NOT NULL,"
                                "userpassword TEXT NOT NULL,"
                                "created_time DATETIME)";

const char *REGISTER_USER = "INSERT INTO user_data (username, userpassword, "
                            "created_time) VALUES (?1, ?2, ?3);";

const char *LOGIN_USER =
    "SELECT 1 FROM user_data WHERE username = ? AND userpassword = ?;";

const char *SELECT_USER =
    "SELECT id, username, userpassword, created_time FROM user_data "
    "ORDER BY id DESC;";

const char *UPDATE_USER =
    "UPDATE user_data SET username = ?1, userpassword = ?3 WHERE id = ?2";

const char *DELETE_USER = "DELETE FROM user_data WHERE id = ?1;";

void init_user_data(sqlite3 *db) {
  char *err_msg = nullptr;

  int rc = sqlite3_exec(db, CREATE_USER_TABLE, nullptr, nullptr, &err_msg);
  if (rc != SQLITE_OK) {
    std::cerr << "创建表失败: " << err_msg << std::endl;
    sqlite3_free(err_msg);
    exit(1);
  };
};

bool is_user_table_empty(sqlite3 *db) {
  const char *query = "SELECT COUNT(*) FROM user_data;";
  sqlite3_stmt *stmt;
  bool isEmpty = true;

  // 准备SQL语句
  if (sqlite3_prepare_v2(db, query, -1, &stmt, nullptr) != SQLITE_OK) {
    std::cerr << "Failed to prepare statement: " << sqlite3_errmsg(db)
              << std::endl;
    return true; // 假设出错时表为空（根据你的需求可能需要调整）
  }

  // 执行查询
  if (sqlite3_step(stmt) == SQLITE_ROW) {
    int count = sqlite3_column_int(stmt, 0);
    isEmpty = (count == 0);
  } else {
    std::cerr << "Failed to execute query: " << sqlite3_errmsg(db) << std::endl;
  }

  // 释放语句
  sqlite3_finalize(stmt);

  return isEmpty;
}

void register_user(sqlite3 *db, const string &username,
                   const string &userpassword) {
  sqlite3_stmt *stmt;

  int rc = sqlite3_prepare_v2(db, REGISTER_USER, -1, &stmt, nullptr);

  if (rc != SQLITE_OK) {
    std::cerr << "准备语句失败: " << sqlite3_errmsg(db) << std::endl;
    exit(1);
  }

  sqlite3_bind_text(stmt, 1, username.c_str(), -1, SQLITE_STATIC);
  sqlite3_bind_text(stmt, 2, userpassword.c_str(), -1, SQLITE_STATIC);

  string timestamp = get_utc8_time();
  sqlite3_bind_text(stmt, 3, timestamp.c_str(), -1, SQLITE_STATIC);

  rc = sqlite3_step(stmt);
  if (rc != SQLITE_DONE) {
    std::cerr << "执行失败" << sqlite3_errmsg(db) << std::endl;
    exit(1);
  }

  sqlite3_finalize(stmt);
};

bool login_user(sqlite3 *db, const std::string &username,
                const std::string &password) {
  sqlite3_stmt *stmt;
  bool authResult = false;

  if (sqlite3_prepare_v2(db, LOGIN_USER, -1, &stmt, nullptr) != SQLITE_OK) {
    std::cerr << "准备语句失败: " << sqlite3_errmsg(db) << std::endl;
    sqlite3_close(db);
    return false;
  }

  sqlite3_bind_text(stmt, 1, username.c_str(), -1, SQLITE_STATIC);
  sqlite3_bind_text(stmt, 2, password.c_str(), -1, SQLITE_STATIC);

  // 执行查询
  if (sqlite3_step(stmt) == SQLITE_ROW) {
    authResult = true; // 找到匹配记录
  }

  sqlite3_finalize(stmt);
  return authResult;
}

vector<UserStruct> get_user_inputs(sqlite3 *db) {
  vector<UserStruct> results;
  sqlite3_stmt *stmt;

  int rc = sqlite3_prepare_v2(db, SELECT_USER, -1, &stmt, nullptr);
  if (rc != SQLITE_OK) {
    std::cerr << "准备查询失败: " << sqlite3_errmsg(db) << std::endl;
    exit(1);
  }

  while ((rc = sqlite3_step(stmt)) == SQLITE_ROW) {
    UserStruct record;
    record.id = sqlite3_column_int(stmt, 0);

    const unsigned char *text1 = sqlite3_column_text(stmt, 1);
    record.username = std::string(reinterpret_cast<const char *>(text1));

    const unsigned char *text2 = sqlite3_column_text(stmt, 2);
    record.userpassword = std::string(reinterpret_cast<const char *>(text2));

    const unsigned char *time = sqlite3_column_text(stmt, 3);
    record.created_time = time ? reinterpret_cast<const char *>(time) : "N/A";

    results.push_back(record);
  }

  if (rc != SQLITE_DONE) {
    std::cerr << "查询错误失败: " << sqlite3_errmsg(db) << std::endl;
  }

  sqlite3_finalize(stmt);
  return results;
}

bool update_user(sqlite3 *db, int id, const string &new_username,
                 const string &new_password) {
  sqlite3_stmt *stmt;

  int rc = sqlite3_prepare_v2(db, UPDATE_USER, -1, &stmt, nullptr);
  if (rc != SQLITE_OK) {
    std::cerr << "准备更新语句失败: " << sqlite3_errmsg(db) << std::endl;
    return false;
  }

  sqlite3_bind_text(stmt, 1, new_username.c_str(), -1, SQLITE_STATIC);
  sqlite3_bind_text(stmt, 3, new_password.c_str(), -1, SQLITE_STATIC);
  sqlite3_bind_int(stmt, 2, id);

  rc = sqlite3_step(stmt);
  sqlite3_finalize(stmt);

  if (rc != SQLITE_DONE) {
    std::cerr << "更新执行失败: " << sqlite3_errmsg(db) << std::endl;
    return false;
  }

  return true;
}

bool delete_user(sqlite3 *db, int id) {
  sqlite3_stmt *stmt = nullptr;

  int rc = sqlite3_prepare_v2(db, DELETE_USER, -1, &stmt, nullptr);
  if (rc != SQLITE_OK) {
    std::cerr << "准备删除语句失败: " << sqlite3_errmsg(db) << std::endl;
    return false;
  }

  rc = sqlite3_bind_int(stmt, 1, id);

  if (rc != SQLITE_OK) {
    std::cerr << "绑定参数失败: " << sqlite3_errmsg(db) << std::endl;
    sqlite3_finalize(stmt);
    return false;
  }

  rc = sqlite3_step(stmt);
  const bool success = (rc == SQLITE_DONE);
  const char *errorMsg = sqlite3_errmsg(db);

  sqlite3_finalize(stmt);

  if (!success) {
    std::cerr << "删除执行失败： " << errorMsg << std::endl;
    return false;
  }
  return true;
};