MFCで在庫管理システム作成　２

class CInventoryManagementSystemApp : public CWinApp
{
// 略

private:
	CString m_strUserID;	// ユーザID
	CString m_strUserName;	// ユーザ名
	BOOL m_bUserAdmin;		// ユーザ権限　TRUE=管理者、FALSE=一般
	const int m_iCategoryCnt = 3;	// カテゴリー数
	const CString m_strArrayCategory[3] = {_T("Food), _T("Stationery"), _T("Electronic equipment")};	// カテゴリー配列

public:
	// ユーザID設定
	void SetUserID(CString strUserID) { m_strUserID = strUserID; }
	// ユーザID取得
	CString strGetUserID() { return m_strUserID; }

	// ユーザ名設定
	void SetUserName(CString strUserName) { m_strUserName = strUserName; }
	// ユーザ名取得
	CString strGetUserName() { return m_strUserName; }

	// ユーザ権限設定
	void SetUserAdmin(BOOL bUserAdmin) { m_bUserAdmin = bUserAdmin; }
	// ユーザ権限取得
	BOOL bGetUserAdmin() { return m_bUserAdmin; }

	// カテゴリー数返却
	const int iGetCategoryCnt() { return m_iCategoryCnt; }
	// カテゴリー返却
	const CString strGetCategory(int iId) { return m_strArrayCategory[iId]; }
};

CInventoryManagementSystemApp::CInventoryManagementSystemApp()
{
	// 略

	m_strUserID = _T("");	// ユーザID
	m_strUserName = _T("");	// ユーザ名
	m_bUserAdmin = FALSE;	// ユーザ権限
}

IDD_DIALOG_LOGIN DIALOGEX 0, 0, 166, 134
STYLE DS_SETFONT | DS_MODALFRAME | DS_FIXEDSYS | WS_POPUP | WS_CAPTION | WS_SYSMENU
CAPTION "Dialog"
FONT 8, "MS Shell Dlg", 400, 0, 0x1
BEGIN
    DEFPUSHBUTTON   "ログイン",IDOK,20,100,50,14
    PUSHBUTTON      "キャンセル",IDCANCEL,90,100,50,14
    LTEXT           "ユーザID",IDC_STATIC_LOGIN_ID,20,20,30,8
    EDITTEXT        IDC_EDIT_LOGIN_ID,20,30,100,14,ES_AUTOHSCROLL | ES_NUMBER
    LTEXT           "パスワード",IDC_STATIC_LOGIN_PASSWORD,20,50,29,8
    EDITTEXT        IDC_EDIT_LOGIN_PASSWORD,20,60,100,14,ES_PASSWORD | ES_AUTOHSCROLL
END

class CLogin : public CDialogEx
{
	// 略

private:
	CString m_strLoginUserID;		// ユーザID
	CString m_strLoginPassword;		// ログインパスワード
};

CLogin::CLogin(CWnd* pParent /*=nullptr*/)
	: CDialogEx(IDD_DIALOG_LOGIN, pParent)
{
	m_strLoginUserID = _T("");		// ログインユーザID
	m_strLoginPassword = _T("");	// ログインパスワード
}

void CLogin::DoDataExchange(CDataExchange* pDX)
{
	CDialogEx::DoDataExchange(pDX);
	DDX_Text(pDX, IDC_EDIT_LOGIN_ID, m_strLoginUserID);					// ユーザID
	DDX_Text(pDX, IDC_EDIT_LOGIN_PASSWORD, m_strLoginPassword);			// パスワード
}

/**
 * @fn
 * @brief	キャンセルボタン押下時の処理
 * @param	なし
 * @return	なし
 */
void CLogin::OnBnClickedCancel()
{
	// TODO: ここにコントロール通知ハンドラー コードを追加します。
	CDialogEx::OnCancel();
	PostQuitMessage(0); // アプリケーション終了
}

/**
 * @fn
 * @brief	ダイアログ初期処理
 * @param	なし
 * @return	TRUE	成功
 *			FALSE	失敗
 */
BOOL CLogin::OnInitDialog()
{
	CDialogEx::OnInitDialog();

	// TODO: ここに初期化を追加してください

	SetWindowText(_T("ログイン画面"));

	return TRUE;  // return TRUE unless you set the focus to a control
	// 例外 : OCX プロパティ ページは必ず FALSE を返します。
}

#include "pch.h"
#include "framework.h"
#include "InventoryManagementSystem.h"
#include "InventoryManagementSystemDlg.h"
#include "Login.h"	// ログイン画面

BOOL CInventoryManagementSystemApp::InitInstance()
{
	// 略

	CLogin dlgLogin;
	if (dlgLogin.DoModal() != IDOK)
	{
		// キャンセルが押された場合、アプリケーションを終了
		return FALSE;
	}

	CInventoryManagementSystemDlg dlg;
	m_pMainWnd = &dlg;
	INT_PTR nResponse = dlg.DoModal();
	// 略

	// ダイアログは閉じられました。アプリケーションのメッセージ ポンプを開始しないで
	//  アプリケーションを終了するために FALSE を返してください。
	return FALSE;
}

#pragma once
#include <mysql/jdbc.h>
#include <string>

class CMySQLManager
{
	// 変数
private:
	// DB接続関連
	const sql::SQLString HOST = "tcp://127.0.0.1:3306";
	const sql::SQLString USER = "root";
	const sql::SQLString PASSWORD = "〇〇〇"; //←ここは自身のパスワード

	// スキーマ
	const sql::SQLString SCHEMA = "inventory_management";
	// テーブル
	const sql::SQLString TABLE_USER = "user";		// ユーザテーブル
	const sql::SQLString TABLE_PRODUCT = "product";	// 商品テーブル
	const sql::SQLString TABLE_LOGS = "logs";		// 履歴テーブル

	// 列名
	// ユーザテーブル
	const sql::SQLString COL_USER_ID = "user_id";
	const sql::SQLString COL_USER_NAME = "user_name";
	const sql::SQLString COL_USER_PASSWORD = "password";
	const sql::SQLString COL_USER_ADMIN = "admin";

	// 商品テーブル
	const sql::SQLString COL_PRODUCT_ID = "product_id";
	const sql::SQLString COL_PRODUCT_NAME = "product_name";
	const sql::SQLString COL_PRODUCT_CATEGORY = "category";
	const sql::SQLString COL_PRODUCT_UP = "unit_price";
	const sql::SQLString COL_PRODUCT_COUNT = "product_count";
	const sql::SQLString COL_PRODUCT_ALERT = "product_alert";
	const sql::SQLString COL_PRODUCT_DISCONTINUE = "discontinue";

	// 履歴テーブル
	const sql::SQLString COL_LOGS_ID = "log_id";
	const sql::SQLString COL_LOGS_OPERATION = "operation";
	const sql::SQLString COL_LOGS_PRODUCT_ID = "product_id";
	const sql::SQLString COL_LOGS_TIME_STAMP = "time_stamp";

	// 接続状況
	std::shared_ptr<sql::Connection> m_con;


	// 関数
private:
	// エラーひとまとめ
	void ErrSQLException(sql::SQLException& e);
	// CString→SQLString変換処理
	sql::SQLString strConverter(CString str);
	// SQLString→CString変換処理
	CString strConvertFromUTF8ToUTF16(sql::SQLString strCol);
	//ハッシュ化
	sql::SQLString strHashing(CString strPassword);

public:
	// コンストラクタ
	CMySQLManager();

	//デストラクタ
	~CMySQLManager();

	//接続
	std::shared_ptr<sql::Connection> GetConnection();

	//ログインチェック
	BOOL bCheckLogin(CString strId, CString strPassword);
};

#include "pch.h"
#include "InventoryManagementSystem.h"	//ここに当プロジェクトのヘッダー
#include "MySQLManager.h"

#include "pch.h"
#include "InventoryManagementSystem.h"	//ここに当PJのヘッダー
#include "MySQLManager.h"

/**
 * @fn
 * @brief	コンストラクタ
 * @param	なし
 * @return	なし
 */
CMySQLManager::CMySQLManager()
{

}

/**
 * @fn
 * @brief	デストラクタ
 * @param	なし
 * @return	なし
 */
CMySQLManager::~CMySQLManager()
{
    // デストラクタで接続をクローズ
    if (m_con && m_con->isClosed() == false)
    {
        m_con->close();
    }
}

/**
 * @fn
 * @brief	エラー情報出力関数
 * @param	SQLException    SQLの例外
 * @return	なし
 */
void CMySQLManager::ErrSQLException(sql::SQLException& e)
{
    //接続できなかったため、エラーを表示させる
    CString errorMessage = _T("エラー：") + CString(e.what());
    CString errMsgtemp;
    errMsgtemp.Format(_T("\r\nエラーコード: %d"), e.getErrorCode());
    errorMessage += errMsgtemp;
    AfxMessageBox(errorMessage, MB_OK | MB_ICONERROR);
}

/**
 * @fn
 * @brief	CString→SQLString変換関数
 * @param	CString     変換元文字列
 * @return	SQLString   変換後文字列
 */
sql::SQLString CMySQLManager::strConverter(CString str)
{
    CT2CA pszConvertedAnsiString(str);
    std::string strStd(pszConvertedAnsiString);
    sql::SQLString sqlStrUserID(strStd);

    return sqlStrUserID;
}

/**
 * @fn
 * @brief	SQLString→CString変換関数
 * @param	SQLString   変換元文字列
 * @return	CString     変換後文字列
 */
CString CMySQLManager::strConvertFromUTF8ToUTF16(sql::SQLString strCol)
{
    CString strRtn = _T("");
    int len = MultiByteToWideChar(CP_UTF8, 0, strCol.c_str(), -1, NULL, 0);
    if (len > 0)
    {
        std::wstring wideName(len, 0);
        MultiByteToWideChar(CP_UTF8, 0, strCol.c_str(), -1, &wideName[0], len);
        // std::wstringからCStringに変換
        CString cstrCol(wideName.c_str());
        strRtn = cstrCol;
    }

    return strRtn;
}

/**
 * @fn
 * @brief	パスワードハッシュ化関数
 * @param	CString     生パスワード
 * @return	SQLString   ハッシュ化済文字列
 */
sql::SQLString CMySQLManager::strHashing(CString strPassword)
{
    CT2CA pszConvertedAnsiString(strPassword);
    std::string strStd(pszConvertedAnsiString);
    std::hash<std::string> hash_fn;

    std::string strValue = std::to_string(hash_fn(strStd)); //ハッシュ化処理
    return sql::SQLString(strValue);    //SQLString型に変更
}

/**
 * @fn
 * @brief	接続確立関数
 * @param	なし
 * @return	std::shared_ptr<sql::Connection>    コネクション(?)
 */
std::shared_ptr<sql::Connection> CMySQLManager::GetConnection()
{
    if (!m_con || m_con->isClosed())
    {
        try
        {
            sql::mysql::MySQL_Driver* driver = sql::mysql::get_mysql_driver_instance();

            // 新しい接続を作成
            m_con = std::shared_ptr<sql::Connection>(driver->connect(HOST, USER, PASSWORD));
        }
        catch (sql::SQLException& e)
        {
            // 接続でエラーが生じた
            ErrSQLException(e);
        }
    }

    return m_con;
}

/**
 * @fn
 * @brief	ログインチェック
 * @param	CString         ユーザID
 * @param   CString         パスワード
 * @return	BOOL    TRUE    ログイン成功
 *                  FALSE   ログイン失敗
 */
BOOL CMySQLManager::bCheckLogin(CString strId, CString strPassword)
{
    try
    {
        // スキーマセット
        m_con->setSchema(SCHEMA);

        sql::SQLString sqlStrId = strConverter(strId);          // ユーザID    SQLString化
        sql::SQLString sqlStrPass = strHashing(strPassword);    // パスワード  ハッシュ化

        sql::SQLString strSqlQuery = "SELECT * FROM " + TABLE_USER + " WHERE " + COL_USER_ID + " = '" + sqlStrId + "' AND " + COL_USER_PASSWORD + " = '" + sqlStrPass + "'";
        // SELECT * FROM user WHERE user_id = '00000' AND password = 'ハッシュ化文字列'

        //クエリ実行
        std::unique_ptr<sql::PreparedStatement> pstmt(m_con->prepareStatement(strSqlQuery));

        std::unique_ptr<sql::ResultSet> res(pstmt->executeQuery());

        CString strUserName = _T("");
        BOOL bUserAuth = FALSE;
        if (!res->next())
        {
            AfxMessageBox(_T("Idまたはパスワードが見当たらない"), MB_ICONERROR);
            return FALSE;
        }
        else
        {
            // IDとパスワードが見つかった！！

            //ユーザ名取得
            std::string strCol = res->getString(COL_USER_NAME);
            strUserName = strConvertFromUTF8ToUTF16(strCol);
            //ユーザ権限取得
            int iAuth = res->getInt(COL_USER_ADMIN);
            if (iAuth == 1)
            {
                bUserAuth = TRUE;
            }

            //ログイン情報保持
            CInventoryManagementSystemApp* pApp = static_cast<CInventoryManagementSystemApp*>(AfxGetApp());
            pApp->SetUserID(strId);         // ユーザID
            pApp->SetUserName(strUserName); // ユーザ名
            pApp->SetUserAdmin(bUserAuth);  // ユーザ権限
        }
    }
    catch (sql::SQLException& e)
    {
        ErrSQLException(e);
        return FALSE;
    }

    return TRUE;
}

#include "MySQLManager.h"

/**
 * @fn
 * @brief	ログインボタン押下時の処理
 * @param	なし
 * @return	なし
 */
void CLogin::OnBnClickedOk()
{
	//メンバ変数に、エディットボックスの文字列を入れる
	UpdateData(TRUE);

	if (m_strLoginUserID == _T("") || m_strLoginPassword == _T(""))
	{
		// ログインIDまたはパスワードが埋まっていない
		AfxMessageBox(_T("IDまたはパスワードを入力してください。"), MB_ICONERROR);
		return;
	}

	//両方埋まっている
	CMySQLManager Manager;		// コンストラクタ
	Manager.GetConnection();	// 接続確立

	if (Manager.bCheckLogin(m_strLoginUserID, m_strLoginPassword) == TRUE)
	{
		// ログイン成功！！
		CDialog::OnOK();
	}
	else
	{
		//何らかの形でログイン失敗
		return;
	}
}

m_con = std::shared_ptr<sql::Connection>(driver->connect(HOST, USER, PASSWORD));