minetest-flow-cd2025/generate_docs.py

179 lines
5.7 KiB
Python

from ruamel.yaml import YAML
import collections, functools, re, requests
yaml = YAML(typ='safe')
hide_elements = {
'background', 'background9', 'scroll_container', 'scrollbar', 'tabheader'
}
hide_comments = {'x', 'y', 'w', 'h', 'name', 'selected'}
special_case_names = {'tablecolumns': 'TableColumns',
'tableoptions': 'TableOptions'}
def fetch_elements():
res = requests.get('https://github.com/luk3yx/minetest-formspec_ast/raw/'
'master/elements.yaml')
return yaml.load(res.text)
@functools.cache
def fetch_lua_api():
return requests.get('https://github.com/minetest/minetest/raw/'
'master/doc/lua_api.md').text
Field = collections.namedtuple('Field', 'name type is_list comment',
defaults=(False, ''))
special_case_types = {
'field': (
Field('enter_after_edit', 'boolean', False,
('Makes changing the field submit it on mobile devices.\n'
'Requires a recent version of formspec_ast.')),
),
}
def search_for_fields(obj, *, is_list=False):
assert isinstance(obj, (list, tuple))
if len(obj) == 2:
if obj[1] == '...':
yield from search_for_fields(obj[0], is_list=True)
return
if isinstance(obj[0], str) and isinstance(obj[1], str):
yield Field(obj[0], obj[1], is_list)
return
for e in obj:
yield from search_for_fields(e)
def optional(element_name, field_name):
if field_name in ('w', 'h'):
return (element_name not in ('list', 'hypertext', 'model') and
'image' not in element_name)
return field_name == 'name'
def get_comment(element_name, field):
if field.comment:
return field.comment
if field.name in hide_comments:
return
# Try and get something useful from lua_api.md
_, _, docs = fetch_lua_api().partition(f'\n### `{element_name}[')
doc_lines = docs.split('\n###', 1)[0].split('\n')
s = (f'*_`{field.name}`:_', f'*_`{field.name}`_(optional)')
for line in (it := iter(doc_lines)):
if line.lower().replace(' ', '_').startswith(s):
lines = [line.split(': ', 1)[-1]]
for line2 in it:
if not line2.startswith(' ') or 'comma' in line2:
break
lines.append(line2.lstrip())
return '\n'.join(lines)
def element_to_docs(element_name, variants):
if element_name in special_case_names:
flow_name = special_case_names[element_name]
else:
flow_name = re.sub(r'_(.)', lambda m: m.group(1).upper(),
element_name.capitalize())
fields = collections.Counter(search_for_fields(variants))
if ((element_name in hide_elements or Field('x', 'number') not in fields or
all(field.name in ('x', 'y') for field in fields)) and
element_name not in special_case_names):
return ''
if element_name in special_case_types:
fields.update(special_case_types[element_name])
lines = [
f'### `gui.{flow_name}`\n',
f"Equivalent to Minetest's `{element_name}[]` element.\n",
'**Example**',
'```lua',
f'gui.{flow_name} {{'
]
num = 1
indent = ' ' * 4
if element_name == 'tablecolumns':
lines.append(f'{indent}tablecolumns = {{')
lines.append(f'{indent}{indent}{{')
indent += ' ' * 8
for field, count in fields.items():
if (field.name in ('x', 'y') or (element_name == 'tooltip' and
field.name in ('w', 'h'))):
continue
if field.type == 'number':
value = num
num += 1
elif field.type == 'string':
if field.name == 'name':
value = f'"my_{element_name}"'
elif field.name == 'orientation':
value = '"vertical"'
elif 'color' in field.name:
value = '"#FF0000"'
elif field.name == 'type':
value = '"text"'
elif field.name == 'gui_element_name':
value = '"my_button"'
elif 'texture' in field.name:
value = '"texture.png"'
else:
value = '"Hello world!"'
elif field.type in ('boolean', 'fullscreen'):
value = 'false'
elif field.type == 'table':
value = '{field = "value"}'
else:
value = '<?>'
if field.is_list and field.type != 'table':
value = f'{{{value}, ...}}'
if comment := get_comment(element_name, field):
if lines and lines[-1].startswith(indent):
lines.append('')
for comment_line in comment.split('\n'):
lines.append(f'{indent}-- {comment_line}')
line = f'{indent}{field.name} = {value},'
if ((count < len(variants) or optional(element_name, field.name)) and
field.name != 'gui_element_name'):
line += ' -- Optional'
lines.append(line)
if element_name == 'tablecolumns':
lines.append(' ' * 8 + '},')
lines.append(' ' * 8 + '...')
lines.append(' ' * 4 + '}')
lines.append('}')
lines.append('```')
return '\n'.join(lines)
if __name__ == '__main__':
print('Fetching data...')
elements = fetch_elements()
print('Done.')
with open('elements.md', 'w') as f:
f.write('# Auto-generated elements list\n\n')
f.write('This is probably broken.')
for element_name, variants in elements.items():
docs = element_to_docs(element_name, variants)
if docs:
f.write('\n\n')
f.write(docs)