90 lines
3.4 KiB
Python
90 lines
3.4 KiB
Python
import io
|
|
import os
|
|
import re
|
|
import urllib
|
|
|
|
from Plugin import PluginManager
|
|
from Config import config
|
|
from Translate import Translate
|
|
|
|
plugin_dir = os.path.dirname(__file__)
|
|
|
|
if "_" not in locals():
|
|
_ = Translate(plugin_dir + "/languages/")
|
|
|
|
|
|
@PluginManager.registerTo("UiRequest")
|
|
class UiFileManagerPlugin(object):
|
|
def actionWrapper(self, path, extra_headers=None):
|
|
match = re.match("/list/(.*?)(/.*|)$", path)
|
|
if not match:
|
|
return super().actionWrapper(path, extra_headers)
|
|
|
|
if not extra_headers:
|
|
extra_headers = {}
|
|
|
|
request_address, inner_path = match.groups()
|
|
|
|
script_nonce = self.getScriptNonce()
|
|
|
|
self.sendHeader(extra_headers=extra_headers, script_nonce=script_nonce)
|
|
|
|
site = self.server.site_manager.need(request_address)
|
|
|
|
if not site:
|
|
return super().actionWrapper(path, extra_headers)
|
|
|
|
request_params = urllib.parse.urlencode(
|
|
{"address": site.address, "site": request_address, "inner_path": inner_path.strip("/")}
|
|
)
|
|
|
|
is_content_loaded = "content.json" in site.content_manager.contents
|
|
|
|
return iter([super().renderWrapper(
|
|
site, path, "uimedia/plugins/uifilemanager/list.html?%s" % request_params,
|
|
"List", extra_headers, show_loadingscreen=not is_content_loaded, script_nonce=script_nonce
|
|
)])
|
|
|
|
def actionUiMedia(self, path, *args, **kwargs):
|
|
if path.startswith("/uimedia/plugins/uifilemanager/"):
|
|
file_path = path.replace("/uimedia/plugins/uifilemanager/", plugin_dir + "/media/")
|
|
if config.debug and (file_path.endswith("all.js") or file_path.endswith("all.css")):
|
|
# If debugging merge *.css to all.css and *.js to all.js
|
|
from Debug import DebugMedia
|
|
DebugMedia.merge(file_path)
|
|
|
|
if file_path.endswith("js"):
|
|
data = _.translateData(open(file_path).read(), mode="js").encode("utf8")
|
|
elif file_path.endswith("html"):
|
|
if self.get.get("address"):
|
|
site = self.server.site_manager.need(self.get.get("address"))
|
|
if "content.json" not in site.content_manager.contents:
|
|
site.needFile("content.json")
|
|
data = _.translateData(open(file_path).read(), mode="html").encode("utf8")
|
|
else:
|
|
data = open(file_path, "rb").read()
|
|
|
|
return self.actionFile(file_path, file_obj=io.BytesIO(data), file_size=len(data))
|
|
else:
|
|
return super().actionUiMedia(path)
|
|
|
|
def error404(self, path=""):
|
|
if not path.endswith("index.html") and not path.endswith("/"):
|
|
return super().error404(path)
|
|
|
|
path_parts = self.parsePath(path)
|
|
if not path_parts:
|
|
return super().error404(path)
|
|
|
|
site = self.server.site_manager.get(path_parts["request_address"])
|
|
|
|
if not site or not site.content_manager.contents.get("content.json"):
|
|
return super().error404(path)
|
|
|
|
if path_parts["inner_path"] in site.content_manager.contents.get("content.json").get("files", {}):
|
|
return super().error404(path)
|
|
|
|
self.sendHeader(200)
|
|
path_redirect = "/list" + re.sub("^/media/", "/", path)
|
|
self.log.debug("Index.html not found: %s, redirecting to: %s" % (path, path_redirect))
|
|
return self.formatRedirect(path_redirect)
|