This commit is contained in:
Azalea (on HyDEV-Daisy) 2022-08-23 11:01:05 -04:00
commit 25ca95df8a
No known key found for this signature in database
GPG key ID: E289FAC0DA92DD2B
5 changed files with 75 additions and 13 deletions

4
hyfetch/__main__.py Normal file
View file

@ -0,0 +1,4 @@
from hyfetch import main
if __name__ == '__main__':
main.run()

View file

@ -46,3 +46,5 @@ class GlobalConfig:
GLOBAL_CFG = GlobalConfig(color_mode='8bit', override_distro=None, debug=False, is_light=False) GLOBAL_CFG = GlobalConfig(color_mode='8bit', override_distro=None, debug=False, is_light=False)
MINGIT_URL = 'https://github.com/git-for-windows/git/releases/download/v2.37.2.windows.2/MinGit-2.37.2.2-busybox-32-bit.zip'

View file

@ -336,6 +336,7 @@ def run():
parser.add_argument('--debug', action='store_true', help=f'Debug mode') parser.add_argument('--debug', action='store_true', help=f'Debug mode')
parser.add_argument('--test-distro', help=f'Test for a specific distro') parser.add_argument('--test-distro', help=f'Test for a specific distro')
parser.add_argument('--test-print', action='store_true', help=f'Test print distro ascii art only') parser.add_argument('--test-print', action='store_true', help=f'Test print distro ascii art only')
parser.add_argument('--ask-exit', action='store_true', help=f'Ask before exitting')
args = parser.parse_args() args = parser.parse_args()
@ -343,6 +344,10 @@ def run():
print(f'Version is {VERSION}') print(f'Version is {VERSION}')
return return
# Ensure git bash for windows
ensure_git_bash()
check_windows_cmd()
# Test distro ascii art # Test distro ascii art
if args.test_distro: if args.test_distro:
print(f'Setting distro to {args.test_distro}') print(f'Setting distro to {args.test_distro}')
@ -385,3 +390,6 @@ def run():
# Run # Run
run_neofetch(preset, config.color_align) run_neofetch(preset, config.color_align)
if args.ask_exit:
input('Press any key to exit...')

View file

