mirror/v
1
0
Fork 0
mirror of https://github.com/vlang/v.git synced 2023-08-10 21:13:21 +03:00
Code Issues Projects Releases Wiki Activity

readline: add custom prompt feature

Browse Source
This commit is contained in:
Henrixounez 2019-09-02 15:29:40 +02:00 committed by Alexander Medvednikov
parent 93716ee944
commit 510fb2a981
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
vlib/readline/readline_lin.v
View File

@ -39,6 +39,7 @@ mut:
cursor int // Cursor position cursor int // Cursor position
overwrite bool overwrite bool
cursor_row_offset int cursor_row_offset int
prompt string
} }
@ -107,10 +108,12 @@ fn (r Readline) read_char() byte {
// Main function of the readline module // Main function of the readline module
// Will loop and ingest characters until EOF or Enter // Will loop and ingest characters until EOF or Enter
// Returns the completed line // Returns the completed line
pub fn (r mut Readline) read_line() string { pub fn (r mut Readline) read_line(prompt string) string {
r.current = '' r.current = ''
r.cursor = 0 r.cursor = 0
r.prompt = prompt
print(r.prompt)
for { for {
c := r.read_char() c := r.read_char()
a := r.analyse(c) a := r.analyse(c)
@ -235,15 +238,17 @@ fn calculate_screen_position(x_in int, y_in int, screen_columns int, char_count
} }
// Will redraw the line // Will redraw the line
// TODO: Fix if prompt is longer than terminal columns
fn (r mut Readline) refresh_line() { fn (r mut Readline) refresh_line() {
mut end_of_input := [0, 0] mut end_of_input := [0, 0]
calculate_screen_position(0, 0, get_screen_columns(), r.current.len, mut end_of_input) calculate_screen_position(r.prompt.len, 0, get_screen_columns(), r.current.len, mut end_of_input)
end_of_input[1] += r.current.count('\n') end_of_input[1] += r.current.count('\n')
mut cursor_pos := [0, 0] mut cursor_pos := [0, 0]
calculate_screen_position(0, 0, get_screen_columns(), r.cursor, mut cursor_pos) calculate_screen_position(r.prompt.len, 0, get_screen_columns(), r.cursor, mut cursor_pos)
shift_cursor(0, -r.cursor_row_offset) shift_cursor(0, -r.cursor_row_offset)
term.erase_toend() term.erase_toend()
print(r.prompt)
print(r.current) print(r.current)
if end_of_input[0] == 0 && end_of_input[1] > 0 { if end_of_input[0] == 0 && end_of_input[1] > 0 {
print('\n') print('\n')
@ -264,13 +269,8 @@ fn (r mut Readline) insert_character(c byte) {
// Overwriting // Overwriting
} }
r.cursor++ r.cursor++
// Simply print new character if at end of line // Refresh the line to add the new character
// Otherwise refresh the line if cursor != r.current.len
if r.cursor == r.current.len && r.current.len < get_screen_columns() {
print(c.str())
} else {
r.refresh_line() r.refresh_line()
}
} }
// Removes the character behind cursor. // Removes the character behind cursor.