440 lines
12 KiB
Python
440 lines
12 KiB
Python
# Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors
|
|
# MIT License. See license.txt
|
|
|
|
from __future__ import print_function, unicode_literals
|
|
|
|
import os
|
|
import re
|
|
import json
|
|
import shutil
|
|
import warnings
|
|
import tempfile
|
|
from distutils.spawn import find_executable
|
|
|
|
import frappe
|
|
from frappe.utils.minify import JavascriptMinify
|
|
|
|
import click
|
|
from requests import get
|
|
from six import iteritems, text_type
|
|
from six.moves.urllib.parse import urlparse
|
|
|
|
|
|
timestamps = {}
|
|
app_paths = None
|
|
sites_path = os.path.abspath(os.getcwd())
|
|
|
|
|
|
def download_file(url, prefix):
|
|
filename = urlparse(url).path.split("/")[-1]
|
|
local_filename = os.path.join(prefix, filename)
|
|
with get(url, stream=True, allow_redirects=True) as r:
|
|
r.raise_for_status()
|
|
with open(local_filename, "wb") as f:
|
|
for chunk in r.iter_content(chunk_size=8192):
|
|
f.write(chunk)
|
|
return local_filename
|
|
|
|
|
|
def build_missing_files():
|
|
# check which files dont exist yet from the build.json and tell build.js to build only those!
|
|
missing_assets = []
|
|
current_asset_files = []
|
|
|
|
for type in ["css", "js"]:
|
|
current_asset_files.extend(
|
|
[
|
|
"{0}/{1}".format(type, name)
|
|
for name in os.listdir(os.path.join(sites_path, "assets", type))
|
|
]
|
|
)
|
|
|
|
with open(os.path.join(sites_path, "assets", "frappe", "build.json")) as f:
|
|
all_asset_files = json.load(f).keys()
|
|
|
|
for asset in all_asset_files:
|
|
if asset.replace("concat:", "") not in current_asset_files:
|
|
missing_assets.append(asset)
|
|
|
|
if missing_assets:
|
|
from subprocess import check_call
|
|
from shlex import split
|
|
|
|
click.secho("\nBuilding missing assets...\n", fg="yellow")
|
|
command = split(
|
|
"node rollup/build.js --files {0} --no-concat".format(",".join(missing_assets))
|
|
)
|
|
check_call(command, cwd=os.path.join("..", "apps", "frappe"))
|
|
|
|
|
|
def get_assets_link(frappe_head):
|
|
from subprocess import getoutput
|
|
from requests import head
|
|
|
|
tag = getoutput(
|
|
"cd ../apps/frappe && git show-ref --tags -d | grep %s | sed -e 's,.*"
|
|
" refs/tags/,,' -e 's/\^{}//'"
|
|
% frappe_head
|
|
)
|
|
|
|
if tag:
|
|
# if tag exists, download assets from github release
|
|
url = "https://github.com/frappe/frappe/releases/download/{0}/assets.tar.gz".format(tag)
|
|
else:
|
|
url = "http://assets.frappeframework.com/{0}.tar.gz".format(frappe_head)
|
|
|
|
if not head(url):
|
|
raise ValueError("URL {0} doesn't exist".format(url))
|
|
|
|
return url
|
|
|
|
|
|
def download_frappe_assets(verbose=True):
|
|
"""Downloads and sets up Frappe assets if they exist based on the current
|
|
commit HEAD.
|
|
Returns True if correctly setup else returns False.
|
|
"""
|
|
from simple_chalk import green
|
|
from subprocess import getoutput
|
|
from tempfile import mkdtemp
|
|
|
|
assets_setup = False
|
|
frappe_head = getoutput("cd ../apps/frappe && git rev-parse HEAD")
|
|
|
|
if frappe_head:
|
|
try:
|
|
url = get_assets_link(frappe_head)
|
|
click.secho("Retreiving assets...", fg="yellow")
|
|
prefix = mkdtemp(prefix="frappe-assets-", suffix=frappe_head)
|
|
assets_archive = download_file(url, prefix)
|
|
print("\n{0} Downloaded Frappe assets from {1}".format(green('✔'), url))
|
|
|
|
if assets_archive:
|
|
import tarfile
|
|
|
|
click.secho("\nExtracting assets...\n", fg="yellow")
|
|
with tarfile.open(assets_archive) as tar:
|
|
for file in tar:
|
|
if not file.isdir():
|
|
dest = "." + file.name.replace("./frappe-bench/sites", "")
|
|
show = dest.replace("./assets/", "")
|
|
tar.makefile(file, dest)
|
|
print("{0} Restored {1}".format(green('✔'), show))
|
|
|
|
build_missing_files()
|
|
return True
|
|
else:
|
|
raise
|
|
except Exception:
|
|
# TODO: log traceback in bench.log
|
|
click.secho("An Error occurred while downloading assets...", fg="red")
|
|
assets_setup = False
|
|
finally:
|
|
try:
|
|
shutil.rmtree(os.path.dirname(assets_archive))
|
|
except Exception:
|
|
pass
|
|
|
|
return assets_setup
|
|
|
|
|
|
def symlink(target, link_name, overwrite=False):
|
|
"""
|
|
Create a symbolic link named link_name pointing to target.
|
|
If link_name exists then FileExistsError is raised, unless overwrite=True.
|
|
When trying to overwrite a directory, IsADirectoryError is raised.
|
|
|
|
Source: https://stackoverflow.com/a/55742015/10309266
|
|
"""
|
|
|
|
if not overwrite:
|
|
return os.symlink(target, link_name)
|
|
|
|
# os.replace() may fail if files are on different filesystems
|
|
link_dir = os.path.dirname(link_name)
|
|
|
|
# Create link to target with temporary filename
|
|
while True:
|
|
temp_link_name = tempfile.mktemp(dir=link_dir)
|
|
|
|
# os.* functions mimic as closely as possible system functions
|
|
# The POSIX symlink() returns EEXIST if link_name already exists
|
|
# https://pubs.opengroup.org/onlinepubs/9699919799/functions/symlink.html
|
|
try:
|
|
os.symlink(target, temp_link_name)
|
|
break
|
|
except FileExistsError:
|
|
pass
|
|
|
|
# Replace link_name with temp_link_name
|
|
try:
|
|
# Pre-empt os.replace on a directory with a nicer message
|
|
if os.path.isdir(link_name):
|
|
raise IsADirectoryError("Cannot symlink over existing directory: '{}'".format(link_name))
|
|
try:
|
|
os.replace(temp_link_name, link_name)
|
|
except AttributeError:
|
|
os.renames(temp_link_name, link_name)
|
|
except:
|
|
if os.path.islink(temp_link_name):
|
|
os.remove(temp_link_name)
|
|
raise
|
|
|
|
|
|
def setup():
|
|
global app_paths
|
|
pymodules = []
|
|
for app in frappe.get_all_apps(True):
|
|
try:
|
|
pymodules.append(frappe.get_module(app))
|
|
except ImportError:
|
|
pass
|
|
app_paths = [os.path.dirname(pymodule.__file__) for pymodule in pymodules]
|
|
|
|
|
|
def get_node_pacman():
|
|
exec_ = find_executable("yarn")
|
|
if exec_:
|
|
return exec_
|
|
raise ValueError("Yarn not found")
|
|
|
|
|
|
def bundle(no_compress, app=None, make_copy=False, restore=False, verbose=False, skip_frappe=False):
|
|
"""concat / minify js files"""
|
|
setup()
|
|
make_asset_dirs(make_copy=make_copy, restore=restore)
|
|
|
|
pacman = get_node_pacman()
|
|
mode = "build" if no_compress else "production"
|
|
command = "{pacman} run {mode}".format(pacman=pacman, mode=mode)
|
|
|
|
if app:
|
|
command += " --app {app}".format(app=app)
|
|
|
|
if skip_frappe:
|
|
command += " --skip_frappe"
|
|
|
|
frappe_app_path = os.path.abspath(os.path.join(app_paths[0], ".."))
|
|
check_yarn()
|
|
frappe.commands.popen(command, cwd=frappe_app_path)
|
|
|
|
|
|
def watch(no_compress):
|
|
"""watch and rebuild if necessary"""
|
|
setup()
|
|
|
|
pacman = get_node_pacman()
|
|
|
|
frappe_app_path = os.path.abspath(os.path.join(app_paths[0], ".."))
|
|
check_yarn()
|
|
frappe_app_path = frappe.get_app_path("frappe", "..")
|
|
frappe.commands.popen("{pacman} run watch".format(pacman=pacman), cwd=frappe_app_path)
|
|
|
|
|
|
def check_yarn():
|
|
if not find_executable("yarn"):
|
|
print("Please install yarn using below command and try again.\nnpm install -g yarn")
|
|
|
|
|
|
def make_asset_dirs(make_copy=False, restore=False):
|
|
# don't even think of making assets_path absolute - rm -rf ahead.
|
|
assets_path = os.path.join(frappe.local.sites_path, "assets")
|
|
|
|
for dir_path in [os.path.join(assets_path, "js"), os.path.join(assets_path, "css")]:
|
|
if not os.path.exists(dir_path):
|
|
os.makedirs(dir_path)
|
|
|
|
for app_name in frappe.get_all_apps(True):
|
|
pymodule = frappe.get_module(app_name)
|
|
app_base_path = os.path.abspath(os.path.dirname(pymodule.__file__))
|
|
|
|
symlinks = []
|
|
app_public_path = os.path.join(app_base_path, "public")
|
|
# app/public > assets/app
|
|
symlinks.append([app_public_path, os.path.join(assets_path, app_name)])
|
|
# app/node_modules > assets/app/node_modules
|
|
if os.path.exists(os.path.abspath(app_public_path)):
|
|
symlinks.append(
|
|
[
|
|
os.path.join(app_base_path, "..", "node_modules"),
|
|
os.path.join(assets_path, app_name, "node_modules"),
|
|
]
|
|
)
|
|
|
|
app_doc_path = None
|
|
if os.path.isdir(os.path.join(app_base_path, "docs")):
|
|
app_doc_path = os.path.join(app_base_path, "docs")
|
|
|
|
elif os.path.isdir(os.path.join(app_base_path, "www", "docs")):
|
|
app_doc_path = os.path.join(app_base_path, "www", "docs")
|
|
|
|
if app_doc_path:
|
|
symlinks.append([app_doc_path, os.path.join(assets_path, app_name + "_docs")])
|
|
|
|
for source, target in symlinks:
|
|
source = os.path.abspath(source)
|
|
if os.path.exists(source):
|
|
if restore:
|
|
if os.path.exists(target):
|
|
if os.path.islink(target):
|
|
os.unlink(target)
|
|
else:
|
|
shutil.rmtree(target)
|
|
shutil.copytree(source, target)
|
|
elif make_copy:
|
|
if os.path.exists(target):
|
|
warnings.warn("Target {target} already exists.".format(target=target))
|
|
else:
|
|
shutil.copytree(source, target)
|
|
else:
|
|
if os.path.exists(target):
|
|
if os.path.islink(target):
|
|
os.unlink(target)
|
|
else:
|
|
shutil.rmtree(target)
|
|
try:
|
|
symlink(source, target, overwrite=True)
|
|
except OSError:
|
|
print("Cannot link {} to {}".format(source, target))
|
|
else:
|
|
# warnings.warn('Source {source} does not exist.'.format(source = source))
|
|
pass
|
|
|
|
|
|
def build(no_compress=False, verbose=False):
|
|
assets_path = os.path.join(frappe.local.sites_path, "assets")
|
|
|
|
for target, sources in iteritems(get_build_maps()):
|
|
pack(os.path.join(assets_path, target), sources, no_compress, verbose)
|
|
|
|
|
|
def get_build_maps():
|
|
"""get all build.jsons with absolute paths"""
|
|
# framework js and css files
|
|
|
|
build_maps = {}
|
|
for app_path in app_paths:
|
|
path = os.path.join(app_path, "public", "build.json")
|
|
if os.path.exists(path):
|
|
with open(path) as f:
|
|
try:
|
|
for target, sources in iteritems(json.loads(f.read())):
|
|
# update app path
|
|
source_paths = []
|
|
for source in sources:
|
|
if isinstance(source, list):
|
|
s = frappe.get_pymodule_path(source[0], *source[1].split("/"))
|
|
else:
|
|
s = os.path.join(app_path, source)
|
|
source_paths.append(s)
|
|
|
|
build_maps[target] = source_paths
|
|
except ValueError as e:
|
|
print(path)
|
|
print("JSON syntax error {0}".format(str(e)))
|
|
return build_maps
|
|
|
|
|
|
def pack(target, sources, no_compress, verbose):
|
|
from six import StringIO
|
|
|
|
outtype, outtxt = target.split(".")[-1], ""
|
|
jsm = JavascriptMinify()
|
|
|
|
for f in sources:
|
|
suffix = None
|
|
if ":" in f:
|
|
f, suffix = f.split(":")
|
|
if not os.path.exists(f) or os.path.isdir(f):
|
|
print("did not find " + f)
|
|
continue
|
|
timestamps[f] = os.path.getmtime(f)
|
|
try:
|
|
with open(f, "r") as sourcefile:
|
|
data = text_type(sourcefile.read(), "utf-8", errors="ignore")
|
|
|
|
extn = f.rsplit(".", 1)[1]
|
|
|
|
if (
|
|
outtype == "js"
|
|
and extn == "js"
|
|
and (not no_compress)
|
|
and suffix != "concat"
|
|
and (".min." not in f)
|
|
):
|
|
tmpin, tmpout = StringIO(data.encode("utf-8")), StringIO()
|
|
jsm.minify(tmpin, tmpout)
|
|
minified = tmpout.getvalue()
|
|
if minified:
|
|
outtxt += text_type(minified or "", "utf-8").strip("\n") + ";"
|
|
|
|
if verbose:
|
|
print("{0}: {1}k".format(f, int(len(minified) / 1024)))
|
|
elif outtype == "js" and extn == "html":
|
|
# add to frappe.templates
|
|
outtxt += html_to_js_template(f, data)
|
|
else:
|
|
outtxt += "\n/*\n *\t%s\n */" % f
|
|
outtxt += "\n" + data + "\n"
|
|
|
|
except Exception:
|
|
print("--Error in:" + f + "--")
|
|
print(frappe.get_traceback())
|
|
|
|
with open(target, "w") as f:
|
|
f.write(outtxt.encode("utf-8"))
|
|
|
|
print("Wrote %s - %sk" % (target, str(int(os.path.getsize(target) / 1024))))
|
|
|
|
|
|
def html_to_js_template(path, content):
|
|
"""returns HTML template content as Javascript code, adding it to `frappe.templates`"""
|
|
return """frappe.templates["{key}"] = '{content}';\n""".format(
|
|
key=path.rsplit("/", 1)[-1][:-5], content=scrub_html_template(content))
|
|
|
|
|
|
def scrub_html_template(content):
|
|
"""Returns HTML content with removed whitespace and comments"""
|
|
# remove whitespace to a single space
|
|
content = re.sub("\s+", " ", content)
|
|
|
|
# strip comments
|
|
content = re.sub("(<!--.*?-->)", "", content)
|
|
|
|
return content.replace("'", "\'")
|
|
|
|
|
|
def files_dirty():
|
|
for target, sources in iteritems(get_build_maps()):
|
|
for f in sources:
|
|
if ":" in f:
|
|
f, suffix = f.split(":")
|
|
if not os.path.exists(f) or os.path.isdir(f):
|
|
continue
|
|
if os.path.getmtime(f) != timestamps.get(f):
|
|
print(f + " dirty")
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
|
|
def compile_less():
|
|
if not find_executable("lessc"):
|
|
return
|
|
|
|
for path in app_paths:
|
|
less_path = os.path.join(path, "public", "less")
|
|
if os.path.exists(less_path):
|
|
for fname in os.listdir(less_path):
|
|
if fname.endswith(".less") and fname != "variables.less":
|
|
fpath = os.path.join(less_path, fname)
|
|
mtime = os.path.getmtime(fpath)
|
|
if fpath in timestamps and mtime == timestamps[fpath]:
|
|
continue
|
|
|
|
timestamps[fpath] = mtime
|
|
|
|
print("compiling {0}".format(fpath))
|
|
|
|
css_path = os.path.join(path, "public", "css", fname.rsplit(".", 1)[0] + ".css")
|
|
os.system("lessc {0} > {1}".format(fpath, css_path))
|