commands.py 91 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616
  1. """JupyterLab command handler"""
  2. # Copyright (c) Jupyter Development Team.
  3. # Distributed under the terms of the Modified BSD License.
  4. import contextlib
  5. import errno
  6. import hashlib
  7. import itertools
  8. import json
  9. import logging
  10. import os
  11. import os.path as osp
  12. import re
  13. import shutil
  14. import site
  15. import stat
  16. import subprocess
  17. import sys
  18. import tarfile
  19. from copy import deepcopy
  20. from dataclasses import dataclass
  21. from glob import glob
  22. from pathlib import Path
  23. from tempfile import TemporaryDirectory
  24. from threading import Event
  25. from typing import Optional
  26. from urllib.error import URLError
  27. from urllib.request import Request, quote, urljoin, urlopen
  28. from jupyter_core.paths import jupyter_config_dir
  29. from jupyter_server.extension.serverextension import GREEN_ENABLED, GREEN_OK, RED_DISABLED, RED_X
  30. from jupyterlab_server.config import (
  31. get_allowed_levels,
  32. get_federated_extensions,
  33. get_package_url,
  34. get_page_config,
  35. get_static_page_config,
  36. write_page_config,
  37. )
  38. from jupyterlab_server.process import Process, WatchHelper, list2cmdline, which
  39. from packaging.version import Version
  40. from traitlets import Bool, HasTraits, Instance, List, Unicode, default
  41. from jupyterlab._version import __version__
  42. from jupyterlab.coreconfig import CoreConfig
  43. from jupyterlab.jlpmapp import HERE, YARN_PATH
  44. from jupyterlab.semver import Range, gt, gte, lt, lte, make_semver
  45. # The regex for expecting the webpack output.
  46. WEBPACK_EXPECT = re.compile(r".*theme-light-extension/style/theme.css")
  47. # The repo root directory
  48. REPO_ROOT = osp.abspath(osp.join(HERE, ".."))
  49. # The dev mode directory.
  50. DEV_DIR = osp.join(REPO_ROOT, "dev_mode")
  51. # If we are pinning the package, rename it `pin@<alias>`
  52. PIN_PREFIX = "pin@"
  53. # Default Yarn registry used in default yarn.lock
  54. YARN_DEFAULT_REGISTRY = "https://registry.yarnpkg.com"
  55. class ProgressProcess(Process):
  56. def __init__(self, cmd, logger=None, cwd=None, kill_event=None, env=None):
  57. """Start a subprocess that can be run asynchronously.
  58. Parameters
  59. ----------
  60. cmd: list
  61. The command to run.
  62. logger: :class:`~logger.Logger`, optional
  63. The logger instance.
  64. cwd: string, optional
  65. The cwd of the process.
  66. kill_event: :class:`~threading.Event`, optional
  67. An event used to kill the process operation.
  68. env: dict, optional
  69. The environment for the process.
  70. """
  71. if not isinstance(cmd, (list, tuple)):
  72. msg = "Command must be given as a list"
  73. raise ValueError(msg)
  74. if kill_event and kill_event.is_set():
  75. msg = "Process aborted"
  76. raise ValueError(msg)
  77. self.logger = _ensure_logger(logger)
  78. self._last_line = ""
  79. self.cmd = cmd
  80. self.logger.debug(f"> {list2cmdline(cmd)}")
  81. self.proc = self._create_process(
  82. cwd=cwd,
  83. env=env,
  84. stderr=subprocess.STDOUT,
  85. stdout=subprocess.PIPE,
  86. universal_newlines=True,
  87. encoding="utf-8",
  88. )
  89. self._kill_event = kill_event or Event()
  90. Process._procs.add(self)
  91. def wait(self):
  92. cache = []
  93. proc = self.proc
  94. kill_event = self._kill_event
  95. spinner = itertools.cycle(["-", "\\", "|", "/"])
  96. while proc.poll() is None:
  97. sys.stdout.write(next(spinner)) # write the next character
  98. sys.stdout.flush() # flush stdout buffer (actual character display)
  99. sys.stdout.write("\b")
  100. if kill_event.is_set():
  101. self.terminate()
  102. msg = "Process was aborted"
  103. raise ValueError(msg)
  104. try:
  105. out, _ = proc.communicate(timeout=0.1)
  106. cache.append(out)
  107. except subprocess.TimeoutExpired:
  108. continue
  109. self.logger.debug("\n".join(cache))
  110. sys.stdout.flush()
  111. return self.terminate()
  112. def pjoin(*args):
  113. """Join paths to create a real path."""
  114. return osp.abspath(osp.join(*args))
  115. def get_user_settings_dir():
  116. """Get the configured JupyterLab user settings directory."""
  117. settings_dir = os.environ.get("JUPYTERLAB_SETTINGS_DIR")
  118. settings_dir = settings_dir or pjoin(jupyter_config_dir(), "lab", "user-settings")
  119. return osp.abspath(settings_dir)
  120. def get_workspaces_dir():
  121. """Get the configured JupyterLab workspaces directory."""
  122. workspaces_dir = os.environ.get("JUPYTERLAB_WORKSPACES_DIR")
  123. workspaces_dir = workspaces_dir or pjoin(jupyter_config_dir(), "lab", "workspaces")
  124. return osp.abspath(workspaces_dir)
  125. def get_app_dir():
  126. """Get the configured JupyterLab app directory."""
  127. # Default to the override environment variable.
  128. if os.environ.get("JUPYTERLAB_DIR"):
  129. # We must resolve the path to get the canonical case of the path for
  130. # case-sensitive systems
  131. return str(Path(os.environ["JUPYTERLAB_DIR"]).resolve())
  132. # Use the default locations for data_files.
  133. app_dir = pjoin(sys.prefix, "share", "jupyter", "lab")
  134. # Check for a user level install.
  135. # Ensure that USER_BASE is defined
  136. if hasattr(site, "getuserbase"):
  137. site.getuserbase()
  138. userbase = getattr(site, "USER_BASE", None)
  139. if HERE.startswith(userbase) and not app_dir.startswith(userbase):
  140. app_dir = pjoin(userbase, "share", "jupyter", "lab")
  141. # Check for a system install in '/usr/local/share'.
  142. elif (
  143. sys.prefix.startswith("/usr")
  144. and not osp.exists(app_dir)
  145. and osp.exists("/usr/local/share/jupyter/lab")
  146. ):
  147. app_dir = "/usr/local/share/jupyter/lab"
  148. # Check for a path relative to the site-packages directory, e.g.,
  149. # `<prefix>/lib/python3.13/site-packages/jupyterlab/../../../..` This is
  150. # useful for cases where the the `jupyterlab` module is outside the current
  151. # Python environment, which can occur via various Python path manipulations.
  152. elif not osp.exists(app_dir):
  153. maybe_app_dir = pjoin(
  154. osp.dirname(osp.dirname(osp.dirname(osp.dirname(HERE)))),
  155. "share",
  156. "jupyter",
  157. "lab",
  158. )
  159. if osp.exists(maybe_app_dir):
  160. app_dir = maybe_app_dir
  161. # We must resolve the path to get the canonical case of the path for
  162. # case-sensitive systems
  163. return str(Path(app_dir).resolve())
  164. def dedupe_yarn(path, logger=None):
  165. """`yarn-deduplicate` with the `fewer` strategy to minimize total
  166. packages installed in a given staging directory
  167. This means a extension (or dependency) _could_ cause a downgrade of an
  168. version expected at publication time, but core should aggressively set
  169. pins above, for example, known-bad versions
  170. """
  171. had_dupes = (
  172. ProgressProcess(
  173. [
  174. "node",
  175. YARN_PATH,
  176. "dlx",
  177. "yarn-berry-deduplicate",
  178. "-s",
  179. "fewerHighest",
  180. "--fail",
  181. ],
  182. cwd=path,
  183. logger=logger,
  184. ).wait()
  185. != 0
  186. )
  187. if had_dupes:
  188. yarn_proc = ProgressProcess(["node", YARN_PATH], cwd=path, logger=logger)
  189. yarn_proc.wait()
  190. def ensure_node_modules(cwd, logger=None):
  191. """Ensure that node_modules is up to date.
  192. Returns true if the node_modules was updated.
  193. """
  194. logger = _ensure_logger(logger)
  195. yarn_proc = ProgressProcess(
  196. ["node", YARN_PATH, "--immutable", "--immutable-cache"], cwd=cwd, logger=logger
  197. )
  198. ret = yarn_proc.wait()
  199. # Update node_modules if needed.
  200. if ret != 0:
  201. yarn_proc = ProgressProcess(["node", YARN_PATH], cwd=cwd, logger=logger)
  202. yarn_proc.wait()
  203. dedupe_yarn(REPO_ROOT, logger)
  204. return ret != 0
  205. def ensure_dev(logger=None):
  206. """Ensure that the dev assets are available."""
  207. logger = _ensure_logger(logger)
  208. target = pjoin(DEV_DIR, "static")
  209. # Determine whether to build.
  210. if ensure_node_modules(REPO_ROOT, logger) or not osp.exists(target):
  211. yarn_proc = ProgressProcess(["node", YARN_PATH, "build"], cwd=REPO_ROOT, logger=logger)
  212. yarn_proc.wait()
  213. def ensure_core(logger=None):
  214. """Ensure that the core assets are available."""
  215. staging = pjoin(HERE, "staging")
  216. logger = _ensure_logger(logger)
  217. # Determine whether to build.
  218. target = pjoin(HERE, "static", "index.html")
  219. if not osp.exists(target):
  220. ensure_node_modules(staging, logger)
  221. yarn_proc = ProgressProcess(["node", YARN_PATH, "build"], cwd=staging, logger=logger)
  222. yarn_proc.wait()
  223. def ensure_app(app_dir):
  224. """Ensure that an application directory is available.
  225. If it does not exist, return a list of messages to prompt the user.
  226. """
  227. if osp.exists(pjoin(app_dir, "static", "index.html")):
  228. return
  229. msgs = [
  230. f'JupyterLab application assets not found in "{app_dir}"',
  231. "Please run `jlpm run build:core` then `jupyter lab build` ",
  232. "or use a different app directory",
  233. ]
  234. return msgs
  235. def watch_packages(logger=None):
  236. """Run watch mode for the source packages.
  237. Parameters
  238. ----------
  239. logger: :class:`~logger.Logger`, optional
  240. The logger instance.
  241. Returns
  242. -------
  243. A list of `WatchHelper` objects.
  244. """
  245. logger = _ensure_logger(logger)
  246. ensure_node_modules(REPO_ROOT, logger)
  247. ts_dir = osp.abspath(osp.join(REPO_ROOT, "packages", "metapackage"))
  248. # Run typescript watch and wait for the string indicating it is done.
  249. ts_regex = r".* Found 0 errors\. Watching for file changes\."
  250. ts_proc = WatchHelper(
  251. ["node", YARN_PATH, "run", "watch"], cwd=ts_dir, logger=logger, startup_regex=ts_regex
  252. )
  253. return [ts_proc]
  254. def watch_dev(logger=None):
  255. """Run watch mode in a given directory.
  256. Parameters
  257. ----------
  258. logger: :class:`~logger.Logger`, optional
  259. The logger instance.
  260. Returns
  261. -------
  262. A list of `WatchHelper` objects.
  263. """
  264. logger = _ensure_logger(logger)
  265. package_procs = watch_packages(logger)
  266. # Run webpack watch and wait for compilation.
  267. wp_proc = WatchHelper(
  268. ["node", YARN_PATH, "run", "watch"],
  269. cwd=DEV_DIR,
  270. logger=logger,
  271. startup_regex=WEBPACK_EXPECT,
  272. )
  273. return [*package_procs, wp_proc]
  274. class AppOptions(HasTraits):
  275. """Options object for build system"""
  276. def __init__(self, logger=None, core_config=None, **kwargs):
  277. if core_config is not None:
  278. kwargs["core_config"] = core_config
  279. if logger is not None:
  280. kwargs["logger"] = logger
  281. # use the default if app_dir is empty
  282. if "app_dir" in kwargs and not kwargs["app_dir"]:
  283. kwargs.pop("app_dir")
  284. super().__init__(**kwargs)
  285. app_dir = Unicode(help="The application directory")
  286. use_sys_dir = Bool(
  287. True,
  288. help=("Whether to shadow the default app_dir if that is set to a non-default value"),
  289. )
  290. logger = Instance(logging.Logger, help="The logger to use")
  291. core_config = Instance(CoreConfig, help="Configuration for core data")
  292. kill_event = Instance(Event, args=(), help="Event for aborting call")
  293. labextensions_path = List(
  294. Unicode(), help="The paths to look in for prebuilt JupyterLab extensions"
  295. )
  296. registry = Unicode(help="NPM packages registry URL")
  297. splice_source = Bool(False, help="Splice source packages into app directory.")
  298. skip_full_build_check = Bool(
  299. False,
  300. help=(
  301. "If true, perform only a quick check that the lab build is up to date."
  302. " If false, perform a thorough check, which verifies extension contents."
  303. ),
  304. )
  305. verbose = Bool(False, help="Increase verbosity level.")
  306. @default("logger")
  307. def _default_logger(self):
  308. return logging.getLogger("jupyterlab")
  309. # These defaults need to be federated to pick up
  310. # any changes to env vars:
  311. @default("app_dir")
  312. def _default_app_dir(self):
  313. return get_app_dir()
  314. @default("core_config")
  315. def _default_core_config(self):
  316. return CoreConfig()
  317. @default("registry")
  318. def _default_registry(self):
  319. config = _yarn_config(self.logger)["yarn config"]
  320. return config.get("registry", YARN_DEFAULT_REGISTRY)
  321. def _ensure_options(options):
  322. """Helper to use deprecated kwargs for AppOption"""
  323. if options is None:
  324. return AppOptions()
  325. elif issubclass(options.__class__, AppOptions):
  326. return options
  327. else:
  328. return AppOptions(**options)
  329. def watch(app_options=None):
  330. """Watch the application.
  331. Parameters
  332. ----------
  333. app_options: :class:`AppOptions`, optional
  334. The application options.
  335. Returns
  336. -------
  337. A list of processes to run asynchronously.
  338. """
  339. app_options = _ensure_options(app_options)
  340. _node_check(app_options.logger)
  341. handler = _AppHandler(app_options)
  342. package_procs = watch_packages(app_options.logger) if app_options.splice_source else []
  343. return package_procs + handler.watch()
  344. def install_extension(extension, app_options=None, pin=None):
  345. """Install an extension package into JupyterLab.
  346. The extension is first validated.
  347. Returns `True` if a rebuild is recommended, `False` otherwise.
  348. """
  349. app_options = _ensure_options(app_options)
  350. _node_check(app_options.logger)
  351. handler = _AppHandler(app_options)
  352. return handler.install_extension(extension, pin=pin)
  353. def uninstall_extension(name=None, app_options=None, all_=False):
  354. """Uninstall an extension by name or path.
  355. Returns `True` if a rebuild is recommended, `False` otherwise.
  356. """
  357. app_options = _ensure_options(app_options)
  358. _node_check(app_options.logger)
  359. handler = _AppHandler(app_options)
  360. if all_ is True:
  361. return handler.uninstall_all_extensions()
  362. return handler.uninstall_extension(name)
  363. def update_extension(name=None, all_=False, app_dir=None, app_options=None):
  364. """Update an extension by name, or all extensions.
  365. Either `name` must be given as a string, or `all_` must be `True`.
  366. If `all_` is `True`, the value of `name` is ignored.
  367. Returns `True` if a rebuild is recommended, `False` otherwise.
  368. """
  369. app_options = _ensure_options(app_options)
  370. _node_check(app_options.logger)
  371. handler = _AppHandler(app_options)
  372. if all_ is True:
  373. return handler.update_all_extensions()
  374. return handler.update_extension(name)
  375. def clean(app_options=None):
  376. """Clean the JupyterLab application directory."""
  377. app_options = _ensure_options(app_options)
  378. logger = app_options.logger
  379. app_dir = app_options.app_dir
  380. logger.info(f"Cleaning {app_dir}...")
  381. if app_dir == pjoin(HERE, "dev"):
  382. msg = "Cannot clean the dev app"
  383. raise ValueError(msg)
  384. if app_dir == pjoin(HERE, "core"):
  385. msg = "Cannot clean the core app"
  386. raise ValueError(msg)
  387. if getattr(app_options, "all", False):
  388. logger.info(f"Removing everything in {app_dir}...")
  389. _rmtree_star(app_dir, logger)
  390. else:
  391. possible_targets = ["extensions", "settings", "staging", "static"]
  392. targets = [t for t in possible_targets if getattr(app_options, t)]
  393. for name in targets:
  394. target = pjoin(app_dir, name)
  395. if osp.exists(target):
  396. logger.info(f"Removing {name}...")
  397. _rmtree(target, logger)
  398. else:
  399. logger.info(f"{name} not present, skipping...")
  400. logger.info("Success!")
  401. if getattr(app_options, "all", False) or getattr(app_options, "extensions", False):
  402. logger.info("All of your extensions have been removed, and will need to be reinstalled")
  403. def build(
  404. name=None,
  405. version=None,
  406. static_url=None,
  407. kill_event=None,
  408. clean_staging=False,
  409. app_options=None,
  410. production=True,
  411. minimize=True,
  412. ):
  413. """Build the JupyterLab application."""
  414. app_options = _ensure_options(app_options)
  415. _node_check(app_options.logger)
  416. handler = _AppHandler(app_options)
  417. return handler.build(
  418. name=name,
  419. version=version,
  420. static_url=static_url,
  421. production=production,
  422. minimize=minimize,
  423. clean_staging=clean_staging,
  424. )
  425. def get_app_info(app_options=None):
  426. """Get a dictionary of information about the app."""
  427. handler = _AppHandler(app_options)
  428. handler._ensure_disabled_info()
  429. return handler.info
  430. def enable_extension(extension, app_options=None, level="sys_prefix"):
  431. """Enable a JupyterLab extension/plugin.
  432. Returns `True` if a rebuild is recommended, `False` otherwise.
  433. """
  434. handler = _AppHandler(app_options)
  435. return handler.toggle_extension(extension, False, level=level)
  436. def disable_extension(extension, app_options=None, level="sys_prefix"):
  437. """Disable a JupyterLab extension/plugin.
  438. Returns `True` if a rebuild is recommended, `False` otherwise.
  439. """
  440. handler = _AppHandler(app_options)
  441. return handler.toggle_extension(extension, True, level=level)
  442. def check_extension(extension, installed=False, app_options=None):
  443. """Check if a JupyterLab extension is enabled or disabled."""
  444. handler = _AppHandler(app_options)
  445. return handler.check_extension(extension, installed)
  446. def lock_extension(extension, app_options=None, level="sys_prefix"):
  447. """Lock a JupyterLab extension/plugin."""
  448. handler = _AppHandler(app_options)
  449. return handler.toggle_extension_lock(extension, True, level=level)
  450. def unlock_extension(extension, app_options=None, level="sys_prefix"):
  451. """Unlock a JupyterLab extension/plugin."""
  452. handler = _AppHandler(app_options)
  453. return handler.toggle_extension_lock(extension, False, level=level)
  454. def build_check(app_options=None):
  455. """Determine whether JupyterLab should be built.
  456. Returns a list of messages.
  457. """
  458. app_options = _ensure_options(app_options)
  459. _node_check(app_options.logger)
  460. handler = _AppHandler(app_options)
  461. return handler.build_check()
  462. def list_extensions(app_options=None):
  463. """List the extensions."""
  464. handler = _AppHandler(app_options)
  465. return handler.list_extensions()
  466. def link_package(path, app_options=None):
  467. """Link a package against the JupyterLab build.
  468. Returns `True` if a rebuild is recommended, `False` otherwise.
  469. """
  470. handler = _AppHandler(app_options)
  471. return handler.link_package(path)
  472. def unlink_package(package, app_options=None):
  473. """Unlink a package from JupyterLab by path or name.
  474. Returns `True` if a rebuild is recommended, `False` otherwise.
  475. """
  476. handler = _AppHandler(app_options)
  477. return handler.unlink_package(package)
  478. def get_app_version(app_options=None):
  479. """Get the application version."""
  480. handler = _AppHandler(app_options)
  481. return handler.info["version"]
  482. def get_latest_compatible_package_versions(names, app_options=None):
  483. """Get the latest compatible version of a list of packages."""
  484. handler = _AppHandler(app_options)
  485. return handler.latest_compatible_package_versions(names)
  486. def read_package(target):
  487. """Read the package data in a given target tarball."""
  488. with tarfile.open(target, "r") as tar:
  489. with tar.extractfile("package/package.json") as f:
  490. data = json.loads(f.read().decode("utf8"))
  491. data["jupyterlab_extracted_files"] = [f.path[len("package/") :] for f in tar.getmembers()]
  492. return data
  493. # ----------------------------------------------------------------------
  494. # Implementation details
  495. # ----------------------------------------------------------------------
  496. class _AppHandler:
  497. def __init__(self, options):
  498. """Create a new _AppHandler object"""
  499. options = _ensure_options(options)
  500. self._options = options
  501. self.app_dir = options.app_dir
  502. self.sys_dir = get_app_dir() if options.use_sys_dir else self.app_dir
  503. self.logger = options.logger
  504. # Make a deep copy of the core data so we don't influence the original copy
  505. self.core_data = deepcopy(options.core_config._data)
  506. self.labextensions_path = options.labextensions_path
  507. self.verbose = options.verbose
  508. self.kill_event = options.kill_event
  509. self.registry = options.registry
  510. self.skip_full_build_check = options.skip_full_build_check
  511. # Do this last since it relies on other attributes
  512. self.info = self._get_app_info()
  513. # Migrate from 4.0 which did not have "locked" status
  514. try:
  515. self._maybe_mirror_disabled_in_locked(level="sys_prefix")
  516. except (PermissionError, OSError):
  517. try:
  518. self.logger.info(
  519. "`sys_prefix` level settings are read-only, using `user` level for migration to `lockedExtensions`"
  520. )
  521. self._maybe_mirror_disabled_in_locked(level="user")
  522. except (PermissionError, OSError):
  523. self.logger.warning(
  524. "Both `sys_prefix` and `user` level settings are read-only, cannot auto-migrate `disabledExtensions` to `lockedExtensions`"
  525. )
  526. def install_extension(self, extension, existing=None, pin=None):
  527. """Install an extension package into JupyterLab.
  528. The extension is first validated.
  529. Returns `True` if a rebuild is recommended, `False` otherwise.
  530. """
  531. extension = _normalize_path(extension)
  532. extensions = self.info["extensions"]
  533. # Check for a core extensions.
  534. if extension in self.info["core_extensions"]:
  535. config = self._read_build_config()
  536. uninstalled = config.get("uninstalled_core_extensions", [])
  537. if extension in uninstalled:
  538. self.logger.info(f"Installing core extension {extension}")
  539. uninstalled.remove(extension)
  540. config["uninstalled_core_extensions"] = uninstalled
  541. self._write_build_config(config)
  542. return True
  543. return False
  544. # Create the app dirs if needed.
  545. self._ensure_app_dirs()
  546. # Install the package using a temporary directory.
  547. with TemporaryDirectory() as tempdir:
  548. info = self._install_extension(extension, tempdir, pin=pin)
  549. name = info["name"]
  550. # Local directories get name mangled and stored in metadata.
  551. if info["is_dir"]:
  552. config = self._read_build_config()
  553. local = config.setdefault("local_extensions", {})
  554. local[name] = info["source"]
  555. self._write_build_config(config)
  556. # Remove an existing extension with the same name and different path
  557. if name in extensions:
  558. other = extensions[name]
  559. if other["path"] != info["path"] and other["location"] == "app":
  560. os.remove(other["path"])
  561. return True
  562. def build(
  563. self,
  564. name=None,
  565. version=None,
  566. static_url=None,
  567. clean_staging=False,
  568. production=True,
  569. minimize=True,
  570. ):
  571. """Build the application."""
  572. if production is None:
  573. production = not (self.info["linked_packages"] or self.info["local_extensions"])
  574. if not production:
  575. minimize = False
  576. # If splicing, make sure the source packages are built
  577. if self._options.splice_source:
  578. ensure_node_modules(REPO_ROOT, logger=self.logger)
  579. self._run(["node", YARN_PATH, "build:packages"], cwd=REPO_ROOT)
  580. info = ["production" if production else "development"]
  581. if production:
  582. info.append("minimized" if minimize else "not minimized")
  583. self.logger.info(f"Building jupyterlab assets ({', '.join(info)})")
  584. # Set up the build directory.
  585. app_dir = self.app_dir
  586. self._populate_staging(
  587. name=name, version=version, static_url=static_url, clean=clean_staging
  588. )
  589. staging = pjoin(app_dir, "staging")
  590. # Make sure packages are installed.
  591. ret = self._run(["node", YARN_PATH, "install"], cwd=staging)
  592. if ret != 0:
  593. msg = "npm dependencies failed to install"
  594. self.logger.debug(msg)
  595. raise RuntimeError(msg)
  596. # Build the app.
  597. dedupe_yarn(staging, self.logger)
  598. command = f"build:{'prod' if production else 'dev'}{':minimize' if minimize else ''}"
  599. ret = self._run(["node", YARN_PATH, "run", command], cwd=staging)
  600. if ret != 0:
  601. msg = "JupyterLab failed to build"
  602. self.logger.debug(msg)
  603. raise RuntimeError(msg)
  604. def watch(self):
  605. """Start the application watcher and then run the watch in
  606. the background.
  607. """
  608. staging = pjoin(self.app_dir, "staging")
  609. self._populate_staging()
  610. # Make sure packages are installed.
  611. self._run(["node", YARN_PATH, "install"], cwd=staging)
  612. dedupe_yarn(staging, self.logger)
  613. proc = WatchHelper(
  614. ["node", YARN_PATH, "run", "watch"],
  615. cwd=pjoin(self.app_dir, "staging"),
  616. startup_regex=WEBPACK_EXPECT,
  617. logger=self.logger,
  618. )
  619. return [proc]
  620. def list_extensions(self): # noqa
  621. """Print an output of the extensions."""
  622. self._ensure_disabled_info()
  623. logger = self.logger
  624. info = self.info
  625. version = info["version"]
  626. logger.info(f"JupyterLab v{version}")
  627. if info["federated_extensions"] or info["extensions"]:
  628. info["compat_errors"] = self._get_extension_compat()
  629. if info["federated_extensions"]:
  630. self._list_federated_extensions()
  631. if info["extensions"]:
  632. logger.info("Other labextensions (built into JupyterLab)")
  633. self._list_extensions(info, "app")
  634. self._list_extensions(info, "sys")
  635. local = info["local_extensions"]
  636. if local:
  637. logger.info("\n local extensions:")
  638. for name in sorted(local):
  639. logger.info(f" {name}: {local[name]}")
  640. linked_packages = info["linked_packages"]
  641. if linked_packages:
  642. logger.info("\n linked packages:")
  643. for key in sorted(linked_packages):
  644. source = linked_packages[key]["source"]
  645. logger.info(f" {key}: {source}")
  646. uninstalled_core = info["uninstalled_core"]
  647. if uninstalled_core:
  648. logger.info("\nUninstalled core extensions:")
  649. [logger.info(f" {item}") for item in sorted(uninstalled_core)]
  650. all_exts = (
  651. list(info["federated_extensions"])
  652. + list(info["extensions"])
  653. + list(info["core_extensions"])
  654. )
  655. # Ignore disabled extensions that are not installed
  656. disabled = [i for i in info["disabled"] if i.partition(":")[0] in all_exts]
  657. if disabled:
  658. logger.info("\nDisabled extensions:")
  659. for item in sorted(disabled):
  660. # Show that all plugins will be disabled if the whole extension matches
  661. if item in all_exts:
  662. item += " (all plugins)" # noqa PLW2901
  663. logger.info(f" {item}")
  664. # Here check if modules are improperly shadowed
  665. improper_shadowed = []
  666. for ext_name in self.info["shadowed_exts"]:
  667. source_version = self.info["extensions"][ext_name]["version"]
  668. prebuilt_version = self.info["federated_extensions"][ext_name]["version"]
  669. if not gte(prebuilt_version, source_version, True):
  670. improper_shadowed.append(ext_name)
  671. if improper_shadowed:
  672. logger.info(
  673. "\nThe following source extensions are overshadowed by older prebuilt extensions:"
  674. )
  675. [logger.info(f" {name}") for name in sorted(improper_shadowed)]
  676. messages = self.build_check(fast=True)
  677. if messages:
  678. logger.info("\nBuild recommended, please run `jupyter lab build`:")
  679. [logger.info(f" {item}") for item in messages]
  680. def build_check(self, fast=None): # noqa
  681. """Determine whether JupyterLab should be built.
  682. Returns a list of messages.
  683. """
  684. if fast is None:
  685. fast = self.skip_full_build_check
  686. app_dir = self.app_dir
  687. local = self.info["local_extensions"]
  688. linked = self.info["linked_packages"]
  689. messages = []
  690. # Check for no application.
  691. pkg_path = pjoin(app_dir, "static", "package.json")
  692. if not osp.exists(pkg_path):
  693. return ["No built application"]
  694. static_data = self.info["static_data"]
  695. old_jlab = static_data["jupyterlab"]
  696. old_deps = static_data.get("dependencies", {})
  697. # Look for mismatched version.
  698. static_version = old_jlab.get("version", "")
  699. if not static_version.endswith("-spliced"):
  700. core_version = old_jlab["version"]
  701. if Version(static_version) != Version(core_version):
  702. msg = f"Version mismatch: {static_version} (built), {core_version} (current)"
  703. return [msg]
  704. shadowed_exts = self.info["shadowed_exts"]
  705. # Look for mismatched extensions.
  706. new_package = self._get_package_template(silent=fast)
  707. new_jlab = new_package["jupyterlab"]
  708. new_deps = new_package.get("dependencies", {})
  709. for ext_type in ["extensions", "mimeExtensions"]:
  710. # Extensions that were added.
  711. for ext in new_jlab[ext_type]:
  712. if ext in shadowed_exts:
  713. continue
  714. if ext not in old_jlab[ext_type]:
  715. messages.append(f"{ext} needs to be included in build")
  716. # Extensions that were removed.
  717. for ext in old_jlab[ext_type]:
  718. if ext in shadowed_exts:
  719. continue
  720. if ext not in new_jlab[ext_type]:
  721. messages.append(f"{ext} needs to be removed from build")
  722. # Look for mismatched dependencies
  723. src_pkg_dir = pjoin(REPO_ROOT, "packages")
  724. for pkg, dep in new_deps.items():
  725. if old_deps.get(pkg, "").startswith(src_pkg_dir):
  726. continue
  727. if pkg not in old_deps:
  728. continue
  729. # Skip local and linked since we pick them up separately.
  730. if pkg in local or pkg in linked:
  731. continue
  732. if old_deps[pkg] != dep:
  733. msg = f"{pkg} changed from {old_deps[pkg]} to {new_deps[pkg]}"
  734. messages.append(msg)
  735. # Look for updated local extensions.
  736. for name, source in local.items():
  737. if fast or name in shadowed_exts:
  738. continue
  739. dname = pjoin(app_dir, "extensions")
  740. if self._check_local(name, source, dname):
  741. messages.append(f"{name} content changed")
  742. # Look for updated linked packages.
  743. for name, item in linked.items():
  744. if fast or name in shadowed_exts:
  745. continue
  746. dname = pjoin(app_dir, "staging", "linked_packages")
  747. if self._check_local(name, item["source"], dname):
  748. messages.append(f"{name} content changed")
  749. return messages
  750. def uninstall_extension(self, name):
  751. """Uninstall an extension by name.
  752. Returns `True` if a rebuild is recommended, `False` otherwise.
  753. """
  754. info = self.info
  755. logger = self.logger
  756. if name in info["federated_extensions"]:
  757. if (
  758. info["federated_extensions"][name]
  759. .get("install", {})
  760. .get("uninstallInstructions", None)
  761. ):
  762. instructions = info["federated_extensions"][name]["install"][
  763. "uninstallInstructions"
  764. ]
  765. logger.error(f"JupyterLab cannot uninstall this extension. {instructions}")
  766. else:
  767. logger.error(
  768. f"JupyterLab cannot uninstall {name} since it was installed outside of JupyterLab. Use the same method used to install this extension to uninstall this extension."
  769. )
  770. return False
  771. # Allow for uninstalled core extensions.
  772. if name in info["core_extensions"]:
  773. config = self._read_build_config()
  774. uninstalled = config.get("uninstalled_core_extensions", [])
  775. if name not in uninstalled:
  776. logger.info(f"Uninstalling core extension {name}")
  777. uninstalled.append(name)
  778. config["uninstalled_core_extensions"] = uninstalled
  779. self._write_build_config(config)
  780. return True
  781. return False
  782. local = info["local_extensions"]
  783. for extname, data in info["extensions"].items():
  784. path = data["path"]
  785. if extname == name:
  786. msg = f"Uninstalling {name} from {osp.dirname(path)}"
  787. logger.info(msg)
  788. os.remove(path)
  789. # Handle local extensions.
  790. if extname in local:
  791. config = self._read_build_config()
  792. data = config.setdefault("local_extensions", {}) # noqa PLW2901
  793. del data[extname]
  794. self._write_build_config(config)
  795. return True
  796. logger.warning(f'No labextension named "{name}" installed')
  797. return False
  798. def uninstall_all_extensions(self):
  799. """Uninstalls all extensions
  800. Returns `True` if a rebuild is recommended, `False` otherwise
  801. """
  802. should_rebuild = False
  803. for extname, _ in self.info["extensions"].items():
  804. uninstalled = self.uninstall_extension(extname)
  805. should_rebuild = should_rebuild or uninstalled
  806. return should_rebuild
  807. def update_all_extensions(self):
  808. """Update all non-local extensions.
  809. Returns `True` if a rebuild is recommended, `False` otherwise.
  810. """
  811. should_rebuild = False
  812. for extname, _ in self.info["extensions"].items():
  813. if extname in self.info["local_extensions"]:
  814. continue
  815. updated = self._update_extension(extname)
  816. # Rebuild if at least one update happens:
  817. should_rebuild = should_rebuild or updated
  818. return should_rebuild
  819. def update_extension(self, name):
  820. """Update an extension by name.
  821. Returns `True` if a rebuild is recommended, `False` otherwise.
  822. """
  823. if name not in self.info["extensions"]:
  824. self.logger.warning(f'No labextension named "{name}" installed')
  825. return False
  826. return self._update_extension(name)
  827. def _update_extension(self, name):
  828. """Update an extension by name.
  829. Returns `True` if a rebuild is recommended, `False` otherwise.
  830. """
  831. data = self.info["extensions"][name]
  832. if data["alias_package_source"]:
  833. self.logger.warning(f"Skipping updating pinned extension '{name}'.")
  834. return False
  835. try:
  836. latest = self._latest_compatible_package_version(name)
  837. except URLError:
  838. return False
  839. if latest is None:
  840. self.logger.warning(f"No compatible version found for {name}!")
  841. return False
  842. if latest == data["version"]:
  843. self.logger.info(f"Extension {name!r} already up to date")
  844. return False
  845. self.logger.info(f"Updating {name} to version {latest}")
  846. return self.install_extension(f"{name}@{latest}")
  847. def link_package(self, path):
  848. """Link a package at the given path.
  849. Returns `True` if a rebuild is recommended, `False` otherwise.
  850. """
  851. path = _normalize_path(path)
  852. if not osp.exists(path) or not osp.isdir(path):
  853. msg = f'Cannot install "{path}" only link local directories'
  854. raise ValueError(msg)
  855. with TemporaryDirectory() as tempdir:
  856. info = self._extract_package(path, tempdir)
  857. messages = _validate_extension(info["data"])
  858. if not messages:
  859. return self.install_extension(path)
  860. # Warn that it is a linked package.
  861. self.logger.warning(
  862. f"Installing {path} as a linked package because it does not have extension metadata:"
  863. )
  864. [self.logger.warning(f" {m}") for m in messages]
  865. # Add to metadata.
  866. config = self._read_build_config()
  867. linked = config.setdefault("linked_packages", {})
  868. linked[info["name"]] = info["source"]
  869. self._write_build_config(config)
  870. return True
  871. def unlink_package(self, path):
  872. """Unlink a package by name or at the given path.
  873. A ValueError is raised if the path is not an unlinkable package.
  874. Returns `True` if a rebuild is recommended, `False` otherwise.
  875. """
  876. path = _normalize_path(path)
  877. config = self._read_build_config()
  878. linked = config.setdefault("linked_packages", {})
  879. found = None
  880. for name, source in linked.items():
  881. if path in {name, source}:
  882. found = name
  883. if found:
  884. del linked[found]
  885. else:
  886. local = config.setdefault("local_extensions", {})
  887. for name, source in local.items():
  888. if path in {name, source}:
  889. found = name
  890. if found:
  891. del local[found]
  892. path = self.info["extensions"][found]["path"]
  893. os.remove(path)
  894. if not found:
  895. msg = f"No linked package for {path}"
  896. raise ValueError(msg)
  897. self._write_build_config(config)
  898. return True
  899. def _is_extension_locked(self, extension, level="sys_prefix", include_higher_levels=True):
  900. app_settings_dir = osp.join(self.app_dir, "settings")
  901. page_config = get_static_page_config(
  902. app_settings_dir=app_settings_dir,
  903. logger=self.logger,
  904. level=level,
  905. include_higher_levels=True,
  906. )
  907. locked = page_config.get("lockedExtensions", {})
  908. return locked.get(extension, False)
  909. def toggle_extension(self, extension, value, level="sys_prefix"):
  910. """Enable or disable a lab extension.
  911. Returns `True` if a rebuild is recommended, `False` otherwise.
  912. """
  913. app_settings_dir = osp.join(self.app_dir, "settings")
  914. # If extension is locked at a higher level, we don't toggle it.
  915. # The highest level at which an extension can be locked is system,
  916. # so we do not need to check levels above that one.
  917. if level != "system":
  918. allowed = get_allowed_levels()
  919. if self._is_extension_locked(
  920. extension, level=allowed[allowed.index(level) + 1], include_higher_levels=True
  921. ):
  922. self.logger.info("Extension locked at a higher level, cannot toggle status")
  923. return False
  924. complete_page_config = get_static_page_config(
  925. app_settings_dir=app_settings_dir, logger=self.logger, level="all"
  926. )
  927. level_page_config = get_static_page_config(
  928. app_settings_dir=app_settings_dir, logger=self.logger, level=level
  929. )
  930. disabled = complete_page_config.get("disabledExtensions", {})
  931. disabled_at_level = level_page_config.get("disabledExtensions", {})
  932. did_something = False
  933. is_disabled = disabled.get(extension, False)
  934. if value and not is_disabled:
  935. disabled_at_level[extension] = True
  936. did_something = True
  937. elif not value and is_disabled:
  938. disabled_at_level[extension] = False
  939. did_something = True
  940. if did_something:
  941. level_page_config["disabledExtensions"] = disabled_at_level
  942. write_page_config(level_page_config, level=level)
  943. return did_something
  944. def _maybe_mirror_disabled_in_locked(self, level="sys_prefix"):
  945. """Lock all extensions that were previously disabled.
  946. This exists to facilitate migration from 4.0 (which did not include lock
  947. function) to 4.1 which exposes the plugin management to users in UI.
  948. Returns `True` if migration happened, `False` otherwise.
  949. """
  950. app_settings_dir = osp.join(self.app_dir, "settings")
  951. page_config = get_static_page_config(
  952. app_settings_dir=app_settings_dir, logger=self.logger, level=level
  953. )
  954. if "lockedExtensions" in page_config:
  955. # short-circuit if migration already happened
  956. return False
  957. # copy disabled onto lockedExtensions, ensuring the mapping format
  958. disabled = page_config.get("disabledExtensions", {})
  959. if isinstance(disabled, list):
  960. disabled = dict.fromkeys(disabled, True)
  961. # Short circuit if disabled is empty
  962. if not disabled:
  963. return False
  964. page_config["lockedExtensions"] = disabled
  965. write_page_config(page_config, level=level)
  966. return True
  967. def toggle_extension_lock(self, extension, value, level="sys_prefix"):
  968. """Lock or unlock a lab extension (/plugin)."""
  969. app_settings_dir = osp.join(self.app_dir, "settings")
  970. # The highest level at which an extension can be locked is system,
  971. # so we do not need to check levels above that one.
  972. if level != "system":
  973. allowed = get_allowed_levels()
  974. if self._is_extension_locked(
  975. extension, level=allowed[allowed.index(level) + 1], include_higher_levels=True
  976. ):
  977. self.logger.info("Extension locked at a higher level, cannot toggle")
  978. return False
  979. page_config = get_static_page_config(
  980. app_settings_dir=app_settings_dir, logger=self.logger, level=level
  981. )
  982. locked = page_config.get("lockedExtensions", {})
  983. locked[extension] = value
  984. page_config["lockedExtensions"] = locked
  985. write_page_config(page_config, level=level)
  986. def check_extension(self, extension, check_installed_only=False):
  987. """Check if a lab extension is enabled or disabled"""
  988. self._ensure_disabled_info()
  989. info = self.info
  990. if extension in info["core_extensions"]:
  991. return self._check_core_extension(extension, info, check_installed_only)
  992. if extension in info["linked_packages"]:
  993. self.logger.info(f"{extension}:{GREEN_ENABLED}")
  994. return True
  995. return self._check_common_extension(extension, info, check_installed_only)
  996. def _check_core_extension(self, extension, info, check_installed_only):
  997. """Check if a core extension is enabled or disabled"""
  998. if extension in info["uninstalled_core"]:
  999. self.logger.info(f"{extension}:{RED_X}")
  1000. return False
  1001. if check_installed_only:
  1002. self.logger.info(f"{extension}: {GREEN_OK}")
  1003. return True
  1004. if extension in info["disabled_core"]:
  1005. self.logger.info(f"{extension}: {RED_DISABLED}")
  1006. return False
  1007. self.logger.info(f"{extension}:{GREEN_ENABLED}")
  1008. return True
  1009. def _check_common_extension(self, extension, info, check_installed_only):
  1010. """Check if a common (non-core) extension is enabled or disabled"""
  1011. if extension not in info["extensions"]:
  1012. self.logger.info(f"{extension}:{RED_X}")
  1013. return False
  1014. errors = self._get_extension_compat()[extension]
  1015. if errors:
  1016. self.logger.info(f"{extension}:{RED_X} (compatibility errors)")
  1017. return False
  1018. if check_installed_only:
  1019. self.logger.info(f"{extension}: {GREEN_OK}")
  1020. return True
  1021. if _is_disabled(extension, info["disabled"]):
  1022. self.logger.info(f"{extension}: {RED_DISABLED}")
  1023. return False
  1024. self.logger.info(f"{extension}:{GREEN_ENABLED}")
  1025. return True
  1026. def _get_app_info(self):
  1027. """Get information about the app."""
  1028. info = {}
  1029. info["core_data"] = core_data = self.core_data
  1030. info["extensions"] = extensions = self._get_extensions(core_data)
  1031. info["local_extensions"] = self._get_local_extensions()
  1032. info["linked_packages"] = self._get_linked_packages()
  1033. info["app_extensions"] = app = []
  1034. info["sys_extensions"] = sys = []
  1035. for name, data in extensions.items():
  1036. data["is_local"] = name in info["local_extensions"]
  1037. if data["location"] == "app":
  1038. app.append(name)
  1039. else:
  1040. sys.append(name)
  1041. info["uninstalled_core"] = self._get_uninstalled_core_extensions()
  1042. info["static_data"] = _get_static_data(self.app_dir)
  1043. app_data = info["static_data"] or core_data
  1044. info["version"] = app_data["jupyterlab"]["version"]
  1045. info["staticUrl"] = app_data["jupyterlab"].get("staticUrl", "")
  1046. info["sys_dir"] = self.sys_dir
  1047. info["app_dir"] = self.app_dir
  1048. info["core_extensions"] = _get_core_extensions(self.core_data)
  1049. info["federated_extensions"] = get_federated_extensions(self.labextensions_path)
  1050. info["shadowed_exts"] = [
  1051. ext for ext in info["extensions"] if ext in info["federated_extensions"]
  1052. ]
  1053. return info
  1054. def _ensure_disabled_info(self):
  1055. info = self.info
  1056. if "disabled" in info:
  1057. return
  1058. labextensions_path = self.labextensions_path
  1059. app_settings_dir = osp.join(self.app_dir, "settings")
  1060. page_config = get_page_config(
  1061. labextensions_path, app_settings_dir=app_settings_dir, logger=self.logger
  1062. )
  1063. disabled = page_config.get("disabledExtensions", {})
  1064. # handle disabledExtensions specified as a list (jupyterlab_server < 2.10)
  1065. # see https://github.com/jupyterlab/jupyterlab_server/pull/192 for more info
  1066. if isinstance(disabled, list):
  1067. disabled = dict.fromkeys(disabled, True)
  1068. info["disabled"] = disabled
  1069. locked = page_config.get("lockedExtensions", {})
  1070. if isinstance(locked, list):
  1071. locked = dict.fromkeys(locked, True)
  1072. info["locked"] = locked
  1073. disabled_core = []
  1074. for key in info["core_extensions"]:
  1075. if key in info["disabled"]:
  1076. disabled_core.append(key)
  1077. info["disabled_core"] = disabled_core
  1078. def _populate_staging(self, name=None, version=None, static_url=None, clean=False): # noqa
  1079. """Set up the assets in the staging directory."""
  1080. app_dir = self.app_dir
  1081. staging = pjoin(app_dir, "staging")
  1082. if clean and osp.exists(staging):
  1083. self.logger.info(f"Cleaning {staging}")
  1084. _rmtree(staging, self.logger)
  1085. self._ensure_app_dirs()
  1086. if not version:
  1087. version = self.info["core_data"]["jupyterlab"]["version"]
  1088. splice_source = self._options.splice_source
  1089. if splice_source:
  1090. self.logger.debug("Splicing dev packages into app directory.")
  1091. source_dir = DEV_DIR
  1092. version = __version__ + "-spliced"
  1093. else:
  1094. source_dir = pjoin(HERE, "staging")
  1095. # Look for mismatched version.
  1096. pkg_path = pjoin(staging, "package.json")
  1097. if osp.exists(pkg_path):
  1098. with open(pkg_path) as fid:
  1099. data = json.load(fid)
  1100. if data["jupyterlab"].get("version", "") != version:
  1101. _rmtree(staging, self.logger)
  1102. os.makedirs(staging)
  1103. for fname in [
  1104. "index.js",
  1105. "bootstrap.js",
  1106. "publicpath.js",
  1107. "webpack.config.js",
  1108. "webpack.prod.config.js",
  1109. "webpack.prod.minimize.config.js",
  1110. ]:
  1111. target = pjoin(staging, fname)
  1112. shutil.copy(pjoin(source_dir, fname), target)
  1113. for fname in [".yarnrc.yml", "yarn.js"]:
  1114. target = pjoin(staging, fname)
  1115. shutil.copy(pjoin(HERE, "staging", fname), target)
  1116. # Ensure a clean templates directory
  1117. templates = pjoin(staging, "templates")
  1118. if osp.exists(templates):
  1119. _rmtree(templates, self.logger)
  1120. try:
  1121. shutil.copytree(pjoin(source_dir, "templates"), templates)
  1122. except shutil.Error as error:
  1123. # `copytree` throws an error if copying to + from NFS even though
  1124. # the copy is successful (see https://bugs.python.org/issue24564
  1125. # and https://github.com/jupyterlab/jupyterlab/issues/5233)
  1126. real_error = "[Errno 22]" not in str(error) and "[Errno 5]" not in str(error)
  1127. if real_error or not osp.exists(templates):
  1128. raise
  1129. # Ensure a clean linked packages directory.
  1130. linked_dir = pjoin(staging, "linked_packages")
  1131. if osp.exists(linked_dir):
  1132. _rmtree(linked_dir, self.logger)
  1133. os.makedirs(linked_dir)
  1134. # Template the package.json file.
  1135. # Update the local extensions.
  1136. extensions = self.info["extensions"]
  1137. removed = False
  1138. for key, source in self.info["local_extensions"].items():
  1139. # Handle a local extension that was removed.
  1140. if key not in extensions:
  1141. config = self._read_build_config()
  1142. data = config.setdefault("local_extensions", {})
  1143. del data[key]
  1144. self._write_build_config(config)
  1145. removed = True
  1146. continue
  1147. dname = pjoin(app_dir, "extensions")
  1148. self._update_local(key, source, dname, extensions[key], "local_extensions")
  1149. # Update the list of local extensions if any were removed.
  1150. if removed:
  1151. self.info["local_extensions"] = self._get_local_extensions()
  1152. # Update the linked packages.
  1153. linked = self.info["linked_packages"]
  1154. for key, item in linked.items():
  1155. dname = pjoin(staging, "linked_packages")
  1156. self._update_local(key, item["source"], dname, item, "linked_packages")
  1157. # Then get the package template.
  1158. data = self._get_package_template()
  1159. jlab = data["jupyterlab"]
  1160. if version:
  1161. jlab["version"] = version
  1162. if name:
  1163. jlab["name"] = name
  1164. if static_url:
  1165. jlab["staticUrl"] = static_url
  1166. # Handle splicing of packages
  1167. if splice_source:
  1168. # Splice workspace tree as linked dependencies
  1169. for path in glob(pjoin(REPO_ROOT, "packages", "*", "package.json")):
  1170. local_path = osp.dirname(osp.abspath(path))
  1171. pkg_data = json.loads(Path(path).read_text(encoding="utf-8"))
  1172. name = pkg_data["name"]
  1173. if name in data["dependencies"]:
  1174. data["dependencies"][name] = local_path
  1175. jlab["linkedPackages"][name] = local_path
  1176. if name in data["resolutions"]:
  1177. data["resolutions"][name] = local_path
  1178. # splice the builder as well
  1179. local_path = osp.abspath(pjoin(REPO_ROOT, "builder"))
  1180. data["devDependencies"]["@jupyterlab/builder"] = local_path
  1181. target = osp.join(staging, "node_modules", "@jupyterlab", "builder")
  1182. # Remove node_modules so it gets re-populated
  1183. node_modules = pjoin(staging, "node_modules")
  1184. if osp.exists(node_modules):
  1185. shutil.rmtree(node_modules, ignore_errors=True)
  1186. # Write the package file
  1187. pkg_path = pjoin(staging, "package.json")
  1188. with open(pkg_path, "w") as fid:
  1189. json.dump(data, fid, indent=4)
  1190. # copy known-good yarn.lock if missing
  1191. lock_path = pjoin(staging, "yarn.lock")
  1192. lock_template = pjoin(HERE, "staging", "yarn.lock")
  1193. if not osp.exists(lock_path):
  1194. shutil.copy(lock_template, lock_path)
  1195. os.chmod(lock_path, stat.S_IWRITE | stat.S_IREAD)
  1196. def _get_package_template(self, silent=False): # noqa
  1197. """Get the template the for staging package.json file."""
  1198. logger = self.logger
  1199. # make a deep copy of the data so we don't influence the core data
  1200. data = deepcopy(self.info["core_data"])
  1201. local = self.info["local_extensions"]
  1202. linked = self.info["linked_packages"]
  1203. extensions = self.info["extensions"]
  1204. shadowed_exts = self.info["shadowed_exts"]
  1205. jlab = data["jupyterlab"]
  1206. def format_path(path):
  1207. path = osp.relpath(path, osp.abspath(osp.realpath(pjoin(self.app_dir, "staging"))))
  1208. path = "file:" + path.replace(os.sep, "/")
  1209. if os.name == "nt":
  1210. path = path.lower()
  1211. return path
  1212. jlab["linkedPackages"] = {}
  1213. # Handle local extensions.
  1214. for key, source in local.items():
  1215. if key in shadowed_exts:
  1216. continue
  1217. jlab["linkedPackages"][key] = source
  1218. data["resolutions"][key] = "file:" + self.info["extensions"][key]["path"]
  1219. # Handle linked packages.
  1220. for key, item in linked.items():
  1221. if key in shadowed_exts:
  1222. continue
  1223. path = pjoin(self.app_dir, "staging", "linked_packages")
  1224. path = pjoin(path, item["filename"])
  1225. data["dependencies"][key] = format_path(path)
  1226. jlab["linkedPackages"][key] = item["source"]
  1227. data["resolutions"][key] = format_path(path)
  1228. data["jupyterlab"]["extensionMetadata"] = {}
  1229. # Handle extensions
  1230. compat_errors = self._get_extension_compat()
  1231. for key, value in extensions.items():
  1232. # Reject incompatible extensions with a message.
  1233. errors = compat_errors[key]
  1234. if errors:
  1235. if not silent:
  1236. _log_single_compat_errors(logger, key, value["version"], errors)
  1237. continue
  1238. data["dependencies"][key] = format_path(value["path"])
  1239. jlab_data = value["jupyterlab"]
  1240. for item in ["extension", "mimeExtension"]:
  1241. ext = jlab_data.get(item, False)
  1242. if not ext:
  1243. continue
  1244. if ext is True:
  1245. ext = ""
  1246. jlab[item + "s"][key] = ext
  1247. # Add metadata for the extension
  1248. data["jupyterlab"]["extensionMetadata"][key] = jlab_data
  1249. # Handle uninstalled core extensions.
  1250. for item in self.info["uninstalled_core"]:
  1251. if item in jlab["extensions"]:
  1252. data["jupyterlab"]["extensions"].pop(item)
  1253. elif item in jlab["mimeExtensions"]:
  1254. data["jupyterlab"]["mimeExtensions"].pop(item)
  1255. # Remove from dependencies as well.
  1256. if item in data["dependencies"]:
  1257. data["dependencies"].pop(item)
  1258. return data
  1259. def _check_local(self, name, source, dname):
  1260. """Check if a local package has changed.
  1261. `dname` is the directory name of existing package tar archives.
  1262. """
  1263. # Extract the package in a temporary directory.
  1264. with TemporaryDirectory() as tempdir:
  1265. info = self._extract_package(source, tempdir)
  1266. # Test if the file content has changed.
  1267. # This relies on `_extract_package` adding the hashsum
  1268. # to the filename, allowing a simple exist check to
  1269. # compare the hash to the "cache" in dname.
  1270. target = pjoin(dname, info["filename"])
  1271. return not osp.exists(target)
  1272. def _update_local(self, name, source, dname, data, dtype):
  1273. """Update a local dependency. Return `True` if changed."""
  1274. # Extract the package in a temporary directory.
  1275. existing = data["filename"]
  1276. if not osp.exists(pjoin(dname, existing)):
  1277. existing = ""
  1278. with TemporaryDirectory() as tempdir:
  1279. info = self._extract_package(source, tempdir)
  1280. # Bail if the file content has not changed.
  1281. if info["filename"] == existing:
  1282. return existing
  1283. shutil.move(info["path"], pjoin(dname, info["filename"]))
  1284. # Remove the previous tarball and return the new file name.
  1285. if existing:
  1286. os.remove(pjoin(dname, existing))
  1287. data["filename"] = info["filename"]
  1288. data["path"] = pjoin(data["tar_dir"], data["filename"])
  1289. return info["filename"]
  1290. def _get_extensions(self, core_data):
  1291. """Get the extensions for the application."""
  1292. app_dir = self.app_dir
  1293. extensions = {}
  1294. # Get system level packages.
  1295. sys_path = pjoin(self.sys_dir, "extensions")
  1296. app_path = pjoin(self.app_dir, "extensions")
  1297. extensions = self._get_extensions_in_dir(self.sys_dir, core_data)
  1298. # Look in app_dir if different.
  1299. app_path = pjoin(app_dir, "extensions")
  1300. if app_path == sys_path or not osp.exists(app_path):
  1301. return extensions
  1302. extensions.update(self._get_extensions_in_dir(app_dir, core_data))
  1303. return extensions
  1304. def _get_extensions_in_dir(self, dname, core_data):
  1305. """Get the extensions in a given directory."""
  1306. extensions = {}
  1307. location = "app" if dname == self.app_dir else "sys"
  1308. for target in glob(pjoin(dname, "extensions", "*.tgz")):
  1309. data = read_package(target)
  1310. deps = data.get("dependencies", {})
  1311. name = data["name"]
  1312. jlab = data.get("jupyterlab", {})
  1313. path = osp.abspath(target)
  1314. filename = osp.basename(target)
  1315. if filename.startswith(PIN_PREFIX):
  1316. alias = filename[len(PIN_PREFIX) : -len(".tgz")]
  1317. else:
  1318. alias = None
  1319. url = get_package_url(data)
  1320. extensions[alias or name] = {
  1321. "description": data.get("description", ""),
  1322. "path": path,
  1323. "filename": osp.basename(path),
  1324. "url": url,
  1325. "version": data["version"],
  1326. # Only save the package name if the extension name is an alias
  1327. "alias_package_source": name if alias else None,
  1328. "jupyterlab": jlab,
  1329. "dependencies": deps,
  1330. "tar_dir": osp.dirname(path),
  1331. "location": location,
  1332. }
  1333. return extensions
  1334. def _get_extension_compat(self):
  1335. """Get the extension compatibility info."""
  1336. compat = {}
  1337. core_data = self.info["core_data"]
  1338. seen = set()
  1339. for name, data in self.info["federated_extensions"].items():
  1340. deps = data["dependencies"]
  1341. compat[name] = _validate_compatibility(name, deps, core_data)
  1342. seen.add(name)
  1343. for name, data in self.info["extensions"].items():
  1344. if name in seen:
  1345. continue
  1346. deps = data["dependencies"]
  1347. compat[name] = _validate_compatibility(name, deps, core_data)
  1348. return compat
  1349. def _get_local_extensions(self):
  1350. """Get the locally installed extensions."""
  1351. return self._get_local_data("local_extensions")
  1352. def _get_linked_packages(self):
  1353. """Get the linked packages."""
  1354. info = self._get_local_data("linked_packages")
  1355. dname = pjoin(self.app_dir, "staging", "linked_packages")
  1356. for name, source in info.items():
  1357. info[name] = {"source": source, "filename": "", "tar_dir": dname}
  1358. if not osp.exists(dname):
  1359. return info
  1360. for path in glob(pjoin(dname, "*.tgz")):
  1361. path = osp.abspath(path) # noqa PLW2901
  1362. data = read_package(path)
  1363. name = data["name"]
  1364. if name not in info:
  1365. self.logger.warning(f"Removing orphaned linked package {name}")
  1366. os.remove(path)
  1367. continue
  1368. item = info[name]
  1369. item["filename"] = osp.basename(path)
  1370. item["path"] = path
  1371. item["version"] = data["version"]
  1372. item["data"] = data
  1373. return info
  1374. def _get_uninstalled_core_extensions(self):
  1375. """Get the uninstalled core extensions."""
  1376. config = self._read_build_config()
  1377. return config.get("uninstalled_core_extensions", [])
  1378. def _ensure_app_dirs(self):
  1379. """Ensure that the application directories exist"""
  1380. dirs = ["extensions", "settings", "staging", "schemas", "themes"]
  1381. for dname in dirs:
  1382. path = pjoin(self.app_dir, dname)
  1383. if not osp.exists(path):
  1384. try:
  1385. os.makedirs(path)
  1386. except OSError as e:
  1387. if e.errno != errno.EEXIST:
  1388. raise
  1389. def _list_extensions(self, info, ext_type):
  1390. """List the extensions of a given type."""
  1391. self._ensure_disabled_info()
  1392. logger = self.logger
  1393. names = info[f"{ext_type}_extensions"]
  1394. if not names:
  1395. return
  1396. dname = info[f"{ext_type}_dir"]
  1397. error_accumulator = {}
  1398. logger.info(f" {ext_type} dir: {dname}")
  1399. for name in sorted(names):
  1400. if name in info["federated_extensions"]:
  1401. continue
  1402. data = info["extensions"][name]
  1403. version = data["version"]
  1404. errors = info["compat_errors"][name]
  1405. extra = self._compose_extra_status(name, info, data, errors)
  1406. # If we have the package name in the data, this means this extension's name is the alias name
  1407. alias_package_source = data["alias_package_source"]
  1408. if alias_package_source:
  1409. logger.info(f" {name} {alias_package_source} v{version}{extra}")
  1410. else:
  1411. logger.info(f" {name} v{version}{extra}")
  1412. if errors:
  1413. error_accumulator[name] = (version, errors)
  1414. # Write all errors at end:
  1415. _log_multiple_compat_errors(logger, error_accumulator, self.verbose)
  1416. # Write a blank line separator
  1417. logger.info("")
  1418. def _list_federated_extensions(self):
  1419. self._ensure_disabled_info()
  1420. info = self.info
  1421. logger = self.logger
  1422. error_accumulator = {}
  1423. ext_dirs = dict.fromkeys(self.labextensions_path, False)
  1424. for value in info["federated_extensions"].values():
  1425. ext_dirs[value["ext_dir"]] = True
  1426. for ext_dir, has_exts in ext_dirs.items():
  1427. if not has_exts:
  1428. continue
  1429. logger.info(ext_dir)
  1430. for name in info["federated_extensions"]:
  1431. data = info["federated_extensions"][name]
  1432. if data["ext_dir"] != ext_dir:
  1433. continue
  1434. version = data["version"]
  1435. errors = info["compat_errors"][name]
  1436. extra = self._compose_extra_status(name, info, data, errors)
  1437. install = data.get("install")
  1438. if install:
  1439. extra += " ({}, {})".format(install["packageManager"], install["packageName"])
  1440. logger.info(f" {name} v{version}{extra}")
  1441. if errors:
  1442. error_accumulator[name] = (version, errors)
  1443. # Add a spacer line after
  1444. logger.info("")
  1445. # Write all errors at end:
  1446. _log_multiple_compat_errors(logger, error_accumulator, self.verbose)
  1447. def _compose_extra_status(self, name: str, info: dict, data: dict, errors) -> str:
  1448. extra = ""
  1449. if _is_disabled(name, info["disabled"]):
  1450. extra += f" {RED_DISABLED}"
  1451. else:
  1452. extra += f" {GREEN_ENABLED}"
  1453. if errors:
  1454. extra += f" {RED_X}"
  1455. else:
  1456. extra += f" {GREEN_OK}"
  1457. if data["is_local"]:
  1458. extra += "*"
  1459. lock_status = _is_locked(name, info["locked"])
  1460. if lock_status.entire_extension_locked:
  1461. extra += " 🔒 (all plugins locked)"
  1462. elif lock_status.locked_plugins:
  1463. plugin_list = ", ".join(sorted(lock_status.locked_plugins))
  1464. extra += f" 🔒 (plugins: {plugin_list} locked)"
  1465. return extra
  1466. def _read_build_config(self):
  1467. """Get the build config data for the app dir."""
  1468. target = pjoin(self.app_dir, "settings", "build_config.json")
  1469. if not osp.exists(target):
  1470. return {}
  1471. else:
  1472. with open(target) as fid:
  1473. return json.load(fid)
  1474. def _write_build_config(self, config):
  1475. """Write the build config to the app dir."""
  1476. self._ensure_app_dirs()
  1477. target = pjoin(self.app_dir, "settings", "build_config.json")
  1478. with open(target, "w") as fid:
  1479. json.dump(config, fid, indent=4)
  1480. def _get_local_data(self, source):
  1481. """Get the local data for extensions or linked packages."""
  1482. config = self._read_build_config()
  1483. data = config.setdefault(source, {})
  1484. dead = []
  1485. for name, source_ in data.items():
  1486. if not osp.exists(source_):
  1487. dead.append(name)
  1488. for name in dead:
  1489. link_type = source.replace("_", " ")
  1490. msg = f'**Note: Removing dead {link_type} "{name}"'
  1491. self.logger.warning(msg)
  1492. del data[name]
  1493. if dead:
  1494. self._write_build_config(config)
  1495. return data
  1496. def _install_extension(self, extension, tempdir, pin=None):
  1497. """Install an extension with validation and return the name and path."""
  1498. info = self._extract_package(extension, tempdir, pin=pin)
  1499. data = info["data"]
  1500. # Check for compatible version unless:
  1501. # - A specific version was requested (@ in name,
  1502. # but after first char to allow for scope marker).
  1503. # - Package is locally installed.
  1504. allow_fallback = "@" not in extension[1:] and not info["is_dir"]
  1505. name = info["name"]
  1506. # Verify that the package is an extension.
  1507. messages = _validate_extension(data)
  1508. if messages:
  1509. all_messages = "\n".join(messages)
  1510. msg = f'"{extension}" is not a valid extension:\n{all_messages}'
  1511. if allow_fallback:
  1512. try:
  1513. version = self._latest_compatible_package_version(name)
  1514. except URLError:
  1515. raise ValueError(msg) from None
  1516. else:
  1517. raise ValueError(msg)
  1518. # Verify package compatibility.
  1519. deps = data.get("dependencies", {})
  1520. errors = _validate_compatibility(extension, deps, self.core_data)
  1521. if errors:
  1522. msg = _format_compatibility_errors(data["name"], data["version"], errors)
  1523. if allow_fallback:
  1524. try:
  1525. version = self._latest_compatible_package_version(name)
  1526. except URLError:
  1527. # We cannot add any additional information to error message
  1528. raise ValueError(msg) from None
  1529. if version and name:
  1530. self.logger.debug(f"Incompatible extension:\n{name}")
  1531. self.logger.debug(f"Found compatible version: {version}")
  1532. with TemporaryDirectory() as tempdir2:
  1533. return self._install_extension(f"{name}@{version}", tempdir2)
  1534. # Extend message to better guide the user what to do:
  1535. conflicts = "\n".join(msg.splitlines()[2:])
  1536. msg = "".join((self._format_no_compatible_package_version(name), "\n\n", conflicts))
  1537. raise ValueError(msg)
  1538. # Move the file to the app directory.
  1539. target = pjoin(self.app_dir, "extensions", info["filename"])
  1540. if osp.exists(target):
  1541. os.remove(target)
  1542. shutil.move(info["path"], target)
  1543. info["path"] = target
  1544. return info
  1545. def _extract_package(self, source, tempdir, pin=None):
  1546. """Call `npm pack` for an extension.
  1547. The pack command will download the package tar if `source` is
  1548. a package name, or run `npm pack` locally if `source` is a
  1549. directory.
  1550. """
  1551. is_dir = osp.exists(source) and osp.isdir(source)
  1552. if is_dir and not osp.exists(pjoin(source, "node_modules")):
  1553. self._run(["node", YARN_PATH, "install"], cwd=source)
  1554. info = {"source": source, "is_dir": is_dir}
  1555. ret = self._run([which("npm"), "pack", source], cwd=tempdir)
  1556. if ret != 0:
  1557. msg = f'"{source}" is not a valid npm package'
  1558. raise ValueError(msg)
  1559. path = glob(pjoin(tempdir, "*.tgz"))[0]
  1560. info["data"] = read_package(path)
  1561. if is_dir:
  1562. info["sha"] = sha = _tarsum(path)
  1563. target = path.replace(".tgz", f"-{sha}.tgz")
  1564. shutil.move(path, target)
  1565. info["path"] = target
  1566. else:
  1567. info["path"] = path
  1568. if pin:
  1569. old_path = info["path"]
  1570. new_path = pjoin(osp.dirname(old_path), f"{PIN_PREFIX}{pin}.tgz")
  1571. shutil.move(old_path, new_path)
  1572. info["path"] = new_path
  1573. info["filename"] = osp.basename(info["path"])
  1574. info["name"] = info["data"]["name"]
  1575. info["version"] = info["data"]["version"]
  1576. return info
  1577. def _latest_compatible_package_version(self, name):
  1578. """Get the latest compatible version of a package"""
  1579. core_data = self.info["core_data"]
  1580. try:
  1581. metadata = _fetch_package_metadata(self.registry, name, self.logger)
  1582. except URLError:
  1583. return
  1584. versions = metadata.get("versions", {})
  1585. # Sort pre-release first, as we will reverse the sort:
  1586. def sort_key(key_value):
  1587. return _semver_key(key_value[0], prerelease_first=True)
  1588. for version, data in sorted(versions.items(), key=sort_key, reverse=True):
  1589. deps = data.get("dependencies", {})
  1590. errors = _validate_compatibility(name, deps, core_data)
  1591. if not errors:
  1592. # Found a compatible version
  1593. # skip deprecated versions
  1594. if "deprecated" in data:
  1595. self.logger.debug(
  1596. f"Disregarding compatible version of package as it is deprecated: {name}@{version}"
  1597. )
  1598. continue
  1599. # Verify that the version is a valid extension.
  1600. with TemporaryDirectory() as tempdir:
  1601. info = self._extract_package(f"{name}@{version}", tempdir)
  1602. if _validate_extension(info["data"]):
  1603. # Invalid, do not consider other versions
  1604. return
  1605. # Valid
  1606. return version
  1607. def latest_compatible_package_versions(self, names):
  1608. """Get the latest compatible versions of several packages
  1609. Like _latest_compatible_package_version, but optimized for
  1610. retrieving the latest version for several packages in one go.
  1611. """
  1612. core_data = self.info["core_data"]
  1613. keys = []
  1614. for name in names:
  1615. try:
  1616. metadata = _fetch_package_metadata(self.registry, name, self.logger)
  1617. except URLError:
  1618. continue
  1619. versions = metadata.get("versions", {})
  1620. # Sort pre-release first, as we will reverse the sort:
  1621. def sort_key(key_value):
  1622. return _semver_key(key_value[0], prerelease_first=True)
  1623. for version, data in sorted(versions.items(), key=sort_key, reverse=True):
  1624. # skip deprecated versions
  1625. if "deprecated" in data:
  1626. continue
  1627. deps = data.get("dependencies", {})
  1628. errors = _validate_compatibility(name, deps, core_data)
  1629. if not errors:
  1630. # Found a compatible version
  1631. keys.append(f"{name}@{version}")
  1632. break # break inner for
  1633. versions = {}
  1634. if not keys:
  1635. return versions
  1636. with TemporaryDirectory() as tempdir:
  1637. ret = self._run([which("npm"), "pack", *keys], cwd=tempdir)
  1638. if ret != 0:
  1639. msg = f'"{keys}" is not a valid npm package'
  1640. raise ValueError(msg)
  1641. for key in keys:
  1642. fname = (
  1643. key[0].replace("@", "") + key[1:].replace("@", "-").replace("/", "-") + ".tgz"
  1644. )
  1645. data = read_package(osp.join(tempdir, fname))
  1646. # Verify that the version is a valid extension.
  1647. if not _validate_extension(data):
  1648. # Valid
  1649. versions[data["name"]] = data["version"]
  1650. return versions
  1651. def _format_no_compatible_package_version(self, name):
  1652. """Get the latest compatible version of a package"""
  1653. core_data = self.info["core_data"]
  1654. # Whether lab version is too new:
  1655. lab_newer_than_latest = False
  1656. # Whether the latest version of the extension depend on a "future" version
  1657. # of a singleton package (from the perspective of current lab version):
  1658. latest_newer_than_lab = False
  1659. try:
  1660. metadata = _fetch_package_metadata(self.registry, name, self.logger)
  1661. except URLError:
  1662. pass
  1663. else:
  1664. versions = metadata.get("versions", {})
  1665. # Sort pre-release first, as we will reverse the sort:
  1666. def sort_key(key_value):
  1667. return _semver_key(key_value[0], prerelease_first=True)
  1668. store = tuple(sorted(versions.items(), key=sort_key, reverse=True))
  1669. latest_deps = store[0][1].get("dependencies", {})
  1670. core_deps = core_data["resolutions"]
  1671. singletons = core_data["jupyterlab"]["singletonPackages"]
  1672. for key, value in latest_deps.items():
  1673. if key in singletons:
  1674. # Drop prereleases in comparisons to allow extension authors
  1675. # to not have to update their versions for each
  1676. # Jupyterlab prerelease version.
  1677. c = _compare_ranges(core_deps[key], value, drop_prerelease1=True)
  1678. lab_newer_than_latest = lab_newer_than_latest or c < 0
  1679. latest_newer_than_lab = latest_newer_than_lab or c > 0
  1680. if lab_newer_than_latest:
  1681. # All singleton deps in current version of lab are newer than those
  1682. # in the latest version of the extension
  1683. return (
  1684. f'The extension "{name}" does not yet support the current version of JupyterLab.\n'
  1685. )
  1686. parts = [
  1687. f"No version of {name} could be found that is compatible with "
  1688. "the current version of JupyterLab."
  1689. ]
  1690. if latest_newer_than_lab:
  1691. parts.extend(
  1692. (
  1693. "However, it seems to support a new version of JupyterLab.",
  1694. "Consider upgrading JupyterLab.",
  1695. )
  1696. )
  1697. return " ".join(parts)
  1698. def _run(self, cmd, **kwargs):
  1699. """Run the command using our logger and abort callback.
  1700. Returns the exit code.
  1701. """
  1702. if self.kill_event.is_set():
  1703. msg = "Command was killed"
  1704. raise ValueError(msg)
  1705. kwargs["logger"] = self.logger
  1706. kwargs["kill_event"] = self.kill_event
  1707. proc = ProgressProcess(cmd, **kwargs)
  1708. return proc.wait()
  1709. def _node_check(logger):
  1710. """Check for the existence of nodejs with the correct version."""
  1711. node = which("node")
  1712. try:
  1713. output = subprocess.check_output([node, "node-version-check.js"], cwd=HERE) # noqa S603
  1714. logger.debug(output.decode("utf-8"))
  1715. except Exception:
  1716. data = CoreConfig()._data
  1717. ver = data["engines"]["node"]
  1718. msg = f"Please install nodejs {ver} before continuing. nodejs may be installed using conda or directly from the nodejs website."
  1719. raise ValueError(msg) from None
  1720. def _yarn_config(logger):
  1721. """Get the yarn configuration.
  1722. Returns
  1723. -------
  1724. {"yarn config": dict, "npm config": dict}
  1725. if unsuccessful, the subdictionaries are empty
  1726. """
  1727. configuration = {"yarn config": {}, "npm config": {}}
  1728. try:
  1729. node = which("node")
  1730. except ValueError: # Node not found == user with no need for building jupyterlab
  1731. logger.debug("NodeJS was not found. Yarn user configuration is ignored.")
  1732. return configuration
  1733. try:
  1734. output_binary = subprocess.check_output( # noqa S603
  1735. [node, YARN_PATH, "config", "--json"],
  1736. stderr=subprocess.PIPE,
  1737. cwd=HERE,
  1738. )
  1739. output = output_binary.decode("utf-8")
  1740. lines = iter(output.splitlines())
  1741. try:
  1742. for line in lines:
  1743. info = json.loads(line)
  1744. if info["type"] == "info":
  1745. key = info["data"]
  1746. inspect = json.loads(next(lines))
  1747. if inspect["type"] == "inspect":
  1748. configuration[key] = inspect["data"]
  1749. except StopIteration:
  1750. pass
  1751. logger.debug("Yarn configuration loaded.")
  1752. except subprocess.CalledProcessError as e:
  1753. logger.error(
  1754. "Fail to get yarn configuration. {!s}{!s}".format(
  1755. e.stderr.decode("utf-8"), e.output.decode("utf-8")
  1756. )
  1757. )
  1758. except Exception as e:
  1759. logger.error(f"Fail to get yarn configuration. {e!s}")
  1760. return configuration
  1761. def _ensure_logger(logger=None):
  1762. """Ensure that we have a logger"""
  1763. return logger or logging.getLogger("jupyterlab")
  1764. def _normalize_path(extension):
  1765. """Normalize a given extension if it is a path."""
  1766. extension = osp.expanduser(extension)
  1767. if osp.exists(extension):
  1768. extension = osp.abspath(extension)
  1769. return extension
  1770. def _rmtree(path, logger):
  1771. """Remove a tree, logging errors"""
  1772. def onerror(*exc_info):
  1773. logger.debug("Error in shutil.rmtree", exc_info=exc_info)
  1774. shutil.rmtree(path, onerror=onerror)
  1775. def _unlink(path, logger):
  1776. """Remove a file, logging errors"""
  1777. try:
  1778. os.unlink(path)
  1779. except Exception:
  1780. logger.debug("Error in os.unlink", exc_info=sys.exc_info())
  1781. def _rmtree_star(path, logger):
  1782. """Remove all files/trees within a dir, logging errors"""
  1783. for filename in os.listdir(path):
  1784. file_path = osp.join(path, filename)
  1785. if osp.isfile(file_path) or osp.islink(file_path):
  1786. _unlink(file_path, logger)
  1787. elif osp.isdir(file_path):
  1788. _rmtree(file_path, logger)
  1789. def _validate_extension(data): # noqa
  1790. """Detect if a package is an extension using its metadata.
  1791. Returns any problems it finds.
  1792. """
  1793. jlab = data.get("jupyterlab", None)
  1794. if jlab is None:
  1795. return ["No `jupyterlab` key"]
  1796. if not isinstance(jlab, dict):
  1797. return ["The `jupyterlab` key must be a JSON object"]
  1798. extension = jlab.get("extension", False)
  1799. mime_extension = jlab.get("mimeExtension", False)
  1800. theme_path = jlab.get("themePath", "")
  1801. schema_dir = jlab.get("schemaDir", "")
  1802. messages = []
  1803. if not extension and not mime_extension:
  1804. messages.append("No `extension` or `mimeExtension` key present")
  1805. if extension == mime_extension:
  1806. msg = "`mimeExtension` and `extension` must point to different modules"
  1807. messages.append(msg)
  1808. files = data["jupyterlab_extracted_files"]
  1809. main = data.get("main", "index.js")
  1810. if not main.endswith(".js"):
  1811. main += ".js"
  1812. if extension is True:
  1813. extension = main
  1814. elif extension and not extension.endswith(".js"):
  1815. extension += ".js"
  1816. if mime_extension is True:
  1817. mime_extension = main
  1818. elif mime_extension and not mime_extension.endswith(".js"):
  1819. mime_extension += ".js"
  1820. if extension and extension not in files:
  1821. messages.append(f'Missing extension module "{extension}"')
  1822. if mime_extension and mime_extension not in files:
  1823. messages.append(f'Missing mimeExtension module "{mime_extension}"')
  1824. if theme_path and not any(f.startswith(str(Path(theme_path))) for f in files):
  1825. messages.append(f'themePath is empty: "{theme_path}"')
  1826. if schema_dir and not any(f.startswith(str(Path(schema_dir))) for f in files):
  1827. messages.append(f'schemaDir is empty: "{schema_dir}"')
  1828. return messages
  1829. def _tarsum(input_file):
  1830. """
  1831. Compute the recursive sha sum of a tar file.
  1832. """
  1833. chunk_size = 100 * 1024
  1834. h = hashlib.new("sha1") # noqa: S324
  1835. with tarfile.open(input_file, "r") as tar:
  1836. for member in tar:
  1837. if not member.isfile():
  1838. continue
  1839. with tar.extractfile(member) as f:
  1840. if f: # Check if f is not None (safety check)
  1841. data = f.read(chunk_size)
  1842. while data:
  1843. h.update(data)
  1844. data = f.read(chunk_size)
  1845. return h.hexdigest()
  1846. def _get_static_data(app_dir):
  1847. """Get the data for the app static dir."""
  1848. target = pjoin(app_dir, "static", "package.json")
  1849. if osp.exists(target):
  1850. with open(target) as fid:
  1851. return json.load(fid)
  1852. else:
  1853. return None
  1854. def _validate_compatibility(extension, deps, core_data):
  1855. """Validate the compatibility of an extension."""
  1856. core_deps = core_data["resolutions"]
  1857. singletons = core_data["jupyterlab"]["singletonPackages"]
  1858. errors = []
  1859. for key, value in deps.items():
  1860. if key in singletons:
  1861. # Drop prereleases in comparisons to allow extension authors
  1862. # to not have to update their versions for each
  1863. # Jupyterlab prerelease version.
  1864. overlap = _test_overlap(core_deps[key], value, drop_prerelease1=True)
  1865. if overlap is False:
  1866. errors.append((key, core_deps[key], value))
  1867. return errors
  1868. def _test_overlap(spec1, spec2, drop_prerelease1=False, drop_prerelease2=False):
  1869. """Test whether two version specs overlap.
  1870. Returns `None` if we cannot determine compatibility,
  1871. otherwise whether there is an overlap
  1872. """
  1873. cmp = _compare_ranges(
  1874. spec1, spec2, drop_prerelease1=drop_prerelease1, drop_prerelease2=drop_prerelease2
  1875. )
  1876. if cmp is None:
  1877. return
  1878. return cmp == 0
  1879. def _compare_ranges(spec1, spec2, drop_prerelease1=False, drop_prerelease2=False): # noqa
  1880. """Test whether two version specs overlap.
  1881. Returns `None` if we cannot determine compatibility,
  1882. otherwise return 0 if there is an overlap, 1 if
  1883. spec1 is lower/older than spec2, and -1 if spec1
  1884. is higher/newer than spec2.
  1885. """
  1886. # Test for overlapping semver ranges.
  1887. r1 = Range(spec1, True)
  1888. r2 = Range(spec2, True)
  1889. # If either range is empty, we cannot verify.
  1890. if not r1.range or not r2.range:
  1891. return
  1892. # Set return_value to a sentinel value
  1893. return_value = False
  1894. # r1.set may be a list of ranges if the range involved an ||, so we need to test for overlaps between each pair.
  1895. for r1set, r2set in itertools.product(r1.set, r2.set):
  1896. x1 = r1set[0].semver
  1897. x2 = r1set[-1].semver
  1898. y1 = r2set[0].semver
  1899. y2 = r2set[-1].semver
  1900. if x1.prerelease and drop_prerelease1:
  1901. x1 = x1.inc("patch")
  1902. if y1.prerelease and drop_prerelease2:
  1903. y1 = y1.inc("patch")
  1904. o1 = r1set[0].operator
  1905. o2 = r2set[0].operator
  1906. # We do not handle (<) specifiers.
  1907. if o1.startswith("<") or o2.startswith("<"):
  1908. continue
  1909. # Handle single value specifiers.
  1910. lx = lte if x1 == x2 else lt
  1911. ly = lte if y1 == y2 else lt
  1912. gx = gte if x1 == x2 else gt
  1913. gy = gte if x1 == x2 else gt
  1914. # Handle unbounded (>) specifiers.
  1915. def noop(x, y, z):
  1916. return True
  1917. if x1 == x2 and o1.startswith(">"):
  1918. lx = noop
  1919. if y1 == y2 and o2.startswith(">"):
  1920. ly = noop
  1921. # Check for overlap.
  1922. if (
  1923. (gte(x1, y1, True) and ly(x1, y2, True))
  1924. or (gy(x2, y1, True) and ly(x2, y2, True))
  1925. or (gte(y1, x1, True) and lx(y1, x2, True))
  1926. or (gx(y2, x1, True) and lx(y2, x2, True))
  1927. ):
  1928. # if we ever find an overlap, we can return immediately
  1929. return 0
  1930. if gte(y1, x2, True):
  1931. if return_value is False:
  1932. # We can possibly return 1
  1933. return_value = 1
  1934. elif return_value == -1:
  1935. # conflicting information, so we must return None
  1936. return_value = None
  1937. continue
  1938. if gte(x1, y2, True):
  1939. if return_value is False:
  1940. return_value = -1
  1941. elif return_value == 1:
  1942. # conflicting information, so we must return None
  1943. return_value = None
  1944. continue
  1945. msg = "Unexpected case comparing version ranges"
  1946. raise AssertionError(msg)
  1947. if return_value is False:
  1948. return_value = None
  1949. return return_value
  1950. def _is_disabled(name, disabled=None):
  1951. """Test whether the package is disabled."""
  1952. disabled = disabled or {}
  1953. for pattern, value in disabled.items():
  1954. # skip packages explicitly marked as not disabled
  1955. if value is False:
  1956. continue
  1957. if name == pattern:
  1958. return True
  1959. if re.compile(pattern).match(name) is not None:
  1960. return True
  1961. return False
  1962. @dataclass(frozen=True)
  1963. class LockStatus:
  1964. entire_extension_locked: bool
  1965. # locked plugins are only given if extension is not locked as a whole
  1966. locked_plugins: Optional[frozenset[str]] = None
  1967. def _is_locked(name, locked=None) -> LockStatus:
  1968. """Test whether the package is locked.
  1969. If only a subset of extension plugins is locked return them.
  1970. """
  1971. locked = locked or {}
  1972. locked_plugins = set()
  1973. for lock, value in locked.items():
  1974. # skip packages explicitly marked as not locked
  1975. if value is False:
  1976. continue
  1977. if name == lock:
  1978. return LockStatus(entire_extension_locked=True)
  1979. extension_part = lock.partition(":")[0]
  1980. if name == extension_part:
  1981. locked_plugins.add(lock)
  1982. return LockStatus(entire_extension_locked=False, locked_plugins=locked_plugins)
  1983. def _format_compatibility_errors(name, version, errors):
  1984. """Format a message for compatibility errors."""
  1985. msgs = []
  1986. l0 = 10
  1987. l1 = 10
  1988. for error in errors:
  1989. pkg, jlab, ext = error
  1990. jlab = str(Range(jlab, True))
  1991. ext = str(Range(ext, True))
  1992. msgs.append((pkg, jlab, ext))
  1993. l0 = max(l0, len(pkg) + 1)
  1994. l1 = max(l1, len(jlab) + 1)
  1995. msg = f'\n"{name}@{version}" is not compatible with the current JupyterLab'
  1996. msg += "\nConflicting Dependencies:\n"
  1997. msg += "JupyterLab".ljust(l0)
  1998. msg += "Extension".ljust(l1)
  1999. msg += "Package\n"
  2000. for pkg, jlab, ext in msgs:
  2001. msg += jlab.ljust(l0) + ext.ljust(l1) + pkg + "\n"
  2002. return msg
  2003. def _log_multiple_compat_errors(logger, errors_map, verbose: bool):
  2004. """Log compatibility errors for multiple extensions at once"""
  2005. outdated = []
  2006. for name, (_, errors) in errors_map.items():
  2007. age = _compat_error_age(errors)
  2008. if age > 0:
  2009. outdated.append(name)
  2010. if outdated:
  2011. logger.warning(
  2012. "\n ".join(
  2013. [
  2014. "\n The following extensions may be outdated or specify dependencies that are incompatible with the current version of jupyterlab:",
  2015. *outdated,
  2016. "\n If you are a user, check if an update is available for these packages.\n"
  2017. + (
  2018. " If you are a developer, re-run with `--verbose` flag for more details.\n"
  2019. if not verbose
  2020. else " See below for the details.\n"
  2021. ),
  2022. ]
  2023. )
  2024. )
  2025. # Print out compatibility errors for all extensions, even the ones inferred
  2026. # to be possibly outdated, to guide developers upgrading their extensions.
  2027. for name, (version, errors) in errors_map.items():
  2028. if name in outdated and not verbose:
  2029. continue
  2030. msg = _format_compatibility_errors(name, version, errors)
  2031. logger.warning(f"{msg}\n")
  2032. def _log_single_compat_errors(logger, name, version, errors):
  2033. """Log compatibility errors for a single extension"""
  2034. age = _compat_error_age(errors)
  2035. if age > 0:
  2036. logger.warning(f'The extension "{name}" is outdated.\n')
  2037. else:
  2038. msg = _format_compatibility_errors(name, version, errors)
  2039. logger.warning(f"{msg}\n")
  2040. def _compat_error_age(errors):
  2041. """Compare all incompatibilities for an extension.
  2042. Returns a number > 0 if all extensions are older than that supported by lab.
  2043. Returns a number < 0 if all extensions are newer than that supported by lab.
  2044. Returns 0 otherwise (i.e. a mix).
  2045. """
  2046. # Do any extensions depend on too old lab packages?
  2047. any_older = False
  2048. # Do any extensions depend on too new lab packages?
  2049. any_newer = False
  2050. for _, jlab, ext in errors:
  2051. # Drop prereleases in comparisons to allow extension authors
  2052. # to not have to update their versions for each
  2053. # Jupyterlab prerelease version.
  2054. c = _compare_ranges(ext, jlab, drop_prerelease1=True)
  2055. any_newer = any_newer or c < 0
  2056. any_older = any_older or c > 0
  2057. if any_older and not any_newer:
  2058. return 1
  2059. elif any_newer and not any_older:
  2060. return -1
  2061. return 0
  2062. def _get_core_extensions(core_data):
  2063. """Get the core extensions."""
  2064. data = core_data["jupyterlab"]
  2065. return list(data["extensions"]) + list(data["mimeExtensions"])
  2066. def _semver_prerelease_key(prerelease):
  2067. """Sort key for prereleases.
  2068. Precedence for two pre-release versions with the same
  2069. major, minor, and patch version MUST be determined by
  2070. comparing each dot separated identifier from left to
  2071. right until a difference is found as follows:
  2072. identifiers consisting of only digits are compare
  2073. numerically and identifiers with letters or hyphens
  2074. are compared lexically in ASCII sort order. Numeric
  2075. identifiers always have lower precedence than non-
  2076. numeric identifiers. A larger set of pre-release
  2077. fields has a higher precedence than a smaller set,
  2078. if all of the preceding identifiers are equal.
  2079. """
  2080. for entry in prerelease:
  2081. if isinstance(entry, int):
  2082. # Assure numerics always sort before string
  2083. yield ("", entry)
  2084. else:
  2085. # Use ASCII compare:
  2086. yield (entry,)
  2087. def _semver_key(version, prerelease_first=False):
  2088. """A sort key-function for sorting semver version string.
  2089. The default sorting order is ascending (0.x -> 1.x -> 2.x).
  2090. If `prerelease_first`, pre-releases will come before
  2091. ALL other semver keys (not just those with same version).
  2092. I.e (1.0-pre, 2.0-pre -> 0.x -> 1.x -> 2.x).
  2093. Otherwise it will sort in the standard way that it simply
  2094. comes before any release with shared version string
  2095. (0.x -> 1.0-pre -> 1.x -> 2.0-pre -> 2.x).
  2096. """
  2097. v = make_semver(version, True)
  2098. key = ((0,) if v.prerelease else (1,)) if prerelease_first else ()
  2099. key = (*key, v.major, v.minor, v.patch)
  2100. if not prerelease_first:
  2101. # NOT having a prerelease is > having one
  2102. key = (*key, 0) if v.prerelease else (1,)
  2103. if v.prerelease:
  2104. key = key + tuple(_semver_prerelease_key(v.prerelease))
  2105. return key
  2106. def _fetch_package_metadata(registry, name, logger):
  2107. """Fetch the metadata for a package from the npm registry"""
  2108. req = Request( # noqa S310
  2109. urljoin(registry, quote(name, safe="@")),
  2110. headers={
  2111. "Accept": ("application/vnd.npm.install-v1+json; q=1.0, application/json; q=0.8, */*")
  2112. },
  2113. )
  2114. try:
  2115. logger.debug(f"Fetching URL: {req.full_url}")
  2116. except AttributeError:
  2117. logger.debug(f"Fetching URL: {req.get_full_url()}")
  2118. try:
  2119. with contextlib.closing(urlopen(req)) as response: # noqa S310
  2120. return json.loads(response.read().decode("utf-8"))
  2121. except URLError as exc:
  2122. logger.warning(f"Failed to fetch package metadata for {name!r}: {exc!r}")
  2123. raise
  2124. if __name__ == "__main__":
  2125. watch_dev(HERE)