54 lines
2.1 KiB
Python
54 lines
2.1 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
from typing import TYPE_CHECKING
|
||
|
|
|
||
|
|
from argparse import Namespace, ArgumentParser
|
||
|
|
|
||
|
|
from ...lib.log import *
|
||
|
|
from ...lib.ExecContext import InputMode
|
||
|
|
from ..Cmd import Cmd
|
||
|
|
from ..CmdProjects import CmdProjects
|
||
|
|
from ...App import Scope
|
||
|
|
|
||
|
|
if TYPE_CHECKING:
|
||
|
|
from ...lib.ExecContext import Result
|
||
|
|
|
||
|
|
class CmdCanonicalizeRemotes(Cmd): # export
|
||
|
|
|
||
|
|
def __rewrite_url(self, url: str) -> str:
|
||
|
|
return url.replace('/srv/git', '')
|
||
|
|
|
||
|
|
def __init__(self, parent: CmdProjects) -> None:
|
||
|
|
super().__init__(parent, 'canonicalize-remotes', help='Streamline janware Git remotes')
|
||
|
|
|
||
|
|
def add_arguments(self, parser: ArgumentParser) -> None:
|
||
|
|
super().add_arguments(parser)
|
||
|
|
parser.add_argument('-n', '--dry-run', default=False, action='store_true', help='Only log what would be done')
|
||
|
|
|
||
|
|
async def _run(self, args: Namespace) -> None:
|
||
|
|
|
||
|
|
async def git(cmd: list[str], ro=False, throw=True) -> Result:
|
||
|
|
cmd = ['/usr/bin/git', *cmd]
|
||
|
|
log(NOTICE, f'-- {" ".join(cmd)}')
|
||
|
|
if ro or not args.dry_run:
|
||
|
|
return await self.app.exec_context.run(cmd, cmd_input=InputMode.NonInteractive, throw=throw)
|
||
|
|
remotes: dict[str, dict[str, str]] = {}
|
||
|
|
stdout, stderr, status = await git(['remote', '-v'], ro=True)
|
||
|
|
for line in stdout.decode().splitlines():
|
||
|
|
name, url, fp = line.split()
|
||
|
|
remote = remotes.setdefault(name, {})
|
||
|
|
key = 'url' if fp == '(fetch)' else 'pushurl'
|
||
|
|
fpurls = remote.setdefault(key, [])
|
||
|
|
fpurls.append(url)
|
||
|
|
for remote, config in remotes.items():
|
||
|
|
dirty_keys: set[str] = set()
|
||
|
|
for key, urls in config.items():
|
||
|
|
for url in urls:
|
||
|
|
if url != self.__rewrite_url(url):
|
||
|
|
dirty_keys.add(key)
|
||
|
|
for key in dirty_keys:
|
||
|
|
urls = config[key]
|
||
|
|
await git(['config', '--unset-all', f'remote.{remote}.{key}'], throw=False)
|
||
|
|
for url in urls:
|
||
|
|
await git(['config', '--add', f'remote.{remote}.{key}', self.__rewrite_url(url)])
|