tools/jj/new_workspace.py

#!/usr/bin/env python3
# Copyright 2026 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Creates a new jj chromium workspace in a separate directory."""

import argparse
import os
import pathlib
import shutil
import sys

import util

_SUCCESS_MSG = \
'''A new jj workspace has been created, but submodules have not been synced.
Run `gclient sync` to sync submodules.'''


def main(args):
  parser = argparse.ArgumentParser(
      description=__doc__, formatter_class=argparse.RawTextHelpFormatter)
  parser.add_argument('workspace',
                      help='The directory to create the new workspace in.',
                      type=pathlib.Path)
  parser.add_argument('--name', help='The name of the new workspace.')
  parser.add_argument('-r',
                      '--revision',
                      help='The revision to create the workspace at',
                      default='@')

  opts = parser.parse_args(args)

  ws = opts.workspace
  name = opts.name or ws.name
  if ws.exists():
    print(f'Error: {ws} already exists.', file=sys.stderr)
    return 1

  source = pathlib.Path(os.environ['JJ_WORKSPACE_ROOT']).parent

  try:
    ws.mkdir()
    (ws / '.gclient').symlink_to((source / '.gclient').resolve())
    util.run_jj(
        ['workspace', 'add', ws / 'src', '-r', opts.revision, '--name', name])

  except:
    # Ensure that we can't have a partially created workspace.
    shutil.rmtree(opts.workspace)
    raise

  print(_SUCCESS_MSG)
  return 0


if __name__ == '__main__':
  sys.exit(main(sys.argv[1:]))