484 lines
19 KiB
C++
484 lines
19 KiB
C++
//
|
||
// Created by AmirS on 24.04.2025.
|
||
//
|
||
|
||
#include "Script.h"
|
||
#include "AdbUtils.h"
|
||
#include "XmlParser.h"
|
||
#include "QDebug"
|
||
#include <regex>
|
||
#include <string>
|
||
#include <thread>
|
||
#include <chrono>
|
||
|
||
// Вспомогательные функции
|
||
|
||
void wait(const int milliseconds) {
|
||
std::this_thread::sleep_for(std::chrono::milliseconds(milliseconds));
|
||
}
|
||
|
||
Script::Script(QObject *parent) : QObject(parent) {
|
||
}
|
||
|
||
Script::~Script() = default;
|
||
|
||
void Script::start() {
|
||
qDebug("Запуск скрипта...");
|
||
|
||
processUiLoop();
|
||
}
|
||
|
||
void Script::processUiLoop() {
|
||
while (true) {
|
||
QString uiDumpPath = "ui_dump.xml";
|
||
AdbUtils::dumpAndPullUiXml(uiDumpPath);
|
||
|
||
QList<UiElement> elements = XmlParser::parseUiDumpUsingQXml(uiDumpPath);
|
||
|
||
switch (currentState) {
|
||
case State::WaitingForPinInput:
|
||
if (XmlParser::isPinCodeScreen(elements)) {
|
||
qDebug() << "Экран ввода ПИН-кода обнаружен";
|
||
inputPinCode("1211");
|
||
currentState = State::ClosingBanner;
|
||
}
|
||
break;
|
||
|
||
case State::ClosingBanner:
|
||
handleClosingBanner(elements);
|
||
break;
|
||
|
||
case State::OnMainScreen:
|
||
if (XmlParser::isMainScreen(elements)) {
|
||
qDebug() << "State: OnMainScreen -> Поиск кнопки 'Все продукты'";
|
||
swipeToFindElement("Все продукты");
|
||
currentState = State::SearchingCard;
|
||
}
|
||
break;
|
||
|
||
case State::SearchingCard:
|
||
qDebug() << "State: SearchingCard -> Поиск блока карты";
|
||
// Добавить сюда поиск карты по последним 4 цифрам, известным для выбора карты
|
||
currentState = State::OnCardDetails;
|
||
break;
|
||
|
||
case State::OnCardDetails:
|
||
// Обработка экрана перевода
|
||
handleTransferScreen(elements);
|
||
break;
|
||
|
||
case State::OnTransferScreen:
|
||
qDebug() << "State: OnTransferScreen -> Перевод выполнен успешно.";
|
||
currentState = State::Idle;
|
||
break;
|
||
|
||
case State::Idle:
|
||
default:
|
||
qDebug() << "State: Idle -> Waiting for action";
|
||
if (XmlParser::isPinCodeScreen(elements)) {
|
||
currentState = State::WaitingForPinInput;
|
||
}
|
||
break;
|
||
}
|
||
|
||
QThread::sleep(1); // Пауза между проверками экрана
|
||
}
|
||
}
|
||
|
||
|
||
// Функция для запуска скрипта с передачей параметров
|
||
// void Script::runScript(const QString& pin, const QString& last4digits) {
|
||
// qDebug("Запуск скрипта для ru.rshb.dbo");
|
||
//
|
||
// if (!AdbUtils::checkDevices()) {
|
||
// qDebug("ADB: Нет устройств");
|
||
// return;
|
||
// }
|
||
//
|
||
// if (!AdbUtils::startApp("ru.rshb.dbo")) {
|
||
// qDebug("Не удалось запустить приложение");
|
||
// return;
|
||
// }
|
||
//
|
||
// wait(4500);
|
||
// AdbUtils::dumpAndPullUiXml("ui_01.xml");
|
||
//
|
||
// auto ui = XmlParser::parseUiDumpUsingQXml("ui_01.xml");
|
||
//
|
||
// if (XmlParser::isPinCodeScreen(ui)) {
|
||
// qDebug("Экран PIN-кода");
|
||
//
|
||
// // Находим координаты цифр на экране и вводим ПИН
|
||
// std::vector<UiElement> pinButtons = getPinButtons(ui);
|
||
// for (size_t i = 0; i < pin.length(); ++i) {
|
||
// char digit = pin[i];
|
||
// for (const auto& el : pinButtons) {
|
||
// if (el.text == std::string(1, digit)) {
|
||
// AdbUtils::tap((el.x1 + el.x2) / 2, (el.y1 + el.y2) / 2);
|
||
// break;
|
||
// }
|
||
// }
|
||
// }
|
||
// wait(1000);
|
||
// }
|
||
//
|
||
// // Появляющиеся после входа по ПИН баннеры — цикл для обработки разных состояний показов модальных окон
|
||
// bool postLoginScreenFound = false;
|
||
// int attempts = 5; // Максимальное количество попыток парсинга xml
|
||
// while (attempts-- > 0) {
|
||
// AdbUtils::dumpAndPullUiXml("ui_02.xml");
|
||
// auto ui2 = XmlParser::parseUiDumpUsingQXml("ui_02.xml");
|
||
//
|
||
// // Закрытие баннеров или диалогов
|
||
// if (handlePostLoginScreen(ui2)) {
|
||
// postLoginScreenFound = true;
|
||
// break;
|
||
// }
|
||
//
|
||
// // Проверка, если мы на главном экране
|
||
// if (isMainScreen(ui2)) {
|
||
// qDebug("Главный экран подтверждён");
|
||
// postLoginScreenFound = true;
|
||
// break;
|
||
// }
|
||
//
|
||
// // Ждем немного перед следующим запросом дампа экрана
|
||
// wait(1000);
|
||
// }
|
||
//
|
||
// if (!postLoginScreenFound) {
|
||
// qDebug("Не удалось определить пост-логин баннер.");
|
||
// return;
|
||
// }
|
||
//
|
||
// // Пролистывание до кнопки "Все продукты"
|
||
// AdbUtils::swipe(1000, 2000, 1000, 1300);
|
||
// wait(500);
|
||
//
|
||
// // Нажимаем на кнопку "Все продукты"
|
||
// AdbUtils::dumpAndPullUiXml("ui_03.xml");
|
||
// auto ui3 = XmlParser::parseUiDumpUsingQXml("ui_03.xml");
|
||
//
|
||
// // Ищем кнопку "Все продукты" и нажимаем на неё
|
||
// tapByText(ui3, "Все продукты");
|
||
// wait(800);
|
||
//
|
||
// // Шаг 6: Выбор карты на экране "Все продукты"
|
||
// // Запрашиваем дамп UI для поиска карт
|
||
// AdbUtils::dumpAndPullUiXml("ui_04.xml");
|
||
// auto ui4 = XmlParser::parseUiDumpUsingQXml("ui_04.xml");
|
||
//
|
||
// // Определим, какую карту выбрать, например, с 4 цифрами "0823"
|
||
// UiElement cardButton = findCardByLast4Digits(ui4, last4digits);
|
||
// if (cardButton.text.isEmpty()) {
|
||
// qDebug("Не найдена карта с номером, содержащим " + last4digits);
|
||
// return;
|
||
// }
|
||
//
|
||
// // Нажимаем на карту
|
||
// AdbUtils::tap((cardButton.x1 + cardButton.x2) / 2, (cardButton.y1 + cardButton.y2) / 2);
|
||
// wait(1000);
|
||
//
|
||
// // Нажимаем на кнопку "Оплатить"
|
||
// // Запрашиваем дамп UI для поиска кнопки "Оплатить"
|
||
// AdbUtils::dumpAndPullUiXml("ui_05.xml");
|
||
// auto ui5 = XmlParser::parseUiDumpUsingQXml("ui_05.xml");
|
||
//
|
||
// // Ищем кнопку "Оплатить" и нажимаем на неё
|
||
// UiElement payBtn = findPayButton(ui5);
|
||
// if (payBtn.text.isEmpty()) {
|
||
// qDebug("Не найдена кнопка 'Оплатить'");
|
||
// return;
|
||
// }
|
||
//
|
||
// // Нажимаем на кнопку "Оплатить"
|
||
// AdbUtils::tap((payBtn.x1 + payBtn.x2) / 2, (payBtn.y1 + payBtn.y2) / 2);
|
||
// qDebug("Нажата кнопка 'Оплатить'");
|
||
// wait(1000);
|
||
// }
|
||
|
||
// Функции для определения экранов и действий
|
||
|
||
void Script::inputPinCode(const QString &pinCode) {
|
||
// Перебираем каждый символ из ПИН-кода
|
||
for (int i = 0; i < pinCode.length(); ++i) {
|
||
QString digit = pinCode.mid(i, 1);
|
||
qDebug() << "Finding button for digit: " << digit;
|
||
|
||
// Поиск кнопки с соответствующим текстом
|
||
QList<UiElement> elements = XmlParser::parseUiDumpUsingQXml("ui_dump.xml");
|
||
|
||
// Ищем кнопку с цифрой
|
||
bool found = false;
|
||
for (const UiElement &el: elements) {
|
||
if (el.text() == digit && el.isClickable()) {
|
||
int x = (el.x1() + el.x2()) / 2;
|
||
int y = (el.y1() + el.y2()) / 2;
|
||
AdbUtils::tap(x, y); // Тапаем по кнопке
|
||
found = true;
|
||
break;
|
||
}
|
||
}
|
||
|
||
// Если не нашли кнопку для цифры, выводим предупреждение
|
||
if (!found) {
|
||
qWarning() << "Digit button for " << digit << " not found!";
|
||
}
|
||
|
||
// Пауза между нажатиями, чтобы избежать слишком быстрого ввода
|
||
QThread::sleep(1);
|
||
}
|
||
}
|
||
|
||
void Script::handleClosingBanner(QList<UiElement> &elements) {
|
||
while (true) {
|
||
// Проверка на наличие баннера или диалога
|
||
if (XmlParser::isBannerOrDialog(elements)) {
|
||
qDebug() << "State: ClosingBanner -> Found banner/dialog. Closing it.";
|
||
|
||
// Найдем кнопку "Закрыть" или кнопку без текста в правом верхнем углу
|
||
bool closed = false;
|
||
|
||
for (const UiElement &el: elements) {
|
||
if (el.isClickable()) {
|
||
// Флаг для отслеживания, если мы нашли хотя бы одно окно для закрытия
|
||
bool bannerClosed = false;
|
||
// Пробуем закрыть баннер, если есть кнопка "Закрыть"
|
||
if ((el.text() == "Закрыть" || el.text() == "Позже") && el.className() == "android.widget.Button") {
|
||
int x = (el.x1() + el.x2()) / 2;
|
||
int y = (el.y1() + el.y2()) / 2;
|
||
AdbUtils::tap(x, y); // Тапаем по кнопке "Закрыть"
|
||
closed = true;
|
||
bannerClosed = true;
|
||
qDebug() << "Banner/dialog closed!";
|
||
break;
|
||
}
|
||
// Если кнопка без текста и в правой верхней части
|
||
if (el.text().isEmpty() && el.className() == "android.widget.Button" &&
|
||
el.x1() > 900 && el.y1() < 100) {
|
||
int x = (el.x1() + el.x2()) / 2;
|
||
int y = (el.y1() + el.y2()) / 2;
|
||
AdbUtils::tap(x, y); // Тапаем по кнопке в правом верхнем углу
|
||
closed = true;
|
||
bannerClosed = true;
|
||
qDebug() << "Banner/dialog closed by tapping button in the top right corner.";
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
|
||
if (!closed) {
|
||
qWarning() << "Невозможно закрыть баннер или диалог!";
|
||
break; // Прерываем цикл, если не удалось закрыть окно
|
||
}
|
||
|
||
// Пауза перед следующим закрытием окна
|
||
QThread::sleep(1);
|
||
} else {
|
||
qDebug() << "State: ClosingBanner -> No banners/dialogs left to close.";
|
||
break; // Заканчиваем цикл, если баннеров/диалогов больше нет
|
||
}
|
||
}
|
||
|
||
// Переход к следующему состоянию
|
||
currentState = State::OnMainScreen; // Если все модальные окна закрыты, переходим к основному экрану
|
||
}
|
||
|
||
void Script::inputAmount(const QString &amount) {
|
||
// Разделяем сумму на отдельные цифры
|
||
for (int i = 0; i < amount.length(); ++i) {
|
||
QString digit = amount.mid(i, 1);
|
||
qDebug() << "Inputting digit: " << digit;
|
||
|
||
// Поиск кнопки с соответствующей цифрой
|
||
QList<UiElement> elements = XmlParser::parseUiDumpUsingQXml("ui_dump.xml");
|
||
|
||
bool found = false;
|
||
for (const UiElement &el: elements) {
|
||
if (el.text() == digit && el.isClickable()) {
|
||
int x = (el.x1() + el.x2()) / 2;
|
||
int y = (el.y1() + el.y2()) / 2;
|
||
AdbUtils::tap(x, y); // Тапаем по кнопке цифры
|
||
found = true;
|
||
break;
|
||
}
|
||
}
|
||
|
||
if (!found) {
|
||
qWarning() << "Digit button for " << digit << " not found!";
|
||
}
|
||
|
||
QThread::sleep(1); // Пауза между нажатиями
|
||
}
|
||
|
||
// Если нужно, нажимаем кнопку для завершения ввода суммы
|
||
QList<UiElement> elements = XmlParser::parseUiDumpUsingQXml("ui_dump.xml");
|
||
for (const UiElement &el: elements) {
|
||
if (el.text() == "ОК" && el.isClickable()) {
|
||
int x = (el.x1() + el.x2()) / 2;
|
||
int y = (el.y1() + el.y2()) / 2;
|
||
AdbUtils::tap(x, y); // Тапаем по кнопке "ОК"
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
|
||
void Script::swipeToFindElement(const QString &searchText) {
|
||
bool found = false;
|
||
int attempts = 0;
|
||
|
||
// Прокручиваем экран несколько раз, пока не найдем элемент или не превысим попытки
|
||
while (attempts < 5 && !found) {
|
||
QList<UiElement> elements = XmlParser::parseUiDumpUsingQXml("ui_dump.xml");
|
||
|
||
// Ищем нужный элемент на экране
|
||
for (const UiElement &el: elements) {
|
||
if (el.text() == searchText && el.isClickable()) {
|
||
qDebug() << "Element found: " << searchText;
|
||
int x = (el.x1() + el.x2()) / 2;
|
||
int y = (el.y1() + el.y2()) / 2;
|
||
AdbUtils::tap(x, y); // Тапаем по найденному элементу
|
||
found = true;
|
||
break;
|
||
}
|
||
}
|
||
|
||
// Если элемент не найден, выполняем свайп
|
||
if (!found) {
|
||
qDebug() << "Element not found, swiping...";
|
||
AdbUtils::swipe(500, 1500, 500, 500); // Свайп вверх
|
||
attempts++;
|
||
QThread::sleep(2); // Пауза между свайпами
|
||
}
|
||
}
|
||
|
||
if (!found) {
|
||
qWarning() << "Element " << searchText << " not found after " << attempts << " swipe attempts.";
|
||
}
|
||
}
|
||
|
||
void Script::handleTransferScreen(QList<UiElement> &elements) {
|
||
// Проверка на экран перевода
|
||
if (XmlParser::isTransferScreen(elements)) {
|
||
qDebug() << "State: OnTransferScreen -> Looking for 'Enter amount' field or 'Confirm' button.";
|
||
|
||
bool found = false;
|
||
|
||
// Если поле для ввода суммы найдено, вводим сумму
|
||
for (const UiElement &el: elements) {
|
||
if (el.text() == "Сумма" && el.isClickable()) {
|
||
qDebug() << "Found 'Amount' input field";
|
||
inputAmount("1000"); // Пример суммы
|
||
found = true;
|
||
break;
|
||
}
|
||
}
|
||
|
||
if (!found) {
|
||
qWarning() << "'Amount' input field not found!";
|
||
}
|
||
|
||
// После ввода суммы ищем кнопку 'Оплатить' или 'Подтвердить'
|
||
for (const UiElement &el: elements) {
|
||
if (el.text() == "Оплатить" && el.isClickable()) {
|
||
int x = (el.x1() + el.x2()) / 2;
|
||
int y = (el.y1() + el.y2()) / 2;
|
||
AdbUtils::tap(x, y); // Тапаем по кнопке "Оплатить"
|
||
qDebug() << "Payment confirmed.";
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
void tapByText(const std::vector<UiElement>& ui, const std::string& text) {
|
||
for (const auto& el : ui) {
|
||
if (el.text == QString::fromStdString(text)) {
|
||
tap((el.x1 + el.x2) / 2, (el.y1 + el.y2) / 2);
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
|
||
// Тап по полю ввода
|
||
void tapInputField(const std::vector<UiElement>& ui, const std::string& fieldText) {
|
||
for (const auto& el : ui) {
|
||
if (el.text == QString::fromStdString(fieldText)) {
|
||
tap((el.x1 + el.x2) / 2, (el.y1 + el.y2) / 2);
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
|
||
// Тап по кнопке "Далее"
|
||
void tapNextButton(const std::vector<UiElement>& ui) {
|
||
for (const auto& el : ui) {
|
||
if (el.text == "Войти" || el.text == "Далее" || el.text == "Продолжить") {
|
||
tap((el.x1 + el.x2) / 2, (el.y1 + el.y2) / 2);
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
|
||
bool handlePostLoginScreen(const std::vector<UiElement>& ui) {
|
||
for (const auto& el : ui) {
|
||
// Ищем баннеры и модальные окна, которые нужно закрыть
|
||
if (el.text == "Получайте новости с помощью" || el.text == "Не пропускайте специальные предложения и рекомендации") {
|
||
// Тапаем по пустой области
|
||
tap((el.x1 + el.x2) / 2, (el.y1 + el.y2) / 2);
|
||
wait(900);
|
||
return true;
|
||
}
|
||
if (el.text == "Позже" || el.text == "Окей" || el.text == "Закрыть") {
|
||
// Тапаем по кнопке закрытия или кнопке "Позже"
|
||
tap((el.x1 + el.x2) / 2, (el.y1 + el.y2) / 2);
|
||
wait(900);
|
||
return true;
|
||
}
|
||
}
|
||
return false;
|
||
}
|
||
|
||
bool isMainScreen(const std::vector<UiElement>& ui) {
|
||
for (const auto& el : ui) {
|
||
if (el.text == "Счета и карты" || el.text == "Доступно баллов") {
|
||
return true;
|
||
}
|
||
}
|
||
return false;
|
||
}
|
||
|
||
UiElement findCardByLast4Digits(const std::vector<UiElement>& ui, const std::string& last4digits) {
|
||
// Проходим по всем элементам UI, которые парсились из XML
|
||
for (const auto& el : ui) {
|
||
// Используем регулярное выражение для поиска 4 цифр в тексте
|
||
std::smatch m;
|
||
std::string str = el.text.toUtf8().constData(); // Преобразуем QString в std::string
|
||
|
||
// Если в тексте найдено совпадение с регулярным выражением для 4 цифр
|
||
if (std::regex_search(str, m, std::regex(R"(\*\*\s?(\d{4}))"))) {
|
||
// Проверяем, совпадают ли последние 4 цифры с переданными
|
||
if (m[1] == last4digits) {
|
||
// Если нашли совпадение, возвращаем элемент
|
||
return el;
|
||
}
|
||
}
|
||
}
|
||
|
||
// Если карта с такими последними цифрами не найдена, возвращаем пустой элемент
|
||
return {};
|
||
}
|
||
|
||
UiElement findPayButton(const std::vector<UiElement>& ui) {
|
||
for (const auto& el : ui) {
|
||
if (el.text == "Оплатить") {
|
||
return el;
|
||
}
|
||
}
|
||
return {};
|
||
}
|
||
|
||
void Script::stop() {
|
||
m_running = false;
|
||
}
|