#include "data.h"
#include "render_ui.h"
#include "stdlib.h"
#include "tools.h"
#include "userAuth.h"
#include <cstdlib>
#include <sqlite3.h>

using namespace std;

const char *DB_NAME = "user_data.db";

sqlite3 *create_connection() {
  sqlite3 *db;
  int rc = sqlite3_open(DB_NAME, &db);
  if (rc != SQLITE_OK) {
    std::cerr << "无法打开数据库：" << sqlite3_errmsg(db) << std::endl;
    exit(1);
  }
  return db;
}

int main() {
  sqlite3 *db = create_connection();
  init_pa_data(db);
  init_user_data(db);
  string login_username;
  string login_password;
  bool user_authorized = false;

  cout << "请输入用户名登录: ";
  cin >> login_username;
  cout << "请输入对应的密码: ";
  cin >> login_password;
  cout << endl;

  user_authorized = login_user(db, login_username, login_password);

  if (user_authorized) {
    cout << "登录成功！" << endl;
  } else {
    cout << "登录失败! " << endl;
    return 1;
  }

  while (user_authorized) {
    system("clear");
    switch (show_main_menu()) {
    case 1:
      system("clear");
      handle_add_pa_record(db);
      break;
    case 2:
      system("clear");
      handle_leave_car(db);
      break;
    case 3:
      system("clear");
      handle_show_pa_records(db);
      break;
    case 4:
      system("clear");
      handle_show_pa_leftcar_records(db);
      break;
    case 5:
      system("clear");
      handle_add_user_record(db);
      break;
    case 6:
      system("clear");
      handle_show_user_records(db);
      break;
    case 7:
      system("clear");
      handle_update_userpassword_record(db);
      break;
    case 8:
      system("clear");
      handle_delete_user_record(db);
      break;
    case 9:
      sqlite3_close(db);
      std::cout << "退出应用" << std::endl;
      return 0;
    }
  }
}