@ -6,16 +6,19 @@ import platform
import re import re
import shlex import shlex
import subprocess import subprocess
import zipfile
from dataclasses import dataclass from dataclasses import dataclass
from pathlib import Path from pathlib import Path
from subprocess import check_output from subprocess import check_output
from tempfile import TemporaryDirectory from tempfile import TemporaryDirectory
from urllib.request import urlretrieve
import pkg_resources import pkg_resources
import psutil
from typing_extensions import Literal from typing_extensions import Literal
from hyfetch.color_util import color from hyfetch.color_util import color
from .constants import GLOBAL_CFG from .constants import GLOBAL_CFG, MINGIT_URL
from .presets import ColorProfile from .presets import ColorProfile
from .serializer import from_dict from .serializer import from_dict
@ -142,6 +145,53 @@ def get_command_path() -> str:
return pkg_resources.resource_filename(__name__, 'scripts/neowofetch') return pkg_resources.resource_filename(__name__, 'scripts/neowofetch')
def ensure_git_bash() -> Path:
"""
Ensure git bash installation for windows
:returns git bash path
"""
if platform.system() == 'Windows':
# Find installation in default path
def_path = Path(r'C:\Program Files\Git\bin\bash.exe')
if def_path.is_file():
return def_path
# Find installation in PATH (C:\Program Files\Git\cmd should be in path)
pth = (os.environ.get('PATH') or '').lower().split(';')
pth = [p for p in pth if p.endswith(r'\git\cmd')]
if pth:
return Path(pth[0]).parent / r'bin\bash.exe'
# Previously downloaded portable installation
path = Path(__file__).parent / 'min_git'
pkg_path = path / 'package.zip'
if path.is_dir():
return path / r'bin\bash.exe'
# No installation found, download a portable installation
print('Git installation not found. Git is required to use HyFetch/neofetch on Windows')
print('Downloading a minimal portable package for Git...')
urlretrieve(MINGIT_URL, pkg_path)
print('Download finished! Extracting...')
with zipfile.ZipFile(pkg_path, 'r') as zip_ref:
zip_ref.extractall(path)
print('Done!')
return path / r'bin\bash.exe'
def check_windows_cmd():
"""
Check if this script is running under cmd.exe. If so, launch an external window with git bash
since cmd doesn't support RGB colors.
"""
if psutil.Process(os.getppid()).name().lower().strip() == 'cmd.exe':
print("cmd.exe doesn't support RGB colors, restarting in MinTTY...")
cmd = f'"{ensure_git_bash().parent.parent / "usr/bin/mintty.exe"}" -s 110,40 -e python -m hyfetch --ask-exit'
os.system(cmd)
exit()
def run_command(args: str, pipe: bool = False) -> str | None: def run_command(args: str, pipe: bool = False) -> str | None:
""" """
Run neofetch command Run neofetch command
@ -153,7 +203,7 @@ def run_command(args: str, pipe: bool = False) -> str | None:
cmd = get_command_path().replace("\\", "/").replace("C:/", "/c/") cmd = get_command_path().replace("\\", "/").replace("C:/", "/c/")
args = args.replace('\\', '/').replace('C:/', '/c/') args = args.replace('\\', '/').replace('C:/', '/c/')
full_cmd = ['C:\\Program Files\\Git\\bin\\bash.exe', '-c', f'{cmd} {args}'] full_cmd = [ensure_git_bash(), '-c', f'{cmd} {args}']
# print(full_cmd) # print(full_cmd)
if pipe: if pipe:
@ -162,7 +212,6 @@ def run_command(args: str, pipe: bool = False) -> str | None:
subprocess.run(full_cmd) subprocess.run(full_cmd)
def get_distro_ascii(distro: str | None = None) -> str: def get_distro_ascii(distro: str | None = None) -> str:
""" """
Get the distro ascii of the current distro. Or if distro is specified, get the specific distro's Get the distro ascii of the current distro. Or if distro is specified, get the specific distro's

View file

@ -1139,12 +1139,6 @@ get_distro() {
elif [[ -d /system/app/ && -d /system/priv-app ]]; then elif [[ -d /system/app/ && -d /system/priv-app ]]; then
distro="Android $(getprop ro.build.version.release)" distro="Android $(getprop ro.build.version.release)"
# Chrome OS doesn't conform to the /etc/*-release standard.
# While the file is a series of variables they can't be sourced
# by the shell since the values aren't quoted.
elif [[ -f /etc/lsb-release && $(< /etc/lsb-release) == *CHROMEOS* ]]; then
distro='Chrome OS'
elif [[ -f /etc/vzlinux-release ]]; then elif [[ -f /etc/vzlinux-release ]]; then
distro='VzLinux' distro='VzLinux'
@ -1208,14 +1202,19 @@ get_distro() {
*) distro+=" on Windows $windows_version" ;; *) distro+=" on Windows $windows_version" ;;
esac esac
# Chrome OS doesn't conform to the /etc/*-release standard.
# While the file is a series of variables they can't be sourced
# by the shell since the values aren't quoted.
elif [[ -f /etc/lsb-release && $(< /etc/lsb-release) == *CHROMEOS* ]]; then
distro='Chrome OS'
elif [[ $(< /proc/version) == *chrome-bot* || -f /dev/cros_ec ]]; then elif [[ $(< /proc/version) == *chrome-bot* || -f /dev/cros_ec ]]; then
[[ $distro != *Chrome* ]] && [[ $distro != *Chrome* ]] &&
case $distro_shorthand in case $distro_shorthand in
on) distro+=" [Chrome OS]" ;; on) distro="Chrome OS $distro" ;;
tiny) distro="Chrome OS" ;; tiny) distro="Chrome OS" ;;
*) distro+=" on Chrome OS" ;; *) distro="Chrome OS $distro" ;;
esac esac
distro=${distro## on }
fi fi
distro=$(trim_quotes "$distro") distro=$(trim_quotes "$distro")