ZeroNet/update.py
caryoscelus 52bd46f3d7 disable update.py as unsafe
users should use git to update until there's safety mechanisms in place
for upgrading via update zite
2022-01-08 11:05:07 +00:00

127 lines
4.8 KiB
Python

import os
import sys
import json
import re
import shutil
def update():
from Config import config
config.parse(silent=True)
if getattr(sys, 'source_update_dir', False):
if not os.path.isdir(sys.source_update_dir):
os.makedirs(sys.source_update_dir)
source_path = sys.source_update_dir.rstrip("/")
else:
source_path = os.getcwd().rstrip("/")
if config.dist_type.startswith("bundle_linux"):
runtime_path = os.path.normpath(os.path.dirname(sys.executable) + "/../..")
else:
runtime_path = os.path.dirname(sys.executable)
updatesite_path = config.data_dir + "/" + config.updatesite
sites_json = json.load(open(config.data_dir + "/sites.json"))
updatesite_bad_files = sites_json.get(config.updatesite, {}).get("cache", {}).get("bad_files", {})
print(
"Update site path: %s, bad_files: %s, source path: %s, runtime path: %s, dist type: %s" %
(updatesite_path, len(updatesite_bad_files), source_path, runtime_path, config.dist_type)
)
updatesite_content_json = json.load(open(updatesite_path + "/content.json"))
inner_paths = list(updatesite_content_json.get("files", {}).keys())
inner_paths += list(updatesite_content_json.get("files_optional", {}).keys())
# Keep file only in ZeroNet directory
inner_paths = [inner_path for inner_path in inner_paths if re.match("^(core|bundle)", inner_path)]
# Checking plugins
plugins_enabled = []
plugins_disabled = []
if os.path.isdir("%s/plugins" % source_path):
for dir in os.listdir("%s/plugins" % source_path):
if dir.startswith("disabled-"):
plugins_disabled.append(dir.replace("disabled-", ""))
else:
plugins_enabled.append(dir)
print("Plugins enabled:", plugins_enabled, "disabled:", plugins_disabled)
update_paths = {}
for inner_path in inner_paths:
if ".." in inner_path:
continue
inner_path = inner_path.replace("\\", "/").strip("/") # Make sure we have unix path
print(".", end=" ")
if inner_path.startswith("core"):
dest_path = source_path + "/" + re.sub("^core/", "", inner_path)
elif inner_path.startswith(config.dist_type):
dest_path = runtime_path + "/" + re.sub("^bundle[^/]+/", "", inner_path)
else:
continue
if not dest_path:
continue
# Keep plugin disabled/enabled status
match = re.match(re.escape(source_path) + "/plugins/([^/]+)", dest_path)
if match:
plugin_name = match.group(1).replace("disabled-", "")
if plugin_name in plugins_enabled: # Plugin was enabled
dest_path = dest_path.replace("plugins/disabled-" + plugin_name, "plugins/" + plugin_name)
elif plugin_name in plugins_disabled: # Plugin was disabled
dest_path = dest_path.replace("plugins/" + plugin_name, "plugins/disabled-" + plugin_name)
print("P", end=" ")
dest_dir = os.path.dirname(dest_path)
if dest_dir and not os.path.isdir(dest_dir):
os.makedirs(dest_dir)
if dest_dir != dest_path.strip("/"):
update_paths[updatesite_path + "/" + inner_path] = dest_path
num_ok = 0
num_rename = 0
num_error = 0
for path_from, path_to in update_paths.items():
print("-", path_from, "->", path_to)
if not os.path.isfile(path_from):
print("Missing file")
continue
data = open(path_from, "rb").read()
try:
open(path_to, 'wb').write(data)
num_ok += 1
except Exception as err:
try:
print("Error writing: %s. Renaming old file as workaround..." % err)
path_to_tmp = path_to + "-old"
if os.path.isfile(path_to_tmp):
os.unlink(path_to_tmp)
os.rename(path_to, path_to_tmp)
num_rename += 1
open(path_to, 'wb').write(data)
shutil.copymode(path_to_tmp, path_to) # Copy permissions
print("Write done after rename!")
num_ok += 1
except Exception as err:
print("Write error after rename: %s" % err)
num_error += 1
print("* Updated files: %s, renamed: %s, error: %s" % (num_ok, num_rename, num_error))
if __name__ == "__main__":
print('please update zeronet-conservancy via git. usually it can be done via single commnad')
print(' git pull')
print('although it depends on your branches setup')
print('updating through 1update site is not considered safe at the moment')
print('if you really want to use it, edit this file')
if False:
sys.path.insert(0, os.path.join(os.path.dirname(__file__), "src")) # Imports relative to src
update()