5 * Permission is hereby granted, free of charge, to any person obtaining a copy of
6 * this software and associated documentation files (the "Software"), to deal in
7 * the Software without restriction, including without limitation the rights to
8 * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
9 * of the Software, and to permit persons to whom the Software is furnished to do
10 * so, subject to the following conditions:
12 * The above copyright notice and this permission notice shall be included in all
13 * copies or substantial portions of the Software.
15 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
26 * isatty/STDERR_FILENO/STDOUT_FILNO
27 * + some other things likewise.
33 #define GMQCC_IS_STDOUT(X) ((FILE*)((void*)X) == stdout)
34 #define GMQCC_IS_STDERR(X) ((FILE*)((void*)X) == stderr)
35 #define GMQCC_IS_DEFINE(X) (GMQCC_IS_STDERR(X) || GMQCC_IS_STDOUT(X))
46 * Doing colored output on windows is fucking stupid. The linux way is
47 * the real way. So we emulate it on windows :)
50 #define WIN32_LEAN_AND_MEAN
54 * Windows doesn't have constants for FILENO, sadly but the docs tell
55 * use the constant values.
59 #define STDERR_FILENO 2
60 #define STDOUT_FILENO 1
63 * Windows and it's posix underscore bullshit. We simply fix this
64 * with yay, another macro :P
66 #define isatty _isatty
88 WCYAN = WBLUE | WGREEN,
89 WMAGENTA = WBLUE | WRED,
90 WYELLOW = WGREEN | WRED,
91 WWHITE = WBLUE | WGREEN | WRED
94 static const ansi2win[] = {
105 static void win_fputs(char *str, FILE *f) {
106 /* state for translate */
116 int colors[] = {-1, -1 };
119 CONSOLE_SCREEN_BUFFER_INFO cinfo;
120 GetConsoleScreenBufferInfo(
121 (GMQCC_IS_STDOUT(h)) ?
122 GetStdHandle(STD_OUTPUT_HANDLE) :
123 GetStdHandle(STD_ERROR_HANDLE), &cinfo
125 icolor = cinfo.wAttributes;
130 else if (state == '\e' && *str == '[')
132 else if (state == '[') {
134 colors[colorpos] = *str;
139 for (find = colorpos + 1, acolor = 0, mult = 1; find < 2; find++) {
140 acolor += (colors[find] - 48) * mult;
144 /* convert to windows color */
147 else if (acolor == RESET) {
151 else if (BLACK < acolor && acolor <= WHITE)
152 wcolor = ansi2win[acolor - 30];
153 else if (acolor == 90) {
154 /* special gray really white man */
159 SetConsoleTextattribute(
161 GetStdHandle(STD_OUTPUT_HANDLE) :
162 GetStdHandle(STD_ERROR_HANDLE),
164 wcolor | intense | (icolor & 0xF0)
174 SetConsoleTextAttribute(
175 (GMQCC_IS_STDOUT(h)) ?
176 GetStdHandle(STD_OUTPUT_HANDLE) :
177 GetStdHandle(STD_ERROR_HANDLE),
184 * We use standard files as default. These can be changed at any time
185 * with con_change(F, F)
187 static con_t console;
190 * Enables color on output if supported.
191 * NOTE: The support for checking colors is NULL. On windows this will
192 * always work, on *nix it depends if the term has colors.
194 * NOTE: This prevents colored output to piped stdout/err via isatty
197 static void con_enablecolor() {
198 if (console.handle_err == stderr || console.handle_err == stdout)
199 console.color_err = !!(isatty(STDERR_FILENO));
200 if (console.handle_out == stderr || console.handle_out == stdout)
201 console.color_out = !!(isatty(STDOUT_FILENO));
205 * Does a write to the handle with the format string and list of
206 * arguments. This colorizes for windows as well via translate
209 static int con_write(FILE *handle, const char *fmt, va_list va) {
212 ln = vfprintf(handle, fmt, va);
216 ln = _vscprintf(fmt, va);
217 data = malloc(ln + 1);
219 vsprintf(data, fmt, va);
220 if (GMQCC_IS_DEFINE(handle))
221 ln = win_fputs(data, handle);
223 ln = fputs(data, handle);
230 /**********************************************************************
231 * EXPOSED INTERFACE BEGINS
232 *********************************************************************/
235 if (!GMQCC_IS_DEFINE(console.handle_err))
236 fclose(console.handle_err);
237 if (!GMQCC_IS_DEFINE(console.handle_out))
238 fclose(console.handle_out);
241 void con_color(int state) {
245 console.color_err = 0;
246 console.color_out = 0;
251 console.handle_err = stderr;
252 console.handle_out = stdout;
262 * This is clever, say you want to change the console to use two
263 * files for out/err. You pass in two strings, it will properly
264 * close the existing handles (if they're not std* handles) and
265 * open them. Now say you want TO use stdout and stderr, this
266 * allows you to do that so long as you cast them to (char*).
267 * Say you need stdout for out, but want a file for error, you can
268 * do this too, just cast the stdout for (char*) and stick to a
269 * string for the error file.
271 int con_change(const char *out, const char *err) {
274 if (GMQCC_IS_DEFINE(out)) {
275 console.handle_out = GMQCC_IS_STDOUT(out) ? stdout : stderr;
277 } else if (!(console.handle_out = fopen(out, "w"))) return 0;
279 if (GMQCC_IS_DEFINE(err)) {
280 console.handle_err = GMQCC_IS_STDOUT(err) ? stdout : stderr;
282 } else if (!(console.handle_err = fopen(err, "w"))) return 0;
285 setvbuf(console.handle_out, NULL, _IONBF, 0);
286 setvbuf(console.handle_err, NULL, _IONBF, 0);
291 int con_verr(const char *fmt, va_list va) {
292 return con_write(console.handle_err, fmt, va);
294 int con_vout(const char *fmt, va_list va) {
295 return con_write(console.handle_out, fmt, va);
299 * Standard stdout/stderr printf functions used generally where they need
302 int con_err(const char *fmt, ...) {
310 int con_out(const char *fmt, ...) {
320 * Utility console message writes for lexer contexts. These will allow
321 * for reporting of file:line based on lexer context, These are used
322 * heavily in the parser/ir/ast.
324 void con_vprintmsg(int level, const char *name, size_t line, const char *msgtype, const char *msg, va_list ap) {
325 /* color selection table */
332 int err = !!(level == LVL_ERROR);
333 int color = (err) ? console.color_err : console.color_out;
334 int (*print)(const char *, ...) = (err) ? &con_err : &con_out;
335 int (*vprint)(const char *, va_list) = (err) ? &con_verr : &con_vout;
338 print("\033[0;%dm%s:%d: \033[0;%dm%s: \033[0m", CON_CYAN, name, (int)line, sel[level], msgtype);
340 print("%s:%d: %s: ", name, (int)line, msgtype);
346 void con_printmsg(int level, const char *name, size_t line, const char *msgtype, const char *msg, ...) {
349 con_vprintmsg(level, name, line, msgtype, msg, va);
353 void con_cvprintmsg(void *ctx, int lvl, const char *msgtype, const char *msg, va_list ap) {
354 con_vprintmsg(lvl, ((lex_ctx*)ctx)->file, ((lex_ctx*)ctx)->line, msgtype, msg, ap);
357 void con_cprintmsg (void *ctx, int lvl, const char *msgtype, const char *msg, ...) {
360 con_cvprintmsg(ctx, lvl, msgtype, msg, va);