56 lines
1.8 KiB
Python
56 lines
1.8 KiB
Python
from typing import List
|
|
|
|
from .iobase import pipe_interaction
|
|
|
|
|
|
class CheckboxOption:
|
|
def __init__(self, id, text, group: str = None, checked: bool = False):
|
|
# it will show as: [] (id): text
|
|
self._id = id
|
|
self._text = text
|
|
self._group = group
|
|
self._checked = checked
|
|
|
|
|
|
def make_checkbox_control(title: str, options: List[CheckboxOption]) -> (str, str, List[str]):
|
|
_NT = "\n"
|
|
groups = list({option._group: 1 for option in options if option._group}.keys())
|
|
|
|
def _check_option_message(option):
|
|
return f"> [{'x' if option._checked else ''}]({option._id}) {option._text}"
|
|
|
|
def _check_option_group_message(group):
|
|
s = _NT.join(
|
|
[_check_option_message(option) for option in options if option._group == group]
|
|
)
|
|
return f"{group}:{_NT}{s}"
|
|
|
|
ui_message = f"""
|
|
{title}
|
|
{_NT.join([_check_option_group_message(group) for group in groups])}
|
|
"""
|
|
return ("checkbox", ui_message, [option._id for option in options])
|
|
|
|
|
|
def checkbox_answer(response: dict, ids: List[str]) -> List[str]:
|
|
return [key for key, value in response.items() if value == "checked" and key in ids]
|
|
|
|
|
|
def ui_checkbox_select(title: str, options: List[CheckboxOption]) -> List[str]:
|
|
"""
|
|
send text to UI as:
|
|
```chatmark
|
|
Which files would you like to commit? I've suggested a few.
|
|
> [x](file1) devchat/engine/prompter.py
|
|
> [x](file2) devchat/prompt.py
|
|
> [](file3) tests/test_cli_prompt.py
|
|
```
|
|
"""
|
|
_1, ui_message, ids = make_checkbox_control(title, options)
|
|
ui_message = f"""```chatmark type=form\n{ui_message}\n```\n"""
|
|
|
|
# print(ui_message)
|
|
# return [option._id for option in options]
|
|
response = pipe_interaction(ui_message)
|
|
return checkbox_answer(response, ids)
|