Jan Lindemann
888c0495ec
Add lib.base to provide basic definitions.
For now, move the definiions of Result, Input and InputMode from ExecContext into lib.base. Having to import them from the ExecContect module is too heavy-handed for those simple types.
Signed-off-by: Jan Lindemann <jan@janware.com>
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.base import InputMode
|
|
from ..Cmd import Cmd
|
|
from ..CmdProjects import CmdProjects
|
|
from ...App import Scope
|
|
|
|
if TYPE_CHECKING:
|
|
from ...lib.base 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)])
|