mirror of https://gitlab.com/bashrc2/epicyon
915 lines
34 KiB
Python
915 lines
34 KiB
Python
__filename__ = "theme.py"
|
|
__author__ = "Bob Mottram"
|
|
__license__ = "AGPL3+"
|
|
__version__ = "1.2.0"
|
|
__maintainer__ = "Bob Mottram"
|
|
__email__ = "bob@libreserver.org"
|
|
__status__ = "Production"
|
|
__module_group__ = "Web Interface"
|
|
|
|
import os
|
|
from utils import is_account_dir
|
|
from utils import load_json
|
|
from utils import save_json
|
|
from utils import get_image_extensions
|
|
from utils import copytree
|
|
from utils import acct_dir
|
|
from utils import dangerous_svg
|
|
from utils import local_actor_url
|
|
from shutil import copyfile
|
|
from shutil import make_archive
|
|
from shutil import unpack_archive
|
|
from shutil import rmtree
|
|
from content import dangerous_css
|
|
|
|
|
|
def import_theme(base_dir: str, filename: str) -> bool:
|
|
"""Imports a theme
|
|
"""
|
|
if not os.path.isfile(filename):
|
|
return False
|
|
temp_theme_dir = base_dir + '/imports/files'
|
|
if os.path.isdir(temp_theme_dir):
|
|
rmtree(temp_theme_dir, ignore_errors=False, onerror=None)
|
|
os.mkdir(temp_theme_dir)
|
|
unpack_archive(filename, temp_theme_dir, 'zip')
|
|
essential_theme_files = ('name.txt', 'theme.json')
|
|
for theme_file in essential_theme_files:
|
|
if not os.path.isfile(temp_theme_dir + '/' + theme_file):
|
|
print('WARN: ' + theme_file +
|
|
' missing from imported theme')
|
|
return False
|
|
new_theme_name = None
|
|
with open(temp_theme_dir + '/name.txt', 'r') as fp_theme:
|
|
new_theme_name = fp_theme.read().replace('\n', '').replace('\r', '')
|
|
if len(new_theme_name) > 20:
|
|
print('WARN: Imported theme name is too long')
|
|
return False
|
|
if len(new_theme_name) < 2:
|
|
print('WARN: Imported theme name is too short')
|
|
return False
|
|
new_theme_name = new_theme_name.lower()
|
|
forbidden_chars = (
|
|
' ', ';', '/', '\\', '?', '!', '#', '@',
|
|
':', '%', '&', '"', '+', '<', '>', '$'
|
|
)
|
|
for char in forbidden_chars:
|
|
if char in new_theme_name:
|
|
print('WARN: theme name contains forbidden character')
|
|
return False
|
|
if not new_theme_name:
|
|
return False
|
|
|
|
# if the theme name in the default themes list?
|
|
default_themes_filename = base_dir + '/defaultthemes.txt'
|
|
if os.path.isfile(default_themes_filename):
|
|
if new_theme_name.title() + '\n' in \
|
|
open(default_themes_filename).read():
|
|
new_theme_name = new_theme_name + '2'
|
|
|
|
theme_dir = base_dir + '/theme/' + new_theme_name
|
|
if not os.path.isdir(theme_dir):
|
|
os.mkdir(theme_dir)
|
|
copytree(temp_theme_dir, theme_dir)
|
|
if os.path.isdir(temp_theme_dir):
|
|
rmtree(temp_theme_dir, ignore_errors=False, onerror=None)
|
|
if scan_themes_for_scripts(theme_dir):
|
|
rmtree(theme_dir, ignore_errors=False, onerror=None)
|
|
return False
|
|
return os.path.isfile(theme_dir + '/theme.json')
|
|
|
|
|
|
def export_theme(base_dir: str, theme: str) -> bool:
|
|
"""Exports a theme as a zip file
|
|
"""
|
|
theme_dir = base_dir + '/theme/' + theme
|
|
if not os.path.isfile(theme_dir + '/theme.json'):
|
|
return False
|
|
if not os.path.isdir(base_dir + '/exports'):
|
|
os.mkdir(base_dir + '/exports')
|
|
export_filename = base_dir + '/exports/' + theme + '.zip'
|
|
if os.path.isfile(export_filename):
|
|
try:
|
|
os.remove(export_filename)
|
|
except OSError:
|
|
print('EX: export_theme unable to delete ' + str(export_filename))
|
|
try:
|
|
make_archive(base_dir + '/exports/' + theme, 'zip', theme_dir)
|
|
except BaseException:
|
|
print('EX: export_theme unable to archive ' +
|
|
base_dir + '/exports/' + str(theme))
|
|
return os.path.isfile(export_filename)
|
|
|
|
|
|
def _get_theme_files() -> []:
|
|
"""Gets the list of theme style sheets
|
|
"""
|
|
return ('epicyon.css', 'login.css', 'follow.css',
|
|
'suspended.css', 'calendar.css', 'blog.css',
|
|
'options.css', 'search.css', 'links.css',
|
|
'welcome.css', 'graph.css', 'podcast.css')
|
|
|
|
|
|
def is_news_theme_name(base_dir: str, theme_name: str) -> bool:
|
|
"""Returns true if the given theme is a news instance
|
|
"""
|
|
theme_dir = base_dir + '/theme/' + theme_name
|
|
if os.path.isfile(theme_dir + '/is_news_instance'):
|
|
return True
|
|
return False
|
|
|
|
|
|
def get_themes_list(base_dir: str) -> []:
|
|
"""Returns the list of available themes
|
|
Note that these should be capitalized, since they're
|
|
also used to create the web interface dropdown list
|
|
and to lookup function names
|
|
"""
|
|
themes = []
|
|
for _, dirs, _ in os.walk(base_dir + '/theme'):
|
|
for theme_name in dirs:
|
|
if '~' not in theme_name and \
|
|
theme_name != 'icons' and theme_name != 'fonts':
|
|
themes.append(theme_name.title())
|
|
break
|
|
themes.sort()
|
|
print('Themes available: ' + str(themes))
|
|
return themes
|
|
|
|
|
|
def _copy_theme_help_files(base_dir: str, theme_name: str,
|
|
system_language: str) -> None:
|
|
"""Copies any theme specific help files from the welcome subdirectory
|
|
"""
|
|
if not system_language:
|
|
system_language = 'en'
|
|
theme_dir = base_dir + '/theme/' + theme_name + '/welcome'
|
|
if not os.path.isdir(theme_dir):
|
|
theme_dir = base_dir + '/defaultwelcome'
|
|
for _, _, files in os.walk(theme_dir):
|
|
for help_markdown_file in files:
|
|
if not help_markdown_file.endswith('_' + system_language + '.md'):
|
|
continue
|
|
dest_help_markdown_file = \
|
|
help_markdown_file.replace('_' + system_language + '.md',
|
|
'.md')
|
|
if dest_help_markdown_file in ('profile.md', 'final.md'):
|
|
dest_help_markdown_file = 'welcome_' + dest_help_markdown_file
|
|
if os.path.isdir(base_dir + '/accounts'):
|
|
copyfile(theme_dir + '/' + help_markdown_file,
|
|
base_dir + '/accounts/' + dest_help_markdown_file)
|
|
break
|
|
|
|
|
|
def _set_theme_in_config(base_dir: str, name: str) -> bool:
|
|
"""Sets the theme with the given name within config.json
|
|
"""
|
|
config_filename = base_dir + '/config.json'
|
|
if not os.path.isfile(config_filename):
|
|
return False
|
|
config_json = load_json(config_filename, 0)
|
|
if not config_json:
|
|
return False
|
|
config_json['theme'] = name
|
|
return save_json(config_json, config_filename)
|
|
|
|
|
|
def _set_newswire_publish_as_icon(base_dir: str, use_icon: bool) -> bool:
|
|
"""Shows the newswire publish action as an icon or a button
|
|
"""
|
|
config_filename = base_dir + '/config.json'
|
|
if not os.path.isfile(config_filename):
|
|
return False
|
|
config_json = load_json(config_filename, 0)
|
|
if not config_json:
|
|
return False
|
|
config_json['showPublishAsIcon'] = use_icon
|
|
return save_json(config_json, config_filename)
|
|
|
|
|
|
def _set_icons_as_buttons(base_dir: str, use_buttons: bool) -> bool:
|
|
"""Whether to show icons in the header (inbox, outbox, etc)
|
|
as buttons
|
|
"""
|
|
config_filename = base_dir + '/config.json'
|
|
if not os.path.isfile(config_filename):
|
|
return False
|
|
config_json = load_json(config_filename, 0)
|
|
if not config_json:
|
|
return False
|
|
config_json['iconsAsButtons'] = use_buttons
|
|
return save_json(config_json, config_filename)
|
|
|
|
|
|
def _set_rss_icon_at_top(base_dir: str, at_top: bool) -> bool:
|
|
"""Whether to show RSS icon at the top of the timeline
|
|
"""
|
|
config_filename = base_dir + '/config.json'
|
|
if not os.path.isfile(config_filename):
|
|
return False
|
|
config_json = load_json(config_filename, 0)
|
|
if not config_json:
|
|
return False
|
|
config_json['rssIconAtTop'] = at_top
|
|
return save_json(config_json, config_filename)
|
|
|
|
|
|
def _set_publish_button_at_top(base_dir: str, at_top: bool) -> bool:
|
|
"""Whether to show the publish button above the title image
|
|
in the newswire column
|
|
"""
|
|
config_filename = base_dir + '/config.json'
|
|
if not os.path.isfile(config_filename):
|
|
return False
|
|
config_json = load_json(config_filename, 0)
|
|
if not config_json:
|
|
return False
|
|
config_json['publishButtonAtTop'] = at_top
|
|
return save_json(config_json, config_filename)
|
|
|
|
|
|
def _set_full_width_timeline_button_header(base_dir: str,
|
|
full_width: bool) -> bool:
|
|
"""Shows the timeline button header containing inbox, outbox,
|
|
calendar, etc as full width
|
|
"""
|
|
config_filename = base_dir + '/config.json'
|
|
if not os.path.isfile(config_filename):
|
|
return False
|
|
config_json = load_json(config_filename, 0)
|
|
if not config_json:
|
|
return False
|
|
config_json['fullWidthTlButtonHeader'] = full_width
|
|
return save_json(config_json, config_filename)
|
|
|
|
|
|
def get_theme(base_dir: str) -> str:
|
|
"""Gets the current theme name from config.json
|
|
"""
|
|
config_filename = base_dir + '/config.json'
|
|
if os.path.isfile(config_filename):
|
|
config_json = load_json(config_filename, 0)
|
|
if config_json:
|
|
if config_json.get('theme'):
|
|
return config_json['theme']
|
|
return 'default'
|
|
|
|
|
|
def _remove_theme(base_dir: str):
|
|
"""Removes the current theme style sheets
|
|
"""
|
|
theme_files = _get_theme_files()
|
|
for filename in theme_files:
|
|
if not os.path.isfile(base_dir + '/' + filename):
|
|
continue
|
|
try:
|
|
os.remove(base_dir + '/' + filename)
|
|
except OSError:
|
|
print('EX: _remove_theme unable to delete ' +
|
|
base_dir + '/' + filename)
|
|
|
|
|
|
def set_cs_sparam(css: str, param: str, value: str) -> str:
|
|
"""Sets a CSS parameter to a given value
|
|
"""
|
|
# is this just a simple string replacement?
|
|
if ';' in param:
|
|
return css.replace(param, value)
|
|
# color replacement
|
|
if param.startswith('rgba('):
|
|
return css.replace(param, value)
|
|
# if the parameter begins with * then don't prepend --
|
|
once_only = False
|
|
if param.startswith('*'):
|
|
if param.startswith('**'):
|
|
once_only = True
|
|
search_str = param.replace('**', '') + ':'
|
|
else:
|
|
search_str = param.replace('*', '') + ':'
|
|
else:
|
|
search_str = '--' + param + ':'
|
|
if search_str not in css:
|
|
return css
|
|
if once_only:
|
|
sstr = css.split(search_str, 1)
|
|
else:
|
|
sstr = css.split(search_str)
|
|
newcss = ''
|
|
for section_str in sstr:
|
|
# handle font-family which is a variable
|
|
next_section = section_str
|
|
if ';' in next_section:
|
|
next_section = next_section.split(';')[0] + ';'
|
|
if search_str == 'font-family:' and "var(--" in next_section:
|
|
newcss += search_str + ' ' + section_str
|
|
continue
|
|
|
|
if not newcss:
|
|
if section_str:
|
|
newcss = section_str
|
|
else:
|
|
newcss = ' '
|
|
else:
|
|
if ';' in section_str:
|
|
newcss += \
|
|
search_str + ' ' + value + ';' + \
|
|
section_str.split(';', 1)[1]
|
|
else:
|
|
newcss += search_str + ' ' + section_str
|
|
return newcss.strip()
|
|
|
|
|
|
def _set_theme_from_dict(base_dir: str, name: str,
|
|
theme_params: {}, bg_params: {},
|
|
allow_local_network_access: bool) -> None:
|
|
"""Uses a dictionary to set a theme
|
|
"""
|
|
if name:
|
|
_set_theme_in_config(base_dir, name)
|
|
theme_files = _get_theme_files()
|
|
for filename in theme_files:
|
|
# check for custom css within the theme directory
|
|
template_filename = \
|
|
base_dir + '/theme/' + name + '/epicyon-' + filename
|
|
if filename == 'epicyon.css':
|
|
template_filename = \
|
|
base_dir + '/theme/' + name + '/epicyon-profile.css'
|
|
|
|
# Ensure that any custom CSS is mostly harmless.
|
|
# If not then just use the defaults
|
|
if dangerous_css(template_filename, allow_local_network_access) or \
|
|
not os.path.isfile(template_filename):
|
|
# use default css
|
|
template_filename = base_dir + '/epicyon-' + filename
|
|
if filename == 'epicyon.css':
|
|
template_filename = base_dir + '/epicyon-profile.css'
|
|
|
|
if not os.path.isfile(template_filename):
|
|
continue
|
|
|
|
with open(template_filename, 'r') as cssfile:
|
|
css = cssfile.read()
|
|
for param_name, param_value in theme_params.items():
|
|
if param_name == 'newswire-publish-icon':
|
|
if param_value.lower() == 'true':
|
|
_set_newswire_publish_as_icon(base_dir, True)
|
|
else:
|
|
_set_newswire_publish_as_icon(base_dir, False)
|
|
continue
|
|
if param_name == 'full-width-timeline-buttons':
|
|
if param_value.lower() == 'true':
|
|
_set_full_width_timeline_button_header(base_dir, True)
|
|
else:
|
|
_set_full_width_timeline_button_header(base_dir, False)
|
|
continue
|
|
if param_name == 'icons-as-buttons':
|
|
if param_value.lower() == 'true':
|
|
_set_icons_as_buttons(base_dir, True)
|
|
else:
|
|
_set_icons_as_buttons(base_dir, False)
|
|
continue
|
|
if param_name == 'rss-icon-at-top':
|
|
if param_value.lower() == 'true':
|
|
_set_rss_icon_at_top(base_dir, True)
|
|
else:
|
|
_set_rss_icon_at_top(base_dir, False)
|
|
continue
|
|
if param_name == 'publish-button-at-top':
|
|
if param_value.lower() == 'true':
|
|
_set_publish_button_at_top(base_dir, True)
|
|
else:
|
|
_set_publish_button_at_top(base_dir, False)
|
|
continue
|
|
css = set_cs_sparam(css, param_name, param_value)
|
|
filename = base_dir + '/' + filename
|
|
with open(filename, 'w+') as cssfile:
|
|
cssfile.write(css)
|
|
|
|
screen_name = (
|
|
'login', 'follow', 'options', 'search', 'welcome'
|
|
)
|
|
for scr in screen_name:
|
|
if bg_params.get(scr):
|
|
_set_background_format(base_dir, name, scr, bg_params[scr])
|
|
|
|
|
|
def _set_background_format(base_dir: str, name: str,
|
|
background_type: str, extension: str) -> None:
|
|
"""Sets the background file extension
|
|
"""
|
|
if extension == 'jpg':
|
|
return
|
|
css_filename = base_dir + '/' + background_type + '.css'
|
|
if not os.path.isfile(css_filename):
|
|
return
|
|
with open(css_filename, 'r') as cssfile:
|
|
css = cssfile.read()
|
|
css = css.replace('background.jpg', 'background.' + extension)
|
|
with open(css_filename, 'w+') as cssfile2:
|
|
cssfile2.write(css)
|
|
|
|
|
|
def enable_grayscale(base_dir: str) -> None:
|
|
"""Enables grayscale for the current theme
|
|
"""
|
|
theme_files = _get_theme_files()
|
|
for filename in theme_files:
|
|
template_filename = base_dir + '/' + filename
|
|
if not os.path.isfile(template_filename):
|
|
continue
|
|
with open(template_filename, 'r') as cssfile:
|
|
css = cssfile.read()
|
|
if 'grayscale' not in css:
|
|
css = \
|
|
css.replace('body, html {',
|
|
'body, html {\n filter: grayscale(100%);')
|
|
filename = base_dir + '/' + filename
|
|
with open(filename, 'w+') as cssfile:
|
|
cssfile.write(css)
|
|
grayscale_filename = base_dir + '/accounts/.grayscale'
|
|
if not os.path.isfile(grayscale_filename):
|
|
with open(grayscale_filename, 'w+') as grayfile:
|
|
grayfile.write(' ')
|
|
|
|
|
|
def disable_grayscale(base_dir: str) -> None:
|
|
"""Disables grayscale for the current theme
|
|
"""
|
|
theme_files = _get_theme_files()
|
|
for filename in theme_files:
|
|
template_filename = base_dir + '/' + filename
|
|
if not os.path.isfile(template_filename):
|
|
continue
|
|
with open(template_filename, 'r') as cssfile:
|
|
css = cssfile.read()
|
|
if 'grayscale' in css:
|
|
css = \
|
|
css.replace('\n filter: grayscale(100%);', '')
|
|
filename = base_dir + '/' + filename
|
|
with open(filename, 'w+') as cssfile:
|
|
cssfile.write(css)
|
|
grayscale_filename = base_dir + '/accounts/.grayscale'
|
|
if os.path.isfile(grayscale_filename):
|
|
try:
|
|
os.remove(grayscale_filename)
|
|
except OSError:
|
|
print('EX: disable_grayscale unable to delete ' +
|
|
grayscale_filename)
|
|
|
|
|
|
def _set_custom_font(base_dir: str):
|
|
"""Uses a dictionary to set a theme
|
|
"""
|
|
custom_font_ext = None
|
|
custom_font_type = None
|
|
font_extension = {
|
|
'woff': 'woff',
|
|
'woff2': 'woff2',
|
|
'otf': 'opentype',
|
|
'ttf': 'truetype'
|
|
}
|
|
for ext, ext_type in font_extension.items():
|
|
filename = base_dir + '/fonts/custom.' + ext
|
|
if os.path.isfile(filename):
|
|
custom_font_ext = ext
|
|
custom_font_type = ext_type
|
|
if not custom_font_ext:
|
|
return
|
|
|
|
theme_files = _get_theme_files()
|
|
for filename in theme_files:
|
|
template_filename = base_dir + '/' + filename
|
|
if not os.path.isfile(template_filename):
|
|
continue
|
|
with open(template_filename, 'r') as cssfile:
|
|
css = cssfile.read()
|
|
css = \
|
|
set_cs_sparam(css, "*src",
|
|
"url('./fonts/custom." +
|
|
custom_font_ext +
|
|
"') format('" +
|
|
custom_font_type + "')")
|
|
css = set_cs_sparam(css, "*font-family", "'CustomFont'")
|
|
filename = base_dir + '/' + filename
|
|
with open(filename, 'w+') as cssfile:
|
|
cssfile.write(css)
|
|
|
|
|
|
def set_theme_from_designer(base_dir: str, theme_name: str, domain: str,
|
|
theme_params: {},
|
|
allow_local_network_access: bool,
|
|
system_language: str):
|
|
custom_theme_filename = base_dir + '/accounts/theme.json'
|
|
save_json(theme_params, custom_theme_filename)
|
|
set_theme(base_dir, theme_name, domain,
|
|
allow_local_network_access, system_language)
|
|
|
|
|
|
def reset_theme_designer_settings(base_dir: str, theme_name: str, domain: str,
|
|
allow_local_network_access: bool,
|
|
system_language: str) -> None:
|
|
"""Resets the theme designer settings
|
|
"""
|
|
custom_variables_file = base_dir + '/accounts/theme.json'
|
|
if os.path.isfile(custom_variables_file):
|
|
try:
|
|
os.remove(custom_variables_file)
|
|
except OSError:
|
|
print('EX: unable to remove theme designer settings on reset')
|
|
|
|
|
|
def _read_variables_file(base_dir: str, theme_name: str,
|
|
variables_file: str,
|
|
allow_local_network_access: bool) -> None:
|
|
"""Reads variables from a file in the theme directory
|
|
"""
|
|
theme_params = load_json(variables_file, 0)
|
|
if not theme_params:
|
|
return
|
|
|
|
# set custom theme parameters
|
|
custom_variables_file = base_dir + '/accounts/theme.json'
|
|
if os.path.isfile(custom_variables_file):
|
|
custom_theme_params = load_json(custom_variables_file, 0)
|
|
if custom_theme_params:
|
|
for variable_name, value in custom_theme_params.items():
|
|
theme_params[variable_name] = value
|
|
|
|
bg_params = {
|
|
"login": "jpg",
|
|
"follow": "jpg",
|
|
"options": "jpg",
|
|
"search": "jpg"
|
|
}
|
|
_set_theme_from_dict(base_dir, theme_name, theme_params, bg_params,
|
|
allow_local_network_access)
|
|
|
|
|
|
def _set_theme_default(base_dir: str, allow_local_network_access: bool):
|
|
name = 'default'
|
|
_remove_theme(base_dir)
|
|
_set_theme_in_config(base_dir, name)
|
|
bg_params = {
|
|
"login": "jpg",
|
|
"follow": "jpg",
|
|
"options": "jpg",
|
|
"search": "jpg"
|
|
}
|
|
theme_params = {
|
|
"newswire-publish-icon": True,
|
|
"full-width-timeline-buttons": False,
|
|
"icons-as-buttons": False,
|
|
"rss-icon-at-top": True,
|
|
"publish-button-at-top": False,
|
|
"banner-height": "20vh",
|
|
"banner-height-mobile": "10vh",
|
|
"search-banner-height-mobile": "15vh"
|
|
}
|
|
_set_theme_from_dict(base_dir, name, theme_params, bg_params,
|
|
allow_local_network_access)
|
|
|
|
|
|
def _set_theme_fonts(base_dir: str, theme_name: str) -> None:
|
|
"""Adds custom theme fonts
|
|
"""
|
|
theme_name_lower = theme_name.lower()
|
|
fonts_dir = base_dir + '/fonts'
|
|
theme_fonts_dir = \
|
|
base_dir + '/theme/' + theme_name_lower + '/fonts'
|
|
if not os.path.isdir(theme_fonts_dir):
|
|
return
|
|
for _, _, files in os.walk(theme_fonts_dir):
|
|
for filename in files:
|
|
if filename.endswith('.woff2') or \
|
|
filename.endswith('.woff') or \
|
|
filename.endswith('.ttf') or \
|
|
filename.endswith('.otf'):
|
|
dest_filename = fonts_dir + '/' + filename
|
|
if os.path.isfile(dest_filename):
|
|
# font already exists in the destination location
|
|
continue
|
|
copyfile(theme_fonts_dir + '/' + filename,
|
|
dest_filename)
|
|
break
|
|
|
|
|
|
def get_text_mode_banner(base_dir: str) -> str:
|
|
"""Returns the banner used for shell browsers, like Lynx
|
|
"""
|
|
text_mode_banner_filename = base_dir + '/accounts/banner.txt'
|
|
if os.path.isfile(text_mode_banner_filename):
|
|
with open(text_mode_banner_filename, 'r') as fp_text:
|
|
banner_str = fp_text.read()
|
|
if banner_str:
|
|
return banner_str.replace('\n', '<br>')
|
|
return None
|
|
|
|
|
|
def get_text_mode_logo(base_dir: str) -> str:
|
|
"""Returns the login screen logo used for shell browsers, like Lynx
|
|
"""
|
|
text_mode_logo_filename = base_dir + '/accounts/logo.txt'
|
|
if not os.path.isfile(text_mode_logo_filename):
|
|
text_mode_logo_filename = base_dir + '/img/logo.txt'
|
|
|
|
with open(text_mode_logo_filename, 'r') as fp_text:
|
|
logo_str = fp_text.read()
|
|
if logo_str:
|
|
return logo_str.replace('\n', '<br>')
|
|
return None
|
|
|
|
|
|
def _set_text_mode_theme(base_dir: str, name: str) -> None:
|
|
# set the text mode logo which appears on the login screen
|
|
# in browsers such as Lynx
|
|
text_mode_logo_filename = \
|
|
base_dir + '/theme/' + name + '/logo.txt'
|
|
if os.path.isfile(text_mode_logo_filename):
|
|
try:
|
|
copyfile(text_mode_logo_filename,
|
|
base_dir + '/accounts/logo.txt')
|
|
except OSError:
|
|
print('EX: _set_text_mode_theme unable to copy ' +
|
|
text_mode_logo_filename + ' ' +
|
|
base_dir + '/accounts/logo.txt')
|
|
else:
|
|
try:
|
|
copyfile(base_dir + '/img/logo.txt',
|
|
base_dir + '/accounts/logo.txt')
|
|
except OSError:
|
|
print('EX: _set_text_mode_theme unable to copy ' +
|
|
base_dir + '/img/logo.txt ' +
|
|
base_dir + '/accounts/logo.txt')
|
|
|
|
# set the text mode banner which appears in browsers such as Lynx
|
|
text_mode_banner_filename = \
|
|
base_dir + '/theme/' + name + '/banner.txt'
|
|
if os.path.isfile(base_dir + '/accounts/banner.txt'):
|
|
try:
|
|
os.remove(base_dir + '/accounts/banner.txt')
|
|
except OSError:
|
|
print('EX: _set_text_mode_theme unable to delete ' +
|
|
base_dir + '/accounts/banner.txt')
|
|
if os.path.isfile(text_mode_banner_filename):
|
|
try:
|
|
copyfile(text_mode_banner_filename,
|
|
base_dir + '/accounts/banner.txt')
|
|
except OSError:
|
|
print('EX: _set_text_mode_theme unable to copy ' +
|
|
text_mode_banner_filename + ' ' +
|
|
base_dir + '/accounts/banner.txt')
|
|
|
|
|
|
def _set_theme_images(base_dir: str, name: str) -> None:
|
|
"""Changes the profile background image
|
|
and banner to the defaults
|
|
"""
|
|
theme_name_lower = name.lower()
|
|
|
|
profile_image_filename = \
|
|
base_dir + '/theme/' + theme_name_lower + '/image.png'
|
|
banner_filename = \
|
|
base_dir + '/theme/' + theme_name_lower + '/banner.png'
|
|
search_banner_filename = \
|
|
base_dir + '/theme/' + theme_name_lower + '/search_banner.png'
|
|
left_col_image_filename = \
|
|
base_dir + '/theme/' + theme_name_lower + '/left_col_image.png'
|
|
right_col_image_filename = \
|
|
base_dir + '/theme/' + theme_name_lower + '/right_col_image.png'
|
|
|
|
_set_text_mode_theme(base_dir, theme_name_lower)
|
|
|
|
background_names = ('login', 'shares', 'delete', 'follow',
|
|
'options', 'block', 'search', 'calendar',
|
|
'welcome')
|
|
extensions = get_image_extensions()
|
|
|
|
for _, dirs, _ in os.walk(base_dir + '/accounts'):
|
|
for acct in dirs:
|
|
if not is_account_dir(acct):
|
|
continue
|
|
account_dir = os.path.join(base_dir + '/accounts', acct)
|
|
|
|
for background_type in background_names:
|
|
for ext in extensions:
|
|
if theme_name_lower == 'default':
|
|
background_image_filename = \
|
|
base_dir + '/theme/default/' + \
|
|
background_type + '_background.' + ext
|
|
else:
|
|
background_image_filename = \
|
|
base_dir + '/theme/' + theme_name_lower + '/' + \
|
|
background_type + '_background' + '.' + ext
|
|
|
|
if os.path.isfile(background_image_filename):
|
|
try:
|
|
copyfile(background_image_filename,
|
|
base_dir + '/accounts/' +
|
|
background_type + '-background.' + ext)
|
|
continue
|
|
except OSError:
|
|
print('EX: _set_theme_images unable to copy ' +
|
|
background_image_filename)
|
|
# background image was not found
|
|
# so remove any existing file
|
|
if os.path.isfile(base_dir + '/accounts/' +
|
|
background_type + '-background.' + ext):
|
|
try:
|
|
os.remove(base_dir + '/accounts/' +
|
|
background_type + '-background.' + ext)
|
|
except OSError:
|
|
print('EX: _set_theme_images unable to delete ' +
|
|
base_dir + '/accounts/' +
|
|
background_type + '-background.' + ext)
|
|
|
|
if os.path.isfile(profile_image_filename) and \
|
|
os.path.isfile(banner_filename):
|
|
try:
|
|
copyfile(profile_image_filename,
|
|
account_dir + '/image.png')
|
|
except OSError:
|
|
print('EX: _set_theme_images unable to copy ' +
|
|
profile_image_filename)
|
|
|
|
try:
|
|
copyfile(banner_filename,
|
|
account_dir + '/banner.png')
|
|
except OSError:
|
|
print('EX: _set_theme_images unable to copy ' +
|
|
banner_filename)
|
|
|
|
try:
|
|
if os.path.isfile(search_banner_filename):
|
|
copyfile(search_banner_filename,
|
|
account_dir + '/search_banner.png')
|
|
except OSError:
|
|
print('EX: _set_theme_images unable to copy ' +
|
|
search_banner_filename)
|
|
|
|
try:
|
|
if os.path.isfile(left_col_image_filename):
|
|
copyfile(left_col_image_filename,
|
|
account_dir + '/left_col_image.png')
|
|
elif os.path.isfile(account_dir +
|
|
'/left_col_image.png'):
|
|
try:
|
|
os.remove(account_dir + '/left_col_image.png')
|
|
except OSError:
|
|
print('EX: _set_theme_images unable to delete ' +
|
|
account_dir + '/left_col_image.png')
|
|
except OSError:
|
|
print('EX: _set_theme_images unable to copy ' +
|
|
left_col_image_filename)
|
|
|
|
try:
|
|
if os.path.isfile(right_col_image_filename):
|
|
copyfile(right_col_image_filename,
|
|
account_dir + '/right_col_image.png')
|
|
else:
|
|
if os.path.isfile(account_dir +
|
|
'/right_col_image.png'):
|
|
try:
|
|
os.remove(account_dir + '/right_col_image.png')
|
|
except OSError:
|
|
print('EX: _set_theme_images ' +
|
|
'unable to delete ' +
|
|
account_dir + '/right_col_image.png')
|
|
except OSError:
|
|
print('EX: _set_theme_images unable to copy ' +
|
|
right_col_image_filename)
|
|
break
|
|
|
|
|
|
def set_news_avatar(base_dir: str, name: str,
|
|
http_prefix: str,
|
|
domain: str, domain_full: str) -> None:
|
|
"""Sets the avatar for the news account
|
|
"""
|
|
nickname = 'news'
|
|
new_filename = base_dir + '/theme/' + name + '/icons/avatar_news.png'
|
|
if not os.path.isfile(new_filename):
|
|
new_filename = base_dir + '/theme/default/icons/avatar_news.png'
|
|
if not os.path.isfile(new_filename):
|
|
return
|
|
avatar_filename = \
|
|
local_actor_url(http_prefix, domain_full, nickname) + '.png'
|
|
avatar_filename = avatar_filename.replace('/', '-')
|
|
filename = base_dir + '/cache/avatars/' + avatar_filename
|
|
|
|
if os.path.isfile(filename):
|
|
try:
|
|
os.remove(filename)
|
|
except OSError:
|
|
print('EX: set_news_avatar unable to delete ' + filename)
|
|
if os.path.isdir(base_dir + '/cache/avatars'):
|
|
copyfile(new_filename, filename)
|
|
account_dir = acct_dir(base_dir, nickname, domain)
|
|
copyfile(new_filename, account_dir + '/avatar.png')
|
|
|
|
|
|
def _set_clear_cache_flag(base_dir: str) -> None:
|
|
"""Sets a flag which can be used by an external system
|
|
(eg. a script in a cron job) to clear the browser cache
|
|
"""
|
|
if not os.path.isdir(base_dir + '/accounts'):
|
|
return
|
|
flag_filename = base_dir + '/accounts/.clear_cache'
|
|
with open(flag_filename, 'w+') as fp_flag:
|
|
fp_flag.write('\n')
|
|
|
|
|
|
def set_theme(base_dir: str, name: str, domain: str,
|
|
allow_local_network_access: bool, system_language: str) -> bool:
|
|
"""Sets the theme with the given name as the current theme
|
|
"""
|
|
result = False
|
|
|
|
prev_theme_name = get_theme(base_dir)
|
|
|
|
# if the theme has changed then remove any custom settings
|
|
if prev_theme_name != name:
|
|
reset_theme_designer_settings(base_dir, name, domain,
|
|
allow_local_network_access,
|
|
system_language)
|
|
|
|
_remove_theme(base_dir)
|
|
|
|
themes = get_themes_list(base_dir)
|
|
for theme_name in themes:
|
|
theme_name_lower = theme_name.lower()
|
|
if name == theme_name_lower:
|
|
allow_access = allow_local_network_access
|
|
try:
|
|
globals()['set_theme' + theme_name](base_dir, allow_access)
|
|
except BaseException:
|
|
print('EX: set_theme unable to set theme ' + theme_name)
|
|
|
|
if prev_theme_name:
|
|
if prev_theme_name.lower() != theme_name_lower:
|
|
# change the banner and profile image
|
|
# to the default for the theme
|
|
_set_theme_images(base_dir, name)
|
|
_set_theme_fonts(base_dir, name)
|
|
result = True
|
|
|
|
if not result:
|
|
# default
|
|
_set_theme_default(base_dir, allow_local_network_access)
|
|
result = True
|
|
|
|
variables_file = base_dir + '/theme/' + name + '/theme.json'
|
|
if os.path.isfile(variables_file):
|
|
_read_variables_file(base_dir, name, variables_file,
|
|
allow_local_network_access)
|
|
|
|
_set_custom_font(base_dir)
|
|
|
|
# set the news avatar
|
|
news_avatar_theme_filename = \
|
|
base_dir + '/theme/' + name + '/icons/avatar_news.png'
|
|
if os.path.isdir(base_dir + '/accounts/news@' + domain):
|
|
if os.path.isfile(news_avatar_theme_filename):
|
|
news_avatar_filename = \
|
|
base_dir + '/accounts/news@' + domain + '/avatar.png'
|
|
copyfile(news_avatar_theme_filename, news_avatar_filename)
|
|
|
|
grayscale_filename = base_dir + '/accounts/.grayscale'
|
|
if os.path.isfile(grayscale_filename):
|
|
enable_grayscale(base_dir)
|
|
else:
|
|
disable_grayscale(base_dir)
|
|
|
|
_copy_theme_help_files(base_dir, name, system_language)
|
|
_set_theme_in_config(base_dir, name)
|
|
_set_clear_cache_flag(base_dir)
|
|
return result
|
|
|
|
|
|
def update_default_themes_list(base_dir: str) -> None:
|
|
"""Recreates the list of default themes
|
|
"""
|
|
theme_names = get_themes_list(base_dir)
|
|
default_themes_filename = base_dir + '/defaultthemes.txt'
|
|
with open(default_themes_filename, 'w+') as fp_def:
|
|
for name in theme_names:
|
|
fp_def.write(name + '\n')
|
|
|
|
|
|
def scan_themes_for_scripts(base_dir: str) -> bool:
|
|
"""Scans the theme directory for any svg files containing scripts
|
|
"""
|
|
for subdir, _, files in os.walk(base_dir + '/theme'):
|
|
for fname in files:
|
|
if not fname.endswith('.svg'):
|
|
continue
|
|
svg_filename = os.path.join(subdir, fname)
|
|
content = ''
|
|
with open(svg_filename, 'r') as fp_svg:
|
|
content = fp_svg.read()
|
|
svg_dangerous = dangerous_svg(content, False)
|
|
if svg_dangerous:
|
|
print('svg file contains script: ' + svg_filename)
|
|
return True
|
|
# deliberately no break - should resursively scan
|
|
return False
|