git-get 0.1.3__py3-none-any.whl

git_get-0.1.3.dist-info/METADATA

@@ -0,0 +1,9 @@
+Metadata-Version: 2.1
+Name: git-get
+Version: 0.1.3
+Summary: Automatically clone Git repos.
+Author-email: Daniel Rocha <daniel@rocha.io>
+Requires-Python: >=3.10
+Requires-Dist: typer >= 0.6.1
+Requires-Dist: rich >= 12.6.0
+Project-URL: homepage, https://github.com/danroc/git-get

git_get-0.1.3.dist-info/RECORD

@@ -0,0 +1,5 @@
+git_get.py,sha256=kZk8n77WCA5it9UJMuZLOdw3Goe1q8imgU-BiMXe6SM,2284
+git_get-0.1.3.dist-info/entry_points.txt,sha256=_WPI8vSIOd7cPrkzukgO5O_SJYcYWtPHJD5e-DBiQJw,38
+git_get-0.1.3.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
+git_get-0.1.3.dist-info/METADATA,sha256=TNfp-qaDiKXs6B00cNkN9tlVuEaY6nzPUwAvWGh0ONk,277
+git_get-0.1.3.dist-info/RECORD,,

git_get-0.1.3.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: flit 3.9.0
+Root-Is-Purelib: true
+Tag: py3-none-any

git_get-0.1.3.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+git-get=gitget:app
+

git_get.py

@@ -0,0 +1,101 @@
+"""Automatically clone Git repos."""
+
+from __future__ import annotations
+
+import enum
+import os
+import re
+import subprocess  # noqa: S404
+import sys
+from dataclasses import dataclass
+
+import typer
+
+__version__ = "0.1.3"
+
+app = typer.Typer()
+
+
+class Options(str, enum.Enum):
+    REPOS_DIR = "GIT_GET_REPOS_DIR"
+    SSH_USERS = "GIT_GET_SSH_USERS"
+    DEFAULT_PREFIX = "GIT_GET_DEFAULT_PREFIX"
+
+
+# Default configuration, which can be overridden by environment variables.
+config = {
+    Options.REPOS_DIR: "~/repos",
+    Options.SSH_USERS: "danroc,metamask",
+    Options.DEFAULT_PREFIX: "https://github.com/",
+}
+
+config = {k: os.getenv(k, config[k]) for k in config}
+
+
+class InvalidURL(Exception):
+    pass
+
+
+class Schema(enum.Enum):
+    SSH = enum.auto()
+    HTTP = enum.auto()
+
+
+FORMATS = {
+    Schema.SSH: re.compile(
+        r"^(.*?)@(?P<host>.*?):(?P<user>.*?)/(?P<repo>.*?)\.git$",
+    ),
+    Schema.HTTP: re.compile(
+        r"^https?://(?P<host>.*?)(?P<port>:\d+)?/(?P<user>.*?)/(?P<repo>.*?)\.git$",
+    ),
+}
+
+
+@dataclass(frozen=True)
+class Repo:
+    host: str
+    user: str
+    name: str
+
+
+def parse(url: str) -> Repo:
+    for format in FORMATS.values():
+        m = format.match(url)
+        if m:
+            return Repo(
+                host=m.group("host"),
+                user=m.group("user"),
+                name=m.group("repo"),
+            )
+    raise InvalidURL
+
+
+@app.command()
+def main(repo_url: str) -> None:
+    clone_url = repo_url.strip()
+    if not clone_url.endswith(".git"):
+        clone_url += ".git"
+
+    try:
+        # Try to parse the URL
+        repo = parse(clone_url)
+    except InvalidURL:
+        # Retry with the prefix
+        clone_url = config[Options.DEFAULT_PREFIX] + clone_url
+        repo = parse(clone_url)
+
+        # Check if should force SSH
+        use_ssh_on = config[Options.SSH_USERS].lower().split(",")
+        if repo.user.lower() in use_ssh_on:
+            clone_url = f"git@{repo.host}:{repo.user}/{repo.name}.git"
+
+    path = os.path.join(config[Options.REPOS_DIR], repo.host, repo.user, repo.name)
+    path = os.path.expanduser(path)
+
+    print(f"Cloning repo '{clone_url}'...")
+    out = subprocess.run(["git", "clone", clone_url, path])  # noqa: S603,S607
+    sys.exit(out.returncode)
+
+
+if __name__ == "__main__":
+    app()