arc/android/ozon/LoginAndCheckAccountsScript.cpp

#include "LoginAndCheckAccountsScript.h"

#include <QThread>

#include "adb/AdbUtils.h"
#include "dao/MaterialDAO.h"
#include "dao/BankProfileDAO.h"
#include "dao/DeviceDAO.h"
#include "dump/TaskDumpRecorder.h"
#include "AppLogger.h"
#include "GetCardInfoScript.h"

namespace Ozon {

LoginAndCheckAccountsScript::LoginAndCheckAccountsScript(
    QString deviceId,
    QString appCode,
    QObject *parent,
    bool fetchProfileOnly,
    bool pinCheckOnly
) : CommonScript(parent),
    m_deviceId(std::move(deviceId)),
    m_appCode(std::move(appCode)),
    m_fetchProfileOnly(fetchProfileOnly),
    m_pinCheckOnly(pinCheckOnly) {
}

LoginAndCheckAccountsScript::~LoginAndCheckAccountsScript() = default;


void LoginAndCheckAccountsScript::doStart() {
    // m_deviceId = ADB serial, device.id = android_id (для БД)
    DeviceInfo device = DeviceDAO::getDeviceById(m_deviceId); if (device.id.isEmpty()) device = DeviceDAO::findByAdbSerial(m_deviceId); if (!device.adbSerial.isEmpty()) m_deviceId = device.adbSerial;
    const BankProfileInfo app = BankProfileDAO::getApplication(device.id, m_appCode);

    if (device.id.isEmpty() || app.id == -1) {
        m_error = "Device or app not found: " + m_deviceId + " " + m_appCode;
        qCritical() << m_error;
        emit finishedWithResult(m_error);
        return;
    }

    const TaskDumpMeta dumpMeta{{}, 0, {}, {}, app.bankProfileId, app.phone, device.id, app.appVersion};
    TaskDumpRecorder::Scope dumpScope("ozon", "FETCH_PROFILE", m_deviceId, dumpMeta, &m_error);

    xmlScreenParser.parseAndSaveXml(AdbUtils::getScreenDumpAsXml(m_deviceId, ++m_counter));

    if (!runAppAndGoToHomeScreen(m_deviceId, app.package, app.pinCode, device.screenWidth, device.screenHeight)) {
        m_error = "Cant open the home screen: " + device.name + " (" + m_deviceId + ") " + m_appCode;
        qWarning() << m_error;
        m_resultScreenshot = AdbUtils::captureScreenshotBytes(m_deviceId);
        if (!m_resultScreenshot.isEmpty()) emit screenshotReady(m_resultScreenshot);
        BankProfileDAO::updateComment(app.id,
            "Не прошла проверка " + QDateTime::currentDateTime().toString("dd.MM.yyyy HH:mm:ss"));

        AdbUtils::tryToKillApplication(m_deviceId, app.package);
        emit finishedWithResult(m_error);
        return;
    } else {
        if (m_pinCheckOnly) {
            emit finishedWithResult(m_error);
            return;
        }

        // Скриншот домашнего экрана
        m_resultScreenshot = AdbUtils::captureScreenshotBytes(m_deviceId);
        if (!m_resultScreenshot.isEmpty())
            emit screenshotReady(m_resultScreenshot);

        // Парсим баланс с домашнего экрана
        xmlScreenParser.parseAndSaveXml(AdbUtils::getScreenDumpAsXml(m_deviceId, ++m_counter));
        const QList<MaterialInfo> parsedCards = xmlScreenParser.parseCardsOnHomeScreen();
        double parsedBalance = parsedCards.isEmpty() ? 0.0 : parsedCards[0].amount;
        qDebug() << "[LoginAndCheck] Parsed balance:" << parsedBalance << "cards:" << parsedCards.size();
        if (parsedBalance > 0.0) {
            BankProfileDAO::updateAmount(app.id, parsedBalance);
        }

        // Освежаем amount только для уже существующих материалов (cardNumber/materialId
        // и прочие поля сохраняем). Новые карты здесь не создаём — запись появится
        // только после успешного OCR полного номера в GetCardInfoScript.
        // Если ниже goToAllProducts + parseAccountsInfo отработают — они уточнят
        // per-card сумму поверх.
        for (const MaterialInfo &card : parsedCards) {
            const MaterialInfo existing = MaterialDAO::findAppAccount(device.id, m_appCode, card.lastNumbers);
            if (existing.id == -1) continue;
            MaterialInfo refresh = existing;
            refresh.amount = card.amount;
            refresh.updateTime = QDateTime::currentDateTimeUtc();
            if (!MaterialDAO::upsertAccount(refresh)) {
                qWarning() << "[LoginAndCheck] Failed to refresh amount for card" << card.lastNumbers;
            } else {
                qDebug() << "[LoginAndCheck] Refreshed amount for card"
                         << card.lastNumbers << "to" << card.amount;
            }
        }

        // Создаём/обновляем bank profile на сервере
        if (const QString err = createBankProfileIfNeeded(device.id, m_appCode, parsedBalance); !err.isEmpty()) {
            m_error = err;
            emit finishedWithResult(m_error);
            return;
        }

        if (goToAllProducts(m_deviceId, device.screenWidth, device.screenHeight)) {
            QList<QPair<MaterialInfo, Node> > accounts = xmlScreenParser.parseAccountsInfo();
            for (auto &[account, node]: accounts) {
                account.deviceId = device.id;
                account.appCode = m_appCode;
                if (!MaterialDAO::upsertAccount(account)) {
                    m_error = "Not updated: " + account.lastNumbers;
                    qCritical() << m_error;
                }
            }

            if (!m_fetchProfileOnly) {
                postAccountsData(accounts);
            }
        } else {
            qWarning() << "Cant go to all products: " << m_deviceId << " " << m_appCode;
        }

        if (!m_fetchProfileOnly) {
            // Запускаем сканирование баланса, данных карт и отправку на сервер (синхронно)
            GetCardInfoScript cardScript(m_deviceId, m_appCode, "");
            cardScript.start();
        }
    }

    emit finishedWithResult(m_error);
}

} // namespace Ozon