#include "stdafx.h"
#include "TransferManager.h"
#include <iostream>
#include <sstream>
#include <fstream>
#include <iomanip>
#include <ctime>
#include <stdexcept>

// ¾²Ì¬»¥³âËø³õʼ»¯
std::mutex TransferManager::m_mutex;

// »ñÈ¡ TransferManager µ¥Àý
TransferManager& TransferManager::getInstance() {
    static TransferManager instance;
    return instance;
}

// ¹¹Ô캯Êý
TransferManager::TransferManager() {
    m_pDB = new BL::SQLiteDatabase();
}

// Îö¹¹º¯Êý
TransferManager::~TransferManager() {
    if (m_pDB) {
        delete m_pDB;
        m_pDB = nullptr;
    }
}

// ±¾µØ±àÂëתΪ UTF-8
std::string TransferManager::ansiToUtf8(const std::string& ansiStr) {
    // 1. ANSI ¡ú UTF-16
    int wideLen = MultiByteToWideChar(CP_ACP, 0, ansiStr.c_str(), -1, nullptr, 0);
    std::wstring wideStr(wideLen, 0);
    MultiByteToWideChar(CP_ACP, 0, ansiStr.c_str(), -1, &wideStr[0], wideLen);

    // 2. UTF-16 ¡ú UTF-8
    int utf8Len = WideCharToMultiByte(CP_UTF8, 0, wideStr.c_str(), -1, nullptr, 0, nullptr, nullptr);
    std::string utf8Str(utf8Len, 0);
    WideCharToMultiByte(CP_UTF8, 0, wideStr.c_str(), -1, &utf8Str[0], utf8Len, nullptr, nullptr);

    utf8Str.pop_back(); // È¥µô×îºóµÄ '\0'
    return utf8Str;
}

// UTF-8 תΪ±¾µØ±àÂë
std::string TransferManager::utf8ToAnsi(const std::string& utf8Str) {
    // 1. UTF-8 ¡ú UTF-16
    int wideLen = MultiByteToWideChar(CP_UTF8, 0, utf8Str.c_str(), -1, NULL, 0);
    std::wstring wideStr(wideLen, 0);
    MultiByteToWideChar(CP_UTF8, 0, utf8Str.c_str(), -1, &wideStr[0], wideLen);

    // 2. UTF-16 ¡ú ANSI
    int ansiLen = WideCharToMultiByte(CP_ACP, 0, wideStr.c_str(), -1, NULL, 0, NULL, NULL);
    std::string ansiStr(ansiLen, 0);
    WideCharToMultiByte(CP_ACP, 0, wideStr.c_str(), -1, &ansiStr[0], ansiLen, NULL, NULL);

    ansiStr.pop_back(); // È¥µô×îºóµÄ '\0'
    return ansiStr;
}

// ³õʼ»¯°áÔ˼Ǽ±í
bool TransferManager::initTransferTable() {
    char szPath[MAX_PATH];
    GetModuleFileName(NULL, szPath, MAX_PATH);
    std::string exePath(szPath);
    std::string dbDir = exePath.substr(0, exePath.find_last_of("\\/")) + "\\DB";

    if (!CreateDirectory(dbDir.c_str(), NULL) && GetLastError() != ERROR_ALREADY_EXISTS) {
        throw std::runtime_error("´´½¨Êý¾Ý¿âĿ¼ʧ°Ü");
    }

    std::string dbPath = dbDir + "\\TransferManager.db";

    if (!m_pDB->connect(dbPath, true)) {
        throw std::runtime_error("Á¬½ÓÊý¾Ý¿âʧ°Ü");
    }

    const std::string createTableSQL = R"(
        CREATE TABLE IF NOT EXISTS transfers (
            record_id    INTEGER PRIMARY KEY AUTOINCREMENT,
            class_id     TEXT NOT NULL,
            status       TEXT NOT NULL,
            create_time  TEXT,
            pick_time    TEXT,
            place_time   TEXT,
            end_time     TEXT,
            description  TEXT
        );
    )";

    return m_pDB->executeQuery(createTableSQL);
}

