Merging PR_218 openai_rev package with new streamlit chat app

venv/lib/python3.9/site-packages/streamlit/git_util.py

@@ -0,0 +1,171 @@
+# Copyright (c) Streamlit Inc. (2018-2022) Snowflake Inc. (2022)
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import os
+import re
+from typing import Any, Optional, Tuple
+
+from streamlit import util
+
+# Github has two URLs, one that is https and one that is ssh
+GITHUB_HTTP_URL = r"^https://(www\.)?github.com/(.+)/(.+)(?:.git)?$"
+GITHUB_SSH_URL = r"^git@github.com:(.+)/(.+)(?:.git)?$"
+
+# We don't support git < 2.7, because we can't get repo info without
+# talking to the remote server, which results in the user being prompted
+# for credentials.
+MIN_GIT_VERSION = (2, 7, 0)
+
+
+class GitRepo:
+    def __init__(self, path):
+        # If we have a valid repo, git_version will be a tuple of 3+ ints:
+        # (major, minor, patch, possible_additional_patch_number)
+        self.git_version: Optional[Tuple[int, ...]] = None
+
+        try:
+            import git
+
+            # GitPython is not fully typed, and mypy is outputting inconsistent
+            # type errors on Mac and Linux. We bypass type checking entirely
+            # by re-declaring the `git` import as an "Any".
+            git_package: Any = git
+            self.repo = git_package.Repo(path, search_parent_directories=True)
+            self.git_version = self.repo.git.version_info
+
+            if self.git_version >= MIN_GIT_VERSION:
+                git_root = self.repo.git.rev_parse("--show-toplevel")
+                self.module = os.path.relpath(path, git_root)
+        except Exception:
+            # The git repo must be invalid for the following reasons:
+            #  * git binary or GitPython not installed
+            #  * No .git folder
+            #  * Corrupted .git folder
+            #  * Path is invalid
+            self.repo = None
+
+    def __repr__(self) -> str:
+        return util.repr_(self)
+
+    def is_valid(self) -> bool:
+        """True if there's a git repo here, and git.version >= MIN_GIT_VERSION."""
+        return (
+            self.repo is not None
+            and self.git_version is not None
+            and self.git_version >= MIN_GIT_VERSION
+        )
+
+    @property
+    def tracking_branch(self):
+        if not self.is_valid():
+            return None
+
+        if self.is_head_detached:
+            return None
+
+        return self.repo.active_branch.tracking_branch()
+
+    @property
+    def untracked_files(self):
+        if not self.is_valid():
+            return None
+
+        return self.repo.untracked_files
+
+    @property
+    def is_head_detached(self):
+        if not self.is_valid():
+            return False
+
+        return self.repo.head.is_detached
+
+    @property
+    def uncommitted_files(self):
+        if not self.is_valid():
+            return None
+
+        return [item.a_path for item in self.repo.index.diff(None)]
+
+    @property
+    def ahead_commits(self):
+        if not self.is_valid():
+            return None
+
+        try:
+            remote, branch_name = self.get_tracking_branch_remote()
+            remote_branch = "/".join([remote.name, branch_name])
+
+            return list(self.repo.iter_commits(f"{remote_branch}..{branch_name}"))
+        except Exception:
+            return list()
+
+    def get_tracking_branch_remote(self):
+        if not self.is_valid():
+            return None
+
+        tracking_branch = self.tracking_branch
+
+        if tracking_branch is None:
+            return None
+
+        remote_name, *branch = tracking_branch.name.split("/")
+        branch_name = "/".join(branch)
+
+        return self.repo.remote(remote_name), branch_name
+
+    def is_github_repo(self):
+        if not self.is_valid():
+            return False
+
+        remote_info = self.get_tracking_branch_remote()
+        if remote_info is None:
+            return False
+
+        remote, _branch = remote_info
+
+        for url in remote.urls:
+            if (
+                re.match(GITHUB_HTTP_URL, url) is not None
+                or re.match(GITHUB_SSH_URL, url) is not None
+            ):
+                return True
+
+        return False
+
+    def get_repo_info(self):
+        if not self.is_valid():
+            return None
+
+        remote_info = self.get_tracking_branch_remote()
+        if remote_info is None:
+            return None
+
+        remote, branch = remote_info
+
+        repo = None
+        for url in remote.urls:
+            https_matches = re.match(GITHUB_HTTP_URL, url)
+            ssh_matches = re.match(GITHUB_SSH_URL, url)
+            if https_matches is not None:
+                repo = f"{https_matches.group(2)}/{https_matches.group(3)}"
+                break
+
+            if ssh_matches is not None:
+                repo = f"{ssh_matches.group(1)}/{ssh_matches.group(2)}"
+                break
+
+        if repo is None:
+            return None
+
+        return repo, branch, self.module