sapling/eden/scm/edenscm/eden_dirstate_fs.py

# Copyright (c) Meta Platforms, Inc. and affiliates.
#
# This software may be used and distributed according to the terms of the
# GNU General Public License version 2.

"""Eden implementation for the dirstate filesystem class."""

from typing import Callable, Iterable, Optional, Tuple

from . import filesystem, perftrace, pycompat, util
from .EdenThriftClient import ScmFileStatus
from .i18n import _
from .pycompat import decodeutf8


class eden_filesystem(filesystem.physicalfilesystem):
    def pendingchanges(
        self, match: "Optional[Callable[[str], bool]]" = None, listignored: bool = False
    ) -> "Iterable[Tuple[str, bool]]":
        if match is None:
            match = util.always

        with perftrace.trace("Get EdenFS Status"):
            perftrace.traceflag("status")
            edenstatus = self.dirstate.eden_client.getStatus(
                self.dirstate.p1(), list_ignored=listignored
            )

        MODIFIED = ScmFileStatus.MODIFIED
        REMOVED = ScmFileStatus.REMOVED
        ADDED = ScmFileStatus.ADDED
        IGNORED = ScmFileStatus.IGNORED

        for path, code in pycompat.iteritems(edenstatus):
            if not util.isvalidutf8(path):
                self.ui.warn(_("skipping invalid utf-8 filename: '%s'\n") % path)
                continue

            path = decodeutf8(path)
            if not match(path):
                continue

            if code == MODIFIED or code == ADDED:
                yield (path, True)
            elif code == REMOVED:
                yield (path, False)
            elif code == IGNORED and listignored:
                yield (path, True)
            else:
                raise RuntimeError(
                    "unexpected status code '%s' for '%s'" % (code, path)
                )