135 lines
4.9 KiB
Python
Executable file
135 lines
4.9 KiB
Python
Executable file
"""QMK Doctor
|
|
|
|
Check out the user's QMK environment and make sure it's ready to compile.
|
|
"""
|
|
import platform
|
|
from subprocess import DEVNULL
|
|
|
|
from milc import cli
|
|
from milc.questions import yesno
|
|
|
|
from qmk import submodules
|
|
from qmk.constants import QMK_FIRMWARE, QMK_FIRMWARE_UPSTREAM
|
|
from .check import CheckStatus, check_binaries, check_binary_versions, check_submodules
|
|
from qmk.commands import git_check_repo, git_get_branch, git_is_dirty, git_get_remotes, git_check_deviation, in_virtualenv
|
|
|
|
|
|
def os_tests():
|
|
"""Determine our OS and run platform specific tests
|
|
"""
|
|
platform_id = platform.platform().lower()
|
|
|
|
if 'darwin' in platform_id or 'macos' in platform_id:
|
|
from .macos import os_test_macos
|
|
return os_test_macos()
|
|
elif 'linux' in platform_id:
|
|
from .linux import os_test_linux
|
|
return os_test_linux()
|
|
elif 'windows' in platform_id:
|
|
from .windows import os_test_windows
|
|
return os_test_windows()
|
|
else:
|
|
cli.log.warning('Unsupported OS detected: %s', platform_id)
|
|
return CheckStatus.WARNING
|
|
|
|
|
|
def git_tests():
|
|
"""Run Git-related checks
|
|
"""
|
|
status = CheckStatus.OK
|
|
|
|
# Make sure our QMK home is a Git repo
|
|
git_ok = git_check_repo()
|
|
if not git_ok:
|
|
cli.log.warning("{fg_yellow}QMK home does not appear to be a Git repository! (no .git folder)")
|
|
status = CheckStatus.WARNING
|
|
else:
|
|
git_branch = git_get_branch()
|
|
if git_branch:
|
|
cli.log.info('Git branch: %s', git_branch)
|
|
git_dirty = git_is_dirty()
|
|
if git_dirty:
|
|
cli.log.warning('{fg_yellow}Git has unstashed/uncommitted changes.')
|
|
status = CheckStatus.WARNING
|
|
git_remotes = git_get_remotes()
|
|
if 'upstream' not in git_remotes.keys() or QMK_FIRMWARE_UPSTREAM not in git_remotes['upstream'].get('url', ''):
|
|
cli.log.warning('{fg_yellow}The official repository does not seem to be configured as git remote "upstream".')
|
|
status = CheckStatus.WARNING
|
|
else:
|
|
git_deviation = git_check_deviation(git_branch)
|
|
if git_branch in ['master', 'develop'] and git_deviation:
|
|
cli.log.warning('{fg_yellow}The local "%s" branch contains commits not found in the upstream branch.', git_branch)
|
|
status = CheckStatus.WARNING
|
|
|
|
return status
|
|
|
|
|
|
@cli.argument('-y', '--yes', action='store_true', arg_only=True, help='Answer yes to all questions.')
|
|
@cli.argument('-n', '--no', action='store_true', arg_only=True, help='Answer no to all questions.')
|
|
@cli.subcommand('Basic QMK environment checks')
|
|
def doctor(cli):
|
|
"""Basic QMK environment checks.
|
|
|
|
This is currently very simple, it just checks that all the expected binaries are on your system.
|
|
|
|
TODO(unclaimed):
|
|
* [ ] Compile a trivial program with each compiler
|
|
"""
|
|
cli.log.info('QMK Doctor is checking your environment.')
|
|
cli.log.info('CLI version: %s', cli.version)
|
|
cli.log.info('QMK home: {fg_cyan}%s', QMK_FIRMWARE)
|
|
|
|
status = os_tests()
|
|
|
|
status = git_tests()
|
|
|
|
venv = in_virtualenv()
|
|
if venv:
|
|
cli.log.info('CLI installed in virtualenv.')
|
|
|
|
# Make sure the basic CLI tools we need are available and can be executed.
|
|
bin_ok = check_binaries()
|
|
|
|
if not bin_ok:
|
|
if yesno('Would you like to install dependencies?', default=True):
|
|
cli.run(['util/qmk_install.sh', '-y'], stdin=DEVNULL, capture_output=False)
|
|
bin_ok = check_binaries()
|
|
|
|
if bin_ok:
|
|
cli.log.info('All dependencies are installed.')
|
|
else:
|
|
status = CheckStatus.ERROR
|
|
|
|
# Make sure the tools are at the correct version
|
|
ver_ok = check_binary_versions()
|
|
if CheckStatus.ERROR in ver_ok:
|
|
status = CheckStatus.ERROR
|
|
elif CheckStatus.WARNING in ver_ok and status == CheckStatus.OK:
|
|
status = CheckStatus.WARNING
|
|
|
|
# Check out the QMK submodules
|
|
sub_ok = check_submodules()
|
|
|
|
if sub_ok == CheckStatus.OK:
|
|
cli.log.info('Submodules are up to date.')
|
|
else:
|
|
if yesno('Would you like to clone the submodules?', default=True):
|
|
submodules.update()
|
|
sub_ok = check_submodules()
|
|
|
|
if sub_ok == CheckStatus.ERROR:
|
|
status = CheckStatus.ERROR
|
|
elif sub_ok == CheckStatus.WARNING and status == CheckStatus.OK:
|
|
status = CheckStatus.WARNING
|
|
|
|
# Report a summary of our findings to the user
|
|
if status == CheckStatus.OK:
|
|
cli.log.info('{fg_green}QMK is ready to go')
|
|
return 0
|
|
elif status == CheckStatus.WARNING:
|
|
cli.log.info('{fg_yellow}QMK is ready to go, but minor problems were found')
|
|
return 1
|
|
else:
|
|
cli.log.info('{fg_red}Major problems detected, please fix these problems before proceeding.')
|
|
cli.log.info('{fg_blue}Check out the FAQ (https://docs.qmk.fm/#/faq_build) or join the QMK Discord (https://discord.gg/Uq7gcHh) for help.')
|
|
return 2
|