66 lines
1.5 KiB
Python
Executable file
66 lines
1.5 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
import argparse
|
|
import re
|
|
import sys
|
|
import os
|
|
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",
|
|
)
|
|
|
|
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",
|
|
)
|
|
|
|
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"):
|
|
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]}`")
|