arc/android/black/CommonScript.cpp

#include "CommonScript.h"

#include <QJsonArray>
#include <QJsonObject>
#include <QRandomGenerator>
#include <QThread>

#include "adb/AdbUtils.h"
#include "dao/ApplicationDAO.h"
#include "net/NetworkService.h"

namespace Black {

CommonScript::CommonScript(QObject *parent)
    : QObject(parent) {
}

CommonScript::~CommonScript() = default;

void CommonScript::start() {
    doStart();
    emit finished();
}

Node CommonScript::swipeToButton(
    const QString &deviceId,
    const QString &text, int &counter, const int width, const int height
) {
    for (int i = 0; i < 10; ++i) {
        if (Node button = xmlScreenParser.findButtonNode(text, true); button.x() > 0 && button.y() > 0) {
            return button;
        }
        const int x = width / 2;
        const int offset = height / 4;
        AdbUtils::makeSwipe(deviceId, x, height - offset, x, offset);
        xmlScreenParser.parseAndSaveXml(AdbUtils::getScreenDumpAsXml(deviceId, ++counter));
    }
    return {};
}

bool CommonScript::inputPinCode(const QString &deviceId, const QString &pinCode) {
    // BlackBanx использует EditText с password=true, а не кнопочную сетку
    Node editText = xmlScreenParser.findPinCodeEditText();
    if (editText.isEmpty()) {
        qWarning() << "[Black::inputPinCode] PIN EditText not found";
        return false;
    }

    // Тапаем по полю для фокуса
    AdbUtils::makeTap(deviceId, editText.x(), editText.y());
    QThread::msleep(500);

    // Вводим PIN через текстовый ввод
    AdbUtils::inputText(deviceId, pinCode);
    QThread::msleep(500);

    return true;
}

bool CommonScript::runAppAndGoToHomeScreen(
    const QString &deviceId,
    const QString &packageName,
    const QString &pinCode,
    const int width,
    const int height
) {
    // Убить приложение если запущено
    if (const QString pid = AdbUtils::tryToGetRunningAppPid(deviceId, packageName); !pid.isEmpty()) {
        qDebug() << "[Black] Process already running, pid:" << pid;
        AdbUtils::getScreenDumpAsXml(deviceId, 1000);
        AdbUtils::tryToKillApplication(deviceId, packageName);
    }
    if (!AdbUtils::tryToStartApplication(deviceId, packageName)) {
        AdbUtils::takeScreenshot(deviceId, "run_app_error");
        qWarning() << "[Black] Process has not started";
        return false;
    }

    // Даем время на запуск
    QThread::msleep(4500);
    int counter = 0;
    bool findAndInputPincode = false;

    while (true) {
        counter++;

        if (counter > 10) {
            AdbUtils::takeScreenshot(deviceId, "too_many_attempts");
            return false;
        }
        xmlScreenParser.parseAndSaveXml(AdbUtils::getScreenDumpAsXml(deviceId, ++m_counter));

        if (xmlScreenParser.isHomeScreen()) {
            return true;
        }

        // Проверяем экран ввода PIN-кода
        if (!findAndInputPincode && xmlScreenParser.isPinCodeScreen()) {
            inputPinCode(deviceId, pinCode);
            findAndInputPincode = true;
            QThread::msleep(3000);
            continue;
        }

        QThread::msleep(1000);
    }
}

void CommonScript::postAccountsData(const QList<QPair<AccountInfo, Node>> &accounts) {
    if (accounts.isEmpty()) {
        return;
    }

    auto *thread = new QThread;
    auto *worker = new NetworkService;

    QJsonArray list;
    for (const QPair<AccountInfo, Node> &pair : accounts) {
        QJsonObject obj;
        AccountInfo account = pair.first;
        obj["appName"] = account.appCode;
        obj["lastNumbers"] = account.lastNumbers;
        obj["amount"] = account.amount;
        obj["description"] = account.description;
        obj["currency"] = account.currency;
        obj["status"] = accountStatusToString(account.status);
        list.append(obj);
    }

    worker->setDeviceId(accounts.first().first.deviceId);
    worker->setPayload(QJsonDocument(list).toJson());
    worker->moveToThread(thread);
    connect(thread, &QThread::started, worker, &NetworkService::postAccountsData);
    connect(worker, &NetworkService::finished, thread, &QThread::quit);
    connect(worker, &NetworkService::finished, worker, &QObject::deleteLater);
    connect(thread, &QThread::finished, thread, &QThread::deleteLater);
    thread->start();
}

void CommonScript::postNewTransactionData(const AccountInfo &account, const TransactionInfo &transaction) {
    auto *thread = new QThread;
    auto *worker = new NetworkService;

    QJsonObject obj;
    obj["id"] = transaction.id;
    obj["amount"] = transaction.amount;
    obj["name"] = transaction.name;
    obj["fee"] = transaction.fee;
    obj["status"] = transactionStatusToString(transaction.status);
    obj["type"] = transactionTypeToString(transaction.type);
    obj["phone"] = transaction.phone;
    obj["description"] = transaction.description;
    obj["updatedDescription"] = transaction.updatedDescription;
    obj["bankName"] = transaction.bankName;
    obj["bankTime"] = transaction.bankTime;
    obj["bankTransactionId"] = transaction.bankTrExternalId;

    worker->setDeviceId(account.deviceId);
    worker->addExtra("lastNumbers", account.lastNumbers);
    worker->addExtra("transactionId", transaction.id);
    worker->setPayload(QJsonDocument(obj).toJson());
    worker->moveToThread(thread);
    connect(thread, &QThread::started, worker, &NetworkService::insertTransactionData);
    connect(worker, &NetworkService::finished, thread, &QThread::quit);
    connect(worker, &NetworkService::finished, worker, &QObject::deleteLater);
    connect(thread, &QThread::finished, thread, &QThread::deleteLater);
    thread->start();
}

void CommonScript::updateTransactionData(
    const AccountInfo &account,
    const int transactionId,
    const TransactionStatus status,
    const QString &oldDesc,
    const QString &newDesc
) {
    auto *thread = new QThread;
    auto *worker = new NetworkService;

    QJsonObject obj;
    obj["id"] = transactionId;
    obj["status"] = transactionStatusToString(status);
    obj["description"] = newDesc;
    obj["updatedDescription"] = oldDesc;

    worker->setDeviceId(account.deviceId);
    worker->addExtra("lastNumbers", account.lastNumbers);
    worker->setPayload(QJsonDocument(obj).toJson());
    worker->moveToThread(thread);
    connect(thread, &QThread::started, worker, &NetworkService::updateTransactionData);
    connect(worker, &NetworkService::finished, thread, &QThread::quit);
    connect(worker, &NetworkService::finished, worker, &QObject::deleteLater);
    connect(thread, &QThread::finished, thread, &QThread::deleteLater);
    thread->start();
}

void CommonScript::createBankProfileIfNeeded(const QString &deviceId, const QString &appCode) {
    const ApplicationInfo app = ApplicationDAO::getApplication(deviceId, appCode);

    const QMap<QString, int> currencyCodes = {
        {"RUB", 643}, {"USD", 840}, {"KRW", 410}, {"AZN", 944}, {"ARS", 32}
    };
    const int currencyCode = currencyCodes.value(app.currency.toUpper(), 840);
    const QStringList parts = app.fullName.split(' ', Qt::SkipEmptyParts);

    QJsonObject profileBody;
    profileBody["first_name"]   = parts.value(0);
    profileBody["middle_name"]  = parts.size() >= 3 ? parts.value(1) : QString("");
    profileBody["last_name"]    = parts.size() >= 3 ? parts.value(2) : parts.value(1);
    profileBody["bank_name"]    = appCode;
    profileBody["phone_number"] = app.phone.simplified().remove(' ');
    profileBody["balance"]      = QString::number(qRound64(app.amount * 100));
    profileBody["state"]        = (app.status == "active") ? "enabled" : "disabled";
    profileBody["currency"]     = currencyCode;

    NetworkService ns;
    ns.setDeviceId(deviceId);
    ns.setPayload(QJsonDocument(profileBody).toJson());

    if (app.bankProfileId.isEmpty()) {
        ns.addExtra("app_id", app.id);
        ns.postBankProfile();
        qDebug() << "[Black] createBankProfileIfNeeded: postBankProfile done";
    } else {
        ns.addExtra("bank_profile_id", app.bankProfileId);
        ns.patchBankProfile();
        qDebug() << "[Black] createBankProfileIfNeeded: patchBankProfile done";
    }
}

void CommonScript::stop() {
    m_running = false;
}

} // namespace Black