feat(script): add new aur check

Signed-off-by: 67au <[email protected]>

1 files changed, 110 insertions, 0 deletions

diff --git a/utils/aur/check-aur-update.py b/utils/aur/check-aur-update.py
new file mode 100644
index 0000000..85f02c2
--- /dev/null
+++ b/utils/aur/check-aur-update.py
@@ -0,0 +1,110 @@
+import asyncio
+import argparse
+import logging
+from collections.abc import Coroutine
+from pathlib import Path
+from typing import Optional
+
+import aiohttp
+import aiofiles
+
+from AURUtils.srcinfo import SRCINFO
+from AURUtils.version import vercmp
+
+
+MAX_WORKERS = 16
+
+
+async def check_aur_update(
+        session: aiohttp.ClientSession,
+        item: Path, dry_run: bool,
+        parse_pkgbuild: Optional[Path] = None,
+) -> None:
+    if item.name.startswith('.') or not item.is_dir():
+        return
+    pkgbuild_file = item.joinpath('PKGBUILD')
+    if not pkgbuild_file.exists():
+        logging.warning(f'PKGBUILD not in {str(item)}, SKIP!')
+        return
+    srcinfo_file = item.joinpath('.SRCINFO')
+    if not srcinfo_file.exists():
+        if parse_pkgbuild.exists():
+            logging.warning(f'.SRCINFO not in {str(item)}, try to parse')
+            proc = await asyncio.create_subprocess_exec(
+                f'{str(parse_pkgbuild.resolve())}',
+                f'{str(pkgbuild_file)}',
+                stdout=asyncio.subprocess.PIPE,
+            )
+            async with aiofiles.open(srcinfo_file, 'w') as f:
+                await f.write((await proc.stdout.read()).decode())
+            logging.info(f'extract .SRCINFO in {str(item)}')
+        else:
+            logging.warning(f'.SRCINFO not in {str(item)}, no tool to parse')
+            return
+    async with aiofiles.open(srcinfo_file, 'r') as f:
+        srcinfo = SRCINFO.parse(await f.read())
+        version = srcinfo.get_version()
+    async with session.get(f'https://aur.archlinux.org/cgit/aur.git/plain/.SRCINFO?h={srcinfo.pkgbase}') as resp:
+        if resp.status != 200:
+            logging.info(f'{item.name} not register in AUR')
+            return
+        new_version = SRCINFO.parse(await resp.text()).get_version()
+        result = vercmp(new_version, version)
+        if result == 0:
+            return
+        elif result < 0:
+            logging.warning(f'Warning: {version} is newer than AUR')
+        else:
+            if item.joinpath('.git').exists():
+                if not dry_run:
+                    await asyncio.create_subprocess_exec('git', '-C', f'{str(item)}', 'pull', 'origin', 'master')
+                logging.info(f'Upgrade {item.name} from {version} to {new_version}')
+            else:
+                logging.info(f'Found update {item.name}({new_version}) (local: {version})')
+
+
+async def with_sem(sem: asyncio.Semaphore, coro: Coroutine):
+    async with sem:
+        return await coro
+
+
+async def main() -> int:
+    parser = argparse.ArgumentParser(description='aur-check-update.py')
+    parser.add_argument('--dry', help='dry run', action='store_true')
+    parser.add_argument('--parse-pkgbuild', dest='script', metavar='SCRIPT',
+                        help='a script to parse pkgbuild',
+                        action='store')
+    parser.add_argument('--max-workers', dest='workers', metavar='MAX_WORKERS',
+                        help=f'aur check max workers, default: {MAX_WORKERS}',
+                        default=MAX_WORKERS, action='store', type=int)
+    parser.add_argument(dest='repo_dir', metavar='REPO_DIR', help='repository directory', nargs='?')
+
+    args = parser.parse_args()
+
+    dry_run = args.dry
+    sem = asyncio.Semaphore(args.workers)
+
+    if args.repo_dir is None:
+        logging.error('repository directory not found')
+        return 1
+
+    script_file = Path(args.script)
+    if args.script is None or not (script_file.is_file() and script_file.exists()):
+        logging.info(f'Parse script {str(script_file)+" " if args.script is None else ""} not found')
+        script_file = None
+
+    repo_dir = Path(args.repo_dir)
+
+    async with aiohttp.ClientSession(timeout=aiohttp.ClientTimeout(10)) as session:
+        tasks = [asyncio.create_task(
+            with_sem(sem, check_aur_update(session, item, dry_run, script_file))
+        ) for item in repo_dir.iterdir()]
+        await asyncio.gather(*tasks)
+
+    return 0
+
+
+if __name__ == '__main__':
+    logging.basicConfig(level=logging.DEBUG,
+                        format='%(asctime)s - %(levelname)s - %(funcName)s - %(lineno)d - %(message)s')
+    asyncio.run(main())
\ No newline at end of file