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