mirror of
https://github.com/spalencsar/bearhub.git
synced 2026-07-06 22:54:16 +02:00
146 lines
4.5 KiB
Python
146 lines
4.5 KiB
Python
import re
|
|
import subprocess
|
|
from typing import List
|
|
|
|
import requests
|
|
from bs4 import BeautifulSoup
|
|
|
|
from fpakman.core import system
|
|
from fpakman.core.snap.constants import SNAP_STORE_URL
|
|
|
|
BASE_CMD = 'snap'
|
|
|
|
|
|
def is_installed():
|
|
version = get_snapd_version()
|
|
return False if version is None else True
|
|
|
|
|
|
def get_version():
|
|
res = system.run_cmd('{} --version'.format(BASE_CMD), print_error=False)
|
|
return res.split('\n')[0].split(' ')[-1].strip() if res else None
|
|
|
|
|
|
def get_snapd_version():
|
|
res = system.run_cmd('{} --version'.format(BASE_CMD), print_error=False)
|
|
|
|
if not res:
|
|
return None
|
|
else:
|
|
lines = res.split('\n')
|
|
|
|
if lines and len(lines) >= 2:
|
|
version = lines[1].split(' ')[-1].strip()
|
|
return version if version and version.lower() != 'unavailable' else None
|
|
else:
|
|
return None
|
|
|
|
|
|
def app_str_to_json(app: str) -> dict:
|
|
app_data = [word for word in app.split(' ') if word]
|
|
app_json = {
|
|
'name': app_data[0],
|
|
'version': app_data[1],
|
|
'rev': app_data[2],
|
|
'tracking': app_data[3],
|
|
'publisher': app_data[4],
|
|
'notes': app_data[5]
|
|
}
|
|
|
|
app_json.update(get_info(app_json['name'], ('summary', 'type', 'description')))
|
|
return app_json
|
|
|
|
|
|
def get_info(app_name: str, attrs: tuple = None):
|
|
full_info_lines = system.run_cmd('{} info {}'.format(BASE_CMD, app_name))
|
|
|
|
data = {}
|
|
|
|
if full_info_lines:
|
|
re_attrs = r'\w+' if not attrs else '|'.join(attrs)
|
|
info_map = re.findall(r'({}):\s+(.+)'.format(re_attrs), full_info_lines)
|
|
|
|
for info in info_map:
|
|
data[info[0]] = info[1].strip()
|
|
|
|
if not attrs or 'description' in attrs:
|
|
desc = re.findall(r'\|\n+((\s+.+\n+)+)', full_info_lines)
|
|
data['description'] = ''.join([w.strip() for w in desc[0][0].strip().split('\n')]).replace('.', '.\n') if desc else None
|
|
|
|
if not attrs or 'commands' in attrs:
|
|
commands = re.findall(r'commands:\s*\n*((\s+-\s.+\s*\n)+)', full_info_lines)
|
|
data['commands'] = commands[0][0].replace('-', '').strip().split('\n') if commands else None
|
|
|
|
return data
|
|
|
|
|
|
def read_installed() -> List[dict]:
|
|
res = system.run_cmd('{} list'.format(BASE_CMD), print_error=False)
|
|
|
|
apps = []
|
|
|
|
if res and len(res) > 0:
|
|
lines = res.split('\n')
|
|
|
|
if not lines[0].startswith('error'):
|
|
for idx, app_str in enumerate(lines):
|
|
if idx > 0 and app_str:
|
|
apps.append(app_str_to_json(app_str))
|
|
|
|
return apps
|
|
|
|
|
|
def search(word: str) -> List[dict]:
|
|
apps = []
|
|
|
|
res = system.run_cmd('{} find "{}"'.format(BASE_CMD, word), print_error=False)
|
|
|
|
if res:
|
|
res = res.split('\n')
|
|
|
|
if not res[0].startswith('No matching'):
|
|
for idx, app_str in enumerate(res):
|
|
if idx > 0 and app_str:
|
|
app_data = [word for word in app_str.split(' ') if word]
|
|
apps.append({
|
|
'name': app_data[0],
|
|
'version': app_data[1],
|
|
'publisher': app_data[2],
|
|
'notes': app_data[3] if app_data[3] != '-' else None,
|
|
'summary': app_data[4] if len(app_data) == 5 else '',
|
|
'rev': None,
|
|
'tracking': None,
|
|
'type': None
|
|
})
|
|
|
|
return apps
|
|
|
|
|
|
def uninstall_and_stream(app_name: str, root_password: str):
|
|
return system.cmd_as_root([BASE_CMD, 'remove', app_name], root_password)
|
|
|
|
|
|
def install_and_stream(app_name: str, custom_install_cmd: str, root_password: str) -> subprocess.Popen:
|
|
|
|
install_cmd = [BASE_CMD, 'install', app_name] # default
|
|
|
|
if custom_install_cmd:
|
|
install_cmd = custom_install_cmd.split(' ')
|
|
else: # tries to retrieve the snapstore proper installation command
|
|
res = requests.get('{}/{}'.format(SNAP_STORE_URL, app_name))
|
|
|
|
if res.status_code == 200:
|
|
soup = BeautifulSoup(res.text, 'html.parser')
|
|
input_install_cmd = soup.find("input", {"id": "snap-install"})
|
|
install_cmd = input_install_cmd.get("value").strip().split(' ')
|
|
|
|
return system.cmd_as_root(install_cmd, root_password)
|
|
|
|
|
|
def downgrade_and_stream(app_name: str, root_password: str) -> subprocess.Popen:
|
|
return system.cmd_as_root([BASE_CMD, 'revert', app_name], root_password)
|
|
|
|
|
|
def refresh_and_stream(app_name: str, root_password: str) -> subprocess.Popen:
|
|
return system.cmd_as_root([BASE_CMD, 'refresh', app_name], root_password)
|