info.py 33 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883
  1. """Functions that help us generate and use info.json files.
  2. """
  3. from pathlib import Path
  4. import jsonschema
  5. from dotty_dict import dotty
  6. from milc import cli
  7. from qmk.constants import CHIBIOS_PROCESSORS, LUFA_PROCESSORS, VUSB_PROCESSORS
  8. from qmk.c_parse import find_layouts, parse_config_h_file, find_led_config
  9. from qmk.json_schema import deep_update, json_load, validate
  10. from qmk.keyboard import config_h, rules_mk
  11. from qmk.keymap import list_keymaps, locate_keymap
  12. from qmk.commands import parse_configurator_json
  13. from qmk.makefile import parse_rules_mk_file
  14. from qmk.math import compute
  15. true_values = ['1', 'on', 'yes']
  16. false_values = ['0', 'off', 'no']
  17. def _valid_community_layout(layout):
  18. """Validate that a declared community list exists
  19. """
  20. return (Path('layouts/default') / layout).exists()
  21. def _validate(keyboard, info_data):
  22. """Perform various validation on the provided info.json data
  23. """
  24. # First validate against the jsonschema
  25. try:
  26. validate(info_data, 'qmk.api.keyboard.v1')
  27. except jsonschema.ValidationError as e:
  28. json_path = '.'.join([str(p) for p in e.absolute_path])
  29. cli.log.error('Invalid API data: %s: %s: %s', keyboard, json_path, e.message)
  30. exit(1)
  31. layouts = info_data.get('layouts', {})
  32. layout_aliases = info_data.get('layout_aliases', {})
  33. community_layouts = info_data.get('community_layouts', [])
  34. # Make sure we have at least one layout
  35. if len(layouts) == 0:
  36. _log_error(info_data, 'No LAYOUTs defined! Need at least one layout defined in info.json.')
  37. # Providing only LAYOUT_all "because I define my layouts in a 3rd party tool"
  38. if len(layouts) == 1 and 'LAYOUT_all' in layouts:
  39. _log_warning(info_data, '"LAYOUT_all" should be "LAYOUT" unless additional layouts are provided.')
  40. # Extended layout name checks
  41. name_fragments = keyboard.split('/')
  42. for layout in layouts.keys():
  43. if any(fragment in layout for fragment in name_fragments):
  44. _log_warning(info_data, f'Layout "{layout}" should not contain name of keyboard.')
  45. # Filter out any non-existing community layouts
  46. for layout in community_layouts:
  47. if not _valid_community_layout(layout):
  48. # Ignore layout from future checks
  49. info_data['community_layouts'].remove(layout)
  50. _log_error(info_data, 'Claims to support a community layout that does not exist: %s' % (layout))
  51. # Make sure we supply layout macros for the community layouts we claim to support
  52. for layout in community_layouts:
  53. layout_name = 'LAYOUT_' + layout
  54. if layout_name not in layouts and layout_name not in layout_aliases:
  55. _log_error(info_data, 'Claims to support community layout %s but no %s() macro found' % (layout, layout_name))
  56. def info_json(keyboard):
  57. """Generate the info.json data for a specific keyboard.
  58. """
  59. cur_dir = Path('keyboards')
  60. root_rules_mk = parse_rules_mk_file(cur_dir / keyboard / 'rules.mk')
  61. if 'DEFAULT_FOLDER' in root_rules_mk:
  62. keyboard = root_rules_mk['DEFAULT_FOLDER']
  63. info_data = {
  64. 'keyboard_name': str(keyboard),
  65. 'keyboard_folder': str(keyboard),
  66. 'keymaps': {},
  67. 'layouts': {},
  68. 'parse_errors': [],
  69. 'parse_warnings': [],
  70. 'maintainer': 'qmk',
  71. }
  72. # Populate the list of JSON keymaps
  73. for keymap in list_keymaps(keyboard, c=False, fullpath=True):
  74. info_data['keymaps'][keymap.name] = {'url': f'https://raw.githubusercontent.com/qmk/qmk_firmware/master/{keymap}/keymap.json'}
  75. # Populate layout data
  76. layouts, aliases = _search_keyboard_h(keyboard)
  77. if aliases:
  78. info_data['layout_aliases'] = aliases
  79. for layout_name, layout_json in layouts.items():
  80. if not layout_name.startswith('LAYOUT_kc'):
  81. layout_json['c_macro'] = True
  82. info_data['layouts'][layout_name] = layout_json
  83. # Merge in the data from info.json, config.h, and rules.mk
  84. info_data = merge_info_jsons(keyboard, info_data)
  85. info_data = _process_defaults(info_data)
  86. info_data = _extract_rules_mk(info_data, rules_mk(str(keyboard)))
  87. info_data = _extract_config_h(info_data, config_h(str(keyboard)))
  88. # Ensure that we have matrix row and column counts
  89. info_data = _matrix_size(info_data)
  90. # Merge in data from <keyboard.c>
  91. info_data = _extract_led_config(info_data, str(keyboard))
  92. # Validate
  93. _validate(keyboard, info_data)
  94. # Check that the reported matrix size is consistent with the actual matrix size
  95. _check_matrix(info_data)
  96. return info_data
  97. def _extract_features(info_data, rules):
  98. """Find all the features enabled in rules.mk.
  99. """
  100. # Process booleans rules
  101. for key, value in rules.items():
  102. if key.endswith('_ENABLE'):
  103. key = '_'.join(key.split('_')[:-1]).lower()
  104. value = True if value.lower() in true_values else False if value.lower() in false_values else value
  105. if 'config_h_features' not in info_data:
  106. info_data['config_h_features'] = {}
  107. if 'features' not in info_data:
  108. info_data['features'] = {}
  109. if key in info_data['features']:
  110. _log_warning(info_data, 'Feature %s is specified in both info.json and rules.mk, the rules.mk value wins.' % (key,))
  111. info_data['features'][key] = value
  112. info_data['config_h_features'][key] = value
  113. return info_data
  114. def _pin_name(pin):
  115. """Returns the proper representation for a pin.
  116. """
  117. pin = pin.strip()
  118. if not pin:
  119. return None
  120. elif pin.isdigit():
  121. return int(pin)
  122. elif pin == 'NO_PIN':
  123. return None
  124. return pin
  125. def _extract_pins(pins):
  126. """Returns a list of pins from a comma separated string of pins.
  127. """
  128. return [_pin_name(pin) for pin in pins.split(',')]
  129. def _extract_2d_array(raw):
  130. """Return a 2d array of strings
  131. """
  132. out_array = []
  133. while raw[-1] != '}':
  134. raw = raw[:-1]
  135. for row in raw.split('},{'):
  136. if row.startswith('{'):
  137. row = row[1:]
  138. if row.endswith('}'):
  139. row = row[:-1]
  140. out_array.append([])
  141. for val in row.split(','):
  142. out_array[-1].append(val)
  143. return out_array
  144. def _extract_2d_int_array(raw):
  145. """Return a 2d array of ints
  146. """
  147. ret = _extract_2d_array(raw)
  148. return [list(map(int, x)) for x in ret]
  149. def _extract_direct_matrix(direct_pins):
  150. """extract direct_matrix
  151. """
  152. direct_pin_array = _extract_2d_array(direct_pins)
  153. for i in range(len(direct_pin_array)):
  154. for j in range(len(direct_pin_array[i])):
  155. if direct_pin_array[i][j] == 'NO_PIN':
  156. direct_pin_array[i][j] = None
  157. return direct_pin_array
  158. def _extract_audio(info_data, config_c):
  159. """Populate data about the audio configuration
  160. """
  161. audio_pins = []
  162. for pin in 'B5', 'B6', 'B7', 'C4', 'C5', 'C6':
  163. if config_c.get(f'{pin}_AUDIO'):
  164. audio_pins.append(pin)
  165. if audio_pins:
  166. info_data['audio'] = {'pins': audio_pins}
  167. def _extract_encoders_values(config_c, postfix=''):
  168. """Common encoder extraction logic
  169. """
  170. a_pad = config_c.get(f'ENCODERS_PAD_A{postfix}', '').replace(' ', '')[1:-1]
  171. b_pad = config_c.get(f'ENCODERS_PAD_B{postfix}', '').replace(' ', '')[1:-1]
  172. resolutions = config_c.get(f'ENCODER_RESOLUTIONS{postfix}', '').replace(' ', '')[1:-1]
  173. default_resolution = config_c.get('ENCODER_RESOLUTION', None)
  174. if a_pad and b_pad:
  175. a_pad = list(filter(None, a_pad.split(',')))
  176. b_pad = list(filter(None, b_pad.split(',')))
  177. resolutions = list(filter(None, resolutions.split(',')))
  178. if default_resolution:
  179. resolutions += [default_resolution] * (len(a_pad) - len(resolutions))
  180. encoders = []
  181. for index in range(len(a_pad)):
  182. encoder = {'pin_a': a_pad[index], 'pin_b': b_pad[index]}
  183. if index < len(resolutions):
  184. encoder['resolution'] = int(resolutions[index])
  185. encoders.append(encoder)
  186. return encoders
  187. def _extract_encoders(info_data, config_c):
  188. """Populate data about encoder pins
  189. """
  190. encoders = _extract_encoders_values(config_c)
  191. if encoders:
  192. if 'encoder' not in info_data:
  193. info_data['encoder'] = {}
  194. if 'rotary' in info_data['encoder']:
  195. _log_warning(info_data, 'Encoder config is specified in both config.h and info.json (encoder.rotary) (Value: %s), the config.h value wins.' % info_data['encoder']['rotary'])
  196. info_data['encoder']['rotary'] = encoders
  197. def _extract_split_encoders(info_data, config_c):
  198. """Populate data about split encoder pins
  199. """
  200. encoders = _extract_encoders_values(config_c, '_RIGHT')
  201. if encoders:
  202. if 'split' not in info_data:
  203. info_data['split'] = {}
  204. if 'encoder' not in info_data['split']:
  205. info_data['split']['encoder'] = {}
  206. if 'right' not in info_data['split']['encoder']:
  207. info_data['split']['encoder']['right'] = {}
  208. if 'rotary' in info_data['split']['encoder']['right']:
  209. _log_warning(info_data, 'Encoder config is specified in both config.h and info.json (encoder.rotary) (Value: %s), the config.h value wins.' % info_data['split']['encoder']['right']['rotary'])
  210. info_data['split']['encoder']['right']['rotary'] = encoders
  211. def _extract_secure_unlock(info_data, config_c):
  212. """Populate data about the secure unlock sequence
  213. """
  214. unlock = config_c.get('SECURE_UNLOCK_SEQUENCE', '').replace(' ', '')[1:-1]
  215. if unlock:
  216. unlock_array = _extract_2d_int_array(unlock)
  217. if 'secure' not in info_data:
  218. info_data['secure'] = {}
  219. if 'unlock_sequence' in info_data['secure']:
  220. _log_warning(info_data, 'Secure unlock sequence is specified in both config.h (SECURE_UNLOCK_SEQUENCE) and info.json (secure.unlock_sequence) (Value: %s), the config.h value wins.' % info_data['secure']['unlock_sequence'])
  221. info_data['secure']['unlock_sequence'] = unlock_array
  222. def _extract_split_main(info_data, config_c):
  223. """Populate data about the split configuration
  224. """
  225. # Figure out how the main half is determined
  226. if config_c.get('SPLIT_HAND_PIN') is True:
  227. if 'split' not in info_data:
  228. info_data['split'] = {}
  229. if 'main' in info_data['split']:
  230. _log_warning(info_data, 'Split main hand is specified in both config.h (SPLIT_HAND_PIN) and info.json (split.main) (Value: %s), the config.h value wins.' % info_data['split']['main'])
  231. info_data['split']['main'] = 'pin'
  232. if config_c.get('SPLIT_HAND_MATRIX_GRID'):
  233. if 'split' not in info_data:
  234. info_data['split'] = {}
  235. if 'main' in info_data['split']:
  236. _log_warning(info_data, 'Split main hand is specified in both config.h (SPLIT_HAND_MATRIX_GRID) and info.json (split.main) (Value: %s), the config.h value wins.' % info_data['split']['main'])
  237. info_data['split']['main'] = 'matrix_grid'
  238. info_data['split']['matrix_grid'] = _extract_pins(config_c['SPLIT_HAND_MATRIX_GRID'])
  239. if config_c.get('EE_HANDS') is True:
  240. if 'split' not in info_data:
  241. info_data['split'] = {}
  242. if 'main' in info_data['split']:
  243. _log_warning(info_data, 'Split main hand is specified in both config.h (EE_HANDS) and info.json (split.main) (Value: %s), the config.h value wins.' % info_data['split']['main'])
  244. info_data['split']['main'] = 'eeprom'
  245. if config_c.get('MASTER_RIGHT') is True:
  246. if 'split' not in info_data:
  247. info_data['split'] = {}
  248. if 'main' in info_data['split']:
  249. _log_warning(info_data, 'Split main hand is specified in both config.h (MASTER_RIGHT) and info.json (split.main) (Value: %s), the config.h value wins.' % info_data['split']['main'])
  250. info_data['split']['main'] = 'right'
  251. if config_c.get('MASTER_LEFT') is True:
  252. if 'split' not in info_data:
  253. info_data['split'] = {}
  254. if 'main' in info_data['split']:
  255. _log_warning(info_data, 'Split main hand is specified in both config.h (MASTER_LEFT) and info.json (split.main) (Value: %s), the config.h value wins.' % info_data['split']['main'])
  256. info_data['split']['main'] = 'left'
  257. def _extract_split_transport(info_data, config_c):
  258. # Figure out the transport method
  259. if config_c.get('USE_I2C') is True:
  260. if 'split' not in info_data:
  261. info_data['split'] = {}
  262. if 'transport' not in info_data['split']:
  263. info_data['split']['transport'] = {}
  264. if 'protocol' in info_data['split']['transport']:
  265. _log_warning(info_data, 'Split transport is specified in both config.h (USE_I2C) and info.json (split.transport.protocol) (Value: %s), the config.h value wins.' % info_data['split']['transport'])
  266. info_data['split']['transport']['protocol'] = 'i2c'
  267. # Ignore transport defaults if "SPLIT_KEYBOARD" is unset
  268. elif 'enabled' in info_data.get('split', {}):
  269. if 'split' not in info_data:
  270. info_data['split'] = {}
  271. if 'transport' not in info_data['split']:
  272. info_data['split']['transport'] = {}
  273. if 'protocol' not in info_data['split']['transport']:
  274. info_data['split']['transport']['protocol'] = 'serial'
  275. def _extract_split_right_pins(info_data, config_c):
  276. # Figure out the right half matrix pins
  277. row_pins = config_c.get('MATRIX_ROW_PINS_RIGHT', '').replace('{', '').replace('}', '').strip()
  278. col_pins = config_c.get('MATRIX_COL_PINS_RIGHT', '').replace('{', '').replace('}', '').strip()
  279. direct_pins = config_c.get('DIRECT_PINS_RIGHT', '').replace(' ', '')[1:-1]
  280. if row_pins or col_pins or direct_pins:
  281. if info_data.get('split', {}).get('matrix_pins', {}).get('right', None):
  282. _log_warning(info_data, 'Right hand matrix data is specified in both info.json and config.h, the config.h values win.')
  283. if 'split' not in info_data:
  284. info_data['split'] = {}
  285. if 'matrix_pins' not in info_data['split']:
  286. info_data['split']['matrix_pins'] = {}
  287. if 'right' not in info_data['split']['matrix_pins']:
  288. info_data['split']['matrix_pins']['right'] = {}
  289. if col_pins:
  290. info_data['split']['matrix_pins']['right']['cols'] = _extract_pins(col_pins)
  291. if row_pins:
  292. info_data['split']['matrix_pins']['right']['rows'] = _extract_pins(row_pins)
  293. if direct_pins:
  294. info_data['split']['matrix_pins']['right']['direct'] = _extract_direct_matrix(direct_pins)
  295. def _extract_matrix_info(info_data, config_c):
  296. """Populate the matrix information.
  297. """
  298. row_pins = config_c.get('MATRIX_ROW_PINS', '').replace('{', '').replace('}', '').strip()
  299. col_pins = config_c.get('MATRIX_COL_PINS', '').replace('{', '').replace('}', '').strip()
  300. direct_pins = config_c.get('DIRECT_PINS', '').replace(' ', '')[1:-1]
  301. info_snippet = {}
  302. if 'MATRIX_ROWS' in config_c and 'MATRIX_COLS' in config_c:
  303. if 'matrix_size' in info_data:
  304. _log_warning(info_data, 'Matrix size is specified in both info.json and config.h, the config.h values win.')
  305. info_data['matrix_size'] = {
  306. 'cols': compute(config_c.get('MATRIX_COLS', '0')),
  307. 'rows': compute(config_c.get('MATRIX_ROWS', '0')),
  308. }
  309. if row_pins and col_pins:
  310. if 'matrix_pins' in info_data and 'cols' in info_data['matrix_pins'] and 'rows' in info_data['matrix_pins']:
  311. _log_warning(info_data, 'Matrix pins are specified in both info.json and config.h, the config.h values win.')
  312. info_snippet['cols'] = _extract_pins(col_pins)
  313. info_snippet['rows'] = _extract_pins(row_pins)
  314. if direct_pins:
  315. if 'matrix_pins' in info_data and 'direct' in info_data['matrix_pins']:
  316. _log_warning(info_data, 'Direct pins are specified in both info.json and config.h, the config.h values win.')
  317. info_snippet['direct'] = _extract_direct_matrix(direct_pins)
  318. if config_c.get('CUSTOM_MATRIX', 'no') != 'no':
  319. if 'matrix_pins' in info_data and 'custom' in info_data['matrix_pins']:
  320. _log_warning(info_data, 'Custom Matrix is specified in both info.json and config.h, the config.h values win.')
  321. info_snippet['custom'] = True
  322. if config_c['CUSTOM_MATRIX'] == 'lite':
  323. info_snippet['custom_lite'] = True
  324. if info_snippet:
  325. info_data['matrix_pins'] = info_snippet
  326. return info_data
  327. def _config_to_json(key_type, config_value):
  328. """Convert config value using spec
  329. """
  330. if key_type.startswith('array'):
  331. if '.' in key_type:
  332. key_type, array_type = key_type.split('.', 1)
  333. else:
  334. array_type = None
  335. config_value = config_value.replace('{', '').replace('}', '').strip()
  336. if array_type == 'int':
  337. return list(map(int, config_value.split(',')))
  338. else:
  339. return list(map(str.strip, config_value.split(',')))
  340. elif key_type == 'bool':
  341. return config_value in true_values
  342. elif key_type == 'hex':
  343. return '0x' + config_value[2:].upper()
  344. elif key_type == 'list':
  345. return config_value.split()
  346. elif key_type == 'int':
  347. return int(config_value)
  348. elif key_type == 'str':
  349. return config_value.strip('"').replace('\\"', '"').replace('\\\\', '\\')
  350. elif key_type == 'bcd_version':
  351. major = int(config_value[2:4])
  352. minor = int(config_value[4])
  353. revision = int(config_value[5])
  354. return f'{major}.{minor}.{revision}'
  355. return config_value
  356. def _extract_config_h(info_data, config_c):
  357. """Pull some keyboard information from existing config.h files
  358. """
  359. # Pull in data from the json map
  360. dotty_info = dotty(info_data)
  361. info_config_map = json_load(Path('data/mappings/info_config.hjson'))
  362. for config_key, info_dict in info_config_map.items():
  363. info_key = info_dict['info_key']
  364. key_type = info_dict.get('value_type', 'raw')
  365. try:
  366. replace_with = info_dict.get('replace_with')
  367. if config_key in config_c and info_dict.get('invalid', False):
  368. if replace_with:
  369. _log_error(info_data, '%s in config.h is no longer a valid option and should be replaced with %s' % (config_key, replace_with))
  370. else:
  371. _log_error(info_data, '%s in config.h is no longer a valid option and should be removed' % config_key)
  372. elif config_key in config_c and info_dict.get('deprecated', False):
  373. if replace_with:
  374. _log_warning(info_data, '%s in config.h is deprecated in favor of %s and will be removed at a later date' % (config_key, replace_with))
  375. else:
  376. _log_warning(info_data, '%s in config.h is deprecated and will be removed at a later date' % config_key)
  377. if config_key in config_c and info_dict.get('to_json', True):
  378. if dotty_info.get(info_key) and info_dict.get('warn_duplicate', True):
  379. _log_warning(info_data, '%s in config.h is overwriting %s in info.json' % (config_key, info_key))
  380. dotty_info[info_key] = _config_to_json(key_type, config_c[config_key])
  381. except Exception as e:
  382. _log_warning(info_data, f'{config_key}->{info_key}: {e}')
  383. info_data.update(dotty_info)
  384. # Pull data that easily can't be mapped in json
  385. _extract_matrix_info(info_data, config_c)
  386. _extract_audio(info_data, config_c)
  387. _extract_secure_unlock(info_data, config_c)
  388. _extract_split_main(info_data, config_c)
  389. _extract_split_transport(info_data, config_c)
  390. _extract_split_right_pins(info_data, config_c)
  391. _extract_encoders(info_data, config_c)
  392. _extract_split_encoders(info_data, config_c)
  393. return info_data
  394. def _process_defaults(info_data):
  395. """Process any additional defaults based on currently discovered information
  396. """
  397. defaults_map = json_load(Path('data/mappings/defaults.hjson'))
  398. for default_type in defaults_map.keys():
  399. thing_map = defaults_map[default_type]
  400. if default_type in info_data:
  401. for key, value in thing_map.get(info_data[default_type], {}).items():
  402. info_data[key] = value
  403. return info_data
  404. def _extract_rules_mk(info_data, rules):
  405. """Pull some keyboard information from existing rules.mk files
  406. """
  407. info_data['processor'] = rules.get('MCU', info_data.get('processor', 'atmega32u4'))
  408. if info_data['processor'] in CHIBIOS_PROCESSORS:
  409. arm_processor_rules(info_data, rules)
  410. elif info_data['processor'] in LUFA_PROCESSORS + VUSB_PROCESSORS:
  411. avr_processor_rules(info_data, rules)
  412. else:
  413. cli.log.warning("%s: Unknown MCU: %s" % (info_data['keyboard_folder'], info_data['processor']))
  414. unknown_processor_rules(info_data, rules)
  415. # Pull in data from the json map
  416. dotty_info = dotty(info_data)
  417. info_rules_map = json_load(Path('data/mappings/info_rules.hjson'))
  418. for rules_key, info_dict in info_rules_map.items():
  419. info_key = info_dict['info_key']
  420. key_type = info_dict.get('value_type', 'raw')
  421. try:
  422. replace_with = info_dict.get('replace_with')
  423. if rules_key in rules and info_dict.get('invalid', False):
  424. if replace_with:
  425. _log_error(info_data, '%s in rules.mk is no longer a valid option and should be replaced with %s' % (rules_key, replace_with))
  426. else:
  427. _log_error(info_data, '%s in rules.mk is no longer a valid option and should be removed' % rules_key)
  428. elif rules_key in rules and info_dict.get('deprecated', False):
  429. if replace_with:
  430. _log_warning(info_data, '%s in rules.mk is deprecated in favor of %s and will be removed at a later date' % (rules_key, replace_with))
  431. else:
  432. _log_warning(info_data, '%s in rules.mk is deprecated and will be removed at a later date' % rules_key)
  433. if rules_key in rules and info_dict.get('to_json', True):
  434. if dotty_info.get(info_key) and info_dict.get('warn_duplicate', True):
  435. _log_warning(info_data, '%s in rules.mk is overwriting %s in info.json' % (rules_key, info_key))
  436. dotty_info[info_key] = _config_to_json(key_type, rules[rules_key])
  437. except Exception as e:
  438. _log_warning(info_data, f'{rules_key}->{info_key}: {e}')
  439. info_data.update(dotty_info)
  440. # Merge in config values that can't be easily mapped
  441. _extract_features(info_data, rules)
  442. return info_data
  443. def find_keyboard_c(keyboard):
  444. """Find all <keyboard>.c files
  445. """
  446. keyboard = Path(keyboard)
  447. current_path = Path('keyboards/')
  448. files = []
  449. for directory in keyboard.parts:
  450. current_path = current_path / directory
  451. keyboard_c_path = current_path / f'{directory}.c'
  452. if keyboard_c_path.exists():
  453. files.append(keyboard_c_path)
  454. return files
  455. def _extract_led_config(info_data, keyboard):
  456. """Scan all <keyboard>.c files for led config
  457. """
  458. cols = info_data['matrix_size']['cols']
  459. rows = info_data['matrix_size']['rows']
  460. # Determine what feature owns g_led_config
  461. features = info_data.get("features", {})
  462. feature = None
  463. if features.get("rgb_matrix", False):
  464. feature = "rgb_matrix"
  465. elif features.get("led_matrix", False):
  466. feature = "led_matrix"
  467. if feature:
  468. # Process
  469. for file in find_keyboard_c(keyboard):
  470. try:
  471. ret = find_led_config(file, cols, rows)
  472. if ret:
  473. info_data[feature] = info_data.get(feature, {})
  474. info_data[feature]["layout"] = ret
  475. except Exception as e:
  476. _log_warning(info_data, f'led_config: {file.name}: {e}')
  477. return info_data
  478. def _matrix_size(info_data):
  479. """Add info_data['matrix_size'] if it doesn't exist.
  480. """
  481. if 'matrix_size' not in info_data and 'matrix_pins' in info_data:
  482. info_data['matrix_size'] = {}
  483. if 'direct' in info_data['matrix_pins']:
  484. info_data['matrix_size']['cols'] = len(info_data['matrix_pins']['direct'][0])
  485. info_data['matrix_size']['rows'] = len(info_data['matrix_pins']['direct'])
  486. elif 'cols' in info_data['matrix_pins'] and 'rows' in info_data['matrix_pins']:
  487. info_data['matrix_size']['cols'] = len(info_data['matrix_pins']['cols'])
  488. info_data['matrix_size']['rows'] = len(info_data['matrix_pins']['rows'])
  489. # Assumption of split common
  490. if 'split' in info_data:
  491. if info_data['split'].get('enabled', False):
  492. info_data['matrix_size']['rows'] *= 2
  493. return info_data
  494. def _check_matrix(info_data):
  495. """Check the matrix to ensure that row/column count is consistent.
  496. """
  497. if 'matrix_pins' in info_data and 'matrix_size' in info_data:
  498. actual_col_count = info_data['matrix_size'].get('cols', 0)
  499. actual_row_count = info_data['matrix_size'].get('rows', 0)
  500. col_count = row_count = 0
  501. if 'direct' in info_data['matrix_pins']:
  502. col_count = len(info_data['matrix_pins']['direct'][0])
  503. row_count = len(info_data['matrix_pins']['direct'])
  504. elif 'cols' in info_data['matrix_pins'] and 'rows' in info_data['matrix_pins']:
  505. col_count = len(info_data['matrix_pins']['cols'])
  506. row_count = len(info_data['matrix_pins']['rows'])
  507. if col_count != actual_col_count and col_count != (actual_col_count / 2):
  508. # FIXME: once we can we should detect if split is enabled to do the actual_col_count/2 check.
  509. _log_error(info_data, f'MATRIX_COLS is inconsistent with the size of MATRIX_COL_PINS: {col_count} != {actual_col_count}')
  510. if row_count != actual_row_count and row_count != (actual_row_count / 2):
  511. # FIXME: once we can we should detect if split is enabled to do the actual_row_count/2 check.
  512. _log_error(info_data, f'MATRIX_ROWS is inconsistent with the size of MATRIX_ROW_PINS: {row_count} != {actual_row_count}')
  513. def _search_keyboard_h(keyboard):
  514. keyboard = Path(keyboard)
  515. current_path = Path('keyboards/')
  516. aliases = {}
  517. layouts = {}
  518. for directory in keyboard.parts:
  519. current_path = current_path / directory
  520. keyboard_h = '%s.h' % (directory,)
  521. keyboard_h_path = current_path / keyboard_h
  522. if keyboard_h_path.exists():
  523. new_layouts, new_aliases = find_layouts(keyboard_h_path)
  524. layouts.update(new_layouts)
  525. for alias, alias_text in new_aliases.items():
  526. if alias_text in layouts:
  527. aliases[alias] = alias_text
  528. return layouts, aliases
  529. def _log_error(info_data, message):
  530. """Send an error message to both JSON and the log.
  531. """
  532. info_data['parse_errors'].append(message)
  533. cli.log.error('%s: %s', info_data.get('keyboard_folder', 'Unknown Keyboard!'), message)
  534. def _log_warning(info_data, message):
  535. """Send a warning message to both JSON and the log.
  536. """
  537. info_data['parse_warnings'].append(message)
  538. cli.log.warning('%s: %s', info_data.get('keyboard_folder', 'Unknown Keyboard!'), message)
  539. def arm_processor_rules(info_data, rules):
  540. """Setup the default info for an ARM board.
  541. """
  542. info_data['processor_type'] = 'arm'
  543. info_data['protocol'] = 'ChibiOS'
  544. if 'STM32' in info_data['processor']:
  545. info_data['platform'] = 'STM32'
  546. elif 'MCU_SERIES' in rules:
  547. info_data['platform'] = rules['MCU_SERIES']
  548. elif 'ARM_ATSAM' in rules:
  549. info_data['platform'] = 'ARM_ATSAM'
  550. return info_data
  551. def avr_processor_rules(info_data, rules):
  552. """Setup the default info for an AVR board.
  553. """
  554. info_data['processor_type'] = 'avr'
  555. info_data['platform'] = rules['ARCH'] if 'ARCH' in rules else 'unknown'
  556. info_data['protocol'] = 'V-USB' if info_data['processor'] in VUSB_PROCESSORS else 'LUFA'
  557. # FIXME(fauxpark/anyone): Eventually we should detect the protocol by looking at PROTOCOL inherited from mcu_selection.mk:
  558. # info_data['protocol'] = 'V-USB' if rules.get('PROTOCOL') == 'VUSB' else 'LUFA'
  559. return info_data
  560. def unknown_processor_rules(info_data, rules):
  561. """Setup the default keyboard info for unknown boards.
  562. """
  563. info_data['bootloader'] = 'unknown'
  564. info_data['platform'] = 'unknown'
  565. info_data['processor'] = 'unknown'
  566. info_data['processor_type'] = 'unknown'
  567. info_data['protocol'] = 'unknown'
  568. return info_data
  569. def merge_info_jsons(keyboard, info_data):
  570. """Return a merged copy of all the info.json files for a keyboard.
  571. """
  572. for info_file in find_info_json(keyboard):
  573. # Load and validate the JSON data
  574. new_info_data = json_load(info_file)
  575. if not isinstance(new_info_data, dict):
  576. _log_error(info_data, "Invalid file %s, root object should be a dictionary." % (str(info_file),))
  577. continue
  578. try:
  579. validate(new_info_data, 'qmk.keyboard.v1')
  580. except jsonschema.ValidationError as e:
  581. json_path = '.'.join([str(p) for p in e.absolute_path])
  582. cli.log.error('Not including data from file: %s', info_file)
  583. cli.log.error('\t%s: %s', json_path, e.message)
  584. continue
  585. # Merge layout data in
  586. if 'layout_aliases' in new_info_data:
  587. info_data['layout_aliases'] = {**info_data.get('layout_aliases', {}), **new_info_data['layout_aliases']}
  588. del new_info_data['layout_aliases']
  589. for layout_name, layout in new_info_data.get('layouts', {}).items():
  590. if layout_name in info_data.get('layout_aliases', {}):
  591. _log_warning(info_data, f"info.json uses alias name {layout_name} instead of {info_data['layout_aliases'][layout_name]}")
  592. layout_name = info_data['layout_aliases'][layout_name]
  593. if layout_name in info_data['layouts']:
  594. if len(info_data['layouts'][layout_name]['layout']) != len(layout['layout']):
  595. msg = 'Number of keys for %s does not match! info.json specifies %d keys, C macro specifies %d'
  596. _log_error(info_data, msg % (layout_name, len(layout['layout']), len(info_data['layouts'][layout_name]['layout'])))
  597. else:
  598. for new_key, existing_key in zip(layout['layout'], info_data['layouts'][layout_name]['layout']):
  599. existing_key.update(new_key)
  600. else:
  601. if not all('matrix' in key_data.keys() for key_data in layout['layout']):
  602. _log_error(info_data, f'Layout "{layout_name}" has no "matrix" definition in either "info.json" or "<keyboard>.h"!')
  603. else:
  604. layout['c_macro'] = False
  605. info_data['layouts'][layout_name] = layout
  606. # Update info_data with the new data
  607. if 'layouts' in new_info_data:
  608. del new_info_data['layouts']
  609. deep_update(info_data, new_info_data)
  610. return info_data
  611. def find_info_json(keyboard):
  612. """Finds all the info.json files associated with a keyboard.
  613. """
  614. # Find the most specific first
  615. base_path = Path('keyboards')
  616. keyboard_path = base_path / keyboard
  617. keyboard_parent = keyboard_path.parent
  618. info_jsons = [keyboard_path / 'info.json']
  619. # Add DEFAULT_FOLDER before parents, if present
  620. rules = rules_mk(keyboard)
  621. if 'DEFAULT_FOLDER' in rules:
  622. info_jsons.append(Path(rules['DEFAULT_FOLDER']) / 'info.json')
  623. # Add in parent folders for least specific
  624. for _ in range(5):
  625. if keyboard_parent == base_path:
  626. break
  627. info_jsons.append(keyboard_parent / 'info.json')
  628. keyboard_parent = keyboard_parent.parent
  629. # Return a list of the info.json files that actually exist
  630. return [info_json for info_json in info_jsons if info_json.exists()]
  631. def keymap_json_config(keyboard, keymap):
  632. """Extract keymap level config
  633. """
  634. keymap_folder = locate_keymap(keyboard, keymap).parent
  635. km_info_json = parse_configurator_json(keymap_folder / 'keymap.json')
  636. return km_info_json.get('config', {})
  637. def keymap_json(keyboard, keymap):
  638. """Generate the info.json data for a specific keymap.
  639. """
  640. keymap_folder = locate_keymap(keyboard, keymap).parent
  641. # Files to scan
  642. keymap_config = keymap_folder / 'config.h'
  643. keymap_rules = keymap_folder / 'rules.mk'
  644. keymap_file = keymap_folder / 'keymap.json'
  645. # Build the info.json file
  646. kb_info_json = info_json(keyboard)
  647. # Merge in the data from keymap.json
  648. km_info_json = keymap_json_config(keyboard, keymap) if keymap_file.exists() else {}
  649. deep_update(kb_info_json, km_info_json)
  650. # Merge in the data from config.h, and rules.mk
  651. _extract_rules_mk(kb_info_json, parse_rules_mk_file(keymap_rules))
  652. _extract_config_h(kb_info_json, parse_config_h_file(keymap_config))
  653. return kb_info_json