dotfiles/bin/clone_with_ssh.py

66 lines
1.5 KiB
Python
Raw Normal View History

2023-03-17 13:57:57 -06:00
#!/usr/bin/env python3
import argparse
import re
import sys
import os
2023-03-17 13:57:57 -06:00
from subprocess import Popen
SSH_RE = re.compile(r"(ssh://)?git@([^/]+):([^/]+)/([^/]+)(/[^/]+)?.*")
HTTP_RE = re.compile(r"https://([^/]+)/([^/]+)/([^/]+).*")
parser = argparse.ArgumentParser(
prog="Clone with SSH v1.0.0",
description="Git-clone an http link as ssh instead",
)
parser.add_argument(
"--depth",
type=int,
required=False,
help="Max depth to clone",
)
2023-03-17 13:57:57 -06:00
parser.add_argument(
"url",
type=str,
help="HTTP or ssh url to use",
)
parser.add_argument(
"out_name",
type=str,
nargs="?",
help="Name to use for the directory",
)
2023-03-17 13:57:57 -06:00
args = parser.parse_args()
if m := HTTP_RE.fullmatch(args.url):
ssh_url = f"ssh://git@{m[1]}:22/{m[2]}/{m[3]}"
elif m := SSH_RE.fullmatch(args.url):
ssh_url = f"ssh://git@{m[2]}:22/{m[3]}/{m[4]}"
else:
print(f"Failed to parse URL `{args.url}`", file=sys.stderr)
exit(1)
if "git.sr.ht" not in args.url and not ssh_url.endswith(".git"):
2023-03-17 13:57:57 -06:00
ssh_url += ".git"
if args.depth is not None:
git_cmd = ["git", "clone", "--depth", str(args.depth), ssh_url]
else:
git_cmd = ["git", "clone", ssh_url]
if args.out_name is not None:
git_cmd.append(args.out_name)
initial_dirs = set(filter(lambda x: os.path.isdir(x), os.listdir()))
Popen(git_cmd).wait()
current_dirs = set(filter(lambda x: os.path.isdir(x), os.listdir()))
new_dirs = current_dirs - initial_dirs
if len(new_dirs) == 1:
print(f"====\nCloned in `{list(new_dirs)[0]}`")