mirror of
https://github.com/qmk/qmk_cli.git
synced 2026-07-25 16:42:55 -04:00
zvecr
60 lines
2.4 KiB
Python
60 lines
2.4 KiB
Python
"""Setup qmk_firmware on your computer.
|
|
"""
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
from milc import cli
|
|
from qmk_cli.git import clone
|
|
from qmk_cli.helpers import question
|
|
|
|
default_repo = 'qmk_firmware'
|
|
default_fork = 'qmk/' + default_repo
|
|
default_branch = 'master'
|
|
|
|
|
|
@cli.argument('-n', '--no', arg_only=True, action='store_true', help='Answer no to all questions')
|
|
@cli.argument('-y', '--yes', arg_only=True, action='store_true', help='Answer yes to all questions')
|
|
@cli.argument('--baseurl', default='https://github.com', help='The URL all git operations start from')
|
|
@cli.argument('-b', '--branch', default=default_branch, help='The branch to clone')
|
|
@cli.argument('destination', default=os.environ['QMK_HOME'], nargs='?', help='The directory to clone to')
|
|
@cli.argument('fork', default=default_fork, nargs='?', help='The qmk_firmware fork to clone')
|
|
@cli.subcommand('Setup your computer for qmk_firmware.')
|
|
def setup(cli):
|
|
qmk_firmware = Path(cli.args.destination)
|
|
|
|
# Sanity checks
|
|
if cli.args.yes and cli.args.no:
|
|
cli.log.error("Can't use both --yes and --no at the same time.")
|
|
exit(1)
|
|
|
|
# Check on qmk_firmware, and if it doesn't exist offer to check it out.
|
|
if qmk_firmware.exists() and os.listdir(qmk_firmware):
|
|
cli.log.info('Found qmk_firmware at %s.', str(qmk_firmware))
|
|
else:
|
|
cli.log.error('qmk_firmware not found!')
|
|
if question('Would you like to clone %s?' % cli.args.fork):
|
|
git_url = '/'.join((cli.config.setup.baseurl, cli.args.fork))
|
|
clone(git_url, cli.args.destination, cli.config.setup.branch)
|
|
|
|
# Run `qmk_firmware/bin/qmk doctor` to check the rest of the environment out
|
|
if qmk_firmware.exists():
|
|
qmk_bin = qmk_firmware / 'bin' / 'qmk'
|
|
doctor_cmd = [sys.executable, str(qmk_bin), 'doctor']
|
|
if cli.args.yes:
|
|
doctor_cmd.append('--yes')
|
|
if cli.args.no:
|
|
doctor_cmd.append('--no')
|
|
doctor = subprocess.run(doctor_cmd)
|
|
if doctor.returncode != 0:
|
|
cli.log.error('Your build environment is not setup completely.')
|
|
|
|
if question('Would you like to run util/qmk_install?'):
|
|
curdir = os.getcwd()
|
|
os.chdir(str(qmk_firmware))
|
|
process = subprocess.run(['util/qmk_install.sh'])
|
|
os.chdir(curdir)
|
|
if process.returncode == 0:
|
|
cli.log.info('QMK setup complete!')
|