2021-08-23 08:22:17 +12:00
|
|
|
from logging import getLogger
|
2023-04-10 23:03:45 +12:00
|
|
|
from typing import List, Callable
|
2021-08-23 08:22:17 +12:00
|
|
|
|
|
|
|
from PyQt5.QtCore import pyqtSignal, QObject
|
2023-04-10 23:03:45 +12:00
|
|
|
from PyQt5.QtWidgets import QApplication
|
2021-08-23 08:22:17 +12:00
|
|
|
|
2023-04-10 23:03:45 +12:00
|
|
|
from rare.components.tabs.store.api.debug import DebugDialog
|
2023-12-13 00:02:07 +13:00
|
|
|
from rare.components.tabs.store.constants import (
|
2021-12-24 22:09:50 +13:00
|
|
|
wishlist_query,
|
|
|
|
search_query,
|
2023-04-10 23:03:45 +12:00
|
|
|
wishlist_add_query,
|
|
|
|
wishlist_remove_query,
|
2021-12-24 22:09:50 +13:00
|
|
|
)
|
2023-04-04 23:46:02 +12:00
|
|
|
from rare.utils.paths import cache_dir
|
2023-12-12 23:41:18 +13:00
|
|
|
from rare.utils.qt_requests import QtRequests
|
2023-04-10 23:03:45 +12:00
|
|
|
from .api.models.query import SearchStoreQuery
|
2023-09-24 06:24:28 +13:00
|
|
|
from .api.models.diesel import DieselProduct
|
2023-04-10 23:03:45 +12:00
|
|
|
from .api.models.response import (
|
|
|
|
ResponseModel,
|
|
|
|
CatalogOfferModel,
|
|
|
|
)
|
2021-08-23 08:22:17 +12:00
|
|
|
|
2024-02-13 05:29:39 +13:00
|
|
|
logger = getLogger("StoreAPI")
|
2023-04-04 23:46:02 +12:00
|
|
|
graphql_url = "https://graphql.epicgames.com/graphql"
|
2021-08-23 08:22:17 +12:00
|
|
|
|
|
|
|
|
2023-04-10 23:03:45 +12:00
|
|
|
DEBUG: Callable[[], bool] = lambda: "--debug" in QApplication.arguments()
|
|
|
|
|
|
|
|
|
2023-09-24 06:24:28 +13:00
|
|
|
class StoreAPI(QObject):
|
2021-08-23 08:22:17 +12:00
|
|
|
update_wishlist = pyqtSignal()
|
|
|
|
|
2023-09-24 06:24:28 +13:00
|
|
|
def __init__(self, token, language: str, country: str, installed):
|
|
|
|
super(StoreAPI, self).__init__()
|
2023-04-04 23:46:02 +12:00
|
|
|
self.token = token
|
|
|
|
self.language_code: str = language
|
|
|
|
self.country_code: str = country
|
2022-02-02 10:29:34 +13:00
|
|
|
self.locale = f"{self.language_code}-{self.country_code}"
|
2023-04-10 23:03:45 +12:00
|
|
|
self.locale = "en-US"
|
2023-12-12 23:41:18 +13:00
|
|
|
self.manager = QtRequests(parent=self)
|
2023-04-04 23:46:02 +12:00
|
|
|
self.authed_manager = QtRequests(token=token, parent=self)
|
|
|
|
self.cached_manager = QtRequests(cache=str(cache_dir().joinpath("store")), parent=self)
|
2021-08-23 08:22:17 +12:00
|
|
|
|
2023-09-24 06:24:28 +13:00
|
|
|
self.installed = installed
|
|
|
|
|
2021-08-23 08:22:17 +12:00
|
|
|
self.browse_active = False
|
|
|
|
self.next_browse_request = tuple(())
|
|
|
|
|
2024-02-13 05:29:39 +13:00
|
|
|
def get_free(self, handle_func: callable):
|
2023-04-04 23:46:02 +12:00
|
|
|
url = "https://store-site-backend-static-ipv4.ak.epicgames.com/freeGamesPromotions"
|
|
|
|
params = {
|
|
|
|
"locale": self.locale,
|
|
|
|
"country": self.country_code,
|
|
|
|
"allowCountries": self.country_code,
|
|
|
|
}
|
2023-04-10 23:03:45 +12:00
|
|
|
self.manager.get(url, lambda data: self.__handle_free_games(data, handle_func), params=params)
|
2021-08-23 08:22:17 +12:00
|
|
|
|
2023-04-10 23:03:45 +12:00
|
|
|
@staticmethod
|
|
|
|
def __handle_free_games(data, handle_func):
|
2021-08-23 08:22:17 +12:00
|
|
|
try:
|
2023-04-10 23:03:45 +12:00
|
|
|
response = ResponseModel.from_dict(data)
|
2024-02-13 05:29:39 +13:00
|
|
|
results: List[CatalogOfferModel] = response.data.catalog.searchStore.elements
|
2023-04-10 23:03:45 +12:00
|
|
|
handle_func(results)
|
|
|
|
except KeyError as e:
|
|
|
|
if DEBUG():
|
|
|
|
raise e
|
2021-11-02 08:17:36 +13:00
|
|
|
logger.error("Free games Api request failed")
|
|
|
|
handle_func(["error", "Key error"])
|
|
|
|
return
|
|
|
|
except Exception as e:
|
2023-04-10 23:03:45 +12:00
|
|
|
if DEBUG():
|
|
|
|
raise e
|
2022-02-02 10:29:34 +13:00
|
|
|
logger.error(f"Free games Api request failed: {e}")
|
2021-11-02 08:17:36 +13:00
|
|
|
handle_func(["error", e])
|
|
|
|
return
|
2021-08-23 08:22:17 +12:00
|
|
|
|
|
|
|
def get_wishlist(self, handle_func):
|
2023-04-04 23:46:02 +12:00
|
|
|
self.authed_manager.post(
|
2021-12-24 22:09:50 +13:00
|
|
|
graphql_url,
|
2023-04-10 23:03:45 +12:00
|
|
|
lambda data: self.__handle_wishlist(data, handle_func),
|
2021-12-24 22:09:50 +13:00
|
|
|
{
|
|
|
|
"query": wishlist_query,
|
|
|
|
"variables": {
|
|
|
|
"country": self.country_code,
|
2023-04-04 23:46:02 +12:00
|
|
|
"locale": self.locale,
|
2023-04-10 23:03:45 +12:00
|
|
|
"withPrice": True,
|
2021-12-24 22:09:50 +13:00
|
|
|
},
|
|
|
|
},
|
|
|
|
)
|
2021-08-23 08:22:17 +12:00
|
|
|
|
2023-04-10 23:03:45 +12:00
|
|
|
@staticmethod
|
|
|
|
def __handle_wishlist(data, handle_func):
|
2021-10-15 10:18:13 +13:00
|
|
|
try:
|
2023-04-10 23:03:45 +12:00
|
|
|
response = ResponseModel.from_dict(data)
|
|
|
|
if response.errors:
|
|
|
|
logger.error(response.errors)
|
2024-02-13 05:29:39 +13:00
|
|
|
handle_func(response.data.wishlist.wishlistItems.elements)
|
2023-04-10 23:03:45 +12:00
|
|
|
except KeyError as e:
|
|
|
|
if DEBUG():
|
|
|
|
raise e
|
2021-11-02 08:17:36 +13:00
|
|
|
logger.error("Free games Api request failed")
|
|
|
|
handle_func(["error", "Key error"])
|
|
|
|
return
|
|
|
|
except Exception as e:
|
2023-04-10 23:03:45 +12:00
|
|
|
if DEBUG():
|
|
|
|
raise e
|
2022-02-02 10:29:34 +13:00
|
|
|
logger.error(f"Free games Api request failed: {e}")
|
2021-11-02 08:17:36 +13:00
|
|
|
handle_func(["error", e])
|
|
|
|
return
|
|
|
|
|
2023-04-10 23:03:45 +12:00
|
|
|
def search_game(self, name, handler):
|
2021-08-23 08:22:17 +12:00
|
|
|
payload = {
|
|
|
|
"query": search_query,
|
2021-12-24 22:09:50 +13:00
|
|
|
"variables": {
|
|
|
|
"category": "games/edition/base|bundles/games|editors|software/edition/base",
|
2023-04-10 23:03:45 +12:00
|
|
|
"count": 20,
|
2021-12-24 22:09:50 +13:00
|
|
|
"country": self.country_code,
|
|
|
|
"keywords": name,
|
|
|
|
"locale": self.locale,
|
|
|
|
"sortDir": "DESC",
|
|
|
|
"allowCountries": self.country_code,
|
|
|
|
"start": 0,
|
|
|
|
"tag": "",
|
|
|
|
"withMapping": False,
|
|
|
|
"withPrice": True,
|
|
|
|
},
|
2021-08-23 08:22:17 +12:00
|
|
|
}
|
|
|
|
|
2023-04-10 23:03:45 +12:00
|
|
|
self.manager.post(graphql_url, lambda data: self.__handle_search(data, handler), payload)
|
2021-08-23 08:22:17 +12:00
|
|
|
|
2023-04-10 23:03:45 +12:00
|
|
|
@staticmethod
|
|
|
|
def __handle_search(data, handler):
|
2021-09-13 09:28:54 +12:00
|
|
|
try:
|
2023-04-10 23:03:45 +12:00
|
|
|
response = ResponseModel.from_dict(data)
|
2024-02-13 05:29:39 +13:00
|
|
|
handler(response.data.catalog.searchStore.elements)
|
2021-09-13 09:28:54 +12:00
|
|
|
except KeyError as e:
|
|
|
|
logger.error(str(e))
|
2023-04-10 23:03:45 +12:00
|
|
|
if DEBUG():
|
|
|
|
raise e
|
|
|
|
handler([])
|
2021-12-23 06:30:14 +13:00
|
|
|
except Exception as e:
|
2022-02-02 10:29:34 +13:00
|
|
|
logger.error(f"Search Api request failed: {e}")
|
2023-04-10 23:03:45 +12:00
|
|
|
if DEBUG():
|
|
|
|
raise e
|
|
|
|
handler([])
|
2021-12-23 06:30:14 +13:00
|
|
|
return
|
2021-08-23 08:22:17 +12:00
|
|
|
|
2023-04-10 23:03:45 +12:00
|
|
|
def browse_games(self, browse_model: SearchStoreQuery, handle_func):
|
2021-08-23 08:22:17 +12:00
|
|
|
if self.browse_active:
|
|
|
|
self.next_browse_request = (browse_model, handle_func)
|
|
|
|
return
|
|
|
|
self.browse_active = True
|
2023-04-04 23:46:02 +12:00
|
|
|
payload = {
|
|
|
|
"query": search_query,
|
2023-04-10 23:03:45 +12:00
|
|
|
"variables": browse_model.to_dict()
|
2023-04-04 23:46:02 +12:00
|
|
|
}
|
2023-09-24 06:24:28 +13:00
|
|
|
# debug = DebugDialog(payload["variables"], None)
|
|
|
|
# debug.exec()
|
2023-04-10 23:03:45 +12:00
|
|
|
self.manager.post(graphql_url, lambda data: self.__handle_browse_games(data, handle_func), payload)
|
2021-08-23 08:22:17 +12:00
|
|
|
|
2023-04-10 23:03:45 +12:00
|
|
|
def __handle_browse_games(self, data, handle_func):
|
2023-09-24 06:24:28 +13:00
|
|
|
# debug = DebugDialog(data, None)
|
|
|
|
# debug.exec()
|
2021-08-23 08:22:17 +12:00
|
|
|
self.browse_active = False
|
2021-12-23 06:30:14 +13:00
|
|
|
if data is None:
|
|
|
|
data = {}
|
2021-08-23 08:22:17 +12:00
|
|
|
if not self.next_browse_request:
|
2021-09-13 09:28:54 +12:00
|
|
|
try:
|
2023-04-10 23:03:45 +12:00
|
|
|
response = ResponseModel.from_dict(data)
|
2024-02-13 05:29:39 +13:00
|
|
|
handle_func(response.data.catalog.searchStore.elements)
|
2021-09-13 09:28:54 +12:00
|
|
|
except KeyError as e:
|
2023-04-10 23:03:45 +12:00
|
|
|
if DEBUG():
|
|
|
|
raise e
|
2021-09-13 09:28:54 +12:00
|
|
|
logger.error(str(e))
|
|
|
|
handle_func([])
|
2021-12-23 06:30:14 +13:00
|
|
|
except Exception as e:
|
2023-04-10 23:03:45 +12:00
|
|
|
if DEBUG():
|
|
|
|
raise e
|
2022-02-02 10:29:34 +13:00
|
|
|
logger.error(f"Browse games Api request failed: {e}")
|
2021-12-23 06:30:14 +13:00
|
|
|
handle_func([])
|
|
|
|
return
|
2021-08-23 08:22:17 +12:00
|
|
|
else:
|
2021-09-19 03:21:43 +12:00
|
|
|
self.browse_games(*self.next_browse_request) # pylint: disable=E1120
|
2021-08-23 08:22:17 +12:00
|
|
|
self.next_browse_request = tuple(())
|
|
|
|
|
2023-09-24 06:24:28 +13:00
|
|
|
def get_game_config_graphql(self, namespace: str, handle_func):
|
|
|
|
payload = {
|
|
|
|
"query": config_query,
|
|
|
|
"variables": {
|
|
|
|
"namespace": namespace
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
def __make_graphql_query(self):
|
|
|
|
pass
|
|
|
|
|
|
|
|
def __make_api_query(self):
|
|
|
|
pass
|
|
|
|
|
|
|
|
def get_game_config_cms(self, slug: str, is_bundle: bool, handle_func):
|
2023-04-10 23:03:45 +12:00
|
|
|
url = "https://store-content.ak.epicgames.com/api"
|
|
|
|
url += f"/{self.locale}/content/{'products' if not is_bundle else 'bundles'}/{slug}"
|
|
|
|
self.manager.get(url, lambda data: self.__handle_get_game(data, handle_func))
|
|
|
|
|
|
|
|
@staticmethod
|
|
|
|
def __handle_get_game(data, handle_func):
|
2023-09-24 06:24:28 +13:00
|
|
|
# debug = DebugDialog(data, None)
|
|
|
|
# debug.exec()
|
2021-12-23 06:30:14 +13:00
|
|
|
try:
|
2023-04-10 23:03:45 +12:00
|
|
|
product = DieselProduct.from_dict(data)
|
|
|
|
handle_func(product)
|
2021-12-23 06:30:14 +13:00
|
|
|
except Exception as e:
|
2023-04-10 23:03:45 +12:00
|
|
|
if DEBUG():
|
|
|
|
raise e
|
2021-12-23 06:30:14 +13:00
|
|
|
logger.error(str(e))
|
2023-01-30 12:56:29 +13:00
|
|
|
# handle_func({})
|
2021-08-23 08:22:17 +12:00
|
|
|
|
2021-11-02 08:17:36 +13:00
|
|
|
# needs a captcha
|
2021-08-23 08:22:17 +12:00
|
|
|
def add_to_wishlist(self, namespace, offer_id, handle_func: callable):
|
|
|
|
payload = {
|
2023-04-10 23:03:45 +12:00
|
|
|
"query": wishlist_add_query,
|
2021-08-23 08:22:17 +12:00
|
|
|
"variables": {
|
|
|
|
"offerId": offer_id,
|
|
|
|
"namespace": namespace,
|
2021-09-14 06:24:09 +12:00
|
|
|
"country": self.country_code,
|
2021-12-24 22:09:50 +13:00
|
|
|
"locale": self.locale,
|
2021-08-23 08:22:17 +12:00
|
|
|
},
|
|
|
|
}
|
2023-04-04 23:46:02 +12:00
|
|
|
self.authed_manager.post(graphql_url, lambda data: self._handle_add_to_wishlist(data, handle_func), payload)
|
2021-08-23 08:22:17 +12:00
|
|
|
|
|
|
|
def _handle_add_to_wishlist(self, data, handle_func):
|
2023-09-24 06:24:28 +13:00
|
|
|
# debug = DebugDialog(data, None)
|
|
|
|
# debug.exec()
|
2021-08-23 08:22:17 +12:00
|
|
|
try:
|
2023-04-10 23:03:45 +12:00
|
|
|
response = ResponseModel.from_dict(data)
|
2024-02-13 05:29:39 +13:00
|
|
|
data = response.data.wishlist.addToWishlist
|
2023-04-10 23:03:45 +12:00
|
|
|
handle_func(data.success)
|
2021-08-23 08:22:17 +12:00
|
|
|
except Exception as e:
|
2023-04-10 23:03:45 +12:00
|
|
|
if DEBUG():
|
|
|
|
raise e
|
2021-08-23 08:22:17 +12:00
|
|
|
logger.error(str(e))
|
|
|
|
handle_func(False)
|
|
|
|
self.update_wishlist.emit()
|
|
|
|
|
|
|
|
def remove_from_wishlist(self, namespace, offer_id, handle_func: callable):
|
|
|
|
payload = {
|
2023-04-10 23:03:45 +12:00
|
|
|
"query": wishlist_remove_query,
|
2021-08-23 08:22:17 +12:00
|
|
|
"variables": {
|
|
|
|
"offerId": offer_id,
|
|
|
|
"namespace": namespace,
|
2021-12-24 22:09:50 +13:00
|
|
|
"operation": "REMOVE",
|
2021-08-23 08:22:17 +12:00
|
|
|
},
|
|
|
|
}
|
2023-04-04 23:46:02 +12:00
|
|
|
self.authed_manager.post(graphql_url, lambda data: self._handle_remove_from_wishlist(data, handle_func),
|
|
|
|
payload)
|
2021-08-23 08:22:17 +12:00
|
|
|
|
|
|
|
def _handle_remove_from_wishlist(self, data, handle_func):
|
2023-09-24 06:24:28 +13:00
|
|
|
# debug = DebugDialog(data, None)
|
|
|
|
# debug.exec()
|
2021-08-23 08:22:17 +12:00
|
|
|
try:
|
2023-04-10 23:03:45 +12:00
|
|
|
response = ResponseModel.from_dict(data)
|
2024-02-13 05:29:39 +13:00
|
|
|
data = response.data.wishlist.removeFromWishlist
|
2023-04-10 23:03:45 +12:00
|
|
|
handle_func(data.success)
|
2021-08-23 08:22:17 +12:00
|
|
|
except Exception as e:
|
2023-04-10 23:03:45 +12:00
|
|
|
if DEBUG():
|
|
|
|
raise e
|
2021-08-23 08:22:17 +12:00
|
|
|
logger.error(str(e))
|
|
|
|
handle_func(False)
|
|
|
|
self.update_wishlist.emit()
|