commands.py 79 KB

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