mirror of
https://github.com/spalencsar/bearhub.git
synced 2026-07-07 00:04:15 +02:00
145 lines
4.8 KiB
Python
145 lines
4.8 KiB
Python
import json
|
|
import os
|
|
import re
|
|
from pathlib import Path
|
|
from typing import Set, List
|
|
|
|
from bauh.gems.arch import pacman
|
|
from bauh.gems.arch.model import ArchPackage
|
|
|
|
RE_DESKTOP_ENTRY = re.compile(r'(Exec|Icon)\s*=\s*(.+)')
|
|
RE_CLEAN_NAME = re.compile(r'^(\w+)-?|_?.+')
|
|
|
|
|
|
def write(app: ArchPackage):
|
|
data = app.get_data_to_cache()
|
|
|
|
if data:
|
|
Path(app.get_disk_cache_path()).mkdir(parents=True, exist_ok=True)
|
|
|
|
with open(app.get_disk_data_path(), 'w+') as f:
|
|
f.write(json.dumps(data))
|
|
|
|
|
|
def fill_icon_path(app: ArchPackage, icon_paths: List[str], only_exact_match: bool):
|
|
ends_with = re.compile(r'.+/{}\.(png|svg|xpm)$'.format(app.icon_path if app.icon_path else app.name), re.IGNORECASE)
|
|
|
|
for path in icon_paths:
|
|
if ends_with.match(path):
|
|
app.icon_path = path
|
|
return
|
|
|
|
if not only_exact_match:
|
|
pkg_icons_path = pacman.list_icon_paths({app.name})
|
|
|
|
if pkg_icons_path:
|
|
app.set_icon(pkg_icons_path)
|
|
|
|
|
|
def set_icon_path(app: ArchPackage, icon_name: str = None):
|
|
installed_icons = pacman.list_icon_paths({app.name})
|
|
|
|
if installed_icons:
|
|
exact_match = re.compile(r'.+/{}\..+$'.format(icon_name.split('.')[0] if icon_name else app.name))
|
|
for icon_path in installed_icons:
|
|
if exact_match.match(icon_path):
|
|
app.icon_path = icon_path
|
|
break
|
|
|
|
|
|
def save_several(pkgnames: Set[str], mirror: str, overwrite: bool = True, maintainer: str = None) -> int:
|
|
to_cache = {n for n in pkgnames if overwrite or not os.path.exists(ArchPackage.disk_cache_path(n, mirror))}
|
|
desktop_files = pacman.list_desktop_entries(to_cache)
|
|
|
|
no_desktop_files = {}
|
|
|
|
to_write = []
|
|
if desktop_files:
|
|
desktop_matches, no_exact_match = {}, set()
|
|
for pkg in to_cache: # first try to find exact matches
|
|
ends_with = re.compile('/usr/share/applications/{}.desktop$'.format(pkg), re.IGNORECASE)
|
|
|
|
for f in desktop_files:
|
|
if ends_with.match(f):
|
|
desktop_matches[pkg] = f
|
|
break
|
|
|
|
if pkg not in desktop_matches:
|
|
no_exact_match.add(pkg)
|
|
if no_exact_match: # check every not matched app individually
|
|
for pkg in no_exact_match:
|
|
entries = pacman.list_desktop_entries({pkg})
|
|
|
|
if entries:
|
|
desktop_matches[pkg] = entries[0]
|
|
|
|
if len(entries) > 1:
|
|
for e in entries:
|
|
if e.startswith('/usr/share/applications'):
|
|
desktop_matches[pkg] = e
|
|
break
|
|
|
|
if not desktop_matches:
|
|
no_desktop_files = to_cache
|
|
else:
|
|
if len(desktop_matches) != len(to_cache):
|
|
no_desktop_files = {p for p in to_cache if p not in desktop_matches}
|
|
|
|
pkgs, apps_icons_noabspath = [], []
|
|
|
|
for pkgname, file in desktop_matches.items():
|
|
p = ArchPackage(name=pkgname, mirror=mirror)
|
|
p.desktop_entry = file
|
|
|
|
with open(file) as f:
|
|
desktop_entry = f.read()
|
|
|
|
for field in RE_DESKTOP_ENTRY.findall(desktop_entry):
|
|
if field[0] == 'Exec':
|
|
p.command = field[1].strip().replace('"', '')
|
|
elif field[0] == 'Icon':
|
|
p.icon_path = field[1].strip()
|
|
|
|
if p.icon_path and '/' not in p.icon_path: # if the icon full path is not defined
|
|
apps_icons_noabspath.append(p)
|
|
|
|
pkgs.append(p)
|
|
|
|
if apps_icons_noabspath:
|
|
icon_paths = pacman.list_icon_paths({app.name for app in apps_icons_noabspath})
|
|
if icon_paths:
|
|
for p in apps_icons_noabspath:
|
|
fill_icon_path(p, icon_paths, False)
|
|
|
|
for p in pkgs:
|
|
to_write.append(p)
|
|
|
|
if no_desktop_files:
|
|
pkgs = {ArchPackage(name=n, mirror=mirror) for n in no_desktop_files}
|
|
bin_paths = pacman.list_bin_paths(no_desktop_files)
|
|
|
|
if bin_paths:
|
|
for p in pkgs:
|
|
ends_with = re.compile(r'.+/{}$'.format(p.name), re.IGNORECASE)
|
|
|
|
for path in bin_paths:
|
|
if ends_with.match(path):
|
|
p.command = path
|
|
break
|
|
|
|
icon_paths = pacman.list_icon_paths(no_desktop_files)
|
|
|
|
if icon_paths:
|
|
for p in pkgs:
|
|
fill_icon_path(p, icon_paths, only_exact_match=True)
|
|
|
|
for p in pkgs:
|
|
to_write.append(p)
|
|
|
|
if to_write:
|
|
for p in to_write:
|
|
p.maintainer = maintainer
|
|
write(p)
|
|
return len(to_write)
|
|
return 0
|