--- /dev/null
+#include "vg_engine.h"
+#include "vg_console.h"
+#include "vg_imgui.h"
+#include "vg_log.h"
+#include "vg_string.h"
+#include <string.h>
+
+struct vg_console vg_console;
+
+void vg_console_reg_var( const char *alias, void *ptr, enum vg_var_dtype type,
+ u32 flags )
+{
+ if( vg_thread_purpose() == k_thread_purpose_main )
+ vg_fatal_error( "FIXME: Cannot register VAR from main thread" );
+
+ if( vg_console.var_count > vg_list_size(vg_console.vars) )
+ vg_fatal_error( "Too many vars registered" );
+
+ vg_var *var = &vg_console.vars[ vg_console.var_count ++ ];
+ var->name = alias;
+ var->data = ptr;
+ var->data_type = type;
+ var->flags = flags;
+
+ vg_info( "Console variable '%s' registered\n", alias );
+}
+
+void vg_console_reg_cmd( const char *alias,
+ int (*function)(int argc, const char *argv[]),
+ void (*poll_suggest)(int argc, const char *argv[]) )
+{
+ if( vg_thread_purpose() == k_thread_purpose_main )
+ vg_fatal_error( "FIXME: Cannot register CMD from main thread" );
+
+ if( vg_console.function_count > vg_list_size(vg_console.functions) )
+ vg_fatal_error( "Too many functions registered" );
+
+ vg_cmd *cmd = &vg_console.functions[ vg_console.function_count ++ ];
+
+ cmd->function = function;
+ cmd->poll_suggest = poll_suggest;
+ cmd->name = alias;
+
+ vg_info( "Console function '%s' registered\n", alias );
+}
+
+static int _vg_console_list( int argc, char const *argv[] )
+{
+ for( int i=0; i<vg_console.function_count; i ++ ){
+ struct vg_cmd *cmd = &vg_console.functions[ i ];
+ vg_info( "* %s\n", cmd->name );
+ }
+
+ for( int i=0; i<vg_console.var_count; i ++ ){
+ struct vg_var *cv = &vg_console.vars[ i ];
+ vg_info( "%s %s\n",
+ (const char *[]){ "i32","u32","f32","str" }[cv->data_type],
+ cv->name );
+ }
+
+ return 0;
+}
+
+static void vg_console_write_persistent(void)
+{
+ FILE *fp = fopen( "cfg/auto.conf", "w" );
+
+ if( !fp ){
+ vg_error( "Cannot open cfg/auto.conf\n" );
+ return;
+ }
+
+ for( int i=0; i<vg_console.var_count; i ++ ){
+ struct vg_var *cv = &vg_console.vars[i];
+
+ if( cv->flags & VG_VAR_PERSISTENT ){
+ if( cv->data_type == k_var_dtype_i32 ){
+ fprintf( fp, "%s %d\n", cv->name, *(i32 *)(cv->data) );
+ }
+ else if( cv->data_type == k_var_dtype_u32 ){
+ fprintf( fp, "%s %u\n", cv->name, *(u32 *)(cv->data) );
+ }
+ else if( cv->data_type == k_var_dtype_f32 ){
+ fprintf( fp, "%s %.5f\n", cv->name, *(float *)(cv->data ) );
+ }
+ else if( cv->data_type == k_var_dtype_str ){
+ vg_str *str = cv->data;
+ if( str->buffer && (str->i > 0) )
+ fprintf( fp, "%s %s\n", cv->name, str->buffer );
+ }
+ }
+ }
+
+ fclose( fp );
+}
+
+static void _vg_console_free(void)
+{
+ vg_console_write_persistent();
+}
+
+/*
+ * splits src into tokens and fills out args as pointers to those tokens
+ * returns number of tokens
+ * dst must be as long as src
+ */
+static int vg_console_tokenize( const char *src, char *dst,
+ const char *args[8] )
+{
+ int arg_count = 0,
+ in_token = 0;
+
+ for( int i=0;; i ++ ){
+ if( src[i] ){
+ if( src[i] == ' ' || src[i] == '\t' ){
+ if( in_token )
+ dst[i] = '\0';
+
+ in_token = 0;
+
+ if( arg_count == 8 )
+ break;
+ }
+ else{
+ dst[i] = src[i];
+
+ if( !in_token ){
+ args[ arg_count ++ ] = &dst[i];
+ in_token = 1;
+ }
+ }
+ }
+ else{
+ dst[i] = '\0';
+ break;
+ }
+ }
+
+ return arg_count;
+}
+
+vg_var *vg_console_match_var( const char *kw )
+{
+ for( int i=0; i<vg_console.var_count; i ++ ){
+ struct vg_var *cv = &vg_console.vars[ i ];
+ if( !strcmp( cv->name, kw ) ){
+ return cv;
+ }
+ }
+
+ return NULL;
+}
+
+vg_cmd *vg_console_match_cmd( const char *kw )
+{
+ for( int i=0; i<vg_console.function_count; i ++ ){
+ struct vg_cmd *cmd = &vg_console.functions[ i ];
+ if( !strcmp( cmd->name, kw ) ){
+ return cmd;
+ }
+ }
+
+ return NULL;
+}
+
+void vg_execute_console_input( const char *cmd, bool silent )
+{
+ char temp[512];
+ char const *args[8];
+ int arg_count = vg_console_tokenize( cmd, temp, args );
+
+ if( arg_count == 0 )
+ return;
+
+ vg_var *cv = vg_console_match_var( args[0] );
+ vg_cmd *fn = vg_console_match_cmd( args[0] );
+
+ if( cv ){
+ /* Cvar Matched, try get value */
+ if( arg_count >= 2 ){
+ if( cv->flags & VG_VAR_CHEAT ){
+ if( !vg_console.cheats && !silent ){
+ vg_error( "variable is cheat protected\n" );
+ return;
+ }
+ }
+
+ if( (cv->data_type == k_var_dtype_u32) ||
+ (cv->data_type == k_var_dtype_i32) ){
+ int *ptr = cv->data;
+ *ptr = atoi( args[1] );
+ }
+ else if( cv->data_type == k_var_dtype_f32 ){
+ float *ptr = cv->data;
+ *ptr = atof( args[1] );
+ }
+ else if( cv->data_type == k_var_dtype_str ){
+ vg_str *str = cv->data;
+ vg_strfree( str );
+ vg_strnull( str, NULL, -1 );
+
+ for( int i=1; i<arg_count; i ++ ){
+ vg_strcat( str, args[i] );
+ if( i!=arg_count-1 ) vg_strcatch( str, ' ' );
+ }
+ }
+ }
+ else{
+ if( cv->data_type == k_var_dtype_i32 )
+ vg_info( "= %d\n", *((int *)cv->data) );
+ else if( cv->data_type == k_var_dtype_u32 )
+ vg_info( "= %u\n", *((u32 *)cv->data) );
+ else if( cv->data_type == k_var_dtype_f32 )
+ vg_info( "= %.4f\n", *((float *)cv->data) );
+ else if( cv->data_type == k_var_dtype_str ){
+ vg_str *str = cv->data;
+ vg_info( "= '%s'\n", str->buffer );
+ }
+ }
+
+ return;
+ }
+ else if( fn ){
+ fn->function( arg_count-1, args+1 );
+ return;
+ }
+
+ if( !silent )
+ vg_error( "No command/var named '%s'. Use 'list' to view all\n",args[0]);
+}
+
+u32 str_lev_distance( const char *s1, const char *s2 ){
+ u32 m = strlen( s1 ),
+ n = strlen( s2 );
+
+ if( m==0 ) return n;
+ if( n==0 ) return m;
+
+ u32 costs[ 256 ];
+
+ for( u32 k=0; k<=n; k++ )
+ costs[k] = k;
+
+ u32 i = 0;
+ for( u32 i=0; i<m; i++ ){
+ costs[0] = i+1;
+
+ u32 corner = i;
+
+ for( u32 j=0; j<n; j++ ){
+ u32 upper = costs[j+1];
+
+ if( s1[i] == s2[j] )
+ costs[ j+1 ] = corner;
+ else{
+ u32 t = (upper < corner)? upper: corner;
+ costs[j+1] = ((costs[j] < t)? costs[j]: t) + 1;
+ }
+
+ corner = upper;
+ }
+ }
+
+ return costs[n];
+}
+
+u32 str_lcs( const char *s1, const char *s2 ){
+ u32 m = VG_MIN( 31, strlen( s1 ) ),
+ n = VG_MIN( 31, strlen( s2 ) );
+
+ int suff[32][32],
+ result = 0;
+
+ for( int i=0; i<=m; i++ ){
+ for( int j=0; j<=n; j++ ){
+ if( i == 0 || j == 0 )
+ suff[i][j] = 0;
+ else if( s1[i-1] == s2[j-1] ){
+ suff[i][j] = suff[i-1][j-1] + 1;
+ result = VG_MAX( result, suff[i][j] );
+ }
+ else
+ suff[i][j] = 0;
+ }
+ }
+
+ return result;
+}
+
+/* str must not fuckoff ever! */
+void console_suggest_score_text( const char *str, const char *input,
+ int minscore )
+{
+ /* filter duplicates */
+ for( int i=0; i<vg_console.suggestion_count; i++ )
+ if( !strcmp( vg_console.suggestions[i].str, str ) )
+ return;
+
+ /* calc score */
+ u32 score = str_lcs( str, input );
+
+ if( score < minscore )
+ return;
+
+ int best_pos = vg_console.suggestion_count;
+ for( int j=best_pos-1; j>=0; j -- )
+ if( score > vg_console.suggestions[j].lev_score )
+ best_pos = j;
+
+ /* insert if good score */
+ if( best_pos < vg_list_size( vg_console.suggestions ) ){
+ int start = VG_MIN( vg_console.suggestion_count,
+ vg_list_size( vg_console.suggestions )-1 );
+ for( int j=start; j>best_pos; j -- )
+ vg_console.suggestions[j] = vg_console.suggestions[j-1];
+
+ vg_console.suggestions[ best_pos ].str = str;
+ vg_console.suggestions[ best_pos ].len = strlen( str );
+ vg_console.suggestions[ best_pos ].lev_score = score;
+
+ if( vg_console.suggestion_count <
+ vg_list_size( vg_console.suggestions ) )
+ vg_console.suggestion_count ++;
+ }
+}
+
+static void console_update_suggestions(void)
+{
+ if( vg_ui.focused_control_type != k_ui_control_textbox ||
+ vg_ui.textbuf != vg_console.input )
+ return;
+
+ vg_console.suggestion_count = 0;
+ vg_console.suggestion_select = -1;
+ vg_console.suggestion_maxlen = 0;
+
+ /*
+ * - must be typing something
+ * - must be at the end
+ * - prev char must not be a whitespace
+ * - cursors should match
+ */
+
+ if( vg_ui.textbox.cursor_pos == 0 ) return;
+ if( vg_ui.textbox.cursor_pos != vg_ui.textbox.cursor_user ) return;
+ if( vg_console.input[ vg_ui.textbox.cursor_pos ] != '\0' ) return;
+
+ if( (vg_console.input[ vg_ui.textbox.cursor_pos -1 ] == ' ') ||
+ (vg_console.input[ vg_ui.textbox.cursor_pos -1 ] == '\t') )
+ return;
+
+ char temp[128];
+ const char *args[8];
+
+ int token_count = vg_console_tokenize( vg_console.input, temp, args );
+ if( !token_count ) return;
+ vg_console.suggestion_pastepos = args[token_count-1]-temp;
+
+ /* Score all our commands and cvars */
+ if( token_count == 1 ){
+ for( int i=0; i<vg_console.var_count; i++ ){
+ vg_var *cvar = &vg_console.vars[i];
+ console_suggest_score_text( cvar->name, args[0], 1 );
+ }
+
+ for( int i=0; i<vg_console.function_count; i++ ){
+ vg_cmd *cmd = &vg_console.functions[i];
+ console_suggest_score_text( cmd->name, args[0], 1 );
+ }
+ }
+ else{
+ vg_cmd *cmd = vg_console_match_cmd( args[0] );
+ vg_var *var = vg_console_match_var( args[0] );
+
+ if( cmd )
+ if( cmd->poll_suggest )
+ cmd->poll_suggest( token_count-1, &args[1] );
+ }
+
+ /* some post processing */
+ for( int i=0; i<vg_console.suggestion_count; i++ ){
+ vg_console.suggestion_maxlen = VG_MAX( vg_console.suggestion_maxlen,
+ vg_console.suggestions[i].len );
+
+ if( vg_console.suggestions[i].lev_score <
+ vg_console.suggestions[0].lev_score/2 )
+ {
+ vg_console.suggestion_count = i;
+ return;
+ }
+ }
+}
+
+/*
+ * Suggestion controls
+ */
+static void _console_fetch_suggestion(void)
+{
+ char *target = &vg_console.input[ vg_console.suggestion_pastepos ];
+
+ if( vg_console.suggestion_select == -1 ){
+ strcpy( target, vg_console.input_copy );
+ _ui_textbox_move_cursor( &vg_ui.textbox.cursor_user,
+ &vg_ui.textbox.cursor_pos, 10000, 1 );
+ }
+ else{
+ strncpy( target,
+ vg_console.suggestions[ vg_console.suggestion_select ].str,
+ vg_list_size( vg_console.input )-1 );
+
+ _ui_textbox_move_cursor( &vg_ui.textbox.cursor_user,
+ &vg_ui.textbox.cursor_pos, 10000, 1 );
+ _ui_textbox_put_char( ' ' );
+ }
+}
+
+static void _console_suggest_store_normal(void)
+{
+ if( vg_console.suggestion_select == -1 ){
+ char *target = &vg_console.input[ vg_console.suggestion_pastepos ];
+ strcpy( vg_console.input_copy, target );
+ }
+}
+
+static void console_suggest_next(void)
+{
+ if( vg_console.suggestion_count ){
+ _console_suggest_store_normal();
+
+ vg_console.suggestion_select ++;
+
+ if( vg_console.suggestion_select >= vg_console.suggestion_count )
+ vg_console.suggestion_select = -1;
+
+ _console_fetch_suggestion();
+ }
+}
+
+static void console_suggest_prev(void)
+{
+ if( vg_console.suggestion_count ){
+ _console_suggest_store_normal();
+
+ vg_console.suggestion_select --;
+
+ if( vg_console.suggestion_select < -1 )
+ vg_console.suggestion_select = vg_console.suggestion_count-1;
+
+ _console_fetch_suggestion();
+ }
+}
+
+static void _vg_console_on_update( char *buf, u32 len )
+{
+ if( buf == vg_console.input ){
+ console_update_suggestions();
+ }
+}
+
+static void console_history_get( char* buf, int entry_num )
+{
+ if( !vg_console.history_count )
+ return;
+
+ int offset = VG_MIN( entry_num, vg_console.history_count -1 ),
+ pick = (vg_console.history_last - offset) %
+ vg_list_size( vg_console.history );
+ strcpy( buf, vg_console.history[ pick ] );
+}
+
+static void _vg_console_on_up( char *buf, u32 len )
+{
+ if( buf == vg_console.input ){
+ vg_console.history_pos =
+ VG_MAX
+ (
+ 0,
+ VG_MIN
+ (
+ vg_console.history_pos+1,
+ VG_MIN
+ (
+ vg_list_size( vg_console.history ),
+ vg_console.history_count - 1
+ )
+ )
+ );
+
+ console_history_get( vg_console.input, vg_console.history_pos );
+ _ui_textbox_move_cursor( &vg_ui.textbox.cursor_user,
+ &vg_ui.textbox.cursor_pos,
+ vg_list_size(vg_console.input)-1, 1);
+ }
+}
+
+static void _vg_console_on_down( char *buf, u32 len )
+{
+ if( buf == vg_console.input ){
+ vg_console.history_pos = VG_MAX( 0, vg_console.history_pos-1 );
+ console_history_get( vg_console.input, vg_console.history_pos );
+
+ _ui_textbox_move_cursor( &vg_ui.textbox.cursor_user,
+ &vg_ui.textbox.cursor_pos,
+ vg_list_size(vg_console.input)-1, 1 );
+ }
+}
+
+static void _vg_console_on_enter( char *buf, u32 len )
+{
+ if( buf == vg_console.input ){
+ if( !strlen( vg_console.input ) )
+ return;
+
+ vg_info( "%s\n", vg_console.input );
+
+ if( strcmp( vg_console.input,
+ vg_console.history[ vg_console.history_last ]) )
+ {
+ vg_console.history_last = ( vg_console.history_last + 1) %
+ vg_list_size(vg_console.history );
+ vg_console.history_count =
+ VG_MIN( vg_list_size( vg_console.history ),
+ vg_console.history_count + 1 );
+ strcpy( vg_console.history[ vg_console.history_last ],
+ vg_console.input );
+ }
+
+ vg_console.history_pos = -1;
+ vg_execute_console_input( vg_console.input, 0 );
+ _ui_textbox_move_cursor( &vg_ui.textbox.cursor_user,
+ &vg_ui.textbox.cursor_pos, -10000, 1 );
+
+ vg_console.input[0] = '\0';
+ console_update_suggestions();
+ }
+}
+
+int vg_console_exec( int argc, const char *argv[] )
+{
+ if( argc < 1 ) return 0;
+
+ int silent=0;
+ if( argc == 2 ) silent=1;
+
+ char path[256];
+ strcpy( path, "cfg/" );
+ strncat( path, argv[0], 250 );
+
+ FILE *fp = fopen( path, "r" );
+ if( fp ){
+ char line[256];
+
+ while( fgets( line, sizeof( line ), fp ) ){
+ line[ strcspn( line, "\r\n#" ) ] = 0x00;
+
+ if( line[0] != 0x00 ){
+ vg_execute_console_input( line, silent );
+ }
+ }
+
+ fclose( fp );
+ }
+ else{
+ vg_error( "Could not open '%s'\n", path );
+ }
+
+ return 0;
+}
+
+
+void vg_console_init(void)
+{
+ vg_console_reg_cmd( "list", _vg_console_list, NULL );
+ vg_console_reg_cmd( "exec", vg_console_exec, NULL );
+ vg_console_reg_var( "cheats", &vg_console.cheats, k_var_dtype_i32,
+#ifdef VG_DEVWINDOW
+ VG_VAR_PERSISTENT
+#else
+ 0
+#endif
+ );
+}
+
+void vg_console_load_autos(void)
+{
+ vg_console_exec( 2, (const char *[]){ "auto.conf", "silent" } );
+}
+
+void vg_console_draw(void)
+{
+ if( !vg_console.enabled ) return;
+
+ SDL_AtomicLock( &vg_log.print_sl );
+
+ int ptr = vg_log.log_line_current;
+ int const fh = vg_ui.font->line_height, log_lines = 32;
+ int console_lines = VG_MIN( log_lines, vg_log.log_line_count );
+
+ ui_rect rect_log = { 0, 0, vg.window_x, log_lines*fh },
+ rect_input = { 0, log_lines*fh + 1, vg.window_x, fh*2 },
+ rect_line = { 0, 0, vg.window_x, fh };
+
+ /*
+ * log
+ */
+ u32 bg_colour = (ui_colour( k_ui_bg )&0x00ffffff)|0x9f000000;
+
+ ui_fill( rect_log, bg_colour );
+ rect_line[1] = rect_log[1]+rect_log[3]-fh;
+
+ for( int i=0; i<console_lines; i ++ ){
+ ptr --;
+
+ if( ptr < 0 ) ptr = vg_list_size( vg_log.log )-1;
+
+ ui_text( rect_line, vg_log.log[ptr], 1, k_ui_align_left, 0 );
+ rect_line[1] -= fh;
+ }
+
+ /*
+ * Input area
+ */
+
+ struct ui_textbox_callbacks callbacks = {
+ .up = _vg_console_on_up,
+ .down = _vg_console_on_down,
+ .change = _vg_console_on_update,
+ .enter = _vg_console_on_enter,
+ };
+ ui_textbox( rect_input, NULL,
+ vg_console.input, vg_list_size(vg_console.input), 1,
+ UI_TEXTBOX_AUTOFOCUS, &callbacks );
+
+ /*
+ * suggestions
+ */
+ if( vg_console.suggestion_count ){
+ ui_rect rect_suggest;
+ rect_copy( rect_input, rect_suggest );
+
+ rect_suggest[0] += 6 + vg_ui.font->spacing*vg_console.suggestion_pastepos;
+ rect_suggest[1] += rect_input[3];
+ rect_suggest[2] = vg_ui.font->spacing * vg_console.suggestion_maxlen;
+ rect_suggest[3] = vg_console.suggestion_count * fh;
+
+ ui_fill( rect_suggest, bg_colour );
+
+ rect_suggest[3] = fh;
+
+ for( int i=0; i<vg_console.suggestion_count; i ++ ){
+ u32 text_colour;
+ if( i == vg_console.suggestion_select ){
+ ui_fill( rect_suggest, ui_colour( k_ui_orange ) );
+ text_colour = ui_colourcont( k_ui_orange );
+ }
+ else text_colour = ui_colourcont( k_ui_bg );
+
+ ui_text( rect_suggest, vg_console.suggestions[i].str, 1,
+ k_ui_align_left, text_colour );
+
+ rect_suggest[1] += fh;
+ }
+ }
+
+ SDL_AtomicUnlock( &vg_log.print_sl );
+}