// ²åÈë²âÊÔ°áÔ˼Ǽ
void TransferManager::insertTestTransferRecord(int nCount /*=10000*/) {
    if (nullptr != m_pDB) {
        static const char* STATUS_STR[] = {
            "Unknown", "Ready", "Running", "Picking", "Placing",
            "Restoring", "Error", "Abort", "Completed"
        };

        auto formatTime = [](const std::tm& time) {
            char szTime[64];
            strftime(szTime, sizeof(szTime), "%Y-%m-%d %H:%M:%S", &time);
            return std::string(szTime);
        };

        std::time_t startTime = std::time(nullptr);
        for (int i = 0; i < nCount; ++i) {
            TransferData data;
            data.strClassID = "T-" + std::to_string(1000 + i);
            data.strStatus = STATUS_STR[i % 9];

            std::time_t baseTime = startTime + i * 60;
            std::time_t pickTime = baseTime + 60;
            std::time_t placeTime = pickTime + 60;
            std::time_t endTime = placeTime + 60;

            std::tm tmCreate = {}, tmPick = {}, tmPlace = {}, tmEnd = {};
            localtime_s(&tmCreate, &baseTime);
            localtime_s(&tmPick, &pickTime);
            localtime_s(&tmPlace, &placeTime);
            localtime_s(&tmEnd, &endTime);

            data.strCreateTime = formatTime(tmCreate);
            data.strPickTime = formatTime(tmPick);
            data.strPlaceTime = formatTime(tmPlace);
            data.strEndTime = formatTime(tmEnd);
            data.strDescription = "Mock transfer task " + std::to_string(i);

            int nRecordId = 0;
            if (!addTransferRecord(data, nRecordId)) {
                std::cerr << "[Error] ²åÈëµÚ " << i << " Ìõ¼Ç¼ʧ°Ü" << std::endl;
            }
        }

        std::cout << "[Mock] ³É¹¦²åÈë " << nCount << " Ìõ²âÊÔ°áÔ˼Ǽ¡£" << std::endl;
    }
}

// ¶Ï¿ªÊý¾Ý¿âÁ¬½Ó
void TransferManager::termTransferTable() {
    if (m_pDB) {
        m_pDB->disconnect();
    }
}

// ɾ³ý°áÔ˼Ǽ±í
bool TransferManager::destroyTransferTable() {
    if (!m_pDB) return false;

    const std::string dropTableSQL = "DROP TABLE IF EXISTS transfers;";
    return m_pDB->executeQuery(dropTableSQL);
}

// ²åÈë°áÔ˼Ǽ
bool TransferManager::addTransferRecord(const TransferData& data, int& outRecordId) {
    if (!m_pDB) {
        return false;
    }

    std::ostringstream oss;
    oss << "INSERT INTO transfers (class_id, status, create_time, pick_time, place_time, end_time, description) "
        << "VALUES ('"
        << ansiToUtf8(data.strClassID) << "', '"
        << ansiToUtf8(data.strStatus) << "', '"
        << data.strCreateTime << "', '"
        << data.strPickTime << "', '"
        << data.strPlaceTime << "', '"
        << data.strEndTime << "', '"
        << ansiToUtf8(data.strDescription) << "') RETURNING record_id;";

    std::lock_guard<std::mutex> lock(m_mutex);

    auto results = m_pDB->fetchResults(oss.str());
    if (!results.empty() && !results[0].empty()) {
        try {
            outRecordId = std::stoi(results[0][0]);
            return true;
        }
        catch (const std::exception& e) {
            std::cerr << "½âÎö record_id ³ö´í: " << e.what() << std::endl;
        }
    }

    return false;
}

