bombsquad-plugin-manager/plugin_manager.py

1322 lines
55 KiB
Python

# ba_meta require api 7
import ba
import _ba
import bastd
from bastd.ui import popup
import urllib.request
import json
import os
import asyncio
import re
from typing import Union, Optional
_env = _ba.env()
_uiscale = ba.app.ui.uiscale
INDEX_META = "https://raw.githubusercontent.com/bombsquad-community/mod-manager/main/index.json"
HEADERS = {
"User-Agent": _env["user_agent_string"],
}
PLUGIN_DIRECTORY = _env["python_directory_user"]
REGEXP = {
"plugin_api_version": re.compile(b"(?<=ba_meta require api )(.*)"),
# "plugin_entry_points": re.compile(b"(ba_meta export plugin\n+class )(.*)\("),
"plugin_entry_points": re.compile(b"(ba_meta export .+\n+class )(.*)\("),
}
VERSION = "0.1.1"
GITHUB_REPO_LINK = "https://github.com/bombsquad-community/plugin-manager/"
_CACHE = {}
def setup_config():
is_config_updated = False
if "Community Plugin Manager" not in ba.app.config:
ba.app.config["Community Plugin Manager"] = {}
if "Installed Plugins" not in ba.app.config["Community Plugin Manager"]:
ba.app.config["Community Plugin Manager"]["Installed Plugins"] = {}
is_config_updated = True
for plugin_name in ba.app.config["Community Plugin Manager"]["Installed Plugins"].keys():
plugin = PluginLocal(plugin_name)
if not plugin.is_installed:
del ba.app.config["Community Plugin Manager"]["Installed Plugins"][plugin_name]
is_config_updated = True
if is_config_updated:
ba.app.config.commit()
def send_network_request(request):
return urllib.request.urlopen(request)
async def async_send_network_request(request):
loop = asyncio.get_event_loop()
response = await loop.run_in_executor(None, send_network_request, request)
return response
def stream_network_response_to_file(request, file):
response = urllib.request.urlopen(request)
chunk_size = 16 * 1024
content = b""
with open(file, "wb") as fout:
while True:
chunk = response.read(chunk_size)
if not chunk:
break
fout.write(chunk)
content += chunk
return content
async def async_stream_network_response_to_file(request, file):
loop = asyncio.get_event_loop()
content = await loop.run_in_executor(None, stream_network_response_to_file, request, file)
return content
def play_sound():
ba.playsound(ba.getsound('swish'))
class Category:
def __init__(self, name, base_download_url, meta_url):
self.name = name
self.base_download_url = base_download_url
self.meta_url = meta_url
self.request_headers = HEADERS
self._plugins = _CACHE.get("categories", {}).get(self.name)
async def get_plugins(self):
if self._plugins is None:
request = urllib.request.Request(
self.meta_url,
headers=self.request_headers,
)
response = await async_send_network_request(request)
plugins_info = json.loads(response.read())
self._plugins = ([Plugin(plugin_info, self.base_download_url)
for plugin_info in plugins_info.items()])
self.set_category_plugins_global_cache(self._plugins)
return self._plugins
def set_category_plugins_global_cache(self, plugins):
if "categories" not in _CACHE:
_CACHE["categories"] = {}
_CACHE["categories"][self.name] = plugins
def unset_category_plugins_global_cache(self):
try:
del _CACHE["categories"][self.name]
except KeyError:
pass
def refresh(self):
self._plugins.clear()
self.unset_category_plugins_global_cache()
async def cleanup(self):
self.refresh()
return await self.get_plugins()
class CategoryAll(Category):
def __init__(self, plugins={}):
super().__init__(name="All", base_download_url=None, meta_url=None)
self._plugins = plugins
class PluginLocal:
def __init__(self, name):
"""
Initialize a plugin locally installed on the device.
"""
self.name = name
self.install_path = os.path.join(PLUGIN_DIRECTORY, f"{name}.py")
self._content = None
self._api_version = None
self._entry_points = []
@property
def is_installed(self):
return os.path.isfile(self.install_path)
@property
def is_installed_via_plugin_manager(self):
return self.name in ba.app.config["Community Plugin Manager"]["Installed Plugins"]
def initialize(self):
if self.name not in ba.app.config["Community Plugin Manager"]["Installed Plugins"]:
ba.app.config["Community Plugin Manager"]["Installed Plugins"][self.name] = {}
return self
def uninstall(self):
try:
os.remove(self.install_path)
except FileNotFoundError:
pass
try:
del ba.app.config["Community Plugin Manager"]["Installed Plugins"][self.name]
except KeyError:
pass
else:
ba.app.config.commit()
@property
def version(self):
try:
version = (ba.app.config["Community Plugin Manager"]
["Installed Plugins"][self.name]["version"])
except KeyError:
version = None
return version
def _get_content(self):
with open(self.install_path, "rb") as fin:
return fin.read()
def _set_content(self, content):
with open(self.install_path, "wb") as fout:
fout.write(content)
async def get_content(self):
if self._content is None:
if not self.is_installed:
# TODO: Raise a more fitting exception.
raise TypeError("Plugin is not available locally.")
loop = asyncio.get_event_loop()
self._content = await loop.run_in_executor(None, self._get_content)
return self._content
async def get_api_version(self):
if self._api_version is None:
content = await self.get_content()
self._api_version = REGEXP["plugin_api_version"].search(content).group()
return self._api_version
async def get_entry_points(self):
if not self._entry_points:
content = await self.get_content()
groups = REGEXP["plugin_entry_points"].findall(content)
# Actual entry points are stored in the first index inside the matching groups.
entry_points = tuple(f"{self.name}.{group[1].decode('utf-8')}" for group in groups)
self._entry_points = entry_points
return self._entry_points
@property
def is_enabled(self):
"""
Return True even if a single entry point is enabled.
"""
entry_point_initials = f"{self.name}."
for entry_point, plugin_info in ba.app.config["Plugins"].items():
if entry_point.startswith(entry_point_initials) and plugin_info["enabled"]:
return True
# XXX: The below logic is more accurate but less efficient, since it actually
# reads the local plugin file and parses entry points from it.
# for entry_point in await self.get_entry_points():
# if ba.app.config["Plugins"][entry_point]["enabled"]:
# return True
return False
# XXX: Commenting this out for now, since `enable` and `disable` currently have their
# own separate logic.
# async def _set_status(self, to_enable=True):
# for entry_point in await self.get_entry_points:
# if entry_point not in ba.app.config["Plugins"]:
# ba.app.config["Plugins"][entry_point] = {}
# ba.app.config["Plugins"][entry_point]["enabled"] = to_enable
async def enable(self):
for entry_point in await self.get_entry_points():
if entry_point not in ba.app.config["Plugins"]:
ba.app.config["Plugins"][entry_point] = {}
ba.app.config["Plugins"][entry_point]["enabled"] = True
# await self._set_status(to_enable=True)
ba.screenmessage("Plugin Enabled")
async def disable(self):
entry_point_initials = f"{self.name}."
for entry_point, plugin_info in ba.app.config["Plugins"].items():
if entry_point.startswith(entry_point_initials):
plugin_info["enabled"] = False
# XXX: The below logic is more accurate but less efficient, since it actually
# reads the local plugin file and parses entry points from it.
# await self._set_status(to_enable=False)
ba.screenmessage("Plugin Disabled")
def set_version(self, version):
ba.app.config["Community Plugin Manager"]["Installed Plugins"][self.name]["version"] = version
return self
# def set_entry_points(self):
# if not "entry_points" in ba.app.config["Community Plugin Manager"]["Installed Plugins"][self.name]:
# ba.app.config["Community Plugin Manager"]["Installed Plugins"][self.name]["entry_points"] = []
# for entry_point in await self.get_entry_points():
# ba.app.config["Community Plugin Manager"]["Installed Plugins"][self.name]["entry_points"].append(entry_point)
async def set_content(self, content):
loop = asyncio.get_event_loop()
await loop.run_in_executor(None, self._set_content, content)
self._content = content
return self
async def set_content_from_network_response(self, request):
content = await async_stream_network_response_to_file(request, self.install_path)
self._content = content
return self
def save(self):
ba.app.config.commit()
return self
class PluginVersion:
pass
class Plugin:
def __init__(self, plugin, base_download_url):
"""
Initialize a plugin from network repository.
"""
self.name, self.info = plugin
self.install_path = os.path.join(PLUGIN_DIRECTORY, f"{self.name}.py")
self.download_url = f"{base_download_url}/{self.name}.py"
self._local_plugin = None
def __repr__(self):
return f"<Plugin({self.name})>"
@property
def is_installed(self):
return os.path.isfile(self.install_path)
@property
def latest_version(self):
# TODO: Return an instance of `PluginVersion`.
return next(iter(self.info["versions"]))
async def _download(self):
local_plugin = self.create_local()
await local_plugin.set_content_from_network_response(self.download_url)
local_plugin.set_version(self.latest_version)
local_plugin.save()
return local_plugin
def get_local(self):
if not self.is_installed:
raise ValueError("Plugin is not installed")
if self._local_plugin is None:
self._local_plugin = PluginLocal(self.name)
return self._local_plugin
def create_local(self):
return (
PluginLocal(self.name)
.initialize()
)
async def install(self):
local_plugin = await self._download()
await local_plugin.enable()
ba.screenmessage("Plugin Installed")
async def uninstall(self):
self.get_local().uninstall()
ba.screenmessage("Plugin Uninstalled")
async def update(self):
await self._download()
ba.screenmessage("Plugin Updated")
class PluginWindow(popup.PopupWindow):
def __init__(self, plugin, origin_widget, button_callback=lambda: None):
play_sound()
self.plugin = plugin
self.button_callback = button_callback
b_text_color = (0.75, 0.7, 0.8)
s = 1.1 if _uiscale is ba.UIScale.SMALL else 1.27 if ba.UIScale.MEDIUM else 1.57
width = 360 * s
height = 100 + 100 * s
color = (1, 1, 1)
text_scale = 0.7 * s
self._transition_out = 'out_scale'
transition = 'in_scale'
scale_origin = origin_widget.get_screen_space_center()
self._root_widget = ba.containerwidget(size=(width, height),
parent=_ba.get_special_widget(
'overlay_stack'),
on_outside_click_call=self._ok,
transition=transition,
scale=(2.1 if _uiscale is ba.UIScale.SMALL else 1.5
if _uiscale is ba.UIScale.MEDIUM else 1.0),
scale_origin_stack_offset=scale_origin)
pos = height * 0.8
plugin_title = f"{plugin.name} (v{plugin.latest_version})"
ba.textwidget(parent=self._root_widget,
position=(width * 0.49, pos), size=(0, 0),
h_align='center', v_align='center', text=plugin_title,
scale=text_scale * 1.25, color=color,
maxwidth=width * 0.9)
pos -= 25
# author =
ba.textwidget(parent=self._root_widget,
position=(width * 0.49, pos),
size=(0, 0),
h_align='center',
v_align='center',
text='by ' + plugin.info["authors"][0]["name"],
scale=text_scale * 0.8,
color=color, maxwidth=width * 0.9)
pos -= 35
# status = ba.textwidget(parent=self._root_widget,
# position=(width * 0.49, pos), size=(0, 0),
# h_align='center', v_align='center',
# text=status_text, scale=text_scale * 0.8,
# color=color, maxwidth=width * 0.9)
pos -= 25
# info =
ba.textwidget(parent=self._root_widget,
position=(width * 0.49, pos), size=(0, 0),
h_align='center', v_align='center',
text=plugin.info["description"],
scale=text_scale * 0.6, color=color,
maxwidth=width * 0.95)
b1_color = (0.6, 0.53, 0.63)
b2_color = (0.8, 0.15, 0.35)
b3_color = (0.2, 0.8, 0.3)
pos = height * 0.1
button_size = (80 * s, 40 * s)
if plugin.is_installed:
self.local_plugin = plugin.get_local()
if self.local_plugin.is_enabled:
button1_label = "Disable"
button1_action = self.disable
else:
button1_label = "Enable"
button1_action = self.enable
button2_label = "Uninstall"
button2_action = self.uninstall
has_update = self.local_plugin.version != plugin.latest_version
if has_update:
button3_label = "Update"
button3_action = self.update
else:
button1_label = "Install"
button1_action = self.install
ba.buttonwidget(parent=self._root_widget,
position=(width * 0.1, pos),
size=button_size,
on_activate_call=button1_action,
color=b1_color,
textcolor=b_text_color,
button_type='square',
text_scale=1,
label=button1_label)
if plugin.is_installed:
ba.buttonwidget(parent=self._root_widget,
position=(width * 0.4, pos),
size=button_size,
on_activate_call=button2_action,
color=b2_color,
textcolor=b_text_color,
button_type='square',
text_scale=1,
label=button2_label)
if has_update:
# button3 =
ba.buttonwidget(parent=self._root_widget,
position=(width * 0.7, pos),
size=button_size,
on_activate_call=button3_action,
color=b3_color,
textcolor=b_text_color,
autoselect=True,
button_type='square',
text_scale=1,
label=button3_label)
ba.containerwidget(edit=self._root_widget,
on_cancel_call=self._ok)
# ba.containerwidget(edit=self._root_widget, selected_child=button3)
# ba.containerwidget(edit=self._root_widget, start_button=button3)
def _ok(self) -> None:
play_sound()
ba.containerwidget(edit=self._root_widget, transition='out_scale')
def button(fn):
async def asyncio_handler(fn, self, *args, **kwargs):
await fn(self, *args, **kwargs)
self.button_callback()
def wrapper(self, *args, **kwargs):
self._ok()
if asyncio.iscoroutinefunction(fn):
loop = asyncio.get_event_loop()
loop.create_task(asyncio_handler(fn, self, *args, **kwargs))
else:
fn(self, *args, **kwargs)
self.button_callback()
return wrapper
@button
async def disable(self) -> None:
play_sound()
await self.local_plugin.disable()
@button
async def enable(self) -> None:
play_sound()
await self.local_plugin.enable()
@button
async def install(self):
play_sound()
await self.plugin.install()
@button
async def uninstall(self):
play_sound()
await self.plugin.uninstall()
@button
async def update(self):
play_sound()
await self.plugin.update()
class PluginManager:
def __init__(self):
self.request_headers = HEADERS
self._index = _CACHE.get("index", {})
async def get_index(self):
global _INDEX
if not self._index:
request = urllib.request.Request(
INDEX_META,
headers=self.request_headers,
)
response = await async_send_network_request(request)
self._index = json.loads(response.read())
self.set_index_global_cache(self._index)
return self._index
def cleanup(self):
self._index.clear()
self.unset_index_global_cache()
async def refresh(self):
self.cleanup()
return await self.get_index()
def set_index_global_cache(self, index):
_CACHE["index"] = index
def unset_index_global_cache(self):
try:
del _CACHE["index"]
except KeyError:
pass
async def soft_refresh(self):
pass
class PluginManagerWindow(ba.Window, PluginManager):
def __init__(self, transition: str = "in_right", origin_widget: ba.Widget = None):
PluginManager.__init__(self)
self.categories = {}
self.category_selection_button = None
self.selected_category = None
self.plugins_in_current_view = {}
loop = asyncio.get_event_loop()
loop.create_task(self.plugin_index())
self._width = (490 if _uiscale is ba.UIScale.MEDIUM else 570)
self._height = (500 if _uiscale is ba.UIScale.SMALL
else 380 if _uiscale is ba.UIScale.MEDIUM
else 500)
top_extra = 20 if _uiscale is ba.UIScale.SMALL else 0
if origin_widget:
self._transition_out = "out_scale"
self._scale_origin = origin_widget.get_screen_space_center()
transition = "in_scale"
super().__init__(root_widget=ba.containerwidget(
size=(self._width, self._height + top_extra),
transition=transition,
toolbar_visibility="menu_minimal",
scale_origin_stack_offset=self._scale_origin,
scale=(1.9 if _uiscale is ba.UIScale.SMALL
else 1.5 if _uiscale is ba.UIScale.MEDIUM
else 1.0),
stack_offset=(0, -25) if _uiscale is ba.UIScale.SMALL else (0, 0)
))
back_pos_x = 5 + (10 if _uiscale is ba.UIScale.SMALL else
27 if _uiscale is ba.UIScale.MEDIUM else 68)
back_pos_y = self._height - (115 if _uiscale is ba.UIScale.SMALL else
65 if _uiscale is ba.UIScale.MEDIUM else 50)
self._back_button = back_button = ba.buttonwidget(
parent=self._root_widget,
position=(back_pos_x, back_pos_y),
size=(60, 60),
scale=0.8,
label=ba.charstr(ba.SpecialChar.BACK),
autoselect=True,
button_type='backSmall',
on_activate_call=self._back)
ba.containerwidget(edit=self._root_widget, cancel_button=back_button)
title_pos = self._height - (100 if _uiscale is ba.UIScale.SMALL else
50 if _uiscale is ba.UIScale.MEDIUM else 50)
ba.textwidget(
parent=self._root_widget,
position=(-10, title_pos),
size=(self._width, 25),
text="Community Plugin Manager",
color=ba.app.ui.title_color,
scale=1.05,
h_align="center",
v_align="center",
maxwidth=270,
)
loading_pos_y = self._height - (235 if _uiscale is ba.UIScale.SMALL else
220 if _uiscale is ba.UIScale.MEDIUM else 250)
self._loading_text = ba.textwidget(
parent=self._root_widget,
position=(-5, loading_pos_y),
size=(self._width, 25),
text="Loading...",
color=ba.app.ui.title_color,
scale=0.7,
h_align="center",
v_align="center",
maxwidth=400,
)
scroll_size_x = (400 if _uiscale is ba.UIScale.SMALL else
380 if _uiscale is ba.UIScale.MEDIUM else 420)
scroll_size_y = (225 if _uiscale is ba.UIScale.SMALL else
235 if _uiscale is ba.UIScale.MEDIUM else 335)
scroll_pos_x = (70 if _uiscale is ba.UIScale.SMALL else
40 if _uiscale is ba.UIScale.MEDIUM else 70)
scroll_pos_y = (125 if _uiscale is ba.UIScale.SMALL else
30 if _uiscale is ba.UIScale.MEDIUM else 40)
self._scrollwidget = ba.scrollwidget(parent=self._root_widget,
size=(scroll_size_x, scroll_size_y),
position=(scroll_pos_x, scroll_pos_y))
self._columnwidget = ba.columnwidget(parent=self._scrollwidget,
border=2,
margin=0)
# name = ba.textwidget(parent=self._columnwidget,
# size=(410, 30),
# selectable=True, always_highlight=True,
# color=(1,1,1),
# on_select_call=lambda: None,
# text="ColorScheme",
# on_activate_call=lambda: None,
# h_align='left', v_align='center',
# maxwidth=420)
# v = (self._height - 75) if _uiscale is ba.UIScale.SMALL else (self._height - 59)
# h = 40
# b_textcolor = (0.75, 0.7, 0.8)
# b_color = (0.6, 0.53, 0.63)
# s = (1.0 if _uiscale is ba.UIScale.SMALL else
# 1.27 if _uiscale is ba.UIScale.MEDIUM else 1.57)
# b_size = (90, 60 * s)
# v -= 63 * s
# self.reload_button = ba.buttonwidget(parent=self._root_widget,
# position=(h, v), size=b_size,
# on_activate_call=self._refresh,
# label="Reload List",
# button_type="square",
# color=b_color,
# textcolor=b_textcolor,
# autoselect=True, text_scale=0.7)
# v -= 63 * s
# self.info_button = ba.buttonwidget(parent=self._root_widget,
# position=(h, v), size=b_size,
# on_activate_call=self._get_info,
# label="Mod Info",
# button_type="square", color=b_color,
# textcolor=b_textcolor,
# autoselect=True, text_scale=0.7)
# v -= 63 * s
# self.sort_button = ba.buttonwidget(parent=self._root_widget,
# position=(h, v), size=b_size,
# on_activate_call=self._sort,
# label="Sort\nAlphabetical",
# button_type="square", color=b_color,
# textcolor=b_textcolor,
# text_scale=0.7, autoselect=True)
# v -= 63 * s
# self.settings_button = ba.buttonwidget(parent=self._root_widget,
# position=(h, v), size=b_size,
# on_activate_call=self._settings,
# label="Settings",
# button_type="square",
# color=b_color,
# textcolor=b_textcolor,
# autoselect=True, text_scale=0.7)
# self.column_pos_y = self._height - 75 - self.tab_height
# self._scroll_width = self._width - 180
# self._scroll_height = self._height - 120 - self.tab_height
# self._scrollwidget = ba.scrollwidget(parent=self._root_widget, size=(
# self._scroll_width, self._scroll_height), position=(
# 140, self.column_pos_y - self._scroll_height + 10))
# self._columnwidget = ba.columnwidget(parent=self._scrollwidget,
# border=2, margin=0)
# self._mod_selected = None
# self._refresh()
def _back(self) -> None:
play_sound()
from bastd.ui.settings.allsettings import AllSettingsWindow
ba.containerwidget(edit=self._root_widget,
transition=self._transition_out)
ba.app.ui.set_main_menu_window(
AllSettingsWindow(transition='in_left').get_root_widget())
async def setup_plugin_categories(self, plugin_index):
self.categories["All"] = None
requests = []
for plugin_category in plugin_index["plugin_categories"]:
category = Category(
plugin_category["display_name"],
plugin_category["base_download_url"],
plugin_category["meta"],
)
self.categories[plugin_category["display_name"]] = category
request = category.get_plugins()
requests.append(request)
categories = await asyncio.gather(*requests)
all_plugins = []
for plugins in categories:
all_plugins.extend(plugins)
self.categories["All"] = CategoryAll(plugins=all_plugins)
async def plugin_index(self):
try:
index = await super().get_index()
await asyncio.gather(
self.draw_refresh_icon(),
self.draw_settings_icon(),
self.setup_plugin_categories(index),
)
await self.select_category("All")
await self.draw_search_bar()
except RuntimeError:
# User probably went back before the PluginManagerWindow could finish loading.
pass
except urllib.error.URLError:
ba.textwidget(edit=self._loading_text,
text="Make sure you are connected\n to the Internet and try again.")
else:
self._loading_text.delete()
async def draw_category_selection_button(self, label=None):
# v = (self._height - 75) if _uiscale is ba.UIScale.SMALL else (self._height - 105)
# v = 395
# h = 440
# category_pos_x = 15 + (0 if _uiscale is ba.UIScale.SMALL else
# 17 if _uiscale is ba.UIScale.MEDIUM else 58)
category_pos_x = (330 if _uiscale is ba.UIScale.SMALL else
285 if _uiscale is ba.UIScale.MEDIUM else 350)
category_pos_y = self._height - (145 if _uiscale is ba.UIScale.SMALL else
110 if _uiscale is ba.UIScale.MEDIUM else 110)
# the next 2 lines belong in 1 line
# # s = 1.0 if _uiscale is ba.UIScale.SMALL else
# # 1.27 if _uiscale is ba.UIScale.MEDIUM else 1.57
# s = 1.75
# b_size = (90, 60 * s)
b_size = (140, 30)
b_textcolor = (0.75, 0.7, 0.8)
b_color = (0.6, 0.53, 0.63)
if label is None:
label = self.selected_category
label = f"Category: {label}"
if self.category_selection_button is None:
self.category_selection_button = ba.buttonwidget(parent=self._root_widget,
position=(category_pos_x,
category_pos_y),
size=b_size,
on_activate_call=self.show_categories,
label=label,
button_type="square",
color=b_color,
textcolor=b_textcolor,
autoselect=True,
text_scale=0.6)
else:
self.category_selection_button = ba.buttonwidget(edit=self.category_selection_button,
label=label)
async def draw_search_bar(self):
search_bar_pos_x = (80 if _uiscale is ba.UIScale.SMALL else
55 if _uiscale is ba.UIScale.MEDIUM else 90)
search_bar_pos_y = self._height - (
145 if _uiscale is ba.UIScale.SMALL else
110 if _uiscale is ba.UIScale.MEDIUM else 120)
search_bar_size_x = (250 if _uiscale is ba.UIScale.SMALL else
230 if _uiscale is ba.UIScale.MEDIUM else 250)
search_bar_size_y = (
35 if _uiscale is ba.UIScale.SMALL else
35 if _uiscale is ba.UIScale.MEDIUM else 45)
filter_txt = ba.Lstr(resource='filterText')
self._filter_text = ba.textwidget(parent=self._root_widget,
text="Search",
size=(search_bar_size_x, search_bar_size_y),
position=(search_bar_pos_x, search_bar_pos_y),
h_align='left',
v_align='center',
editable=True,
description=filter_txt)
async def draw_settings_icon(self):
settings_pos_x = (500 if _uiscale is ba.UIScale.SMALL else
440 if _uiscale is ba.UIScale.MEDIUM else 510)
settings_pos_y = (130 if _uiscale is ba.UIScale.SMALL else
60 if _uiscale is ba.UIScale.MEDIUM else 70)
controller_button = ba.buttonwidget(parent=self._root_widget,
autoselect=True,
position=(settings_pos_x, settings_pos_y),
size=(30, 30),
button_type="square",
label="",
on_activate_call=ba.Call(PluginManagerSettingsWindow,
self._root_widget))
ba.imagewidget(parent=self._root_widget,
position=(settings_pos_x, settings_pos_y),
size=(30, 30),
color=(0.8, 0.95, 1),
texture=ba.gettexture("settingsIcon"),
draw_controller=controller_button)
async def draw_refresh_icon(self):
settings_pos_x = (500 if _uiscale is ba.UIScale.SMALL else
440 if _uiscale is ba.UIScale.MEDIUM else 510)
settings_pos_y = (180 if _uiscale is ba.UIScale.SMALL else
105 if _uiscale is ba.UIScale.MEDIUM else 120)
controller_button = ba.buttonwidget(parent=self._root_widget,
autoselect=True,
position=(settings_pos_x, settings_pos_y),
size=(30, 30),
button_type="square",
label="",
on_activate_call=self.refresh)
ba.imagewidget(parent=self._root_widget,
position=(settings_pos_x, settings_pos_y),
size=(30, 30),
color=(0.8, 0.95, 1),
texture=ba.gettexture("replayIcon"),
draw_controller=controller_button)
async def draw_plugin_names(self):
# v = (self._height - 75) if _uiscale is ba.UIScale.SMALL else (self._height - 105)
# h = 440
# next 2 lines belong in 1 line
# # s = 1.0 if _uiscale is ba.UIScale.SMALL else
# # 1.27 if _uiscale is ba.UIScale.MEDIUM else 1.57
# s = 1.75
# # b_size = (90, 60 * s)
# b_size = (150, 30)
# b_textcolor = (0.75, 0.7, 0.8)
# b_color = (0.6, 0.53, 0.63)
for plugin in self._columnwidget.get_children():
plugin.delete()
plugins = await self.categories[self.selected_category].get_plugins()
for plugin in plugins:
self.draw_plugin_name(plugin)
def draw_plugin_name(self, plugin):
if plugin.is_installed:
local_plugin = plugin.get_local()
if local_plugin.is_enabled:
if not local_plugin.is_installed_via_plugin_manager:
color = (0.8, 0.2, 0.2)
elif local_plugin.version == plugin.latest_version:
color = (0, 1, 0)
else:
color = (1, 0.6, 0)
else:
color = (1, 1, 1)
else:
color = (0.5, 0.5, 0.5)
plugin_to_update = self.plugins_in_current_view.get(plugin.name)
if plugin_to_update:
ba.textwidget(edit=plugin_to_update,
color=color)
else:
text_widget = ba.textwidget(parent=self._columnwidget,
size=(410, 30),
selectable=True,
always_highlight=True,
color=color,
# on_select_call=lambda: None,
text=plugin.name,
click_activate=True,
on_activate_call=ba.Call(PluginWindow, plugin,
self._root_widget,
(lambda:
self.draw_plugin_name(plugin))),
h_align='left',
v_align='center',
maxwidth=420)
self.plugins_in_current_view[plugin.name] = text_widget
def show_categories(self):
play_sound()
# On each new entry, change position to y -= 40.
# value = bastd.ui.popup.PopupMenuWindow(
bastd.ui.popup.PopupMenuWindow(
# position=(200, 40),
position=(200, 0),
scale=(2.3 if _uiscale is ba.UIScale.SMALL else
1.65 if _uiscale is ba.UIScale.MEDIUM else 1.23),
choices=self.categories.keys(),
current_choice=self.selected_category,
delegate=self)
async def select_category(self, category):
self.selected_category = category
self.plugins_in_current_view.clear()
await self.draw_category_selection_button(label=category)
await self.draw_plugin_names()
def popup_menu_selected_choice(self, window, choice):
loop = asyncio.get_event_loop()
loop.create_task(self.select_category(choice))
def popup_menu_closing(self, window):
pass
def cleanup(self):
super().cleanup()
self.categories.clear()
for plugin in self._columnwidget.get_children():
plugin.delete()
self.plugins_in_current_view.clear()
async def _refresh(self):
index = await super().refresh()
await self.setup_plugin_categories(index)
await self.select_category(self.selected_category)
def refresh(self):
play_sound()
loop = asyncio.get_event_loop()
loop.create_task(self._refresh())
def soft_refresh(self):
pass
class PluginManagerSettingsWindow(popup.PopupWindow):
def __init__(self, origin_widget):
play_sound()
b_text_color = (0.75, 0.7, 0.8)
s = 1.1 if _uiscale is ba.UIScale.SMALL else 1.27 if ba.UIScale.MEDIUM else 1.57
width = 360 * s
height = 150 + 100 * s
color = (1, 1, 1)
text_scale = 0.7 * s
self._transition_out = 'out_scale'
transition = 'in_scale'
scale_origin = origin_widget.get_screen_space_center()
self._root_widget = ba.containerwidget(size=(width, height),
parent=_ba.get_special_widget(
'overlay_stack'),
on_outside_click_call=self._disappear,
transition=transition,
scale=(2.1 if _uiscale is ba.UIScale.SMALL else 1.5
if _uiscale is ba.UIScale.MEDIUM else 1.0),
scale_origin_stack_offset=scale_origin)
pos = height * 0.9
setting_title = "Settings"
ba.textwidget(parent=self._root_widget,
position=(width * 0.49, pos), size=(0, 0),
h_align='center', v_align='center', text=setting_title,
scale=text_scale * 1.30, color=color,
maxwidth=width * 0.9)
pos -= 40
ba.textwidget(parent=self._root_widget,
position=(width * 0.49, pos),
size=(0, 0),
h_align='center',
v_align='center',
text='Version : ' + VERSION,
scale=text_scale * 1.1,
color=color, maxwidth=width * 0.9)
pos -= 55
ba.textwidget(parent=self._root_widget,
position=(width * 0.49, pos-5), size=(0, 0),
h_align='center', v_align='center',
text='More Updates Coming Soon.',
scale=text_scale, color=color,
maxwidth=width * 0.95)
pos = height * 0.1
button_size = (270 * s, 50 * s)
ba.buttonwidget(parent=self._root_widget,
position=(width * 0.125, pos),
size=button_size,
on_activate_call=self._open,
textcolor=b_text_color,
button_type='square',
text_scale=1,
label='Open Github Repo')
ba.containerwidget(edit=self._root_widget,
on_cancel_call=self._disappear)
# _ba.app.api_version
def _disappear(self) -> None:
play_sound()
ba.containerwidget(edit=self._root_widget, transition='out_scale')
def _open(self) -> None:
ba.open_url(GITHUB_REPO_LINK)
class NewAllSettingsWindow(ba.Window):
def __init__(self,
transition: str = "in_right",
origin_widget: ba.Widget = None):
# pylint: disable=too-many-statements
# pylint: disable=too-many-locals
import threading
# Preload some modules we use in a background thread so we won"t
# have a visual hitch when the user taps them.
threading.Thread(target=self._preload_modules).start()
ba.set_analytics_screen("Settings Window")
scale_origin: Optional[tuple[float, float]]
if origin_widget is not None:
self._transition_out = "out_scale"
scale_origin = origin_widget.get_screen_space_center()
transition = "in_scale"
else:
self._transition_out = "out_right"
scale_origin = None
width = 900 if _uiscale is ba.UIScale.SMALL else 670
x_inset = 75 if _uiscale is ba.UIScale.SMALL else 0
height = 435
self._r = "settingsWindow"
top_extra = 20 if _uiscale is ba.UIScale.SMALL else 0
super().__init__(root_widget=ba.containerwidget(
size=(width, height + top_extra),
transition=transition,
toolbar_visibility="menu_minimal",
scale_origin_stack_offset=scale_origin,
scale=(1.75 if _uiscale is ba.UIScale.SMALL else
1.35 if _uiscale is ba.UIScale.MEDIUM else 1.0),
stack_offset=(0, -8) if _uiscale is ba.UIScale.SMALL else (0, 0)))
if ba.app.ui.use_toolbars and _uiscale is ba.UIScale.SMALL:
self._back_button = None
ba.containerwidget(edit=self._root_widget,
on_cancel_call=self._do_back)
else:
self._back_button = btn = ba.buttonwidget(
parent=self._root_widget,
autoselect=True,
position=(40 + x_inset, height - 55),
size=(130, 60),
scale=0.8,
text_scale=1.2,
label=ba.Lstr(resource="backText"),
button_type="back",
on_activate_call=self._do_back)
ba.containerwidget(edit=self._root_widget, cancel_button=btn)
ba.textwidget(parent=self._root_widget,
position=(0, height - 44),
size=(width, 25),
text=ba.Lstr(resource=self._r + ".titleText"),
color=ba.app.ui.title_color,
h_align="center",
v_align="center",
maxwidth=130)
if self._back_button is not None:
ba.buttonwidget(edit=self._back_button,
button_type="backSmall",
size=(60, 60),
label=ba.charstr(ba.SpecialChar.BACK))
v = height - 80
v -= 145
basew = 200
baseh = 160
x_offs = x_inset + (105 if _uiscale is ba.UIScale.SMALL else
72) - basew # now unused
x_offs2 = x_offs + basew - 7
x_offs3 = x_offs + 2 * (basew - 7)
x_offs4 = x_offs + 3 * (basew - 7)
x_offs5 = x_offs2 + 0.5 * (basew - 7)
x_offs6 = x_offs5 + (basew - 7)
def _b_title(x: float, y: float, button: ba.Widget,
text: Union[str, ba.Lstr]) -> None:
ba.textwidget(parent=self._root_widget,
text=text,
position=(x + basew * 0.47, y + baseh * 0.22),
maxwidth=basew * 0.7, size=(0, 0),
h_align="center",
v_align="center",
draw_controller=button,
color=(0.7, 0.9, 0.7, 1.0))
ctb = self._controllers_button = ba.buttonwidget(parent=self._root_widget,
autoselect=True,
position=(x_offs2, v),
size=(basew, baseh),
button_type="square",
label="",
on_activate_call=self._do_controllers)
if ba.app.ui.use_toolbars and self._back_button is None:
bbtn = _ba.get_special_widget("back_button")
ba.widget(edit=ctb, left_widget=bbtn)
_b_title(x_offs2, v, ctb,
ba.Lstr(resource=self._r + ".controllersText"))
imgw = imgh = 130
ba.imagewidget(parent=self._root_widget,
position=(x_offs2 + basew * 0.49 - imgw * 0.5, v + 35),
size=(imgw, imgh),
texture=ba.gettexture("controllerIcon"),
draw_controller=ctb)
gfxb = self._graphics_button = ba.buttonwidget(parent=self._root_widget,
autoselect=True,
position=(x_offs3, v),
size=(basew, baseh),
button_type="square",
label="",
on_activate_call=self._do_graphics)
if ba.app.ui.use_toolbars:
pbtn = _ba.get_special_widget("party_button")
ba.widget(edit=gfxb, up_widget=pbtn, right_widget=pbtn)
_b_title(x_offs3, v, gfxb, ba.Lstr(resource=self._r + ".graphicsText"))
imgw = imgh = 110
ba.imagewidget(parent=self._root_widget,
position=(x_offs3 + basew * 0.49 - imgw * 0.5, v + 42),
size=(imgw, imgh),
texture=ba.gettexture("graphicsIcon"),
draw_controller=gfxb)
abtn = self._audio_button = ba.buttonwidget(parent=self._root_widget,
autoselect=True,
position=(x_offs4, v),
size=(basew, baseh),
button_type="square",
label="",
on_activate_call=self._do_audio)
_b_title(x_offs4, v, abtn, ba.Lstr(resource=self._r + ".audioText"))
imgw = imgh = 120
ba.imagewidget(parent=self._root_widget,
position=(x_offs4 + basew * 0.49 - imgw * 0.5 + 5, v + 35),
size=(imgw, imgh),
color=(1, 1, 0), texture=ba.gettexture("audioIcon"),
draw_controller=abtn)
v -= (baseh - 5)
avb = self._advanced_button = ba.buttonwidget(parent=self._root_widget,
autoselect=True,
position=(x_offs5, v),
size=(basew, baseh),
button_type="square",
label="",
on_activate_call=self._do_advanced)
_b_title(x_offs5, v, avb, ba.Lstr(resource=self._r + ".advancedText"))
imgw = imgh = 120
ba.imagewidget(parent=self._root_widget,
position=(x_offs5 + basew * 0.49 - imgw * 0.5 + 5,
v + 35),
size=(imgw, imgh),
color=(0.8, 0.95, 1),
texture=ba.gettexture("advancedIcon"),
draw_controller=avb)
mmb = self._modmgr_button = ba.buttonwidget(parent=self._root_widget,
autoselect=True,
position=(x_offs6, v),
size=(basew, baseh),
button_type="square",
label="",
on_activate_call=self._do_modmanager)
_b_title(x_offs6, v, avb, ba.Lstr(value="Plugin Manager"))
imgw = imgh = 120
ba.imagewidget(parent=self._root_widget,
position=(x_offs6 + basew * 0.49 - imgw * 0.5 + 5,
v + 35),
size=(imgw, imgh),
color=(0.8, 0.95, 1),
texture=ba.gettexture("heart"),
draw_controller=mmb)
self._restore_state()
# noinspection PyUnresolvedReferences
@staticmethod
def _preload_modules() -> None:
"""Preload modules we use (called in bg thread)."""
# import bastd.ui.mainmenu as _unused1
# import bastd.ui.settings.controls as _unused2
# import bastd.ui.settings.graphics as _unused3
# import bastd.ui.settings.audio as _unused4
# import bastd.ui.settings.advanced as _unused5
def _do_back(self) -> None:
# pylint: disable=cyclic-import
from bastd.ui.mainmenu import MainMenuWindow
self._save_state()
ba.containerwidget(edit=self._root_widget,
transition=self._transition_out)
ba.app.ui.set_main_menu_window(
MainMenuWindow(transition="in_left").get_root_widget())
def _do_controllers(self) -> None:
# pylint: disable=cyclic-import
from bastd.ui.settings.controls import ControlsSettingsWindow
self._save_state()
ba.containerwidget(edit=self._root_widget, transition="out_left")
ba.app.ui.set_main_menu_window(ControlsSettingsWindow(
origin_widget=self._controllers_button).get_root_widget())
def _do_graphics(self) -> None:
# pylint: disable=cyclic-import
from bastd.ui.settings.graphics import GraphicsSettingsWindow
self._save_state()
ba.containerwidget(edit=self._root_widget, transition="out_left")
ba.app.ui.set_main_menu_window(GraphicsSettingsWindow(
origin_widget=self._graphics_button).get_root_widget())
def _do_audio(self) -> None:
# pylint: disable=cyclic-import
from bastd.ui.settings.audio import AudioSettingsWindow
self._save_state()
ba.containerwidget(edit=self._root_widget, transition="out_left")
ba.app.ui.set_main_menu_window(AudioSettingsWindow(
origin_widget=self._audio_button).get_root_widget())
def _do_advanced(self) -> None:
# pylint: disable=cyclic-import
from bastd.ui.settings.advanced import AdvancedSettingsWindow
self._save_state()
ba.containerwidget(edit=self._root_widget, transition="out_left")
ba.app.ui.set_main_menu_window(AdvancedSettingsWindow(
origin_widget=self._advanced_button).get_root_widget())
def _do_modmanager(self) -> None:
self._save_state()
ba.containerwidget(edit=self._root_widget, transition="out_left")
ba.app.ui.set_main_menu_window(PluginManagerWindow(
origin_widget=self._modmgr_button).get_root_widget())
def _save_state(self) -> None:
try:
sel = self._root_widget.get_selected_child()
if sel == self._controllers_button:
sel_name = "Controllers"
elif sel == self._graphics_button:
sel_name = "Graphics"
elif sel == self._audio_button:
sel_name = "Audio"
elif sel == self._advanced_button:
sel_name = "Advanced"
elif sel == self._modmgr_button:
sel_name = "Mod Manager"
elif sel == self._back_button:
sel_name = "Back"
else:
raise ValueError(f"unrecognized selection \"{sel}\"")
ba.app.ui.window_states[type(self)] = {"sel_name": sel_name}
except Exception:
ba.print_exception(f"Error saving state for {self}.")
def _restore_state(self) -> None:
try:
sel_name = ba.app.ui.window_states.get(type(self),
{}).get("sel_name")
sel: Optional[ba.Widget]
if sel_name == "Controllers":
sel = self._controllers_button
elif sel_name == "Graphics":
sel = self._graphics_button
elif sel_name == "Audio":
sel = self._audio_button
elif sel_name == "Advanced":
sel = self._advanced_button
elif sel_name == "Mod Manager":
sel = self._modmgr_button
elif sel_name == "Back":
sel = self._back_button
else:
sel = self._controllers_button
if sel is not None:
ba.containerwidget(edit=self._root_widget, selected_child=sel)
except Exception:
ba.print_exception(f"Error restoring state for {self}.")
# ba_meta export plugin
class EntryPoint(ba.Plugin):
def on_app_running(self) -> None:
"""Called when the app is being launched."""
setup_config()
from bastd.ui.settings import allsettings
allsettings.AllSettingsWindow = NewAllSettingsWindow
asyncio.set_event_loop(ba._asyncio._asyncio_event_loop)
# loop = asyncio.get_event_loop()
# loop.create_task(do())
# pm = PluginManager()
# pm.plugin_index()
def on_app_pause(self) -> None:
"""Called after pausing game activity."""
print("pause")
def on_app_resume(self) -> None:
"""Called after the game continues."""
print("resume")
def on_app_shutdown(self) -> None:
"""Called before closing the application."""
print("shutdown")
# print(ba.app.config["Community Plugin Manager"])
# with open(_env["config_file_path"], "r") as fin:
# c = fin.read()
# import json
# print(json.loads(c)["Community Plugin Manager"])