#include "dbmanager.h"
#include <QSqlDatabase>
#include <QSqlError>
#include <QDebug>

DBManager::DBManager(QObject *parent) : QObject(parent)
{

}

bool DBManager::connectToDatabase(const QString &dbType, const QString &host, 
                                 int port, const QString &dbName, 
                                 const QString &user, const QString &password)
{
    // 檢查是否已存在相同名稱的連接
    if (QSqlDatabase::contains("limereport_conn")) {
        m_db = QSqlDatabase::database("limereport_conn");
    } else {
        // 創(chuàng)建新連接
        m_db = QSqlDatabase::addDatabase(dbType, "limereport_conn");
        m_db.setHostName(host);
        m_db.setPort(port);
        m_db.setDatabaseName(dbName);
        m_db.setUserName(user);
        m_db.setPassword(password);
    }

    // 測試連接
    if (!m_db.open()) {
        qCritical() << "數(shù)據(jù)庫連接失敗:" << m_db.lastError().text();
        return false;
    }
    
    qDebug() << "數(shù)據(jù)庫連接成功";
    return true;
}

// 針對SQLite的簡化連接方法
bool DBManager::connectToSQLite(const QString &dbPath)
{
    return connectToDatabase("QSQLITE", "", 0, dbPath, "", "");
}

QSqlDatabase DBManager::database() const
{
    return m_db;
}

#include "reportgenerator.h"
#include "limereport/limereport.h"
#include <QSqlQueryModel>
#include <QSqlQuery>

ReportGenerator::ReportGenerator(QObject *parent) : QObject(parent)
{

}

bool ReportGenerator::generateReport(const QString &templatePath, DBManager *dbManager)
{
    if (!dbManager || !dbManager->database().isOpen()) {
        qCritical() << "數(shù)據(jù)庫連接未就緒";
        return false;
    }

    // 創(chuàng)建報表實(shí)例
    LimeReport::Report *report = new LimeReport::Report();

    // 1. 創(chuàng)建并綁定主數(shù)據(jù)集
    QSqlQueryModel *mainModel = new QSqlQueryModel(report);
    mainModel->setQuery("SELECT id, name, price, stock FROM products", dbManager->database());
    
    // 為字段設(shè)置友好名稱（在報表設(shè)計器中顯示）
    mainModel->setHeaderData(0, Qt::Horizontal, "產(chǎn)品ID");
    mainModel->setHeaderData(1, Qt::Horizontal, "產(chǎn)品名稱");
    mainModel->setHeaderData(2, Qt::Horizontal, "單價");
    mainModel->setHeaderData(3, Qt::Horizontal, "庫存");
    
    // 添加到報表數(shù)據(jù)源，"products"為數(shù)據(jù)源名稱（設(shè)計器中需對應(yīng)）
    report->dataManager()->addModel("products", mainModel, true);

    // 2. 可選：添加第二個數(shù)據(jù)集（例如訂單數(shù)據(jù)）
    QSqlQueryModel *orderModel = new QSqlQueryModel(report);
    orderModel->setQuery("SELECT order_no, order_date, customer FROM orders WHERE status=1", 
                         dbManager->database());
    report->dataManager()->addModel("orders", orderModel, true);

    // 3. 可選：添加報表參數(shù)
    report->setParameterValue("companyName", "我的公司");
    report->setParameterValue("reportDate", QDate::currentDate().toString("yyyy-MM-dd"));

    // 加載報表模板
    if (!report->loadFromFile(templatePath)) {
        qCritical() << "加載報表模板失敗:" << report->lastError();
        delete report;
        return false;
    }

    // 預(yù)覽報表
    report->previewReport();
    
    // 報表會在預(yù)覽窗口關(guān)閉后自動釋放
    return true;
}