// ²éѯËùÓаáÔ˼Ǽ
std::vector<TransferData> TransferManager::getAllTransfers() {
    std::vector<TransferData> records;

    if (!m_pDB) {
        return records;
    }

    const std::string query = R"(
        SELECT record_id, class_id, status, create_time, pick_time, place_time, end_time, description
        FROM transfers
        ORDER BY record_id DESC
    )";

    auto results = m_pDB->fetchResults(query);
    for (const auto& row : results) {
        if (row.size() != 8) continue;

        TransferData data;
        data.nRecordId = std::stoi(row[0]);
        data.strClassID = utf8ToAnsi(row[1]);
        data.strStatus = utf8ToAnsi(row[2]);
        data.strCreateTime = row[3];
        data.strPickTime = row[4];
        data.strPlaceTime = row[5];
        data.strEndTime = row[6];
        data.strDescription = utf8ToAnsi(row[7]);

        records.push_back(data);
    }

    return records;
}

// ¸ù¾Ý ID ²éѯ°áÔ˼Ǽ
TransferData TransferManager::getTransferById(int id) {
	TransferData data;
	if (!m_pDB) {
		return data;
	}
	std::ostringstream oss;
	oss << "SELECT record_id, class_id, status, create_time, pick_time, place_time, end_time, description "
		<< "FROM transfers WHERE record_id = " << id;
	auto results = m_pDB->fetchResults(oss.str());
	if (!results.empty() && results[0].size() == 8) {
		data.nRecordId = std::stoi(results[0][0]);
		data.strClassID = utf8ToAnsi(results[0][1]);
		data.strStatus = utf8ToAnsi(results[0][2]);
		data.strCreateTime = results[0][3];
		data.strPickTime = results[0][4];
		data.strPlaceTime = results[0][5];
		data.strEndTime = results[0][6];
		data.strDescription = utf8ToAnsi(results[0][7]);
	}
	return data;
}

// ¸ù¾Ýʱ¼ä·¶Î§²éѯ°áÔ˼Ǽ
std::vector<TransferData> TransferManager::getTransfersByTimeRange(const std::string& startTime, const std::string& endTime) {
	std::vector<TransferData> records;
	if (!m_pDB) {
		return records;
	}
	std::ostringstream oss;
	oss << "SELECT record_id, class_id, status, create_time, pick_time, place_time, end_time, description "
		<< "FROM transfers WHERE 1=1";
	if (!startTime.empty()) {
		oss << " AND create_time >= '" << startTime << "'";
	}
	if (!endTime.empty()) {
		oss << " AND end_time <= '" << endTime << "'";
	}
	auto results = m_pDB->fetchResults(oss.str());
	for (const auto& row : results) {
		if (row.size() != 8) continue;
		TransferData data;
		data.nRecordId = std::stoi(row[0]);
		data.strClassID = utf8ToAnsi(row[1]);
		data.strStatus = utf8ToAnsi(row[2]);
		data.strCreateTime = row[3];
		data.strPickTime = row[4];
		data.strPlaceTime = row[5];
		data.strEndTime = row[6];
		data.strDescription = utf8ToAnsi(row[7]);
		records.push_back(data);
	}
	return records;
}

// ²éѯָ¶¨×´Ì¬µÄ°áÔ˼Ǽ
std::vector<TransferData> TransferManager::getTransfersByStatus(const std::string& status) {
    std::vector<TransferData> records;
    if (!m_pDB) {
        return records;
    }

    std::ostringstream oss;
    oss << "SELECT record_id, class_id, status, create_time, pick_time, place_time, end_time, description "
        << "FROM transfers WHERE status = '" << ansiToUtf8(status) << "' "
        << "ORDER BY create_time DESC";

    auto results = m_pDB->fetchResults(oss.str());
    for (const auto& row : results) {
        if (row.size() != 8) continue;

        TransferData data;
        data.nRecordId = std::stoi(row[0]);
        data.strClassID = utf8ToAnsi(row[1]);
        data.strStatus = utf8ToAnsi(row[2]);
        data.strCreateTime = row[3];
        data.strPickTime = row[4];
        data.strPlaceTime = row[5];
        data.strEndTime = row[6];
        data.strDescription = utf8ToAnsi(row[7]);

        records.push_back(data);
    }

    return records;
}

