]>
Commit | Line | Data |
---|---|---|
1 | /* Cycript - Optimizing JavaScript Compiler/Runtime | |
2 | * Copyright (C) 2009-2015 Jay Freeman (saurik) | |
3 | */ | |
4 | ||
5 | /* GNU Affero General Public License, Version 3 {{{ */ | |
6 | /* | |
7 | * This program is free software: you can redistribute it and/or modify | |
8 | * it under the terms of the GNU Affero General Public License as published by | |
9 | * the Free Software Foundation, either version 3 of the License, or | |
10 | * (at your option) any later version. | |
11 | ||
12 | * This program is distributed in the hope that it will be useful, | |
13 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
14 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
15 | * GNU Affero General Public License for more details. | |
16 | ||
17 | * You should have received a copy of the GNU Affero General Public License | |
18 | * along with this program. If not, see <http://www.gnu.org/licenses/>. | |
19 | **/ | |
20 | /* }}} */ | |
21 | ||
22 | #include <complex> | |
23 | #include <sstream> | |
24 | ||
25 | #ifdef HAVE_READLINE_H | |
26 | #include <readline.h> | |
27 | #else | |
28 | #include <readline/readline.h> | |
29 | #endif | |
30 | ||
31 | #if RL_READLINE_VERSION >= 0x0600 | |
32 | ||
33 | #include <sys/ioctl.h> | |
34 | ||
35 | #include "Highlight.hpp" | |
36 | ||
37 | #include <term.h> | |
38 | ||
39 | typedef std::complex<int> CYCursor; | |
40 | ||
41 | extern "C" int rl_display_fixed; | |
42 | extern "C" int _rl_vis_botlin; | |
43 | extern "C" int _rl_last_c_pos; | |
44 | extern "C" int _rl_last_v_pos; | |
45 | ||
46 | CYCursor current_; | |
47 | int width_; | |
48 | size_t point_; | |
49 | ||
50 | unsigned CYDisplayWidth() { | |
51 | struct winsize info; | |
52 | if (ioctl(1, TIOCGWINSZ, &info) != -1) | |
53 | return info.ws_col; | |
54 | return tgetnum(const_cast<char *>("co")); | |
55 | } | |
56 | ||
57 | void CYDisplayOutput_(int (*put)(int), const char *&data) { | |
58 | for (;; ++data) { | |
59 | char next(*data); | |
60 | if (next == '\0' || next == CYIgnoreEnd) | |
61 | return; | |
62 | if (put != NULL) | |
63 | put(next); | |
64 | } | |
65 | } | |
66 | ||
67 | CYCursor CYDisplayOutput(int (*put)(int), int width, const char *data, ssize_t offset = 0) { | |
68 | CYCursor point(current_); | |
69 | ||
70 | for (;;) { | |
71 | if (offset-- == 0) | |
72 | point = current_; | |
73 | ||
74 | char next(*data++); | |
75 | switch (next) { | |
76 | case '\0': | |
77 | return point; | |
78 | break; | |
79 | ||
80 | case CYIgnoreStart: | |
81 | CYDisplayOutput_(put, data); | |
82 | case CYIgnoreEnd: | |
83 | ++offset; | |
84 | break; | |
85 | ||
86 | default: | |
87 | current_ += CYCursor(0, 1); | |
88 | if (current_.imag() == width) | |
89 | case '\n': | |
90 | current_ = CYCursor(current_.real() + 1, 0); | |
91 | if (put != NULL) | |
92 | put(next); | |
93 | break; | |
94 | ||
95 | } | |
96 | } | |
97 | } | |
98 | ||
99 | void CYDisplayMove_(char *negative, char *positive, int offset) { | |
100 | if (offset < 0) | |
101 | putp(tparm(negative, -offset)); | |
102 | else if (offset > 0) | |
103 | putp(tparm(positive, offset)); | |
104 | } | |
105 | ||
106 | void CYDisplayMove(CYCursor target) { | |
107 | CYCursor offset(target - current_); | |
108 | ||
109 | CYDisplayMove_(parm_up_cursor, parm_down_cursor, offset.real()); | |
110 | ||
111 | if (char *parm = tparm(column_address, target.imag())) | |
112 | putp(parm); | |
113 | else | |
114 | CYDisplayMove_(parm_left_cursor, parm_right_cursor, offset.imag()); | |
115 | ||
116 | current_ = target; | |
117 | } | |
118 | ||
119 | void CYDisplayUpdate() { | |
120 | rl_display_fixed = 1; | |
121 | rl_redisplay(); | |
122 | current_ = CYCursor(_rl_last_v_pos, _rl_last_c_pos); | |
123 | ||
124 | #if RL_READLINE_VERSION >= 0x0600 | |
125 | const char *prompt(rl_display_prompt); | |
126 | #else | |
127 | const char *prompt(rl_prompt); | |
128 | #endif | |
129 | ||
130 | std::ostringstream stream; | |
131 | CYLexerHighlight(rl_line_buffer, rl_end, stream, true); | |
132 | std::string string(stream.str()); | |
133 | const char *buffer(string.c_str()); | |
134 | ||
135 | int width(CYDisplayWidth()); | |
136 | if (width_ != width) { | |
137 | current_ = CYCursor(); | |
138 | CYDisplayOutput(NULL, width, prompt); | |
139 | current_ = CYDisplayOutput(NULL, width, buffer, point_); | |
140 | } | |
141 | ||
142 | CYDisplayMove(CYCursor()); | |
143 | CYDisplayOutput(putchar, width, prompt); | |
144 | CYCursor target(CYDisplayOutput(putchar, width, stream.str().c_str(), rl_point)); | |
145 | ||
146 | _rl_vis_botlin = current_.real(); | |
147 | ||
148 | if (current_.imag() == 0) | |
149 | CYDisplayOutput(putchar, width, " "); | |
150 | putp(clr_eos); | |
151 | ||
152 | CYDisplayMove(target); | |
153 | fflush(stdout); | |
154 | ||
155 | _rl_last_v_pos = current_.real(); | |
156 | _rl_last_c_pos = current_.imag(); | |
157 | ||
158 | width_ = width; | |
159 | point_ = rl_point; | |
160 | } | |
161 | ||
162 | #endif |