#include "..\include_eapi_header.h"
// 获取磁盘驱动器数量
// type - 指定的磁盘驱动器类型,0表示检测所有类型的驱动器
// 返回值 - 满足条件的驱动器数量
static int GetDrivesNum(int type)
{
	//windows仅支持24个硬盘驱动
	int remaining_letters = 26; // 剩余的字母数,A-Z共26个字母,准备遍历
	int drives_count = 0; // 磁盘驱动器数量计数器
	char drive_letter = 'A'; // 当前枚举的字母

	do
	{
		char root_path[3] = { drive_letter, ':', '0円' }; // 根目录字符串
		int drive_type = GetDriveTypeA(root_path);

		// 根据type的值确定是否满足条件
		if (type)
		{
			if (drive_type == type)
			{
				drives_count++;
			}
		}
		else
		{
			// 只统计类型为2, 3, 4, 5的磁盘驱动器
			if (drive_type == 2 || drive_type == 3 || drive_type == 4 || drive_type == 5)
			{
				drives_count++;
			}
		}

		drive_letter++; // 枚举下一个字母
		remaining_letters--;
	} while (remaining_letters);

	return drives_count;
}
// 调用格式: SDT_INT 取驱动器数量, 命令说明: "取当前指定类型的驱动器数量"
// 参数<1>: [驱动器类型 SDT_INT], 参数说明: "可以被省略。驱动器的类型,可以为以下常量值之一:0、#接口常量.所有驱动器;1、#接口常量.硬盘驱动器;2、#接口常量.光盘驱动器;3、#接口常量.可移动驱动器;4、#接口常量.网络驱动器。"
EAPI_EXTERN_C void eapi_GetDrivesNum_4_eapi(PMDATA_INF pRetData, INT nArgCount, PMDATA_INF pArgInf)
{
	INT type = 0, ret = 0;
	if (pArgInf->m_dtDataType && pArgInf->m_int > 0 && pArgInf->m_int < 5)
	{
		type = pArgInf->m_int;
	}
	switch (type)
	{
	case 0:
		ret = GetDrivesNum(0);
		break;
	case 1:
		ret = GetDrivesNum(3);
		break;
	case 2:
		ret = GetDrivesNum(5);
		break;
	case 3:
		ret = GetDrivesNum(2);
		break;
	case 4:
		ret = GetDrivesNum(4);
		break;
	default:
		break;
	}

	pRetData->m_int = ret;

}