// ·ÖÒ³»ñÈ¡°áÔ˼Ǽ
//std::vector<TransferData> TransferManager::getTransfers(int startPosition, int count) {
//	std::vector<TransferData> records;
//	if (!m_pDB) {
//		return records;
//	}
//	std::ostringstream oss;
//	oss << "SELECT record_id, class_id, status, create_time, pick_time, place_time, end_time, description "
//		<< "FROM transfers ORDER BY create_time DESC LIMIT " << count << " OFFSET " << startPosition;
//	auto results = m_pDB->fetchResults(oss.str());
//	for (const auto& row : results) {
//		if (row.size() != 8) continue;
//		TransferData data;
//		data.nRecordId = std::stoi(row[0]);
//		data.strClassID = row[1];
//		data.strStatus = row[2];
//		data.strCreateTime = row[3];
//		data.strPickTime = row[4];
//		data.strPlaceTime = row[5];
//		data.strEndTime = row[6];
//		data.strDescription = row[7];
//		records.push_back(data);
//	}
//	return records;
//}

// »ñÈ¡¼Ç¼×ÜÊý
int TransferManager::getTotalTransferCountAll() {
    if (!m_pDB) {
        return 0;
    }

    const std::string query = "SELECT COUNT(*) FROM transfers;";
    auto results = m_pDB->fetchResults(query);

    if (!results.empty() && !results[0].empty()) {
        try {
            return std::stoi(results[0][0]);
        }
        catch (const std::exception& e) {
            std::cerr << "Error parsing total count: " << e.what() << std::endl;
        }
    }

    return 0;
}

// »ñÈ¡·ûºÏÌõ¼þµÄ¼Ç¼×ÜÊý
int TransferManager::getFilteredTransferCount(const TransferData& filter) {
    if (!m_pDB) {
        return 0;
    }

    std::ostringstream oss;
    oss << "SELECT COUNT(*) FROM transfers WHERE 1=1";

    // ״̬ɸѡ£¨ÍêÈ«Æ¥Å䣩
    if (!filter.strStatus.empty()) {
        oss << " AND status = '" << ansiToUtf8(filter.strStatus) << "'";
    }

    // ÃèÊö¹Ø¼ü×ÖÄ£ºýÆ¥Åä
    if (!filter.strDescription.empty()) {
        oss << " AND description LIKE '%" << ansiToUtf8(filter.strDescription) << "%'";
    }

    // ʱ¼ä·¶Î§É¸Ñ¡
    if (!filter.strCreateTime.empty()) {
        oss << " AND create_time >= '" << filter.strCreateTime << "'";
    }
    if (!filter.strEndTime.empty()) {
        oss << " AND end_time <= '" << filter.strEndTime << "'";
    }

    auto results = m_pDB->fetchResults(oss.str());
    if (!results.empty() && !results[0].empty()) {
        try {
            return std::stoi(results[0][0]);
        }
        catch (const std::exception& e) {
            std::cerr << "Error parsing total count: " << e.what() << std::endl;
        }
    }

    return 0;
}

