sapling/eden/fs/cli/cmd_util.py

#!/usr/bin/env python3
# Copyright (c) Facebook, Inc. and its affiliates.
#
# This software may be used and distributed according to the terms of the
# GNU General Public License version 2.

import argparse
import os
from pathlib import Path
from typing import Optional, Tuple, Union

from . import config as config_mod, subcmd as subcmd_mod
from .config import EdenCheckout, EdenInstance


def get_eden_instance(args: argparse.Namespace) -> EdenInstance:
    return EdenInstance(
        args.config_dir, etc_eden_dir=args.etc_eden_dir, home_dir=args.home_dir
    )


def find_checkout(
    args: argparse.Namespace, path: Union[Path, str, None]
) -> Tuple[EdenInstance, Optional[EdenCheckout], Optional[Path]]:
    if path is None:
        path = os.getcwd()
    return config_mod.find_eden(
        path,
        etc_eden_dir=args.etc_eden_dir,
        home_dir=args.home_dir,
        state_dir=args.config_dir,
    )


def require_checkout(
    args: argparse.Namespace, path: Union[Path, str, None]
) -> Tuple[EdenInstance, EdenCheckout, Path]:
    instance, checkout, rel_path = find_checkout(args, path)
    if checkout is None:
        msg_path = path if path is not None else os.getcwd()
        raise subcmd_mod.CmdError(f"no Eden checkout found at {msg_path}\n")
    assert rel_path is not None
    return instance, checkout, rel_path