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
25 uint32_t opts_warn [1 + (COUNT_WARNINGS / 32)];
26 bool opts_werror = false;
29 * isatty/STDERR_FILENO/STDOUT_FILNO
30 * + some other things likewise.
36 #define GMQCC_IS_STDOUT(X) ((FILE*)((void*)X) == stdout)
37 #define GMQCC_IS_STDERR(X) ((FILE*)((void*)X) == stderr)
38 #define GMQCC_IS_DEFINE(X) (GMQCC_IS_STDERR(X) || GMQCC_IS_STDOUT(X))
49 * Doing colored output on windows is fucking stupid. The linux way is
50 * the real way. So we emulate it on windows :)
53 #define WIN32_LEAN_AND_MEAN
57 * Windows doesn't have constants for FILENO, sadly but the docs tell
58 * use the constant values.
62 #define STDERR_FILENO 2
63 #define STDOUT_FILENO 1
66 * Windows and it's posix underscore bullshit. We simply fix this
67 * with yay, another macro :P
69 #define isatty _isatty
91 WCYAN = WBLUE | WGREEN,
92 WMAGENTA = WBLUE | WRED,
93 WYELLOW = WGREEN | WRED,
94 WWHITE = WBLUE | WGREEN | WRED
97 static const ansi2win[] = {
108 static void win_fputs(char *str, FILE *f) {
109 /* state for translate */
119 int colors[] = {-1, -1 };
122 CONSOLE_SCREEN_BUFFER_INFO cinfo;
123 GetConsoleScreenBufferInfo(
124 (GMQCC_IS_STDOUT(h)) ?
125 GetStdHandle(STD_OUTPUT_HANDLE) :
126 GetStdHandle(STD_ERROR_HANDLE), &cinfo
128 icolor = cinfo.wAttributes;
133 else if (state == '\e' && *str == '[')
135 else if (state == '[') {
137 colors[colorpos] = *str;
142 for (find = colorpos + 1, acolor = 0, mult = 1; find < 2; find++) {
143 acolor += (colors[find] - 48) * mult;
147 /* convert to windows color */
150 else if (acolor == RESET) {
154 else if (BLACK < acolor && acolor <= WHITE)
155 wcolor = ansi2win[acolor - 30];
156 else if (acolor == 90) {
157 /* special gray really white man */
162 SetConsoleTextattribute(
164 GetStdHandle(STD_OUTPUT_HANDLE) :
165 GetStdHandle(STD_ERROR_HANDLE),
167 wcolor | intense | (icolor & 0xF0)
177 SetConsoleTextAttribute(
178 (GMQCC_IS_STDOUT(h)) ?
179 GetStdHandle(STD_OUTPUT_HANDLE) :
180 GetStdHandle(STD_ERROR_HANDLE),
187 * We use standard files as default. These can be changed at any time
188 * with con_change(F, F)
190 static con_t console;
193 * Enables color on output if supported.
194 * NOTE: The support for checking colors is NULL. On windows this will
195 * always work, on *nix it depends if the term has colors.
197 * NOTE: This prevents colored output to piped stdout/err via isatty
200 static void con_enablecolor() {
201 if (console.handle_err == stderr || console.handle_err == stdout)
202 console.color_err = !!(isatty(STDERR_FILENO));
203 if (console.handle_out == stderr || console.handle_out == stdout)
204 console.color_out = !!(isatty(STDOUT_FILENO));
208 * Does a write to the handle with the format string and list of
209 * arguments. This colorizes for windows as well via translate
212 static int con_write(FILE *handle, const char *fmt, va_list va) {
215 ln = vfprintf(handle, fmt, va);
219 ln = _vscprintf(fmt, va);
220 data = malloc(ln + 1);
222 vsprintf(data, fmt, va);
223 if (GMQCC_IS_DEFINE(handle))
224 ln = win_fputs(data, handle);
226 ln = fputs(data, handle);
233 /**********************************************************************
234 * EXPOSED INTERFACE BEGINS
235 *********************************************************************/
238 if (!GMQCC_IS_DEFINE(console.handle_err))
239 fclose(console.handle_err);
240 if (!GMQCC_IS_DEFINE(console.handle_out))
241 fclose(console.handle_out);
244 void con_color(int state) {
248 console.color_err = 0;
249 console.color_out = 0;
254 console.handle_err = stderr;
255 console.handle_out = stdout;
265 * This is clever, say you want to change the console to use two
266 * files for out/err. You pass in two strings, it will properly
267 * close the existing handles (if they're not std* handles) and
268 * open them. Now say you want TO use stdout and stderr, this
269 * allows you to do that so long as you cast them to (char*).
270 * Say you need stdout for out, but want a file for error, you can
271 * do this too, just cast the stdout for (char*) and stick to a
272 * string for the error file.
274 int con_change(const char *out, const char *err) {
277 if (GMQCC_IS_DEFINE(out)) {
278 console.handle_out = GMQCC_IS_STDOUT(out) ? stdout : stderr;
280 } else if (!(console.handle_out = fopen(out, "w"))) return 0;
282 if (GMQCC_IS_DEFINE(err)) {
283 console.handle_err = GMQCC_IS_STDOUT(err) ? stdout : stderr;
285 } else if (!(console.handle_err = fopen(err, "w"))) return 0;
288 setvbuf(console.handle_out, NULL, _IONBF, 0);
289 setvbuf(console.handle_err, NULL, _IONBF, 0);
294 int con_verr(const char *fmt, va_list va) {
295 return con_write(console.handle_err, fmt, va);
297 int con_vout(const char *fmt, va_list va) {
298 return con_write(console.handle_out, fmt, va);
302 * Standard stdout/stderr printf functions used generally where they need
305 int con_err(const char *fmt, ...) {
313 int con_out(const char *fmt, ...) {
323 * Utility console message writes for lexer contexts. These will allow
324 * for reporting of file:line based on lexer context, These are used
325 * heavily in the parser/ir/ast.
327 void con_vprintmsg(int level, const char *name, size_t line, const char *msgtype, const char *msg, va_list ap) {
328 /* color selection table */
335 int err = !!(level == LVL_ERROR);
336 int color = (err) ? console.color_err : console.color_out;
337 int (*print)(const char *, ...) = (err) ? &con_err : &con_out;
338 int (*vprint)(const char *, va_list) = (err) ? &con_verr : &con_vout;
341 print("\033[0;%dm%s:%d: \033[0;%dm%s: \033[0m", CON_CYAN, name, (int)line, sel[level], msgtype);
343 print("%s:%d: %s: ", name, (int)line, msgtype);
349 void con_printmsg(int level, const char *name, size_t line, const char *msgtype, const char *msg, ...) {
352 con_vprintmsg(level, name, line, msgtype, msg, va);
356 void con_cvprintmsg(void *ctx, int lvl, const char *msgtype, const char *msg, va_list ap) {
357 con_vprintmsg(lvl, ((lex_ctx*)ctx)->file, ((lex_ctx*)ctx)->line, msgtype, msg, ap);
360 void con_cprintmsg (void *ctx, int lvl, const char *msgtype, const char *msg, ...) {
363 con_cvprintmsg(ctx, lvl, msgtype, msg, va);
367 /* General error interface */
368 size_t compile_errors = 0;
369 size_t compile_warnings = 0;
371 void compile_error(lex_ctx ctx, const char *msg, ...)
376 con_cvprintmsg((void*)&ctx, LVL_ERROR, "error", msg, ap);
380 bool GMQCC_WARN compile_warning(lex_ctx ctx, int warntype, const char *fmt, ...)
383 int lvl = LVL_WARNING;
385 if (!OPTS_WARN(warntype))
396 con_vprintmsg(lvl, ctx.file, ctx.line, (opts_werror ? "error" : "warning"), fmt, ap);