std::vector<TransferData> TransferManager::getTransfers(const TransferData& filter, int pageNum, int pageSize) {
    std::vector<TransferData> records;

    if (!m_pDB) {
        return records;
    }

    std::ostringstream oss;
    oss << "SELECT record_id, class_id, status, create_time, pick_time, place_time, end_time, description "
        << "FROM transfers WHERE 1=1";

    // Ìõ¼þÆ´½Ó£¨Óë getTotalTransferCount ±£³ÖÒ»Ö£©
    if (!filter.strStatus.empty()) {
        oss << " AND status = '" << ansiToUtf8(filter.strStatus) << "'";
    }
    if (!filter.strDescription.empty()) {
        oss << " AND description LIKE '%" << ansiToUtf8(filter.strDescription) << "%'";
    }
    if (!filter.strCreateTime.empty()) {
        oss << " AND create_time >= '" << filter.strCreateTime << "'";
    }
    if (!filter.strEndTime.empty()) {
        oss << " AND end_time <= '" << filter.strEndTime << "'";
    }

    // ·ÖÒ³¿ØÖÆ
    int offset = (pageNum - 1) * pageSize;
    oss << " ORDER BY create_time DESC";
    oss << " LIMIT " << pageSize << " OFFSET " << offset;

    // ²éѯ
    auto results = m_pDB->fetchResults(oss.str());
    for (const auto& row : results) {
        if (row.size() != 8) continue;

        TransferData data;
        data.nRecordId = std::stoi(row[0]);
        data.strClassID = utf8ToAnsi(row[1]);
        data.strStatus = utf8ToAnsi(row[2]);
        data.strCreateTime = row[3];
        data.strPickTime = row[4];
        data.strPlaceTime = row[5];
        data.strEndTime = row[6];
        data.strDescription = utf8ToAnsi(row[7]);

        records.push_back(data);
    }

    return records;
}

// ÇåÀíÔçÓÚijһʱ¼äµÄ°áÔ˼Ǽ
void TransferManager::cleanOldTransfers(int daysToKeep) {
	if (!m_pDB) {
		return;
	}
	std::ostringstream oss;
	oss << "DELETE FROM transfers WHERE create_time < datetime('now', '-" << daysToKeep << " days')";

	m_pDB->executeQuery(oss.str());
}

// ¶ÁÈ¡°áÔ˼Ǽ CSV Îļþ
bool TransferManager::readTransferFile(const std::string& filename) {
    std::ifstream file(filename);
    if (!file.is_open()) {
        std::cerr << "ÎÞ·¨´ò¿ªÎļþ: " << filename << std::endl;
        return false;
    }

    std::string line;
    bool firstLine = true;
    int insertedCount = 0;

    while (std::getline(file, line)) {
        if (firstLine) {
            firstLine = false;
            continue;
        }

        std::stringstream ss(line);
        std::string cell;
        TransferData data;

        try {
            std::getline(ss, cell, ',');
            data.nRecordId = std::stoi(cell);

            std::getline(ss, data.strClassID, ',');
            std::getline(ss, data.strStatus, ',');
            std::getline(ss, data.strCreateTime, ',');
            std::getline(ss, data.strPickTime, ',');
            std::getline(ss, data.strPlaceTime, ',');
            std::getline(ss, data.strEndTime, ',');
            std::getline(ss, data.strDescription, ',');

            int newId = -1;
            if (addTransferRecord(data, newId)) {
                ++insertedCount;
            }
        }
        catch (const std::exception& e) {
            std::cerr << "¶ÁÈ¡´íÎóÐÐ: " << line << "£¬´íÎó: " << e.what() << std::endl;
            continue;
        }
    }

    file.close();
    std::cout << "³É¹¦µ¼Èë¼Ç¼Êý: " << insertedCount << std::endl;
    return true;
}

// ±£´æ°áÔ˼Ǽµ½ CSV Îļþ
bool TransferManager::saveTransferFile(const std::string& filename) {
    std::ofstream file(filename);
    if (!file.is_open()) {
        std::cerr << "ÎÞ·¨Ð´ÈëÎļþ: " << filename << std::endl;
        return false;
    }

    // дÈë±êÌâ
    file << "RecordID,ClassID,Status,CreateTime,PickTime,PlaceTime,EndTime,Description\n";

    auto records = getAllTransfers();
    for (const auto& data : records) {
        file << data.nRecordId << ","
            << data.strClassID << ","
            << data.strStatus << ","
            << data.strCreateTime << ","
            << data.strPickTime << ","
            << data.strPlaceTime << ","
            << data.strEndTime << ","
            << data.strDescription << "\n";
    }

    file.close();
    std::cout << "Òѵ¼³ö " << records.size() << " Ìõ¼Ç¼µ½: " << filename << std::endl;
    return true;
}