Files
armorpaint/paint/sources/ui/tab_scripts.c
T
2026-06-30 19:55:54 +02:00

566 lines
18 KiB
C

#include "../global.h"
ui_text_coloring_t *tab_scripts_text_coloring = NULL;
i32 tab_scripts_selected = 0;
gpu_texture_t *tab_scripts_minimap_tex = NULL;
i32 tab_scripts_minimap_selected = -1;
extern bool tab_scripts_minimap_dirty;
bool tab_scripts_minimap_scrolling = false;
void tab_scripts_prepare() {
if (g_project->script_datas == NULL) {
g_project->script_datas = string_array_create(0);
g_project->script_names = string_array_create(0);
}
if (g_project->script_datas->length == 0) {
string_array_push(g_project->script_datas, "void main() {\n \n}\n");
string_array_push(g_project->script_names, "main.c");
}
// A script may have been removed
if (tab_scripts_selected >= g_project->script_datas->length) {
tab_scripts_selected = g_project->script_datas->length - 1;
}
}
char *tab_scripts_get() {
tab_scripts_prepare();
return g_project->script_datas->buffer[tab_scripts_selected];
}
void tab_scripts_set(char *s) {
tab_scripts_prepare();
g_project->script_datas->buffer[tab_scripts_selected] = string_copy(s);
tab_scripts_minimap_dirty = true;
}
void tab_scripts_create(char *name) {
tab_scripts_prepare();
i32 i = string_array_index_of(g_project->script_names, name);
if (i < 0) {
string_array_push(g_project->script_names, string_copy(name));
string_array_push(g_project->script_datas, "");
tab_scripts_selected = g_project->script_datas->length - 1;
}
}
void tab_scripts_draw_export(char *path) {
char *str = tab_scripts_get();
char *f = ui_files_filename;
if (string_equals(f, "")) {
f = string_copy(tr("untitled"));
}
path = string("%s%s%s", path, PATH_SEP, f);
if (!ends_with(path, ".c")) {
path = string("%s.c", path);
}
iron_file_save_bytes(path, sys_string_to_buffer(str), 0);
}
void tab_scripts_draw_import(char *path) {
buffer_t *b = data_get_blob(path);
tab_scripts_set(sys_buffer_to_string(b));
data_delete_blob(path);
}
void tab_scripts_draw_edit() {
tab_scripts_prepare();
if (ui_menu_button(tr("Clear"), "", ICON_ERASE)) {
tab_scripts_set("");
}
g_ui->enabled = !string_equals(g_project->script_names->buffer[tab_scripts_selected], "main.c");
if (ui_menu_button(tr("Delete"), "", ICON_DELETE)) {
array_splice((any_array_t *)g_project->script_datas, tab_scripts_selected, 1);
array_splice((any_array_t *)g_project->script_names, tab_scripts_selected, 1);
tab_scripts_selected = 0;
}
g_ui->enabled = true;
if (ui_menu_button(tr("Import"), "", ICON_IMPORT)) {
ui_files_show("c", false, false, &tab_scripts_draw_import);
}
if (ui_menu_button(tr("Export"), "", ICON_EXPORT)) {
ui_files_show("c", true, false, &tab_scripts_draw_export);
}
if (ui_menu_sub_button(ui_handle(__ID__), tr("Templates"))) {
ui_menu_sub_begin(2);
if (ui_menu_button("hello.c", "", ICON_DRAFT)) {
tab_scripts_set("\
void main() {\n\
printf(\"Hello, world!\\n\");\n\
}\n\
");
}
if (ui_menu_button("rotate.c", "", ICON_DRAFT)) {
tab_scripts_set("\
void on_update() {\n\
mesh_object_t *o = context_main_object();\n\
transform_rotate(o->base->transform, vec4_z_axis(), 0.005);\n\
context_t *c = script_get_context();\n\
c->ddirty = 2;\n\
}\n\
void main() {\n\
script_notify_on_update(on_update);\n\
}\n\
");
}
ui_menu_sub_end();
}
}
// Autocomplete popup (ctrl+space) listing the script api from minic_api.c
#define TAB_SCRIPTS_AC_MAX 8
bool tab_scripts_ac_show = false;
i32 tab_scripts_ac_offset = 0;
static bool tab_scripts_is_ident(char c) {
return c == '_' || (c >= '0' && c <= '9') || (c >= 'A' && c <= 'Z') || (c >= 'a' && c <= 'z');
}
static i32 tab_scripts_line_start(char *text, i32 line) {
i32 pos = 0;
i32 l = 0;
while (l < line && text[pos] != '\0') {
if (text[pos] == '\n') {
l++;
}
pos++;
}
return pos;
}
static i32 tab_scripts_prefix_len(char *text, i32 cursor) {
i32 n = 0;
while (cursor - n - 1 >= 0 && tab_scripts_is_ident(text[cursor - n - 1])) {
n++;
}
return n;
}
static i32 tab_scripts_suffix_len(char *text, i32 cursor) {
i32 n = 0;
while (tab_scripts_is_ident(text[cursor + n])) {
n++;
}
return n;
}
// Collect up to TAB_SCRIPTS_AC_MAX api symbols containing prefix
static i32 tab_scripts_autocomplete_matches(char *prefix, char **names) {
char *lower_prefix = to_lower_case(prefix);
i32 func_count = minic_ext_func_count_get();
i32 global_count = minic_global_count_get();
i32 total = func_count + global_count;
i32 n = 0;
for (i32 i = 0; i < total && n < TAB_SCRIPTS_AC_MAX; ++i) {
char *name = (char *)(i < func_count ? minic_ext_func_name_at(i) : minic_global_name_at(i - func_count));
if (starts_with(to_lower_case(name), lower_prefix)) {
names[n++] = name;
}
}
for (i32 i = 0; i < total && n < TAB_SCRIPTS_AC_MAX; ++i) {
char *name = (char *)(i < func_count ? minic_ext_func_name_at(i) : minic_global_name_at(i - func_count));
char *lower = to_lower_case(name);
if (!starts_with(lower, lower_prefix) && string_index_of(lower, lower_prefix) >= 0) {
names[n++] = name;
}
}
return n;
}
// Replace the whole identifier around the cursor with the chosen symbol
static void tab_scripts_autocomplete_complete(char *name, i32 prefix_len, i32 suffix_len) {
tab_scripts_prepare();
char *text = g_project->script_datas->buffer[tab_scripts_selected];
i32 line = tab_scripts_hscript->i;
i32 col = g_ui->cursor_x;
i32 cur = tab_scripts_line_start(text, line) + col;
char *before = substring(text, 0, cur - prefix_len);
char *after = substring(text, cur + suffix_len, string_length(text));
tab_scripts_set(string("%s%s%s", before, name, after));
tab_scripts_hscript->text = g_project->script_datas->buffer[tab_scripts_selected];
i32 new_col = col - prefix_len + string_length(name);
g_ui->cursor_x = new_col;
g_ui->highlight_anchor = new_col;
g_ui->cursor_sticky_x = new_col;
strcpy(g_ui->text_selected, ui_extract_line(tab_scripts_hscript->text, line)); // Keep the active line in sync
tab_scripts_ac_show = false;
}
static void tab_scripts_toggle_comment() {
tab_scripts_prepare();
char *text = g_project->script_datas->buffer[tab_scripts_selected];
i32 line = tab_scripts_hscript->i;
i32 col = g_ui->cursor_x;
i32 start = tab_scripts_line_start(text, line);
i32 end = start;
while (text[end] != '\0' && text[end] != '\n') {
end++;
}
// Indentation level: first non-space char on the line
i32 ind = start;
while (ind < end && text[ind] == ' ') {
ind++;
}
i32 ind_col = ind - start;
i32 delta;
char *before = substring(text, 0, ind);
if (text[ind] == '/' && ind + 1 < end && text[ind + 1] == '/') {
// Remove "// " or "//"
i32 remove = (ind + 2 < end && text[ind + 2] == ' ') ? 3 : 2;
char *after = substring(text, ind + remove, string_length(text));
tab_scripts_set(string("%s%s", before, after));
delta = -remove;
}
else {
// Prepend "// "
char *after = substring(text, ind, string_length(text));
tab_scripts_set(string("%s// %s", before, after));
delta = 3;
}
tab_scripts_hscript->text = g_project->script_datas->buffer[tab_scripts_selected];
// Keep the caret on the same characters
i32 new_col;
if (col <= ind_col) {
new_col = col; // Caret within indentation, unaffected
}
else if (delta < 0 && col < ind_col - delta) {
new_col = ind_col; // Caret was inside the removed "// "
}
else {
new_col = col + delta;
}
g_ui->cursor_x = new_col;
g_ui->highlight_anchor = new_col;
g_ui->cursor_sticky_x = new_col;
strcpy(g_ui->text_selected, ui_extract_line(tab_scripts_hscript->text, line)); // Keep the active line in sync
}
static bool tab_scripts_minimap_visible(f32 *x, f32 *y, f32 *w, f32 *h) {
if (g_ui->_window_w <= 800 * UI_SCALE()) {
return false;
}
f32 mm_w = 150 * UI_SCALE();
*x = g_ui->_window_w - mm_w;
*y = g_ui->window_header_h;
*w = mm_w;
*h = g_ui->_window_h - g_ui->window_header_h;
return true;
}
static void tab_scripts_cache_minimap() {
tab_scripts_minimap_dirty = false;
char *text = tab_scripts_get();
if (text == NULL) {
return;
}
f32 line_h = 2 * UI_SCALE();
f32 char_w = 1 * UI_SCALE();
i32 tex_w = math_floor(150 * UI_SCALE());
string_array_t *lines = string_split(text, "\n");
i32 tex_h = math_floor(lines->length * line_h);
if (tex_h < 1) {
tex_h = 1;
}
// Create the render target
if (tab_scripts_minimap_tex == NULL || tab_scripts_minimap_tex->width != tex_w || tab_scripts_minimap_tex->height != tex_h) {
if (tab_scripts_minimap_tex != NULL) {
gpu_delete_texture(tab_scripts_minimap_tex);
}
gc_unroot(tab_scripts_minimap_tex);
tab_scripts_minimap_tex = gpu_create_render_target(tex_w, tex_h, GPU_TEXTURE_FORMAT_RGBA32);
gc_root(tab_scripts_minimap_tex);
}
// Render each word as a small rect
draw_begin(tab_scripts_minimap_tex, true, 0x00000000);
draw_set_color(g_theme->HOVER_COL);
for (i32 i = 0; i < lines->length; ++i) {
string_array_t *words = string_split(lines->buffer[i], " ");
f32 x = 0;
for (i32 j = 0; j < words->length; ++j) {
i32 len = string_length(words->buffer[j]);
if (len > 0) {
draw_filled_rect(x, i * line_h, len * char_w, line_h);
}
x += (len + 1) * char_w;
}
}
draw_end();
}
static void tab_scripts_draw_minimap(f32 mm_x, f32 mm_y, f32 mm_w, f32 mm_h) {
if (tab_scripts_minimap_tex == NULL) {
return;
}
f32 line_h = 2 * UI_SCALE();
string_array_t *lines = string_split(tab_scripts_hscript->text, "\n");
f32 content_h = lines->length * UI_ELEMENT_H();
f32 full_h = lines->length * line_h;
f32 scroll_progress = content_h > 0 ? -g_ui->current_window->scroll_offset / content_h : 0;
f32 out_of_screen = full_h - mm_h;
if (out_of_screen < 0) {
out_of_screen = 0;
}
// Draw the visible region of the cached minimap
f32 tex_h = tab_scripts_minimap_tex->height;
f32 sh = tex_h < mm_h ? tex_h : mm_h;
f32 sy = out_of_screen * scroll_progress;
if (sy > tex_h - sh) {
sy = tex_h - sh;
}
if (sy < 0) {
sy = 0;
}
draw_set_color(0xffffffff);
draw_sub_image(tab_scripts_minimap_tex, 0, sy, mm_w, sh, mm_x, mm_y);
i32 offset = math_floor(sy / line_h);
// View box
f32 visible_area = out_of_screen > 0 ? mm_h : full_h;
f32 box_h = (mm_h / UI_ELEMENT_H()) * line_h;
f32 box_y = mm_y + scroll_progress * visible_area;
if (box_y < mm_y) {
box_y = mm_y;
}
else if (box_y > mm_y + mm_h - box_h) {
box_y = mm_y + mm_h - box_h;
}
draw_set_color(0x11ffffff);
draw_filled_rect(mm_x, box_y, mm_w, box_h);
// Drag the minimap to scroll the text area
if (tab_scripts_minimap_scrolling && content_h > mm_h) {
f32 my_local = g_ui->input_y - (g_ui->_window_y + mm_y);
i32 picked = offset + math_floor(my_local / line_h);
f32 visible_lines = mm_h / UI_ELEMENT_H();
f32 target = (picked - visible_lines / 2) * UI_ELEMENT_H();
f32 max_scroll = (lines->length - 1) * UI_ELEMENT_H(); // Last line at the top
if (target < 0) {
target = 0;
}
else if (target > max_scroll) {
target = max_scroll;
}
g_ui->current_window->scroll_offset = -target;
}
}
void tab_scripts_draw(ui_handle_t *htab) {
if (ui_tab(htab, tr("Scripts"), false, -1, false)) {
// Cache minimap
f32 mm_x, mm_y, mm_w, mm_h;
bool minimap_on = tab_scripts_minimap_visible(&mm_x, &mm_y, &mm_w, &mm_h);
if (minimap_on && tab_scripts_minimap_dirty) {
draw_end();
tab_scripts_cache_minimap();
draw_begin(&g_ui->current_window->texture, false, 0);
}
ui_begin_sticky();
f32_array_t *row = f32_array_create_from_raw(
(f32[]){
-70,
-70,
-140,
},
3);
// #ifndef NDEBUG
// if (g_config->experimental) {
// f32_array_push(row, -90);
// }
// #endif
ui_row(row);
if (ui_icon_button(tr("Run"), ICON_PLAY, UI_ALIGN_CENTER)) {
minic_ctx_t *ctx = minic_eval(tab_scripts_hscript->text);
// minic_ctx_free(ctx);
}
if (ui_icon_button(tr("Edit"), ICON_EDIT, UI_ALIGN_CENTER)) {
ui_menu_draw(&tab_scripts_draw_edit, -1, -1);
}
tab_scripts_prepare();
ui_handle_t *file_handle = ui_handle(__ID__);
file_handle->i = tab_scripts_selected;
tab_scripts_selected = ui_combo(file_handle, g_project->script_names, tr("File"), false, UI_ALIGN_LEFT, true);
// #ifndef NDEBUG
// if (g_config->experimental && ui_icon_button("Run Tests", ICON_PLAY, UI_ALIGN_CENTER)) {
// minic_tests();
// }
// #endif
ui_end_sticky();
draw_font_t *_font = g_font;
i32 _font_size = g_ui->font_size;
draw_font_t *f = data_get_font("font_mono.ttf");
ui_set_font(g_ui, f);
g_ui->font_size = math_floor(15 * UI_SCALE());
ui_text_area_line_numbers = true;
ui_text_area_scroll_past_end = true;
gc_unroot(ui_text_area_coloring);
ui_text_area_coloring = tab_scripts_get_text_coloring();
gc_root(ui_text_area_coloring);
tab_scripts_prepare();
tab_scripts_hscript->text = g_project->script_datas->buffer[tab_scripts_selected];
bool ac_selected = g_ui->text_selected_handle == tab_scripts_hscript;
if (!ac_selected) {
tab_scripts_ac_show = false;
}
// Open the autocomplete popup on ctrl+space
if (ac_selected && g_ui->is_ctrl_down && g_ui->is_key_pressed && g_ui->key_code == KEY_CODE_SPACE) {
tab_scripts_ac_show = true;
tab_scripts_ac_offset = 0;
g_ui->is_key_pressed = false; // Consume so the editor ignores ctrl+space
minic_register_builtins(); // Ensure the script api is registered
}
// Toggle line comment on ctrl+/
if (ac_selected && g_ui->is_ctrl_down && g_ui->is_key_pressed && g_ui->key_code == KEY_CODE_SLASH) {
tab_scripts_toggle_comment();
g_ui->is_key_pressed = false; // Consume so the editor ignores ctrl+/
g_ui->key_code = 0;
}
bool ac_accept = false;
if (tab_scripts_ac_show && ac_selected && g_ui->is_key_pressed) {
if (g_ui->key_code == KEY_CODE_DOWN) {
tab_scripts_ac_offset++;
g_ui->is_key_pressed = false;
g_ui->key_code = 0;
}
else if (g_ui->key_code == KEY_CODE_UP) {
tab_scripts_ac_offset--;
g_ui->is_key_pressed = false;
g_ui->key_code = 0;
}
else if (g_ui->key_code == KEY_CODE_RETURN || g_ui->key_code == KEY_CODE_TAB) {
ac_accept = true;
g_ui->is_key_pressed = false;
g_ui->key_code = 0;
}
else if (g_ui->key_code == KEY_CODE_ESCAPE) {
tab_scripts_ac_show = false;
g_ui->is_key_pressed = false;
g_ui->key_code = 0;
}
}
if (ac_selected && g_ui->is_key_pressed) {
tab_scripts_minimap_dirty = true;
}
if (tab_scripts_minimap_selected != tab_scripts_selected) {
tab_scripts_minimap_selected = tab_scripts_selected;
tab_scripts_minimap_dirty = true;
}
minimap_on = tab_scripts_minimap_visible(&mm_x, &mm_y, &mm_w, &mm_h);
if (minimap_on && g_ui->input_started && ui_input_in_rect(g_ui->_window_x + mm_x, g_ui->_window_y + mm_y, mm_w, mm_h)) {
tab_scripts_minimap_scrolling = true;
}
if (!g_ui->input_down) {
tab_scripts_minimap_scrolling = false;
}
// Prevent text area clicks while scrolling the minimap
bool _input_enabled = g_ui->input_enabled;
if (tab_scripts_minimap_scrolling) {
g_ui->input_enabled = false;
}
ui_text_area(tab_scripts_hscript, UI_ALIGN_LEFT, true, "", false);
g_ui->input_enabled = _input_enabled;
g_project->script_datas->buffer[tab_scripts_selected] = tab_scripts_hscript->text;
if (minimap_on) {
tab_scripts_draw_minimap(mm_x, mm_y, mm_w, mm_h);
}
// Autocomplete popup
if (tab_scripts_ac_show && ac_selected) {
char *text = tab_scripts_hscript->text;
i32 line = tab_scripts_hscript->i;
i32 col = g_ui->cursor_x;
i32 cur = tab_scripts_line_start(text, line) + col;
i32 plen = tab_scripts_prefix_len(text, cur);
i32 slen = tab_scripts_suffix_len(text, cur);
char *prefix = substring(text, cur - plen, cur);
char *names[TAB_SCRIPTS_AC_MAX];
i32 n = tab_scripts_autocomplete_matches(prefix, names);
if (n == 0) {
tab_scripts_ac_show = false;
}
else {
tab_scripts_ac_offset = tab_scripts_ac_offset < 0 ? 0 : (tab_scripts_ac_offset > n - 1 ? n - 1 : tab_scripts_ac_offset);
if (ac_accept) {
tab_scripts_autocomplete_complete(names[tab_scripts_ac_offset], plen, slen);
}
else {
// Draw the list below the caret
i32 row_h = math_floor(g_ui->font_size + 8 * UI_SCALE());
f32 pad = 8 * UI_SCALE();
f32 max_w = 0;
for (i32 i = 0; i < n; ++i) {
f32 w = draw_string_width(g_font, g_ui->font_size, names[i]);
max_w = w > max_w ? w : max_w;
}
f32 px = g_ui->cursor_screen_x;
f32 py = g_ui->cursor_screen_y + UI_ELEMENT_H();
f32 pw = max_w + pad * 2;
f32 ph = row_h * n;
ui_draw_shadow(px, py, pw, ph);
draw_set_color(g_theme->SEPARATOR_COL);
draw_filled_rect(px, py, pw, ph);
draw_set_font(g_font, g_ui->font_size);
for (i32 i = 0; i < n; ++i) {
f32 iy = py + row_h * i;
if (i == tab_scripts_ac_offset) {
draw_set_color(g_theme->HIGHLIGHT_COL);
draw_filled_rect(px, iy, pw, row_h);
}
draw_set_color(g_theme->TEXT_COL);
draw_string(names[i], px + pad, iy + math_floor((row_h - g_ui->font_size) / 2.0));
}
}
}
}
ui_text_area_line_numbers = false;
ui_text_area_scroll_past_end = false;
gc_unroot(ui_text_area_coloring);
ui_text_area_coloring = NULL;
ui_set_font(g_ui, _font);
g_ui->font_size = _font_size;
}
}
ui_text_coloring_t *tab_scripts_get_text_coloring() {
if (tab_scripts_text_coloring == NULL) {
buffer_t *blob = data_get_blob("text_coloring.json");
gc_unroot(tab_scripts_text_coloring);
tab_scripts_text_coloring = json_parse(sys_buffer_to_string(blob));
gc_root(tab_scripts_text_coloring);
}
return tab_scripts_text_coloring;
}