util/new_keyboard.sh

   1 #!/bin/bash
   2
   3 # This script generates a new keyboard directory under keyboards/,
   4 # and copies the template files from quantum/template/ into it.
   5
   6 # Print an error message with the word "ERROR" in red.
   7 echo_error() {
   8     echo -e "[\033[0;91mERROR\033[m]: $1"
   9 }
  10
  11 # Print a message in bold.
  12 echo_bold() {
  13     echo -e "\033[1m$1\033[m"
  14 }
  15
  16 # Prompt the user for information, showing the default value in brackets.
  17 prompt() {
  18     local message="$1"
  19     local default="$2"
  20
  21     [ -n "$default" ] && message+=" [$default]"
  22     message+=": "
  23
  24     read -rp "$message" prompt_return
  25     [ -z "$prompt_return" ] && prompt_return="$default"
  26 }
  27
  28 # Grab a username from Git config.
  29 set_git_username() {
  30     git_username="$(git config --get user.name)"
  31 }
  32
  33 # Copy the template files to the new keyboard directory.
  34 copy_templates() {
  35     echo -n "Copying base template files..."
  36     cp -r "quantum/template/base" "${keyboard_dir}"
  37     echo " done"
  38
  39     echo -n "Copying $keyboard_type template files..."
  40     cp -r "quantum/template/${keyboard_type}/." "${keyboard_dir}"
  41     echo " done"
  42
  43     echo -n "Renaming keyboard files..."
  44     mv "${keyboard_dir}/template.c" "${keyboard_dir}/${keyboard_name}.c"
  45     mv "${keyboard_dir}/template.h" "${keyboard_dir}/${keyboard_name}.h"
  46     echo " done"
  47 }
  48
  49 # Set the inplace editing parameter for sed.
  50 # macOS/BSD sed expects a file extension immediately following -i.
  51 set_sed_i() {
  52     sed_i=(-i)
  53
  54     case $(uname -a) in
  55         *Darwin*) sed_i=(-i "")
  56     esac
  57 }
  58
  59 # Replace a token with a value in the given list of files.
  60 replace_placeholders() {
  61     local replace_token="$1"
  62     local replace_value="$2"
  63     shift 2
  64     local replace_filenames=("$@")
  65
  66     echo -n "Replacing $replace_token with $replace_value..."
  67     for replace_filename in "${replace_filenames[@]}"; do
  68         sed "${sed_i[@]}" -e "s/${replace_token}/${replace_value}/g" "$replace_filename"
  69     done
  70     echo " done"
  71 }
  72
  73 # Replace %YEAR% with the current year.
  74 replace_year_placeholders() {
  75     local replace_year_filenames=(
  76         "${keyboard_dir}/config.h"
  77         "${keyboard_dir}/${keyboard_name}.c"
  78         "${keyboard_dir}/${keyboard_name}.h"
  79         "${keyboard_dir}/keymaps/default/config.h"
  80         "${keyboard_dir}/keymaps/default/keymap.c"
  81     )
  82     replace_placeholders "%YEAR%" "$(date +%Y)" "${replace_year_filenames[@]}"
  83 }
  84
  85 # Replace %KEYBOARD% with the keyboard name.
  86 replace_keyboard_placeholders() {
  87     local replace_keyboard_filenames=(
  88         "${keyboard_dir}/config.h"
  89         "${keyboard_dir}/readme.md"
  90         "${keyboard_dir}/${keyboard_name}.c"
  91         "${keyboard_dir}/keymaps/default/readme.md"
  92     )
  93     replace_placeholders "%KEYBOARD%" "$keyboard_name" "${replace_keyboard_filenames[@]}"
  94 }
  95
  96 # Replace %YOUR_NAME% with the username.
  97 replace_name_placeholders() {
  98     local replace_name_filenames=(
  99         "${keyboard_dir}/config.h"
 100         "${keyboard_dir}/readme.md"
 101         "${keyboard_dir}/${keyboard_name}.c"
 102         "${keyboard_dir}/${keyboard_name}.h"
 103         "${keyboard_dir}/keymaps/default/config.h"
 104         "${keyboard_dir}/keymaps/default/keymap.c"
 105     )
 106     replace_placeholders "%YOUR_NAME%" "$username" "${replace_name_filenames[@]}"
 107 }
 108
 109 # Check if an array contains an element.
 110 array_contains() {
 111     local e match="$1"
 112     shift
 113     for e; do
 114         [[ "$e" == "$match" ]] && return 0;
 115     done
 116
 117     return 1
 118 }
 119
 120 # If we've been started from util/, we want to be in qmk_firmware/
 121 [[ "$PWD" == *util ]] && cd ..
 122
 123 # The root qmk_firmware/ directory should have a subdirectory called quantum/
 124 if [ ! -d "quantum" ]; then
 125     echo_error "Could not detect the QMK firmware directory!"
 126     echo_error "Are you sure you're in the right place?"
 127     exit 1
 128 fi
 129
 130 echo_bold "Generating a new QMK keyboard directory"
 131 echo
 132
 133 # Keyboard name is required, so keep prompting until we get one
 134 while [ -z "$keyboard_name" ]; do
 135     prompt "Keyboard Name" ""
 136     keyboard_name=$prompt_return
 137 done
 138
 139 keyboard_dir="keyboards/$keyboard_name"
 140
 141 if [ -d "$keyboard_dir" ]; then
 142     echo_error "Keyboard $keyboard_name already exists!"
 143     exit 1
 144 fi
 145
 146 KEYBOARD_TYPES=("avr" "ps2avrgb")
 147
 148 prompt "Keyboard Type" "avr"
 149 keyboard_type=$prompt_return
 150
 151 if ! array_contains "$keyboard_type" "${KEYBOARD_TYPES[@]}"; then
 152     echo_error "Keyboard type must be one of: ${KEYBOARD_TYPES[*]}"
 153     exit 1
 154 fi
 155
 156 set_git_username
 157 prompt "Your Name" "$git_username"
 158 username=$prompt_return
 159
 160 echo
 161
 162 copy_templates
 163 set_sed_i
 164 replace_year_placeholders
 165 replace_keyboard_placeholders
 166 [ -n "$username" ] && replace_name_placeholders
 167
 168 echo
 169 echo_bold "Created a new keyboard called $keyboard_name."
 170 echo
 171 echo_bold "To start working on things, cd into keyboards/$keyboard_name,"
 172 echo_bold "or open the directory in your favourite text editor."