] [input-file]\n\n", name); 45 | fprintf(out, "\t-d, --date\n" 46 | "\t\tSet the date of the manpage (default: now),\n" 47 | "\t-h, --help\n" 48 | "\t\tDisplay this help text and exit without further processing\n" 49 | "\t-s, --section\n" 50 | "\t\tSet the section of the manpage (default: 1)\n" 51 | "\t-t, --title\n" 52 | "\t\tSet the title of the manpage (default: filename)\n"); } 53 | 54 | struct metadata { 55 | char *title; 56 | char *date; 57 | int section; 58 | }; 59 | 60 | static void 61 | man_text_escape(struct buf *ob, char *src, size_t size) { 62 | size_t i = 0, org; 63 | while (i < size) { 64 | /* copying directly unescaped characters */ 65 | org = i; 66 | while (i < size && src[i] != '-') 67 | i += 1; 68 | if (i > org) bufput(ob, src + org, i - org); 69 | 70 | /* escaping */ 71 | if (i >= size) break; 72 | else if (src[i] == '-') BUFPUTSL(ob, "\\-"); 73 | i += 1; } } 74 | 75 | static void 76 | man_prolog(struct buf *ob, void *opaque) { 77 | struct metadata *m = (struct metadata *)opaque; 78 | bufprintf(ob, 79 | ".\\\" Generated by mkd2man\n" 80 | ".Dd %s\n" 81 | ".Dt %s %d\n" 82 | ".Os", 83 | m->date, 84 | m->title, 85 | m->section 86 | ); } 87 | 88 | static void 89 | man_epilog(struct buf *ob, void *opaque) { 90 | BUFPUTSL(ob, "\n"); } 91 | 92 | static void 93 | man_blockcode(struct buf *ob, struct buf *text, void *opaque) { 94 | if (ob->size) bufputc(ob, '\n'); 95 | BUFPUTSL(ob, ".Bd -literal\n"); 96 | if (text) man_text_escape(ob, text->data, text->size); 97 | BUFPUTSL(ob, ".Ed"); } 98 | 99 | static void 100 | man_blockquote(struct buf *ob, struct buf *text, void *opaque) { 101 | if (ob->size) bufputc(ob, '\n'); 102 | BUFPUTSL(ob, ".Eo\n"); 103 | if (text) man_text_escape(ob, text->data, text->size); 104 | BUFPUTSL(ob, "\n.Ec"); } 105 | 106 | static int 107 | man_codespan(struct buf *ob, struct buf *text, void *opaque) { 108 | if (ob->size) bufputc(ob, '\n'); 109 | BUFPUTSL(ob, ".Bd -literal\n"); 110 | if (text) man_text_escape(ob, text->data, text->size); 111 | BUFPUTSL(ob, ".Ed"); 112 | return 1; } 113 | 114 | static void 115 | man_header(struct buf *ob, struct buf *text, int level, void *opaque) { 116 | if (ob->size) bufputc(ob, '\n'); 117 | switch(level) { 118 | case 1: 119 | BUFPUTSL(ob,".Sh "); 120 | break; 121 | case 2: 122 | BUFPUTSL(ob, ".Ss "); 123 | break; 124 | case 3: 125 | BUFPUTSL(ob, ".Pp\n.Em "); 126 | break; 127 | } 128 | if (text) bufput(ob, text->data, text->size); 129 | } 130 | 131 | static int 132 | man_double_emphasis(struct buf *ob, struct buf *text, char c, void *opaque) { 133 | if (!text || !text->size) return 0; 134 | BUFPUTSL(ob, "\\fB"); 135 | bufput(ob, text->data, text->size); 136 | BUFPUTSL(ob, "\\fP"); 137 | return 1; } 138 | 139 | static int 140 | man_emphasis(struct buf *ob, struct buf *text, char c, void *opaque) { 141 | if (!text || !text->size) return 0; 142 | BUFPUTSL(ob, "\\fI"); 143 | if (text) bufput(ob, text->data, text->size); 144 | BUFPUTSL(ob, "\\fP"); 145 | return 1; } 146 | 147 | static int 148 | man_linebreak(struct buf *ob, void *opaque) { 149 | BUFPUTSL(ob, ".br"); 150 | return 1; } 151 | 152 | static void 153 | man_paragraph(struct buf *ob, struct buf *text, void *opaque) { 154 | if (ob->size) bufputc(ob, '\n'); 155 | BUFPUTSL(ob, ".Pp\n"); 156 | if (text) bufput(ob, text->data, text->size); } 157 | 158 | static void 159 | man_list(struct buf *ob, struct buf *text, int flags, void *opaque) { 160 | if (ob->size) bufputc(ob, '\n'); 161 | if (flags & MKD_LIST_ORDERED) 162 | BUFPUTSL(ob,".Bl -enum\n"); 163 | else 164 | BUFPUTSL(ob,".Bl -bullet\n"); 165 | if (text) bufput(ob, text->data, text->size); 166 | BUFPUTSL(ob, ".El"); } 167 | 168 | static void 169 | man_listitem(struct buf *ob, struct buf *text, int flags, void *opaque) { 170 | BUFPUTSL(ob, ".It\n"); 171 | if (text) { 172 | while (text->size && text->data[text->size - 1] == '\n') 173 | text->size -= 1; 174 | bufput(ob, text->data, text->size); } 175 | BUFPUTSL(ob, "\n"); } 176 | 177 | static void 178 | man_normal_text(struct buf *ob, struct buf *text, void *opaque) { 179 | if (text) man_text_escape(ob, text->data, text->size); } 180 | 181 | 182 | /* renderer structure */ 183 | static struct mkd_renderer to_man = { 184 | /* document-level callbacks */ 185 | man_prolog, 186 | man_epilog, 187 | 188 | /* block-level callbacks */ 189 | man_blockcode, 190 | NULL, 191 | man_blockquote, 192 | NULL, 193 | man_header, 194 | NULL, 195 | man_list, 196 | man_listitem, 197 | man_paragraph, 198 | NULL, 199 | NULL, 200 | NULL, 201 | 202 | /* span-level callbacks */ 203 | NULL, 204 | man_codespan, 205 | man_double_emphasis, 206 | man_emphasis, 207 | NULL, 208 | man_linebreak, 209 | NULL, 210 | NULL, 211 | NULL, 212 | 213 | /* low-level callbacks */ 214 | NULL, 215 | man_normal_text, 216 | 217 | /* renderer data */ 218 | 64, 219 | "*_", 220 | NULL }; 221 | 222 | 223 | 224 | /***************** 225 | * MAIN FUNCTION * 226 | *****************/ 227 | 228 | /* main • main function, interfacing STDIO with the parser */ 229 | int 230 | main(int argc, char **argv) { 231 | struct buf *ib, *ob; 232 | size_t ret; 233 | size_t i; 234 | FILE *in = stdin; 235 | int ch, argerr, help; 236 | char *tmp; 237 | char datebuf[64]; 238 | time_t ttm; 239 | struct tm *tm; 240 | struct stat st; 241 | struct metadata man_metadata; 242 | 243 | struct option longopts[] = { 244 | { "date", required_argument, 0, 'd' }, 245 | { "help", no_argument, 0, 'h' }, 246 | { "section", required_argument, 0, 's' }, 247 | { "title", required_argument, 0, 't' }, 248 | { 0, 0, 0, 0} 249 | }; 250 | 251 | man_metadata.section = 1; 252 | man_metadata.title = NULL; 253 | man_metadata.date = NULL; 254 | /* opening the file if given from the command line */ 255 | argerr = help = 0; 256 | while (!argerr && 257 | (ch = getopt_long(argc, argv, "d:hs:t:", longopts, 0)) != -1) 258 | switch (ch) { 259 | case 'd': 260 | man_metadata.date = optarg; 261 | break; 262 | case 'h': 263 | argerr = help = 1; 264 | break; 265 | case 's': 266 | if (strlen(optarg) != 1 && 267 | strspn(optarg, "123456789") != 1) { 268 | argerr = 1; 269 | break; } 270 | man_metadata.section = (int)strtol(optarg, 271 | (char **)NULL, 10); 272 | break; 273 | case 't': 274 | man_metadata.title = optarg; 275 | break; 276 | default: 277 | argerr = 1; } 278 | if (argerr) { 279 | usage(help ? stdout : stderr, argv[0]); 280 | return help ? EXIT_SUCCESS : EXIT_FAILURE; 281 | } 282 | 283 | argc -= optind; 284 | argv += optind; 285 | 286 | if (argc > 0) { 287 | in = fopen(argv[0], "r"); 288 | if (!in) { 289 | fprintf(stderr,"Unable to open input file \"%s\": %s\n", 290 | argv[0], strerror(errno)); 291 | return EXIT_FAILURE; } } 292 | 293 | if (!man_metadata.date) { 294 | if (in == stdin || stat(argv[0], &st) == -1) { 295 | ttm = time(NULL); 296 | tm = localtime(&ttm); } 297 | else 298 | tm = localtime(&st.st_mtime); 299 | strftime(datebuf, sizeof(datebuf), "%B %d, %Y", tm); 300 | man_metadata.date = datebuf; 301 | } 302 | 303 | if (in == stdin && !man_metadata.title) { 304 | fprintf(stderr, "When reading from stdin the title should be " 305 | "specified is expected\n"); 306 | return EXIT_FAILURE; } 307 | 308 | if (!man_metadata.title) { 309 | tmp = strrchr(argv[0], '/'); 310 | man_metadata.title = strrchr(argv[0], '/'); 311 | if (!tmp) 312 | tmp = argv[0]; 313 | else 314 | tmp++; 315 | man_metadata.title = tmp; 316 | tmp = strrchr(man_metadata.title, '.'); 317 | if (tmp) 318 | *tmp = '\0'; } 319 | 320 | /* Ensure the title is uppercase */ 321 | for (i = 0; i < strlen(man_metadata.title); i++) 322 | man_metadata.title[i] = toupper(man_metadata.title[i]); 323 | 324 | /* reading everything */ 325 | ib = bufnew(READ_UNIT); 326 | bufgrow(ib, READ_UNIT); 327 | while ((ret = fread(ib->data + ib->size, 1, 328 | ib->asize - ib->size, in)) > 0) { 329 | ib->size += ret; 330 | bufgrow(ib, ib->size + READ_UNIT); } 331 | if (in != stdin) fclose(in); 332 | 333 | to_man.opaque = &man_metadata; 334 | /* performing markdown to man */ 335 | ob = bufnew(OUTPUT_UNIT); 336 | markdown(ob, ib, &to_man); 337 | 338 | /* writing the result to stdout */ 339 | ret = fwrite(ob->data, 1, ob->size, stdout); 340 | if (ret < ob->size) 341 | fprintf(stderr, "Warning: only %zu output byte written, " 342 | "out of %zu\n", 343 | ret, 344 | ob->size); 345 | 346 | /* cleanup */ 347 | bufrelease(ib); 348 | bufrelease(ob); 349 | return EXIT_SUCCESS; } 350 | 351 | /* vim: set filetype=c: */ 352 | -------------------------------------------------------------------------------- /ConsoleRenderer.cpp: -------------------------------------------------------------------------------- 1 | extern "C" 2 | { 3 | #include "libsoldout/markdown.h" 4 | } 5 | 6 | #include <stack> 7 | #include <iostream> 8 | #include <string> 9 | #include <cstring> 10 | 11 | #include "ColouredBuffer.hpp" 12 | #include "include/console-markdown/CMDMD.hpp" 13 | 14 | #include "include/console-markdown/CPP.hpp" 15 | #include "include/console-markdown/Pawn.hpp" 16 | 17 | #ifndef CONMD_WINDOWS 18 | #include <sys/ioctl.h> 19 | #include <unistd.h> 20 | #define sprintf_s sprintf 21 | #endif 22 | 23 | static int RenderMarkdown(struct buf * ob, char const * style, struct buf const * text) 24 | { 25 | if (text && text->size) 26 | { 27 | size_t 28 | len = strlen(style); 29 | bufput(ob, style, len); 30 | bufput(ob, text->data, text->size); 31 | BUFPUTSL(ob, "\x1B[0m"); 32 | return 1; 33 | } 34 | return 0; 35 | } 36 | 37 | struct console_data_s 38 | { 39 | size_t WindowWidth; 40 | size_t BufferWidth; 41 | int BulletPoint; 42 | }; 43 | 44 | /*static void 45 | rndr_blockquote(struct buf *ob, struct buf *text, void *opaque) 46 | { 47 | if (ob->size) 48 | { 49 | bufputc(ob, '\n'); 50 | } 51 | BUFPUTSL(ob, "<blockquote>\n"); 52 | if (text) 53 | { 54 | bufput(ob, text->data, text->size); 55 | } 56 | BUFPUTSL(ob, "</blockquote>\n"); 57 | }*/ 58 | 59 | static int 60 | rndr_codespan(struct buf *ob, struct buf *text, void *opaque) 61 | { 62 | if (text) 63 | { 64 | BUFPUTSL(ob, "\x1B[47;30m"); 65 | bufput(ob, text->data, text->size); 66 | BUFPUTSL(ob, "\x1B[0m"); 67 | } 68 | return 1; 69 | } 70 | 71 | static int 72 | rndr_double_emphasis(struct buf *ob, struct buf *text, char c, void *opaque) 73 | { 74 | // Double `*` or `_` is `__bold__`. 75 | #ifndef CONMD_WINDOWS 76 | return RenderMarkdown(ob, "\x1B[5;47;30m", text); 77 | #else 78 | return RenderMarkdown(ob, "\x1B[1m", text); 79 | #endif 80 | } 81 | 82 | static int 83 | rndr_emphasis(struct buf *ob, struct buf *text, char c, void *opaque) 84 | { 85 | // Single `*` or `_` is `*italic*`. 86 | #ifdef CONMD_WINDOWS 87 | return RenderMarkdown(ob, "\x1B[37;1m", text); 88 | #else 89 | return RenderMarkdown(ob, "\x1B[4m", text); 90 | #endif 91 | } 92 | 93 | void rndr_blockcode(struct buf *ob, struct buf *text, void *opaque) 94 | { 95 | // Just indent it all 4 spaces. 96 | if (text && text->size) 97 | { 98 | std::string 99 | code(text->data, text->size); 100 | 101 | std::string::size_type 102 | lastPos = 0, 103 | findPos; 104 | 105 | size_t 106 | windowWidth = reinterpret_cast<console_data_s *>(opaque)->WindowWidth; 107 | bool 108 | nl = windowWidth < reinterpret_cast<console_data_s *>(opaque)->BufferWidth; 109 | if (ob->size) 110 | { 111 | bufputc(ob, '\n'); 112 | } 113 | while (std::string::npos != (findPos = code.find("\n", lastPos))) 114 | { 115 | BUFPUTSL(ob, "\x1B[0m \x1B[47;30m"); 116 | bufput(ob, code.c_str() + lastPos, findPos - lastPos); 117 | if (findPos - lastPos < windowWidth - 4) 118 | { 119 | bufputcn(ob, ' ', windowWidth - 4 - findPos + lastPos); 120 | if (nl) 121 | { 122 | bufputc(ob, '\n'); 123 | } 124 | } 125 | else 126 | { 127 | bufputc(ob, '\n'); 128 | } 129 | lastPos = findPos + 1; 130 | } 131 | bufput(ob, code.c_str() + lastPos, code.length() - lastPos); 132 | BUFPUTSL(ob, "\x1B[0m\n"); 133 | } 134 | } 135 | 136 | void rndr_fencedcode(struct buf *ob, struct buf *text, char *name, size_t namelen, void *opaque) 137 | { 138 | if (!text || !text->size) 139 | { 140 | return; 141 | } 142 | if (namelen) 143 | { 144 | if (!strncmp(name, "cpp", 3)) 145 | { 146 | if (ob->size) 147 | { 148 | bufputc(ob, '\n'); 149 | } 150 | std::string 151 | out = cmdmd::CPP(std::string(text->data, text->size)); 152 | bufput(ob, out.c_str(), out.length()); 153 | bufputc(ob, '\n'); 154 | return; 155 | } 156 | else if (!strncmp(name, "pawn", 3)) 157 | { 158 | if (ob->size) 159 | { 160 | bufputc(ob, '\n'); 161 | } 162 | std::string 163 | out = cmdmd::Pawn(std::string(text->data, text->size)); 164 | bufput(ob, out.c_str(), out.length()); 165 | bufputc(ob, '\n'); 166 | return; 167 | } 168 | } 169 | rndr_blockcode(ob, text, opaque); 170 | } 171 | 172 | static void 173 | rndr_header(struct buf *ob, struct buf *text, int level, void *opaque) 174 | { 175 | if (text && text->size) 176 | { 177 | if (ob->size) 178 | { 179 | bufputc(ob, '\n'); 180 | } 181 | size_t 182 | width = reinterpret_cast<console_data_s *>(opaque)->WindowWidth, 183 | len = text->size; 184 | if (len > width - 2) 185 | { 186 | len = width - 2; 187 | } 188 | switch (level) 189 | { 190 | case 1: 191 | BUFPUTSL(ob, "\x1B[42;5;30m "); 192 | bufput(ob, text->data, text->size); 193 | BUFPUTSL(ob, " \x1B[0;49m\n\x1B[42;5;30m"); 194 | bufputcn(ob, '=', len + 2); 195 | BUFPUTSL(ob, "\x1B[0m\n\x1B[0m"); 196 | break; 197 | case 2: 198 | BUFPUTSL(ob, "\x1B[32;1m "); 199 | bufput(ob, text->data, text->size); 200 | BUFPUTSL(ob, "\n "); 201 | bufputcn(ob, '-', len); 202 | BUFPUTSL(ob, "\x1B[0m\n"); 203 | break; 204 | case 3: 205 | BUFPUTSL(ob, "\x1B[33;1m "); 206 | bufput(ob, text->data, text->size); 207 | BUFPUTSL(ob, "\x1B[0m\n"); 208 | break; 209 | case 4: 210 | BUFPUTSL(ob, "\x1B[31;1m "); 211 | bufput(ob, text->data, text->size); 212 | BUFPUTSL(ob, "\x1B[0m\n"); 213 | break; 214 | } 215 | } 216 | } 217 | 218 | /*static int 219 | rndr_link(struct buf *ob, struct buf *link, struct buf *title, struct buf *content, void *opaque) 220 | { 221 | BUFPUTSL(ob, "<a href=\""); 222 | if (link && link->size) 223 | { 224 | lus_attr_escape(ob, link->data, link->size); 225 | } 226 | if (title && title->size) 227 | { 228 | BUFPUTSL(ob, "\" title=\""); 229 | lus_attr_escape(ob, title->data, title->size); 230 | } 231 | BUFPUTSL(ob, "\">"); 232 | if (content && content->size) 233 | { 234 | bufput(ob, content->data, content->size); 235 | } 236 | BUFPUTSL(ob, "</a>"); 237 | return 1; 238 | }*/ 239 | 240 | static void 241 | rndr_list(struct buf *ob, struct buf *text, int flags, void *opaque) 242 | { 243 | if (text) 244 | { 245 | if (ob->size) 246 | { 247 | bufputc(ob, '\n'); 248 | } 249 | bufput(ob, text->data, text->size); 250 | reinterpret_cast<console_data_s *>(opaque)->BulletPoint = 0; 251 | } 252 | } 253 | 254 | static void 255 | rndr_listitem(struct buf *ob, struct buf *text, int flags, void *opaque) 256 | { 257 | char num[4]; 258 | if (text) 259 | { 260 | while (text->size && text->data[text->size - 1] == '\n') 261 | { 262 | text->size -= 1; 263 | } 264 | if (text->size) 265 | { 266 | if (flags & MKD_LIST_ORDERED) 267 | { 268 | BUFPUTSL(ob, " \x1B[35m"); 269 | sprintf_s(num, "%d", ++reinterpret_cast<console_data_s *>(opaque)->BulletPoint); 270 | bufput(ob, num, strlen(num)); 271 | BUFPUTSL(ob, ")\x1B[0m "); 272 | bufput(ob, text->data, text->size); 273 | bufputc(ob, '\n'); 274 | } 275 | else 276 | { 277 | BUFPUTSL(ob, " \x1B[35m*\x1B[0m "); 278 | bufput(ob, text->data, text->size); 279 | bufputc(ob, '\n'); 280 | } 281 | } 282 | } 283 | } 284 | 285 | static void 286 | rndr_normal_text(struct buf *ob, struct buf *text, void *opaque) 287 | { 288 | bufput(ob, text->data, text->size); 289 | } 290 | 291 | static void 292 | null_block(struct buf *ob, struct buf *text, void *opaque) 293 | { 294 | bufput(ob, text->data, text->size); 295 | } 296 | 297 | static int 298 | null_span(struct buf *ob, struct buf *text, void *opaque) 299 | { 300 | bufput(ob, text->data, text->size); 301 | return 1; 302 | } 303 | 304 | static void 305 | rndr_paragraph(struct buf *ob, struct buf *text, void *opaque) 306 | { 307 | if (ob->size) 308 | { 309 | bufputc(ob, '\n'); 310 | } 311 | if (text) 312 | { 313 | size_t 314 | width = reinterpret_cast<console_data_s *>(opaque)->WindowWidth, 315 | rem = width, 316 | len = text->size; 317 | bool 318 | nl = reinterpret_cast<console_data_s *>(opaque)->BufferWidth > width; 319 | char 320 | * start = text->data, 321 | * end; 322 | while ((end = (char *)memchr(start, '\n', len))) 323 | { 324 | size_t 325 | diff = end - start; 326 | len -= diff + 1; 327 | while (diff >= rem) 328 | { 329 | bufput(ob, start, rem); 330 | diff -= rem; 331 | start += rem; 332 | rem = width; 333 | if (nl) 334 | { 335 | bufputc(ob, '\n'); 336 | } 337 | } 338 | if (diff) 339 | { 340 | bufput(ob, start, diff); 341 | rem = width - diff; 342 | } 343 | start = end + 1; 344 | } 345 | if (len) 346 | { 347 | while (len >= rem) 348 | { 349 | bufput(ob, start, rem); 350 | len -= rem; 351 | start += rem; 352 | rem = width; 353 | if (nl) 354 | { 355 | bufputc(ob, '\n'); 356 | } 357 | } 358 | if (len) 359 | { 360 | bufput(ob, start, len); 361 | } 362 | } 363 | } 364 | bufputc(ob, '\n'); 365 | } 366 | 367 | /*static void 368 | rndr_raw_block(struct buf *ob, struct buf *text, void *opaque) 369 | { 370 | size_t org, sz; 371 | if (!text) 372 | { 373 | return; 374 | } 375 | sz = text->size; 376 | while (sz > 0 && text->data[sz - 1] == '\n') 377 | { 378 | sz -= 1; 379 | } 380 | org = 0; 381 | while (org < sz && text->data[org] == '\n') 382 | { 383 | org += 1; 384 | } 385 | if (org >= sz) 386 | { 387 | return; 388 | } 389 | if (ob->size) 390 | { 391 | bufputc(ob, '\n'); 392 | } 393 | bufput(ob, text->data + org, sz - org); 394 | bufputc(ob, '\n'); 395 | } 396 | 397 | static int 398 | rndr_raw_inline(struct buf *ob, struct buf *text, void *opaque) 399 | { 400 | bufput(ob, text->data, text->size); 401 | return 1; 402 | }*/ 403 | 404 | static int 405 | rndr_linebreak(struct buf *ob, void *opaque) 406 | { 407 | bufputc(ob, '\n'); 408 | return 1; 409 | } 410 | 411 | void rndr_hrule(struct buf *ob, void *opaque) 412 | { 413 | if (ob->size) 414 | { 415 | bufputc(ob, '\n'); 416 | } 417 | size_t 418 | len = reinterpret_cast<console_data_s *>(opaque)->WindowWidth; 419 | BUFPUTSL(ob, "\x1B[44;34m"); 420 | bufputcn(ob, ' ', len); 421 | BUFPUTSL(ob, "\x1B[0m\n"); 422 | } 423 | 424 | /* exported renderer structures */ 425 | std::string cmdmd::Render(std::string const & input) 426 | { 427 | return cmdmd::Render(input.c_str(), input.length()); 428 | } 429 | 430 | std::string cmdmd::Render(char const * input) 431 | { 432 | return cmdmd::Render(input, strlen(input)); 433 | } 434 | 435 | std::string cmdmd::Render(char const * input, size_t len) 436 | { 437 | static struct mkd_renderer 438 | consoleRenderer = 439 | { 440 | NULL, 441 | NULL, 442 | 443 | rndr_blockcode,//NULL, 444 | rndr_fencedcode, 445 | null_block,//NULL, //discount_blockquote, 446 | null_block,//NULL, //rndr_raw_block, 447 | rndr_header, 448 | rndr_hrule, 449 | rndr_list, 450 | rndr_listitem, 451 | rndr_paragraph, 452 | NULL, //discount_table, 453 | NULL, //discount_table_cell, 454 | NULL, //discount_table_row, 455 | 456 | NULL, 457 | rndr_codespan, 458 | rndr_double_emphasis, 459 | rndr_emphasis, 460 | NULL, //html_discount_image, 461 | rndr_linebreak, 462 | NULL, //discount_link, 463 | null_span,//NULL, //rndr_raw_inline, 464 | NULL, //rndr_triple_emphasis, 465 | 466 | NULL, 467 | NULL, 468 | 469 | 64, 470 | "*_", 471 | NULL 472 | }; 473 | buf 474 | * ob = bufnew(64), 475 | ib = { 476 | const_cast<char *>(input), 477 | len, 478 | 0, 479 | 0, 480 | 0xFFFF 481 | }; 482 | #ifdef CONMD_WINDOWS 483 | CONSOLE_SCREEN_BUFFER_INFO 484 | console; 485 | size_t 486 | windowWidth = 80, 487 | bufferWidth = 80; 488 | if (GetConsoleScreenBufferInfo(GetStdHandle(STD_OUTPUT_HANDLE), &console)) 489 | { 490 | // Use the console window size, not the line buffer size. 491 | windowWidth = console.srWindow.Right - console.srWindow.Left + 1; 492 | bufferWidth = console.dwSize.X; 493 | } 494 | console_data_s 495 | data = { windowWidth, bufferWidth, 0 }; 496 | #else 497 | struct winsize w; 498 | ioctl(STDOUT_FILENO, TIOCGWINSZ, &w); 499 | 500 | console_data_s 501 | data = { w.ws_col, w.ws_col, 0 }; 502 | #endif 503 | consoleRenderer.opaque = reinterpret_cast<void *>(&data); 504 | markdown(ob, &ib, &consoleRenderer); 505 | 506 | std::string 507 | ret(ob->data, ob->size); 508 | 509 | bufrelease(ob); 510 | 511 | return ret; 512 | } 513 | 514 | void cmdmd::Init() 515 | { 516 | ColouredBuffer<char>::StandardInstall(); 517 | ColouredBuffer<wchar_t>::StandardInstall(); 518 | } 519 | 520 | std::string 521 | cmdmd:: 522 | Literals:: 523 | operator "" _cmdmd(char const * s, size_t len) 524 | { 525 | return ::cmdmd::Render(s, len); 526 | } 527 | 528 | std::string 529 | cmdmd:: 530 | Literals:: 531 | operator "" _md(char const * s, size_t len) 532 | { 533 | return ::cmdmd::Render(s, len); 534 | } 535 | 536 | -------------------------------------------------------------------------------- /libsoldout/mkd2latex.c: -------------------------------------------------------------------------------- 1 | /* mkd2latex.c - LaTeX-formatted output from markdown text */ 2 | 3 | /* 4 | * Copyright (c) 2009, Baptiste Daroussin, Natacha Porté, and Michael Huang 5 | * 6 | * Permission to use, copy, modify, and distribute this software for any 7 | * purpose with or without fee is hereby granted, provided that the above 8 | * copyright notice and this permission notice appear in all copies. 9 | * 10 | * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES 11 | * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF 12 | * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR 13 | * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES 14 | * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN 15 | * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF 16 | * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. 17 | */ 18 | 19 | /* 20 | * Links require the hyperref package, and images require the graphicx 21 | * package. 22 | */ 23 | 24 | #include "markdown.h" 25 | 26 | #include <stdio.h> 27 | #include <stdlib.h> 28 | #include <errno.h> 29 | #include <string.h> 30 | 31 | #define READ_UNIT 1024 32 | #define OUTPUT_UNIT 64 33 | 34 | /********************* 35 | * ENTITY CONVERSION * 36 | *********************/ 37 | 38 | struct str_pair { 39 | const char *entity; 40 | const char *latex; }; 41 | 42 | static struct str_pair entity_latex[] = { 43 | { "Æ", "\\AE{}" }, 44 | { "Á", "\\'A" }, 45 | { "Â", "\\^A" }, 46 | { "À", "\\`A" }, 47 | { "Å", "\\AA{}" }, 48 | { "Ã", "\\~A" }, 49 | { "Ä", "\\\"A" }, 50 | { "Ç", "\\c{C}" }, 51 | // { "Ð", "\\DH{}" }, // not available in OT1 52 | { "É", "\\'E" }, 53 | { "Ê", "\\^E" }, 54 | { "È", "\\`E" }, 55 | { "Ë", "\\\"E" }, 56 | { "Í", "\\'I" }, 57 | { "Î", "\\^I" }, 58 | { "Ì", "\\`I" }, 59 | { "Ï", "\\\"I" }, 60 | { "Ñ", "\\~N" }, 61 | { "Ó", "\\'O" }, 62 | { "Ô", "\\^O" }, 63 | { "Ò", "\\`O" }, 64 | { "Ø", "\\O{}" }, 65 | { "Õ", "\\~O" }, 66 | { "Ö", "\\\"O" }, 67 | // { "Þ", "\\TH{}" }, // not available in OT1 68 | { "Ú", "\\'U" }, 69 | { "Û", "\\^U" }, 70 | { "Ù", "\\`U" }, 71 | { "Ü", "\\\"U" }, 72 | { "&Ygrave;", "\\`Y" }, 73 | { "á", "\\'a" }, 74 | { "â", "\\^a" }, 75 | // { "´", "\\textasciiacute{}" }, // requires textcomp 76 | { "æ", "\\ae{}" }, 77 | { "à", "\\`a" }, 78 | { "&", "\\&" }, 79 | { "'", "'" }, 80 | { "å", "\\aa{}" }, 81 | { "ã", "\\~a" }, 82 | { "ä", "\\\"a" }, 83 | // { "¦", "\\textbrokenbar{}" }, // requires textcomp 84 | { "ç", "\\c{c}" }, 85 | { "¸", "\\c{}" }, 86 | // { "¢", "\\textcent{}" }, // requires textcomp 87 | { "©", "\\copyright{}" }, 88 | // { "¤", "\\textcurrency{}" }, // requires textcomp 89 | // { "°", "\\textdegree{}" }, // requires textcomp 90 | // { "÷", "\\textdiv{}" }, // requires textcomp 91 | { "é", "\\'e" }, 92 | { "ê", "\\^e" }, 93 | { "è", "\\`e" }, 94 | // { "ð", "\\dh{}" }. // not available in OT1 95 | { "ë", "\\\"e" }, 96 | // { "½", "\\textonehalf{}" }, // requires textcomp 97 | // { "¼", "\\textonequarter{}" }, // requires textcomp 98 | // { "¾", "\\textthreequarter{}" }, // requires textcomp 99 | { ">", "$>$" }, 100 | { "í", "\\'\\i{}" }, 101 | { "î", "\\^\\i{}" }, 102 | { "¡", "\\textexclamdown{}" }, 103 | { "ì", "\\`\\i{}" }, 104 | { "¿", "\\" }, 105 | { "ï", "\\\"\\i{}" }, 106 | // { "«", "\\guillemotleft{}" }, // not available in OT1 107 | { "<", "$<$" }, 108 | // { "¯", "\\textasciimacaron{}" }, // requires textcomp 109 | // { "µ", "\\textmu{}"}, // requires textcomp 110 | { "·", "\\textperiodcentered{}" }, 111 | { " ", "~" }, 112 | // { "¬", "\\textlnot{}" }, // requires textcomp 113 | { "ñ", "\\~n" }, 114 | { "ó", "\\'o" }, 115 | { "ô", "\\^o" }, 116 | { "ò", "\\`o" }, 117 | { "ª", "\\textordfeminine{}" }, 118 | { "º", "\\textordmasculine{}" }, 119 | { "ø", "\\o{}" }, 120 | { "õ", "\\~o" }, 121 | { "ö", "\\\"o" }, 122 | { "¶", "\\P{}" }, 123 | // { "±", "\\textpm{}" }, // requires textcomp 124 | { "£", "\\textsterling{}" }, 125 | { """, "\"" }, 126 | // { "»", "\\guillemotright{}" }, // not available in OT1 127 | { "®", "\\textregistered{}" }, 128 | { "§", "\\S{}" }, 129 | { "", "\\-" }, 130 | // { "¹", "\\textonesuperior{}" }, // requires textcomp 131 | // { "²", "\\texttwosuperior{}" }, // requires textcomp 132 | // { "³", "\\textthreesuperior{}" }, // requires textcomp 133 | { "ß", "\\ss{}" }, 134 | // { "þ", "\\th{}" }, // not available in OT1 135 | // { "×", "\\texttimes{}" }, // requires textcomp 136 | { "ú", "\\'u" }, 137 | { "û", "\\^u" }, 138 | { "ù", "\\`u" }, 139 | // { "¨", "\\textasciidieresis{}" }, // requires textcomp 140 | { "ü", "\\\"u" }, 141 | { "ý", "\\'y" }, 142 | // { "¥", "\\textyen{}" }, // requires textcomp 143 | { "ÿ", "\\\"y" }, 144 | }; 145 | 146 | static int cmp_entity(const void *key, const void *element) { 147 | const struct str_pair *pair = element; 148 | const struct buf *entity = key; 149 | return bufcmps(entity, pair->entity); } 150 | 151 | static const char *entity2latex(const struct buf *entity) { 152 | const struct str_pair *pair; 153 | pair = bsearch(entity, entity_latex, 154 | sizeof entity_latex / sizeof *entity_latex, 155 | sizeof *entity_latex, 156 | &cmp_entity); 157 | return pair ? pair->latex : 0; } 158 | 159 | 160 | 161 | /****************************** 162 | * MARKDOWN TO LATEX RENDERER * 163 | ******************************/ 164 | 165 | static void 166 | latex_text_escape(struct buf *ob, char *src, size_t size) { 167 | size_t i = 0, org; 168 | while (i < size) { 169 | /* copying directly unescaped characters */ 170 | org = i; 171 | while (i < size && src[i] != '&' && src[i] != '%' 172 | && src[i] != '$' && src[i] != '#' && src[i] != '_' 173 | && src[i] != '{' && src[i] != '}' && src[i] != '~' 174 | && src[i] != '^' && src[i] != '\\' && src[i] != '<' 175 | && src[i] != '>') 176 | i += 1; 177 | if (i > org) bufput(ob, src + org, i - org); 178 | 179 | /* escaping */ 180 | if (i >= size) break; 181 | else if (src[i] == '&') BUFPUTSL(ob, "\\&"); 182 | else if (src[i] == '%') BUFPUTSL(ob, "\\%"); 183 | else if (src[i] == '$') BUFPUTSL(ob, "\\$"); 184 | else if (src[i] == '#') BUFPUTSL(ob, "\\#"); 185 | else if (src[i] == '_') BUFPUTSL(ob, "\\_"); 186 | else if (src[i] == '{') BUFPUTSL(ob, "\\{"); 187 | else if (src[i] == '}') BUFPUTSL(ob, "\\}"); 188 | else if (src[i] == '<') BUFPUTSL(ob, "$<$"); 189 | else if (src[i] == '>') BUFPUTSL(ob, "$<$"); 190 | else if (src[i] == '~') BUFPUTSL(ob, "\\textasciitilde{}"); 191 | else if (src[i] == '^') BUFPUTSL(ob, "\\textasciicircum{}"); 192 | else if (src[i] == '\\') BUFPUTSL(ob, "\\textbackslash{}"); 193 | i += 1; } } 194 | 195 | static void 196 | latex_prolog(struct buf *ob, void *opaque) { 197 | BUFPUTSL(ob, 198 | "\\documentclass{article}\n" 199 | "\\usepackage{hyperref}\n" 200 | "\\usepackage{graphicx}\n" 201 | "\\begin{document}\n"); } 202 | 203 | static void 204 | latex_epilog(struct buf *ob, void *opaque) { 205 | BUFPUTSL(ob, "\n\\end{document}\n"); } 206 | 207 | static int 208 | latex_autolink(struct buf *ob, struct buf *link, enum mkd_autolink type, 209 | void *opaque) { 210 | if (!link || !link->size) return 0; 211 | BUFPUTSL(ob, "\\href{"); 212 | if (type == MKDA_IMPLICIT_EMAIL) BUFPUTSL(ob, "mailto:"); 213 | bufput(ob, link->data, link->size); 214 | BUFPUTSL(ob, "}{"); 215 | if (type == MKDA_EXPLICIT_EMAIL && link->size > 7) 216 | latex_text_escape(ob, link->data + 7, link->size - 7); 217 | else latex_text_escape(ob, link->data, link->size); 218 | BUFPUTSL(ob, "}"); 219 | return 1; } 220 | 221 | static int 222 | latex_link(struct buf *ob, struct buf *link, struct buf *title, 223 | struct buf *content, void *opaque) { 224 | BUFPUTSL(ob, "\\href{"); 225 | if (link && link->size) bufput(ob, link->data, link->size); 226 | BUFPUTSL(ob, "}{"); 227 | if (content && content->size) 228 | bufput(ob, content->data, content->size); 229 | BUFPUTSL(ob, "}"); 230 | return 1; } 231 | 232 | static int 233 | latex_image(struct buf *ob, struct buf *link, struct buf *title, 234 | struct buf *alt, void *opaque) { 235 | if (!link || !link->size) return 0; 236 | BUFPUTSL(ob, "\\includegraphics{"); 237 | bufput(ob, link->data, link->size); 238 | BUFPUTSL(ob, "}"); 239 | return 1; } 240 | 241 | static void 242 | latex_blockcode(struct buf *ob, struct buf *text, void *opaque) { 243 | if (ob->size) bufputc(ob, '\n'); 244 | BUFPUTSL(ob, "\\begin{verbatim}\n"); 245 | if (text) bufput(ob, text->data, text->size); 246 | BUFPUTSL(ob, "\\end{verbatim}\n"); } 247 | 248 | static void 249 | latex_blockquote(struct buf *ob, struct buf *text, void *opaque) { 250 | if (ob->size) bufputc(ob, '\n'); 251 | BUFPUTSL(ob, "\\begin{quote}\n"); 252 | if (text) bufput(ob, text->data, text->size); 253 | BUFPUTSL(ob, "\\end{quote}\n"); } 254 | 255 | static int 256 | latex_codespan(struct buf *ob, struct buf *text, void *opaque) { 257 | BUFPUTSL(ob, "\\texttt{"); 258 | if (text) latex_text_escape(ob, text->data, text->size); 259 | BUFPUTSL(ob, "}"); 260 | return 1; } 261 | 262 | static void 263 | latex_header(struct buf *ob, struct buf *text, int level, void *opaque) { 264 | if (ob->size) bufputc(ob, '\n'); 265 | switch(level) { 266 | case 1: 267 | BUFPUTSL(ob,"\\section{"); 268 | break; 269 | case 2: 270 | BUFPUTSL(ob, "\\subsection{"); 271 | break; 272 | case 3: 273 | BUFPUTSL(ob, "\\subsubsection{"); 274 | break; 275 | default: 276 | fprintf(stderr, "Warning: ignoring header level %d\n", 277 | level); 278 | } 279 | if (text) bufput(ob, text->data, text->size); 280 | if (level >= 1 && level <= 3) BUFPUTSL(ob, "}\n"); 281 | } 282 | 283 | static int 284 | latex_double_emphasis(struct buf *ob, struct buf *text, char c, void *opaque) { 285 | if (!text || !text->size) return 0; 286 | BUFPUTSL(ob, "\\textbf{"); 287 | bufput(ob, text->data, text->size); 288 | BUFPUTSL(ob, "}"); 289 | return 1; } 290 | 291 | static int 292 | latex_emphasis(struct buf *ob, struct buf *text, char c, void *opaque) { 293 | if (!text || !text->size) return 0; 294 | BUFPUTSL(ob, "\\emph{"); 295 | if (text) bufput(ob, text->data, text->size); 296 | BUFPUTSL(ob, "}"); 297 | return 1; } 298 | 299 | static int 300 | latex_linebreak(struct buf *ob, void *opaque) { 301 | BUFPUTSL(ob, "\\\\"); 302 | return 1; } 303 | 304 | static void 305 | latex_paragraph(struct buf *ob, struct buf *text, void *opaque) { 306 | if (ob->size) bufputc(ob, '\n'); 307 | if (text) bufput(ob, text->data, text->size); 308 | BUFPUTSL(ob, "\n"); } 309 | 310 | static void 311 | latex_list(struct buf *ob, struct buf *text, int flags, void *opaque) { 312 | if (ob->size) bufputc(ob, '\n'); 313 | if (flags & MKD_LIST_ORDERED) 314 | BUFPUTSL(ob, "\\begin{enumerate}\n"); 315 | else 316 | BUFPUTSL(ob, "\\begin{itemize}\n"); 317 | if (text) bufput(ob, text->data, text->size); 318 | if (flags & MKD_LIST_ORDERED) 319 | BUFPUTSL(ob, "\\end{enumerate}\n"); 320 | else 321 | BUFPUTSL(ob, "\\end{itemize}\n"); } 322 | 323 | static void 324 | latex_listitem(struct buf *ob, struct buf *text, int flags, void *opaque) { 325 | BUFPUTSL(ob, "\\item "); 326 | if (text) { 327 | while (text->size && text->data[text->size - 1] == '\n') 328 | text->size -= 1; 329 | bufput(ob, text->data, text->size); } 330 | BUFPUTSL(ob, "\n"); } 331 | 332 | static void 333 | latex_hrule(struct buf *ob, void *opaque) { 334 | if (ob->size) bufputc(ob, '\n'); 335 | BUFPUTSL(ob, "\\hrule"); } 336 | 337 | static int 338 | latex_triple_emphasis(struct buf *ob, struct buf *text, char c, void *opaque) { 339 | if (!text || !text->size) return 0; 340 | BUFPUTSL(ob, "\\textbf{\\emph{"); 341 | bufput(ob, text->data, text->size); 342 | BUFPUTSL(ob, "}}"); 343 | return 1; } 344 | 345 | static void 346 | latex_entity(struct buf *ob, struct buf *entity, void *opaque) { 347 | const char *rendered = entity2latex(entity); 348 | if (rendered) 349 | bufputs(ob, rendered); 350 | else { 351 | BUFPUTSL(ob, "\\texttt{"); 352 | bufput(ob, entity->data, entity->size); 353 | BUFPUTSL(ob, "}"); } } 354 | 355 | static void 356 | latex_normal_text(struct buf *ob, struct buf *text, void *opaque) { 357 | if (text) latex_text_escape(ob, text->data, text->size); } 358 | 359 | 360 | /* renderer structure */ 361 | static struct mkd_renderer to_latex = { 362 | /* document-level callbacks */ 363 | latex_prolog, 364 | latex_epilog, 365 | 366 | /* block-level callbacks */ 367 | latex_blockcode, 368 | NULL, 369 | latex_blockquote, 370 | latex_blockcode, 371 | latex_header, 372 | latex_hrule, 373 | latex_list, 374 | latex_listitem, 375 | latex_paragraph, 376 | NULL, 377 | NULL, 378 | NULL, 379 | 380 | /* span-level callbacks */ 381 | latex_autolink, 382 | latex_codespan, 383 | latex_double_emphasis, 384 | latex_emphasis, 385 | latex_image, 386 | latex_linebreak, 387 | latex_link, 388 | latex_codespan, 389 | latex_triple_emphasis, 390 | 391 | /* low-level callbacks */ 392 | latex_entity, 393 | latex_normal_text, 394 | 395 | /* renderer data */ 396 | 64, 397 | "*_", 398 | NULL }; 399 | 400 | 401 | 402 | /***************** 403 | * MAIN FUNCTION * 404 | *****************/ 405 | 406 | /* main • main function, interfacing STDIO with the parser */ 407 | int 408 | main(int argc, char **argv) { 409 | struct buf *ib, *ob; 410 | size_t ret; 411 | FILE *in = stdin; 412 | 413 | /* opening the file if given from the command line */ 414 | if (argc > 1) { 415 | in = fopen(argv[1], "r"); 416 | if (!in) { 417 | fprintf(stderr,"Unable to open input file \"%s\": %s\n", 418 | argv[1], strerror(errno)); 419 | return 1; } } 420 | 421 | /* reading everything */ 422 | ib = bufnew(READ_UNIT); 423 | bufgrow(ib, READ_UNIT); 424 | while ((ret = fread(ib->data + ib->size, 1, 425 | ib->asize - ib->size, in)) > 0) { 426 | ib->size += ret; 427 | bufgrow(ib, ib->size + READ_UNIT); } 428 | if (in != stdin) fclose(in); 429 | 430 | /* performing markdown to LaTeX */ 431 | ob = bufnew(OUTPUT_UNIT); 432 | markdown(ob, ib, &to_latex); 433 | 434 | /* writing the result to stdout */ 435 | ret = fwrite(ob->data, 1, ob->size, stdout); 436 | if (ret < ob->size) 437 | fprintf(stderr, "Warning: only %zu output byte written, " 438 | "out of %zu\n", 439 | ret, 440 | ob->size); 441 | 442 | /* cleanup */ 443 | bufrelease(ib); 444 | bufrelease(ob); 445 | return 0; } 446 | 447 | /* vim: set filetype=c: */ 448 | -------------------------------------------------------------------------------- /libsoldout/renderers.c: -------------------------------------------------------------------------------- 1 | /* renderers.h - example markdown renderers */ 2 | 3 | /* 4 | * Copyright (c) 2009, Natacha Porté 5 | * 6 | * Permission to use, copy, modify, and distribute this software for any 7 | * purpose with or without fee is hereby granted, provided that the above 8 | * copyright notice and this permission notice appear in all copies. 9 | * 10 | * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES 11 | * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF 12 | * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR 13 | * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES 14 | * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN 15 | * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF 16 | * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. 17 | */ 18 | 19 | #include "renderers.h" 20 | 21 | #include <strings.h> 22 | 23 | 24 | /***************************** 25 | * EXPORTED HELPER FUNCTIONS * 26 | *****************************/ 27 | 28 | /* lus_attr_escape • copy the buffer entity-escaping '<', '>', '&' and '"' */ 29 | void 30 | lus_attr_escape(struct buf *ob, const char *src, size_t size) { 31 | size_t i = 0, org; 32 | while (i < size) { 33 | /* copying directly unescaped characters */ 34 | org = i; 35 | while (i < size && src[i] != '<' && src[i] != '>' 36 | && src[i] != '&' && src[i] != '"') 37 | i += 1; 38 | if (i > org) bufput(ob, src + org, i - org); 39 | 40 | /* escaping */ 41 | if (i >= size) break; 42 | else if (src[i] == '<') BUFPUTSL(ob, "<"); 43 | else if (src[i] == '>') BUFPUTSL(ob, ">"); 44 | else if (src[i] == '&') BUFPUTSL(ob, "&"); 45 | else if (src[i] == '"') BUFPUTSL(ob, """); 46 | i += 1; } } 47 | 48 | 49 | /* lus_body_escape • copy the buffer entity-escaping '<', '>' and '&' */ 50 | void 51 | lus_body_escape(struct buf *ob, const char *src, size_t size) { 52 | size_t i = 0, org; 53 | while (i < size) { 54 | /* copying directly unescaped characters */ 55 | org = i; 56 | while (i < size && src[i] != '<' && src[i] != '>' 57 | && src[i] != '&') 58 | i += 1; 59 | if (i > org) bufput(ob, src + org, i - org); 60 | 61 | /* escaping */ 62 | if (i >= size) break; 63 | else if (src[i] == '<') BUFPUTSL(ob, "<"); 64 | else if (src[i] == '>') BUFPUTSL(ob, ">"); 65 | else if (src[i] == '&') BUFPUTSL(ob, "&"); 66 | i += 1; } } 67 | 68 | 69 | 70 | /******************** 71 | * GENERIC RENDERER * 72 | ********************/ 73 | 74 | static int 75 | rndr_autolink(struct buf *ob, struct buf *link, enum mkd_autolink type, 76 | void *opaque) { 77 | if (!link || !link->size) return 0; 78 | BUFPUTSL(ob, "<a href=\""); 79 | if (type == MKDA_IMPLICIT_EMAIL) BUFPUTSL(ob, "mailto:"); 80 | lus_attr_escape(ob, link->data, link->size); 81 | BUFPUTSL(ob, "\">"); 82 | if (type == MKDA_EXPLICIT_EMAIL && link->size > 7) 83 | lus_body_escape(ob, link->data + 7, link->size - 7); 84 | else lus_body_escape(ob, link->data, link->size); 85 | BUFPUTSL(ob, "</a>"); 86 | return 1; } 87 | 88 | static void 89 | rndr_blockcode(struct buf *ob, struct buf *text, void *opaque) { 90 | if (ob->size) bufputc(ob, '\n'); 91 | BUFPUTSL(ob, "<pre><code>"); 92 | if (text) lus_body_escape(ob, text->data, text->size); 93 | BUFPUTSL(ob, "</code></pre>\n"); } 94 | 95 | static void 96 | rndr_blockquote(struct buf *ob, struct buf *text, void *opaque) { 97 | if (ob->size) bufputc(ob, '\n'); 98 | BUFPUTSL(ob, "<blockquote>\n"); 99 | if (text) bufput(ob, text->data, text->size); 100 | BUFPUTSL(ob, "</blockquote>\n"); } 101 | 102 | static int 103 | rndr_codespan(struct buf *ob, struct buf *text, void *opaque) { 104 | BUFPUTSL(ob, "<code>"); 105 | if (text) lus_body_escape(ob, text->data, text->size); 106 | BUFPUTSL(ob, "</code>"); 107 | return 1; } 108 | 109 | static int 110 | rndr_double_emphasis(struct buf *ob, struct buf *text, char c, void *opaque) { 111 | if (!text || !text->size) return 0; 112 | BUFPUTSL(ob, "<strong>"); 113 | bufput(ob, text->data, text->size); 114 | BUFPUTSL(ob, "</strong>"); 115 | return 1; } 116 | 117 | static int 118 | rndr_emphasis(struct buf *ob, struct buf *text, char c, void *opaque) { 119 | if (!text || !text->size) return 0; 120 | BUFPUTSL(ob, "<em>"); 121 | if (text) bufput(ob, text->data, text->size); 122 | BUFPUTSL(ob, "</em>"); 123 | return 1; } 124 | 125 | static void 126 | rndr_header(struct buf *ob, struct buf *text, int level, void *opaque) { 127 | if (ob->size) bufputc(ob, '\n'); 128 | bufprintf(ob, "<h%d>", level); 129 | if (text) bufput(ob, text->data, text->size); 130 | bufprintf(ob, "</h%d>\n", level); } 131 | 132 | static int 133 | rndr_link(struct buf *ob, struct buf *link, struct buf *title, 134 | struct buf *content, void *opaque) { 135 | BUFPUTSL(ob, "<a href=\""); 136 | if (link && link->size) lus_attr_escape(ob, link->data, link->size); 137 | if (title && title->size) { 138 | BUFPUTSL(ob, "\" title=\""); 139 | lus_attr_escape(ob, title->data, title->size); } 140 | BUFPUTSL(ob, "\">"); 141 | if (content && content->size) bufput(ob, content->data, content->size); 142 | BUFPUTSL(ob, "</a>"); 143 | return 1; } 144 | 145 | static void 146 | rndr_list(struct buf *ob, struct buf *text, int flags, void *opaque) { 147 | if (ob->size) bufputc(ob, '\n'); 148 | bufput(ob, (flags & MKD_LIST_ORDERED) ? "<ol>\n" : "<ul>\n", 5); 149 | if (text) bufput(ob, text->data, text->size); 150 | bufput(ob, (flags & MKD_LIST_ORDERED) ? "</ol>\n" : "</ul>\n", 6); } 151 | 152 | static void 153 | rndr_listitem(struct buf *ob, struct buf *text, int flags, void *opaque) { 154 | BUFPUTSL(ob, "<li>"); 155 | if (text) { 156 | while (text->size && text->data[text->size - 1] == '\n') 157 | text->size -= 1; 158 | bufput(ob, text->data, text->size); } 159 | BUFPUTSL(ob, "</li>\n"); } 160 | 161 | static void 162 | rndr_normal_text(struct buf *ob, struct buf *text, void *opaque) { 163 | if (text) lus_body_escape(ob, text->data, text->size); } 164 | 165 | static void 166 | rndr_paragraph(struct buf *ob, struct buf *text, void *opaque) { 167 | if (ob->size) bufputc(ob, '\n'); 168 | BUFPUTSL(ob, "<p>"); 169 | if (text) bufput(ob, text->data, text->size); 170 | BUFPUTSL(ob, "</p>\n"); } 171 | 172 | static void 173 | rndr_raw_block(struct buf *ob, struct buf *text, void *opaque) { 174 | size_t org, sz; 175 | if (!text) return; 176 | sz = text->size; 177 | while (sz > 0 && text->data[sz - 1] == '\n') sz -= 1; 178 | org = 0; 179 | while (org < sz && text->data[org] == '\n') org += 1; 180 | if (org >= sz) return; 181 | if (ob->size) bufputc(ob, '\n'); 182 | bufput(ob, text->data + org, sz - org); 183 | bufputc(ob, '\n'); } 184 | 185 | static int 186 | rndr_raw_inline(struct buf *ob, struct buf *text, void *opaque) { 187 | bufput(ob, text->data, text->size); 188 | return 1; } 189 | 190 | static int 191 | rndr_triple_emphasis(struct buf *ob, struct buf *text, char c, void *opaque) { 192 | if (!text || !text->size) return 0; 193 | BUFPUTSL(ob, "<strong><em>"); 194 | bufput(ob, text->data, text->size); 195 | BUFPUTSL(ob, "</em></strong>"); 196 | return 1; } 197 | 198 | 199 | 200 | /******************* 201 | * HTML 4 RENDERER * 202 | *******************/ 203 | 204 | static void 205 | html_hrule(struct buf *ob, void *opaque) { 206 | if (ob->size) bufputc(ob, '\n'); 207 | BUFPUTSL(ob, "<hr>\n"); } 208 | 209 | static int 210 | html_image(struct buf *ob, struct buf *link, struct buf *title, 211 | struct buf *alt, void *opaque) { 212 | if (!link || !link->size) return 0; 213 | BUFPUTSL(ob, "<img src=\""); 214 | lus_attr_escape(ob, link->data, link->size); 215 | BUFPUTSL(ob, "\" alt=\""); 216 | if (alt && alt->size) 217 | lus_attr_escape(ob, alt->data, alt->size); 218 | if (title && title->size) { 219 | BUFPUTSL(ob, "\" title=\""); 220 | lus_attr_escape(ob, title->data, title->size); } 221 | BUFPUTSL(ob, "\">"); 222 | return 1; } 223 | 224 | static int 225 | html_linebreak(struct buf *ob, void *opaque) { 226 | BUFPUTSL(ob, "<br>\n"); 227 | return 1; } 228 | 229 | 230 | /* exported renderer structure */ 231 | const struct mkd_renderer mkd_html = { 232 | NULL, 233 | NULL, 234 | 235 | rndr_blockcode, 236 | NULL, 237 | rndr_blockquote, 238 | rndr_raw_block, 239 | rndr_header, 240 | html_hrule, 241 | rndr_list, 242 | rndr_listitem, 243 | rndr_paragraph, 244 | NULL, 245 | NULL, 246 | NULL, 247 | 248 | rndr_autolink, 249 | rndr_codespan, 250 | rndr_double_emphasis, 251 | rndr_emphasis, 252 | html_image, 253 | html_linebreak, 254 | rndr_link, 255 | rndr_raw_inline, 256 | rndr_triple_emphasis, 257 | 258 | NULL, 259 | rndr_normal_text, 260 | 261 | 64, 262 | "*_", 263 | NULL }; 264 | 265 | 266 | 267 | /********************** 268 | * XHTML 1.0 RENDERER * 269 | **********************/ 270 | 271 | static void 272 | xhtml_hrule(struct buf *ob, void *opaque) { 273 | if (ob->size) bufputc(ob, '\n'); 274 | BUFPUTSL(ob, "<hr />\n"); } 275 | 276 | static int 277 | xhtml_image(struct buf *ob, struct buf *link, struct buf *title, 278 | struct buf *alt, void *opaque) { 279 | if (!link || !link->size) return 0; 280 | BUFPUTSL(ob, "<img src=\""); 281 | lus_attr_escape(ob, link->data, link->size); 282 | BUFPUTSL(ob, "\" alt=\""); 283 | if (alt && alt->size) 284 | lus_attr_escape(ob, alt->data, alt->size); 285 | if (title && title->size) { 286 | BUFPUTSL(ob, "\" title=\""); 287 | lus_attr_escape(ob, title->data, title->size); } 288 | BUFPUTSL(ob, "\" />"); 289 | return 1; } 290 | 291 | static int 292 | xhtml_linebreak(struct buf *ob, void *opaque) { 293 | BUFPUTSL(ob, "<br />\n"); 294 | return 1; } 295 | 296 | 297 | /* exported renderer structure */ 298 | const struct mkd_renderer mkd_xhtml = { 299 | NULL, 300 | NULL, 301 | 302 | rndr_blockcode, 303 | NULL, 304 | rndr_blockquote, 305 | rndr_raw_block, 306 | rndr_header, 307 | xhtml_hrule, 308 | rndr_list, 309 | rndr_listitem, 310 | rndr_paragraph, 311 | NULL, 312 | NULL, 313 | NULL, 314 | 315 | rndr_autolink, 316 | rndr_codespan, 317 | rndr_double_emphasis, 318 | rndr_emphasis, 319 | xhtml_image, 320 | xhtml_linebreak, 321 | rndr_link, 322 | rndr_raw_inline, 323 | rndr_triple_emphasis, 324 | 325 | NULL, 326 | rndr_normal_text, 327 | 328 | 64, 329 | "*_", 330 | NULL }; 331 | 332 | 333 | 334 | /********************** 335 | * DISCOUNT RENDERERS * 336 | **********************/ 337 | 338 | static int 339 | print_link_wxh(struct buf *ob, struct buf *link) { 340 | size_t eq, ex, end; 341 | if (link->size < 1) return 0; 342 | eq = link->size - 1; 343 | while (eq > 0 && (link->data[eq - 1] != ' ' || link->data[eq] != '=')) 344 | eq -= 1; 345 | if (!eq) return 0; 346 | ex = eq + 1; 347 | while (ex < link->size 348 | && link->data[ex] >= '0' && link->data[ex] <= '9') 349 | ex += 1; 350 | if (ex >= link->size || ex == eq + 1 || link->data[ex] != 'x') return 0; 351 | end = ex + 1; 352 | while (end < link->size 353 | && link->data[end] >= '0' && link->data[end] <= '9') 354 | end += 1; 355 | if (end == ex + 1) return 0; 356 | /* everything is fine, proceeding to actual printing */ 357 | lus_attr_escape(ob, link->data, eq - 1); 358 | BUFPUTSL(ob, "\" width="); 359 | bufput(ob, link->data + eq + 1, ex - eq - 1); 360 | BUFPUTSL(ob, " height="); 361 | bufput(ob, link->data + ex + 1, end - ex - 1); 362 | return 1; } 363 | 364 | static int 365 | discount_image(struct buf *ob, struct buf *link, struct buf *title, 366 | struct buf *alt, int xhtml) { 367 | if (!link || !link->size) return 0; 368 | BUFPUTSL(ob, "<img src=\""); 369 | if (!print_link_wxh(ob, link)) { 370 | lus_attr_escape(ob, link->data, link->size); 371 | bufputc(ob, '"'); } 372 | BUFPUTSL(ob, " alt=\""); 373 | if (alt && alt->size) 374 | lus_attr_escape(ob, alt->data, alt->size); 375 | if (title && title->size) { 376 | BUFPUTSL(ob, "\" title=\""); 377 | lus_attr_escape(ob, title->data, title->size); } 378 | bufputs(ob, xhtml ? "\" />" : "\">"); 379 | return 1; } 380 | 381 | static int 382 | html_discount_image(struct buf *ob, struct buf *link, struct buf *title, 383 | struct buf *alt, void *opaque) { 384 | return discount_image(ob, link, title, alt, 0); } 385 | 386 | static int 387 | xhtml_discount_image(struct buf *ob, struct buf *link, struct buf *title, 388 | struct buf *alt, void *opaque) { 389 | return discount_image(ob, link, title, alt, 1); } 390 | 391 | static int 392 | discount_link(struct buf *ob, struct buf *link, struct buf *title, 393 | struct buf *content, void *opaque) { 394 | if (!link) return rndr_link(ob, link, title, content, opaque); 395 | else if (link->size > 5 && !strncasecmp(link->data, "abbr:", 5)) { 396 | BUFPUTSL(ob, "<abbr title=\""); 397 | lus_attr_escape(ob, link->data + 5, link->size - 5); 398 | BUFPUTSL(ob, "\">"); 399 | bufput(ob, content->data, content->size); 400 | BUFPUTSL(ob, "</abbr>"); 401 | return 1; } 402 | else if (link->size > 6 && !strncasecmp(link->data, "class:", 6)) { 403 | BUFPUTSL(ob, "<span class=\""); 404 | lus_attr_escape(ob, link->data + 6, link->size - 6); 405 | BUFPUTSL(ob, "\">"); 406 | bufput(ob, content->data, content->size); 407 | BUFPUTSL(ob, "</span>"); 408 | return 1; } 409 | else if (link->size > 3 && !strncasecmp(link->data, "id:", 3)) { 410 | BUFPUTSL(ob, "<span id=\""); 411 | lus_attr_escape(ob, link->data + 3, link->size - 3); 412 | BUFPUTSL(ob, "\">"); 413 | bufput(ob, content->data, content->size); 414 | BUFPUTSL(ob, "</span>"); 415 | return 1; } 416 | else if (link->size > 4 && !strncasecmp(link->data, "raw:", 4)) { 417 | bufput(ob, link->data + 4, link->size - 4); 418 | return 1; } 419 | return rndr_link(ob, link, title, content, opaque); } 420 | 421 | static void 422 | discount_blockquote(struct buf *ob, struct buf *text, void *opaque) { 423 | size_t i = 5, size = text->size; 424 | char *data = text->data; 425 | if (text->size < 5 || strncasecmp(text->data, "<p>%", 4)) { 426 | rndr_blockquote(ob, text, opaque); 427 | return; } 428 | while (i < size && data[i] != '\n' && data[i] != '%') 429 | i += 1; 430 | if (i >= size || data[i] != '%') { 431 | rndr_blockquote(ob, text, opaque); 432 | return; } 433 | BUFPUTSL(ob, "<div class=\""); 434 | bufput(ob, text->data + 4, i - 4); 435 | BUFPUTSL(ob, "\"><p>"); 436 | i += 1; 437 | if (i + 4 >= text->size && !strncasecmp(text->data + i, "</p>", 4)) { 438 | size_t old_i = i; 439 | i += 4; 440 | while (i + 3 < text->size 441 | && (data[i] != '<' || data[i + 1] != 'p' || data[i + 2] != '>')) 442 | i += 1; 443 | if (i + 3 >= text->size) i = old_i; } 444 | bufput(ob, text->data + i, text->size - i); 445 | BUFPUTSL(ob, "</div>\n"); } 446 | 447 | static void 448 | discount_table(struct buf *ob, struct buf *head_row, struct buf *rows, 449 | void *opaque) { 450 | if (ob->size) bufputc(ob, '\n'); 451 | BUFPUTSL(ob, "<table>\n"); 452 | if (head_row) { 453 | BUFPUTSL(ob, "<thead>\n"); 454 | bufput(ob, head_row->data, head_row->size); 455 | BUFPUTSL(ob, "</thead>\n<tbody>\n"); } 456 | if (rows) 457 | bufput(ob, rows->data, rows->size); 458 | if (head_row) 459 | BUFPUTSL(ob, "</tbody>\n"); 460 | BUFPUTSL(ob, "</table>\n"); } 461 | 462 | static void 463 | discount_table_row(struct buf *ob, struct buf *cells, int flags, void *opaque){ 464 | (void)flags; 465 | BUFPUTSL(ob, " <tr>\n"); 466 | if (cells) bufput(ob, cells->data, cells->size); 467 | BUFPUTSL(ob, " </tr>\n"); } 468 | 469 | static void 470 | discount_table_cell(struct buf *ob, struct buf *text, int flags, void *opaque){ 471 | if (flags & MKD_CELL_HEAD) 472 | BUFPUTSL(ob, " <th"); 473 | else 474 | BUFPUTSL(ob, " <td"); 475 | switch (flags & MKD_CELL_ALIGN_MASK) { 476 | case MKD_CELL_ALIGN_LEFT: 477 | BUFPUTSL(ob, " align=\"left\""); 478 | break; 479 | case MKD_CELL_ALIGN_RIGHT: 480 | BUFPUTSL(ob, " align=\"right\""); 481 | break; 482 | case MKD_CELL_ALIGN_CENTER: 483 | BUFPUTSL(ob, " align=\"center\""); 484 | break; } 485 | bufputc(ob, '>'); 486 | if (text) bufput(ob, text->data, text->size); 487 | if (flags & MKD_CELL_HEAD) 488 | BUFPUTSL(ob, "</th>\n"); 489 | else 490 | BUFPUTSL(ob, "</td>\n"); } 491 | 492 | /* exported renderer structures */ 493 | const struct mkd_renderer discount_html = { 494 | NULL, 495 | NULL, 496 | 497 | rndr_blockcode, 498 | NULL, 499 | discount_blockquote, 500 | rndr_raw_block, 501 | rndr_header, 502 | html_hrule, 503 | rndr_list, 504 | rndr_listitem, 505 | rndr_paragraph, 506 | discount_table, 507 | discount_table_cell, 508 | discount_table_row, 509 | 510 | rndr_autolink, 511 | rndr_codespan, 512 | rndr_double_emphasis, 513 | rndr_emphasis, 514 | html_discount_image, 515 | html_linebreak, 516 | discount_link, 517 | rndr_raw_inline, 518 | rndr_triple_emphasis, 519 | 520 | NULL, 521 | rndr_normal_text, 522 | 523 | 64, 524 | "*_", 525 | NULL }; 526 | const struct mkd_renderer discount_xhtml = { 527 | NULL, 528 | NULL, 529 | 530 | rndr_blockcode, 531 | NULL, 532 | discount_blockquote, 533 | rndr_raw_block, 534 | rndr_header, 535 | xhtml_hrule, 536 | rndr_list, 537 | rndr_listitem, 538 | rndr_paragraph, 539 | discount_table, 540 | discount_table_cell, 541 | discount_table_row, 542 | 543 | rndr_autolink, 544 | rndr_codespan, 545 | rndr_double_emphasis, 546 | rndr_emphasis, 547 | xhtml_discount_image, 548 | xhtml_linebreak, 549 | discount_link, 550 | rndr_raw_inline, 551 | rndr_triple_emphasis, 552 | 553 | NULL, 554 | rndr_normal_text, 555 | 556 | 64, 557 | "*_", 558 | NULL }; 559 | 560 | 561 | /**************************** 562 | * NATACHA'S OWN EXTENSIONS * 563 | ****************************/ 564 | 565 | static void 566 | nat_span(struct buf *ob, struct buf *text, char *tag) { 567 | bufprintf(ob, "<%s>", tag); 568 | bufput(ob, text->data, text->size); 569 | bufprintf(ob, "</%s>", tag); } 570 | 571 | static int 572 | nat_emphasis(struct buf *ob, struct buf *text, char c, void *opaque) { 573 | if (!text || !text->size || c == '+' || c == '-') return 0; 574 | if (c == '|') nat_span(ob, text, "span"); 575 | else nat_span(ob, text, "em"); 576 | return 1; } 577 | 578 | static int 579 | nat_double_emphasis(struct buf *ob, struct buf *text, char c, void *opaque) { 580 | if (!text || !text->size || c == '|') return 0; 581 | if (c == '+') nat_span(ob, text, "ins"); 582 | else if (c == '-') nat_span(ob, text, "del"); 583 | else nat_span(ob, text, "strong"); 584 | return 1; } 585 | 586 | static int 587 | nat_triple_emphasis(struct buf *ob, struct buf *text, char c, void *opaque) { 588 | if (!text || !text->size || c == '+' || c == '-' || c == '|') return 0; 589 | BUFPUTSL(ob, "<strong><em>"); 590 | bufput(ob, text->data, text->size); 591 | BUFPUTSL(ob, "</em></strong>"); 592 | return 1; } 593 | 594 | static void 595 | nat_header(struct buf *ob, struct buf *text, int level, void *opaque) { 596 | size_t i = 0; 597 | if (ob->size) bufputc(ob, '\n'); 598 | while (i < text->size && (text->data[i] == '-' || text->data[i] == '_' 599 | || text->data[i] == '.' || text->data[i] == ':' 600 | || (text->data[i] >= 'a' && text->data[i] <= 'z') 601 | || (text->data[i] >= 'A' && text->data[i] <= 'Z') 602 | || (text->data[i] >= '0' && text->data[i] <= '0'))) 603 | i += 1; 604 | bufprintf(ob, "<h%d", level); 605 | if (i < text->size && text->data[i] == '#') { 606 | bufprintf(ob, " id=\"%.*s\">", (int)i, text->data); 607 | i += 1; } 608 | else { 609 | bufputc(ob, '>'); 610 | i = 0; } 611 | bufput(ob, text->data + i, text->size - i); 612 | bufprintf(ob, "</h%d>\n", level); } 613 | 614 | static void 615 | nat_paragraph(struct buf *ob, struct buf *text, void *opaque) { 616 | size_t i = 0; 617 | if (ob->size) bufputc(ob, '\n'); 618 | BUFPUTSL(ob, "<p"); 619 | if (text && text->size && text->data[0] == '(') { 620 | i = 1; 621 | while (i < text->size && (text->data[i] == ' ' 622 | /* this seems to be a bit more restrictive than */ 623 | /* what is allowed for class names */ 624 | || (text->data[i] >= 'a' && text->data[i] <= 'z') 625 | || (text->data[i] >= 'A' && text->data[i] <= 'Z') 626 | || (text->data[i] >= '0' && text->data[i] <= '0'))) 627 | i += 1; 628 | if (i < text->size && text->data[i] == ')') { 629 | bufprintf(ob, " class=\"%.*s\"", 630 | (int)(i - 1), text->data + 1); 631 | i += 1; } 632 | else i = 0; } 633 | bufputc(ob, '>'); 634 | if (text) bufput(ob, text->data + i, text->size - i); 635 | BUFPUTSL(ob, "</p>\n"); } 636 | 637 | 638 | /* exported renderer structures */ 639 | const struct mkd_renderer nat_html = { 640 | NULL, 641 | NULL, 642 | 643 | rndr_blockcode, 644 | NULL, 645 | discount_blockquote, 646 | rndr_raw_block, 647 | nat_header, 648 | html_hrule, 649 | rndr_list, 650 | rndr_listitem, 651 | nat_paragraph, 652 | NULL, 653 | NULL, 654 | NULL, 655 | 656 | rndr_autolink, 657 | rndr_codespan, 658 | nat_double_emphasis, 659 | nat_emphasis, 660 | html_discount_image, 661 | html_linebreak, 662 | discount_link, 663 | rndr_raw_inline, 664 | nat_triple_emphasis, 665 | 666 | NULL, 667 | rndr_normal_text, 668 | 669 | 64, 670 | "*_-+|", 671 | NULL }; 672 | const struct mkd_renderer nat_xhtml = { 673 | NULL, 674 | NULL, 675 | 676 | rndr_blockcode, 677 | NULL, 678 | discount_blockquote, 679 | rndr_raw_block, 680 | nat_header, 681 | xhtml_hrule, 682 | rndr_list, 683 | rndr_listitem, 684 | nat_paragraph, 685 | NULL, 686 | NULL, 687 | NULL, 688 | 689 | rndr_autolink, 690 | rndr_codespan, 691 | nat_double_emphasis, 692 | nat_emphasis, 693 | xhtml_discount_image, 694 | xhtml_linebreak, 695 | discount_link, 696 | rndr_raw_inline, 697 | nat_triple_emphasis, 698 | 699 | NULL, 700 | rndr_normal_text, 701 | 702 | 64, 703 | "*_-+|", 704 | NULL }; 705 | -------------------------------------------------------------------------------- /console-colour.c: -------------------------------------------------------------------------------- 1 | #include "console-colour.h" 2 | 3 | struct console_colour_state_s 4 | gConsoleStreamState = { 5 | 0x80, 6 | STATE_NONE 7 | }; 8 | 9 | #ifdef CONMD_WINDOWS 10 | 11 | #include <stdio.h> 12 | #include <limits.h> 13 | 14 | static int OutputC(wchar_t c, struct console_colour_stream_s * const stream) 15 | { 16 | return stream->Call->OutputC(stream->Call->Data, c, stream); 17 | } 18 | 19 | static int OutputA(char const * c, int len, struct console_colour_stream_s * const stream) 20 | { 21 | return stream->Call->OutputA(stream->Call->Data, c, len, stream); 22 | } 23 | 24 | static int OutputW(wchar_t const * c, int len, struct console_colour_stream_s * const stream) 25 | { 26 | return stream->Call->OutputW(stream->Call->Data, c, len, stream); 27 | } 28 | 29 | static void Colour(struct console_colour_stream_s * const stream) 30 | { 31 | //if (!stream->State->Coloured) 32 | //{ 33 | // // Colouring is disabled. Just do nothing. 34 | //} 35 | //else 36 | if (stream->State->DefaultStyle & COMMON_LVB_REVERSE_VIDEO) 37 | { 38 | stream->Call->OutputColour(stream->Call->Data, stream->State->CurrentStyle >> 4 & 0x0F | stream->State->CurrentStyle << 4 & 0xF0, stream); 39 | } 40 | else 41 | { 42 | stream->Call->OutputColour(stream->Call->Data, stream->State->CurrentStyle, stream); 43 | } 44 | } 45 | 46 | void Backout(struct console_colour_stream_s * const stream) 47 | { 48 | // Back out of the buffering. 49 | char backout[32] = ""; 50 | // Run the state machine. 51 | switch (stream->State->State) 52 | { 53 | default: 54 | // Skip might be an issue - we are in the middle of a multi-byte character... 55 | return; 56 | case STATE_ESC: 57 | sprintf_s(backout, sizeof (backout), "\x1B"); 58 | break; 59 | case STATE_START: 60 | sprintf_s(backout, sizeof (backout), "\x1B["); 61 | break; 62 | case STATE_A00: 63 | sprintf_s(backout, sizeof (backout), "\x1B[%d", stream->State->Attr0); 64 | break; 65 | case STATE_S0: 66 | sprintf_s(backout, sizeof (backout), "\x1B[%d;", stream->State->Attr0); 67 | break; 68 | case STATE_A10: 69 | sprintf_s(backout, sizeof (backout), "\x1B[%d;%d", stream->State->Attr0, stream->State->Attr1); 70 | break; 71 | case STATE_S1: 72 | sprintf_s(backout, sizeof (backout), "\x1B[%d;%d;", stream->State->Attr0, stream->State->Attr1); 73 | break; 74 | case STATE_A20: 75 | sprintf_s(backout, sizeof (backout), "\x1B[%d;%d;%d", stream->State->Attr0, stream->State->Attr1, stream->State->Attr2); 76 | break; 77 | case STATE_S2: 78 | sprintf_s(backout, sizeof (backout), "\x1B[%d;%d;%d;", stream->State->Attr0, stream->State->Attr1, stream->State->Attr2); 79 | break; 80 | case STATE_A30: 81 | sprintf_s(backout, sizeof (backout), "\x1B[%d;%d;%d;%d", stream->State->Attr0, stream->State->Attr1, stream->State->Attr2, stream->State->Attr3); 82 | break; 83 | case STATE_S3: 84 | sprintf_s(backout, sizeof (backout), "\x1B[%d;%d;%d;%d;", stream->State->Attr0, stream->State->Attr1, stream->State->Attr2, stream->State->Attr3); 85 | break; 86 | case STATE_A40: 87 | sprintf_s(backout, sizeof (backout), "\x1B[%d;%d;%d;%d;%d", stream->State->Attr0, stream->State->Attr1, stream->State->Attr2, stream->State->Attr3, stream->State->Attr4); 88 | break; 89 | } 90 | if (backout[0]) 91 | { 92 | Colour(stream); 93 | OutputA(backout, (int)strlen(backout), stream); 94 | } 95 | stream->State->State = STATE_NONE; 96 | stream->State->Attr0 = stream->State->Attr1 = stream->State->Attr2 = stream->State->Attr3 = stream->State->Attr4 = 0; 97 | } 98 | 99 | static const WORD 100 | FOREGROUND = FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE | FOREGROUND_INTENSITY, 101 | BACKGROUND = BACKGROUND_RED | BACKGROUND_GREEN | BACKGROUND_BLUE | BACKGROUND_INTENSITY; 102 | 103 | static WORD NToColour(unsigned char attr) 104 | { 105 | switch (attr) 106 | { 107 | // Original colours. 108 | case 0: 109 | // Black. 110 | return 0; 111 | case 1: 112 | // Red. 113 | return FOREGROUND_RED; 114 | case 2: 115 | // Green. 116 | return FOREGROUND_GREEN; 117 | case 3: 118 | // Yellow. 119 | return FOREGROUND_RED | FOREGROUND_GREEN; 120 | case 4: 121 | // Blue. 122 | return FOREGROUND_BLUE; 123 | case 5: 124 | // Magenta. 125 | return FOREGROUND_RED | FOREGROUND_BLUE; 126 | case 6: 127 | // Cyan. 128 | return FOREGROUND_GREEN | FOREGROUND_BLUE; 129 | case 7: 130 | // White. 131 | return FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE; 132 | case 8 + 0: 133 | // Black. 134 | return FOREGROUND_INTENSITY; 135 | case 8 + 1: 136 | // Red. 137 | return FOREGROUND_RED | FOREGROUND_INTENSITY; 138 | case 8 + 2: 139 | // Green. 140 | return FOREGROUND_GREEN | FOREGROUND_INTENSITY; 141 | case 8 + 3: 142 | // Yellow. 143 | return FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_INTENSITY; 144 | case 8 + 4: 145 | // Blue. 146 | return FOREGROUND_BLUE | FOREGROUND_INTENSITY; 147 | case 8 + 5: 148 | // Magenta. 149 | return FOREGROUND_RED | FOREGROUND_BLUE | FOREGROUND_INTENSITY; 150 | case 8 + 6: 151 | // Cyan. 152 | return FOREGROUND_GREEN | FOREGROUND_BLUE | FOREGROUND_INTENSITY; 153 | case 8 + 7: 154 | // White. 155 | return FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE | FOREGROUND_INTENSITY; 156 | } 157 | return 0; 158 | } 159 | 160 | static unsigned char Shuffle6Colour(unsigned char colour) 161 | { 162 | // The search results are better using a different search order. 163 | const unsigned char 164 | ORDER[16] = { 0, 15, 1, 2, 4, 3, 6, 5, 9, 10, 12, 11, 14, 13, 8, 7 }; 165 | return ORDER[colour]; 166 | } 167 | 168 | static WORD Make6Colour(int r, int g, int b, struct console_colour_stream_s* const stream) 169 | { 170 | // There's no easy way to map 16 colours to 3 components. So we need full colour space mapping. 171 | // I didn't want to have to write that... I KNOW there are better mapping functions based on 172 | // what colours are eyes are better at perceiving, but I'm only mapping to 16 so it hardly 173 | // matters. 174 | // 175 | // I did a LOT of experimenting with orders to get this as the best order I could find. 176 | const int 177 | CONSOLE_COLOURS[16][3] = { 178 | {0, 0, 0}, 179 | {5, 5, 5}, 180 | {2, 0, 0}, 181 | {0, 2, 0}, 182 | {0, 0, 2}, 183 | {2, 2, 0}, 184 | {0, 2, 2}, 185 | {2, 0, 2}, 186 | {4, 0, 0}, 187 | {0, 4, 0}, 188 | {0, 0, 4}, 189 | {4, 4, 0}, 190 | {0, 4, 4}, 191 | {4, 0, 4}, 192 | {2, 2, 2}, 193 | {4, 4, 4}, 194 | }; 195 | 196 | int 197 | dist = INT_MAX; 198 | unsigned char 199 | found = 0; 200 | for (unsigned char i = 0; i != 16; ++i) 201 | { 202 | int 203 | tr = r - CONSOLE_COLOURS[i][0], 204 | tg = g - CONSOLE_COLOURS[i][1], 205 | tb = b - CONSOLE_COLOURS[i][2], 206 | cur = (tr * tr) + (tg * tg) + (tb * tb); 207 | if (cur < dist) 208 | { 209 | dist = cur; 210 | found = i; 211 | } 212 | } 213 | 214 | return NToColour(Shuffle6Colour(found)); 215 | } 216 | 217 | static WORD Make24Colour(int r, int g, int b, struct console_colour_stream_s* const stream) 218 | { 219 | // Use the reduced 6-value colour space mapping, since we've tweaked that a lot already. 220 | r = r + 25 / 51; // Rounding division, to get the component from 0-5 (inclusive). 221 | g = g + 25 / 51; // Rounding division, to get the component from 0-5 (inclusive). 222 | b = b + 25 / 51; // Rounding division, to get the component from 0-5 (inclusive). 223 | return Make6Colour(r, g, b, stream); 224 | } 225 | 226 | static WORD Make256Colour(unsigned char attr, struct console_colour_stream_s * const stream) 227 | { 228 | // Greyscales. Find the nearest. 229 | if (attr >= 250) 230 | return FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE | FOREGROUND_INTENSITY; 231 | else if (attr >= 244) 232 | return FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE; 233 | else if (attr >= 238) 234 | return FOREGROUND_INTENSITY; 235 | else if (attr >= 232) 236 | return 0; 237 | else if (attr >= 16) 238 | { 239 | // 216 colours. Extract the RGB from the colour. Then scale them to 8-bit. 240 | attr -= 16; 241 | unsigned char 242 | r = attr / 36, 243 | g = attr % 36 / 6, 244 | b = attr % 6; 245 | return Make6Colour(r, g, b, stream); 246 | } 247 | 248 | return NToColour(attr); 249 | } 250 | 251 | static WORD GetColour(unsigned char attr, struct console_colour_stream_s* const stream) 252 | { 253 | WORD 254 | current = stream->State->CurrentStyle; 255 | switch (attr) 256 | { 257 | case 1: 258 | return current | FOREGROUND_INTENSITY; 259 | case 2: 260 | return current & ~FOREGROUND_INTENSITY; 261 | case 3: 262 | case 7: 263 | // Inverse. Store the fact that it is inverted somewhere, and only apply it at the very 264 | // last possible moment. 265 | stream->State->DefaultStyle |= COMMON_LVB_REVERSE_VIDEO; 266 | break; 267 | case 23: 268 | case 27: 269 | stream->State->DefaultStyle &= ~COMMON_LVB_REVERSE_VIDEO; 270 | break; 271 | case 4: 272 | case 21: 273 | return current | COMMON_LVB_UNDERSCORE; 274 | case 22: 275 | return current & ~FOREGROUND_INTENSITY; 276 | case 24: 277 | return current & ~COMMON_LVB_UNDERSCORE; 278 | case 39: 279 | return current & ~FOREGROUND | stream->State->DefaultStyle & FOREGROUND; 280 | case 49: 281 | return current & ~BACKGROUND | stream->State->DefaultStyle & BACKGROUND; 282 | case 30: 283 | // Black. 284 | return current & ~FOREGROUND; 285 | case 31: 286 | // Red. 287 | return current & ~FOREGROUND | FOREGROUND_RED; 288 | case 32: 289 | // Green. 290 | return current & ~FOREGROUND | FOREGROUND_GREEN; 291 | case 33: 292 | // Yellow. 293 | return current & ~FOREGROUND | FOREGROUND_RED | FOREGROUND_GREEN; 294 | case 34: 295 | // Blue. 296 | return current & ~FOREGROUND | FOREGROUND_BLUE; 297 | case 35: 298 | // Magenta. 299 | return current & ~FOREGROUND | FOREGROUND_RED | FOREGROUND_BLUE; 300 | case 36: 301 | // Cyan. 302 | return current & ~FOREGROUND | FOREGROUND_GREEN | FOREGROUND_BLUE; 303 | case 37: 304 | // White. 305 | return current & ~FOREGROUND | FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE; 306 | case 40: 307 | // Black. 308 | return current & ~BACKGROUND; 309 | case 41: 310 | // Red. 311 | return current & ~BACKGROUND | BACKGROUND_RED; 312 | case 42: 313 | // Green. 314 | return current & ~BACKGROUND | BACKGROUND_GREEN; 315 | case 43: 316 | // Yellow. 317 | return current & ~BACKGROUND | BACKGROUND_RED | BACKGROUND_GREEN; 318 | case 44: 319 | // Blue. 320 | return current & ~BACKGROUND | BACKGROUND_BLUE; 321 | case 45: 322 | // Magenta. 323 | return current & ~BACKGROUND | BACKGROUND_RED | BACKGROUND_BLUE; 324 | case 46: 325 | // Cyan. 326 | return current & ~BACKGROUND | BACKGROUND_GREEN | BACKGROUND_BLUE; 327 | case 47: 328 | // White. 329 | return current & ~BACKGROUND | BACKGROUND_RED | BACKGROUND_GREEN | BACKGROUND_BLUE; 330 | case 60 + 30: 331 | // Black. 332 | return current & ~FOREGROUND | FOREGROUND_INTENSITY; 333 | case 60 + 31: 334 | // Red. 335 | return current & ~FOREGROUND | FOREGROUND_RED | FOREGROUND_INTENSITY; 336 | case 60 + 32: 337 | // Green. 338 | return current & ~FOREGROUND | FOREGROUND_GREEN | FOREGROUND_INTENSITY; 339 | case 60 + 33: 340 | // Yellow. 341 | return current & ~FOREGROUND | FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_INTENSITY; 342 | case 60 + 34: 343 | // Blue. 344 | return current & ~FOREGROUND | FOREGROUND_BLUE | FOREGROUND_INTENSITY; 345 | case 60 + 35: 346 | // Magenta. 347 | return current & ~FOREGROUND | FOREGROUND_RED | FOREGROUND_BLUE | FOREGROUND_INTENSITY; 348 | case 60 + 36: 349 | // Cyan. 350 | return current & ~FOREGROUND | FOREGROUND_GREEN | FOREGROUND_BLUE | FOREGROUND_INTENSITY; 351 | case 60 + 37: 352 | // White. 353 | return current & ~FOREGROUND | FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE | FOREGROUND_INTENSITY; 354 | case 60 + 40: 355 | // Black. 356 | return current & ~BACKGROUND | BACKGROUND_INTENSITY; 357 | case 60 + 41: 358 | // Red. 359 | return current & ~BACKGROUND | BACKGROUND_RED | BACKGROUND_INTENSITY; 360 | case 60 + 42: 361 | // Green. 362 | return current & ~BACKGROUND | BACKGROUND_GREEN | BACKGROUND_INTENSITY; 363 | case 60 + 43: 364 | // Yellow. 365 | return current & ~BACKGROUND | BACKGROUND_RED | BACKGROUND_GREEN | BACKGROUND_INTENSITY; 366 | case 60 + 44: 367 | // Blue. 368 | return current & ~BACKGROUND | BACKGROUND_BLUE | BACKGROUND_INTENSITY; 369 | case 60 + 45: 370 | // Magenta. 371 | return current & ~BACKGROUND | BACKGROUND_RED | BACKGROUND_BLUE | BACKGROUND_INTENSITY; 372 | case 60 + 46: 373 | // Cyan. 374 | return current & ~BACKGROUND | BACKGROUND_GREEN | BACKGROUND_BLUE | BACKGROUND_INTENSITY; 375 | case 60 + 47: 376 | // White. 377 | return current & ~BACKGROUND | BACKGROUND_RED | BACKGROUND_GREEN | BACKGROUND_BLUE | BACKGROUND_INTENSITY; 378 | } 379 | return current; 380 | } 381 | 382 | static void GetColours(struct console_colour_stream_s * const stream) 383 | { 384 | switch (stream->State->Attr0) 385 | { 386 | case 0: 387 | // Reset. 388 | stream->State->DefaultStyle &= ~COMMON_LVB_REVERSE_VIDEO; 389 | stream->State->CurrentStyle = stream->State->DefaultStyle; 390 | break; 391 | case 38: 392 | // Special foreground colour. 393 | // TODO: Find the closest working colour from the 16 available. 394 | switch (stream->State->Attr1) 395 | { 396 | case 5: 397 | // 256 colours. 398 | stream->State->CurrentStyle = stream->State->CurrentStyle & ~FOREGROUND | Make256Colour(stream->State->Attr2, stream); 399 | break; 400 | case 2: 401 | // 24-bit colours. 402 | stream->State->CurrentStyle = stream->State->CurrentStyle & ~FOREGROUND | Make24Colour(stream->State->Attr2, stream->State->Attr3, stream->State->Attr4, stream); 403 | break; 404 | } 405 | break; 406 | case 48: 407 | // Special background colour. 408 | switch (stream->State->Attr1) 409 | { 410 | case 5: 411 | // 256 colours. 412 | stream->State->CurrentStyle = stream->State->CurrentStyle & ~BACKGROUND | Make256Colour(stream->State->Attr2, stream) << 4; 413 | break; 414 | case 2: 415 | // 24-bit colours. 416 | stream->State->CurrentStyle = stream->State->CurrentStyle & ~BACKGROUND | Make24Colour(stream->State->Attr2, stream->State->Attr3, stream->State->Attr4, stream) << 4; 417 | break; 418 | } 419 | break; 420 | default: 421 | // Normal code. 422 | stream->State->CurrentStyle = GetColour(stream->State->Attr0, stream); 423 | stream->State->CurrentStyle = GetColour(stream->State->Attr1, stream); 424 | stream->State->CurrentStyle = GetColour(stream->State->Attr2, stream); 425 | stream->State->CurrentStyle = GetColour(stream->State->Attr3, stream); 426 | stream->State->CurrentStyle = GetColour(stream->State->Attr4, stream); 427 | break; 428 | } 429 | } 430 | 431 | static int RunStateMachine(wchar_t c, struct console_colour_stream_s * const stream) 432 | { 433 | // Run the stream machine. 434 | switch (stream->State->State) 435 | { 436 | case STATE_NONE: 437 | //if (stream->State->Coloured) 438 | { 439 | switch (c) 440 | { 441 | case 29: // '\x1D', Group separator. 442 | return /*stream->State->*/OutputC(10, stream); // `\n` 443 | case 31: // '\x1F', Unit separator. 444 | return /*stream->State->*/OutputC(10, stream); // `\n` 445 | case 30: // '\x1E', Record separator. 446 | return 0; 447 | } 448 | } 449 | if (c == '\x1B') 450 | { 451 | stream->State->State = STATE_ESC; 452 | return 0; 453 | } 454 | else if ((c & stream->State->UnicodeMask)) 455 | { 456 | // Start of a multi-byte character. Skip all checks on the next one. 457 | stream->State->State = STATE_SKIP; 458 | } 459 | break; 460 | case STATE_SKIP: 461 | if (!(c & stream->State->UnicodeMask)) 462 | { 463 | stream->State->State = STATE_NONE; 464 | } 465 | break; 466 | case STATE_ESC: 467 | if (c == '[') 468 | { 469 | stream->State->State = STATE_START; 470 | } 471 | else 472 | { 473 | Backout(stream); 474 | break; 475 | } 476 | return 0; 477 | case STATE_START: 478 | if ('0' <= c && c <= '9') 479 | { 480 | stream->State->Attr0 = (unsigned char)(c - '0'); 481 | stream->State->State = STATE_A00; 482 | } 483 | else 484 | { 485 | Backout(stream); 486 | break; 487 | } 488 | return 0; 489 | case STATE_A00: 490 | if ('0' <= c && c <= '9') 491 | { 492 | stream->State->Attr0 = stream->State->Attr0 * 10 + (unsigned char)(c - '0'); 493 | } 494 | else if (c == ';') 495 | stream->State->State = STATE_S0; 496 | //else if (c == ':') 497 | // stream->State->State = STATE_T416; 498 | else if (c == 'm') 499 | { 500 | stream->State->State = STATE_DONE; 501 | break; 502 | } 503 | else 504 | { 505 | Backout(stream); 506 | break; 507 | } 508 | return 0; 509 | case STATE_S0: 510 | if ('0' <= c && c <= '9') 511 | { 512 | stream->State->Attr1 = (unsigned char)(c - '0'); 513 | stream->State->State = STATE_A10; 514 | } 515 | else if (c == 'm') 516 | { 517 | stream->State->State = STATE_DONE; 518 | break; 519 | } 520 | else 521 | { 522 | Backout(stream); 523 | break; 524 | } 525 | return 0; 526 | case STATE_A10: 527 | if ('0' <= c && c <= '9') 528 | { 529 | stream->State->Attr1 = stream->State->Attr1 * 10 + (unsigned char)(c - '0'); 530 | } 531 | else if (c == ';') 532 | stream->State->State = STATE_S1; 533 | else if (c == 'm') 534 | { 535 | stream->State->State = STATE_DONE; 536 | break; 537 | } 538 | else 539 | { 540 | Backout(stream); 541 | break; 542 | } 543 | return 0; 544 | case STATE_S1: 545 | if ('0' <= c && c <= '9') 546 | { 547 | stream->State->Attr2 = (unsigned char)(c - '0'); 548 | stream->State->State = STATE_A20; 549 | } 550 | else if (c == 'm') 551 | { 552 | stream->State->State = STATE_DONE; 553 | break; 554 | } 555 | else 556 | { 557 | Backout(stream); 558 | break; 559 | } 560 | return 0; 561 | case STATE_A20: 562 | if ('0' <= c && c <= '9') 563 | { 564 | stream->State->Attr2 = stream->State->Attr2 * 10 + (unsigned char)(c - '0'); 565 | } 566 | else if (c == ';') 567 | stream->State->State = STATE_S2; 568 | else if (c == 'm') 569 | { 570 | stream->State->State = STATE_DONE; 571 | break; 572 | } 573 | else 574 | { 575 | Backout(stream); 576 | break; 577 | } 578 | return 0; 579 | case STATE_S2: 580 | if ('0' <= c && c <= '9') 581 | { 582 | stream->State->Attr3 = (unsigned char)(c - '0'); 583 | stream->State->State = STATE_A30; 584 | } 585 | else if (c == 'm') 586 | { 587 | stream->State->State = STATE_DONE; 588 | break; 589 | } 590 | else 591 | { 592 | Backout(stream); 593 | break; 594 | } 595 | return 0; 596 | case STATE_A30: 597 | if ('0' <= c && c <= '9') 598 | { 599 | stream->State->Attr3 = stream->State->Attr3 * 10 + (unsigned char)(c - '0'); 600 | } 601 | else if (c == ';') 602 | stream->State->State = STATE_S3; 603 | else if (c == 'm') 604 | { 605 | stream->State->State = STATE_DONE; 606 | break; 607 | } 608 | else 609 | { 610 | Backout(stream); 611 | break; 612 | } 613 | return 0; 614 | case STATE_S3: 615 | if ('0' <= c && c <= '9') 616 | { 617 | stream->State->Attr4 = (unsigned char)(c - '0'); 618 | stream->State->State = STATE_A40; 619 | } 620 | else if (c == 'm') 621 | { 622 | stream->State->State = STATE_DONE; 623 | break; 624 | } 625 | else 626 | { 627 | Backout(stream); 628 | break; 629 | } 630 | return 0; 631 | case STATE_A40: 632 | if ('0' <= c && c <= '9') 633 | { 634 | stream->State->Attr4 = stream->State->Attr4 * 10 + (unsigned char)(c - '0'); 635 | } 636 | else if (c == 'm') 637 | { 638 | stream->State->State = STATE_DONE; 639 | break; 640 | } 641 | else 642 | { 643 | Backout(stream); 644 | break; 645 | } 646 | return 0; 647 | } 648 | if (stream->State->State == STATE_DONE) 649 | { 650 | GetColours(stream); 651 | stream->State->State = STATE_NONE; 652 | stream->State->Attr0 = stream->State->Attr1 = stream->State->Attr2 = stream->State->Attr3 = stream->State->Attr4 = 0; 653 | return 0; 654 | } 655 | Colour(stream); 656 | return /*stream->State->*/OutputC(c, stream); 657 | } 658 | 659 | #if defined CONMD_HOOKS 660 | #define MAYBE_REMOVE_HOOKS() RemoveStreamHooks() 661 | #define MAYBE_READD_HOOKS() ReaddStreamHooks() 662 | #else 663 | #define MAYBE_REMOVE_HOOKS() ((void)0) 664 | #define MAYBE_READD_HOOKS() ((void)0) 665 | #endif 666 | 667 | int WriteColouredA(char const * s, int n, struct console_colour_stream_s* const stream) 668 | { 669 | MAYBE_REMOVE_HOOKS(); 670 | char const 671 | * cur, 672 | * start = s, 673 | * end = s + n; 674 | // Just in case. 675 | if (n == 0 || s == NULL) 676 | { 677 | MAYBE_READD_HOOKS(); 678 | return 0; 679 | } 680 | int 681 | ret = 0; 682 | for (; ; ) 683 | { 684 | while (*s && !(stream->State->State == STATE_NONE || stream->State->State == STATE_SKIP)) 685 | { 686 | RunStateMachine(*s++, stream); 687 | } 688 | char const * 689 | esc = s; 690 | //if (stream->State->Coloured) 691 | { 692 | for (cur = s; cur != end; ++cur) 693 | { 694 | switch (*cur) 695 | { 696 | case '\x1B': // ESC. 697 | stream->State->State = STATE_ESC; 698 | esc = cur; 699 | goto WriteColoured_loop_done; 700 | case '\x1D': // Group separator. 701 | stream->State->State = STATE_EXTRA_NL; 702 | esc = cur; 703 | goto WriteColoured_loop_done; 704 | case '\x1F': // Unit separator. 705 | stream->State->State = STATE_EXTRA_2NL; 706 | esc = cur; 707 | goto WriteColoured_loop_done; 708 | case '\x1E': // Record separator. 709 | stream->State->State = STATE_NONE; 710 | esc = cur; 711 | goto WriteColoured_loop_done; 712 | } 713 | } 714 | WriteColoured_loop_done: 715 | if (cur == end) 716 | { 717 | break; 718 | } 719 | } 720 | //else 721 | //{ 722 | // esc = strchr(s, '\x1B'); 723 | // if (esc == NULL) 724 | // break; 725 | // stream->State->State = STATE_ESC; 726 | //} 727 | // Write the bit before the escape sequence. 728 | if (esc != s) 729 | { 730 | // Restore the correct colours. 731 | Colour(stream); 732 | int 733 | wrote = OutputA(s, (int)(esc - s), stream); 734 | ret += wrote; 735 | if (wrote != esc - s) 736 | { 737 | MAYBE_READD_HOOKS(); 738 | return (int)(s - start + wrote); 739 | } 740 | } 741 | s = esc + 1; 742 | switch (stream->State->State) 743 | { 744 | case STATE_EXTRA_2NL: 745 | OutputC(10, stream); 746 | case STATE_EXTRA_NL: 747 | OutputC(10, stream); 748 | break; 749 | /*case STATE_EXTRA_SPACE: 750 | OutputA(' '); 751 | break;*/ 752 | default: 753 | continue; 754 | } 755 | stream->State->State = STATE_NONE; 756 | } 757 | // Restore the correct colours. 758 | Colour(stream); 759 | int 760 | wrote = OutputA(s, (int)(end - s), stream); 761 | MAYBE_READD_HOOKS(); 762 | return (int)(s - start + wrote); 763 | } 764 | 765 | int WriteColouredW(wchar_t const * s, int n, struct console_colour_stream_s* const stream) 766 | { 767 | MAYBE_REMOVE_HOOKS(); 768 | wchar_t const 769 | * cur, 770 | * start = s, 771 | * end = s + n; 772 | // Just in case. 773 | if (n == 0 || s == NULL) 774 | { 775 | MAYBE_READD_HOOKS(); 776 | return 0; 777 | } 778 | int 779 | ret = 0; 780 | for (; ; ) 781 | { 782 | while (*s && !(stream->State->State == STATE_NONE || stream->State->State == STATE_SKIP)) 783 | { 784 | RunStateMachine(*s++, stream); 785 | } 786 | wchar_t const * 787 | esc = s; 788 | //if (stream->State->Coloured) 789 | { 790 | for (cur = s; cur != end; ++cur) 791 | { 792 | switch (*cur) 793 | { 794 | case '\x1B': // ESC. 795 | stream->State->State = STATE_ESC; 796 | esc = cur; 797 | goto WriteColoured_loop_done; 798 | case '\x1D': // Group separator. 799 | stream->State->State = STATE_EXTRA_NL; 800 | esc = cur; 801 | goto WriteColoured_loop_done; 802 | case '\x1F': // Unit separator. 803 | stream->State->State = STATE_EXTRA_2NL; 804 | esc = cur; 805 | goto WriteColoured_loop_done; 806 | case '\x1E': // Record separator. 807 | stream->State->State = STATE_NONE; 808 | esc = cur; 809 | goto WriteColoured_loop_done; 810 | } 811 | } 812 | WriteColoured_loop_done: 813 | if (cur == end) 814 | { 815 | break; 816 | } 817 | } 818 | //else 819 | //{ 820 | // esc = wcschr(s, '\x1B'); 821 | // if (esc == NULL) 822 | // break; 823 | // stream->State->State = STATE_ESC; 824 | //} 825 | // Write the bit before the escape sequence. 826 | if (esc != s) 827 | { 828 | // Restore the correct colours. 829 | Colour(stream); 830 | int 831 | wrote = OutputW(s, (int)(esc - s), stream); 832 | ret += wrote; 833 | if (wrote != esc - s) 834 | { 835 | MAYBE_READD_HOOKS(); 836 | return (int)(s - start + wrote); 837 | } 838 | } 839 | s = esc + 1; 840 | switch (stream->State->State) 841 | { 842 | case STATE_EXTRA_2NL: 843 | OutputC(10, stream); 844 | case STATE_EXTRA_NL: 845 | OutputC(10, stream); 846 | break; 847 | /*case STATE_EXTRA_SPACE: 848 | OutputW(' '); 849 | break;*/ 850 | default: 851 | continue; 852 | } 853 | stream->State->State = STATE_NONE; 854 | } 855 | // Restore the correct colours. 856 | Colour(stream); 857 | int 858 | wrote = OutputW(s, (int)(end - s), stream); 859 | MAYBE_READD_HOOKS(); 860 | return (int)(s - start + wrote); 861 | } 862 | 863 | #endif 864 | 865 | --------------------------------------------------------------------------------

] " 44 | "[ -t ] [input-file]\n\n", name); 45 | fprintf(out, "\t-d, --date\n" 46 | "\t\tSet the date of the manpage (default: now),\n" 47 | "\t-h, --help\n" 48 | "\t\tDisplay this help text and exit without further processing\n" 49 | "\t-s, --section\n" 50 | "\t\tSet the section of the manpage (default: 1)\n" 51 | "\t-t, --title\n" 52 | "\t\tSet the title of the manpage (default: filename)\n"); } 53 | 54 | struct metadata { 55 | char *title; 56 | char *date; 57 | int section; 58 | }; 59 | 60 | static void 61 | man_text_escape(struct buf *ob, char *src, size_t size) { 62 | size_t i = 0, org; 63 | while (i < size) { 64 | /* copying directly unescaped characters */ 65 | org = i; 66 | while (i < size && src[i] != '-') 67 | i += 1; 68 | if (i > org) bufput(ob, src + org, i - org); 69 | 70 | /* escaping */ 71 | if (i >= size) break; 72 | else if (src[i] == '-') BUFPUTSL(ob, "\\-"); 73 | i += 1; } } 74 | 75 | static void 76 | man_prolog(struct buf *ob, void *opaque) { 77 | struct metadata *m = (struct metadata *)opaque; 78 | bufprintf(ob, 79 | ".\\\" Generated by mkd2man\n" 80 | ".Dd %s\n" 81 | ".Dt %s %d\n" 82 | ".Os", 83 | m->date, 84 | m->title, 85 | m->section 86 | ); } 87 | 88 | static void 89 | man_epilog(struct buf *ob, void *opaque) { 90 | BUFPUTSL(ob, "\n"); } 91 | 92 | static void 93 | man_blockcode(struct buf *ob, struct buf *text, void *opaque) { 94 | if (ob->size) bufputc(ob, '\n'); 95 | BUFPUTSL(ob, ".Bd -literal\n"); 96 | if (text) man_text_escape(ob, text->data, text->size); 97 | BUFPUTSL(ob, ".Ed"); } 98 | 99 | static void 100 | man_blockquote(struct buf *ob, struct buf *text, void *opaque) { 101 | if (ob->size) bufputc(ob, '\n'); 102 | BUFPUTSL(ob, ".Eo\n"); 103 | if (text) man_text_escape(ob, text->data, text->size); 104 | BUFPUTSL(ob, "\n.Ec"); } 105 | 106 | static int 107 | man_codespan(struct buf *ob, struct buf *text, void *opaque) { 108 | if (ob->size) bufputc(ob, '\n'); 109 | BUFPUTSL(ob, ".Bd -literal\n"); 110 | if (text) man_text_escape(ob, text->data, text->size); 111 | BUFPUTSL(ob, ".Ed"); 112 | return 1; } 113 | 114 | static void 115 | man_header(struct buf *ob, struct buf *text, int level, void *opaque) { 116 | if (ob->size) bufputc(ob, '\n'); 117 | switch(level) { 118 | case 1: 119 | BUFPUTSL(ob,".Sh "); 120 | break; 121 | case 2: 122 | BUFPUTSL(ob, ".Ss "); 123 | break; 124 | case 3: 125 | BUFPUTSL(ob, ".Pp\n.Em "); 126 | break; 127 | } 128 | if (text) bufput(ob, text->data, text->size); 129 | } 130 | 131 | static int 132 | man_double_emphasis(struct buf *ob, struct buf *text, char c, void *opaque) { 133 | if (!text || !text->size) return 0; 134 | BUFPUTSL(ob, "\\fB"); 135 | bufput(ob, text->data, text->size); 136 | BUFPUTSL(ob, "\\fP"); 137 | return 1; } 138 | 139 | static int 140 | man_emphasis(struct buf *ob, struct buf *text, char c, void *opaque) { 141 | if (!text || !text->size) return 0; 142 | BUFPUTSL(ob, "\\fI"); 143 | if (text) bufput(ob, text->data, text->size); 144 | BUFPUTSL(ob, "\\fP"); 145 | return 1; } 146 | 147 | static int 148 | man_linebreak(struct buf *ob, void *opaque) { 149 | BUFPUTSL(ob, ".br"); 150 | return 1; } 151 | 152 | static void 153 | man_paragraph(struct buf *ob, struct buf *text, void *opaque) { 154 | if (ob->size) bufputc(ob, '\n'); 155 | BUFPUTSL(ob, ".Pp\n"); 156 | if (text) bufput(ob, text->data, text->size); } 157 | 158 | static void 159 | man_list(struct buf *ob, struct buf *text, int flags, void *opaque) { 160 | if (ob->size) bufputc(ob, '\n'); 161 | if (flags & MKD_LIST_ORDERED) 162 | BUFPUTSL(ob,".Bl -enum\n"); 163 | else 164 | BUFPUTSL(ob,".Bl -bullet\n"); 165 | if (text) bufput(ob, text->data, text->size); 166 | BUFPUTSL(ob, ".El"); } 167 | 168 | static void 169 | man_listitem(struct buf *ob, struct buf *text, int flags, void *opaque) { 170 | BUFPUTSL(ob, ".It\n"); 171 | if (text) { 172 | while (text->size && text->data[text->size - 1] == '\n') 173 | text->size -= 1; 174 | bufput(ob, text->data, text->size); } 175 | BUFPUTSL(ob, "\n"); } 176 | 177 | static void 178 | man_normal_text(struct buf *ob, struct buf *text, void *opaque) { 179 | if (text) man_text_escape(ob, text->data, text->size); } 180 | 181 | 182 | /* renderer structure */ 183 | static struct mkd_renderer to_man = { 184 | /* document-level callbacks */ 185 | man_prolog, 186 | man_epilog, 187 | 188 | /* block-level callbacks */ 189 | man_blockcode, 190 | NULL, 191 | man_blockquote, 192 | NULL, 193 | man_header, 194 | NULL, 195 | man_list, 196 | man_listitem, 197 | man_paragraph, 198 | NULL, 199 | NULL, 200 | NULL, 201 | 202 | /* span-level callbacks */ 203 | NULL, 204 | man_codespan, 205 | man_double_emphasis, 206 | man_emphasis, 207 | NULL, 208 | man_linebreak, 209 | NULL, 210 | NULL, 211 | NULL, 212 | 213 | /* low-level callbacks */ 214 | NULL, 215 | man_normal_text, 216 | 217 | /* renderer data */ 218 | 64, 219 | "*_", 220 | NULL }; 221 | 222 | 223 | 224 | /***************** 225 | * MAIN FUNCTION * 226 | *****************/ 227 | 228 | /* main • main function, interfacing STDIO with the parser */ 229 | int 230 | main(int argc, char **argv) { 231 | struct buf *ib, *ob; 232 | size_t ret; 233 | size_t i; 234 | FILE *in = stdin; 235 | int ch, argerr, help; 236 | char *tmp; 237 | char datebuf[64]; 238 | time_t ttm; 239 | struct tm *tm; 240 | struct stat st; 241 | struct metadata man_metadata; 242 | 243 | struct option longopts[] = { 244 | { "date", required_argument, 0, 'd' }, 245 | { "help", no_argument, 0, 'h' }, 246 | { "section", required_argument, 0, 's' }, 247 | { "title", required_argument, 0, 't' }, 248 | { 0, 0, 0, 0} 249 | }; 250 | 251 | man_metadata.section = 1; 252 | man_metadata.title = NULL; 253 | man_metadata.date = NULL; 254 | /* opening the file if given from the command line */ 255 | argerr = help = 0; 256 | while (!argerr && 257 | (ch = getopt_long(argc, argv, "d:hs:t:", longopts, 0)) != -1) 258 | switch (ch) { 259 | case 'd': 260 | man_metadata.date = optarg; 261 | break; 262 | case 'h': 263 | argerr = help = 1; 264 | break; 265 | case 's': 266 | if (strlen(optarg) != 1 && 267 | strspn(optarg, "123456789") != 1) { 268 | argerr = 1; 269 | break; } 270 | man_metadata.section = (int)strtol(optarg, 271 | (char **)NULL, 10); 272 | break; 273 | case 't': 274 | man_metadata.title = optarg; 275 | break; 276 | default: 277 | argerr = 1; } 278 | if (argerr) { 279 | usage(help ? stdout : stderr, argv[0]); 280 | return help ? EXIT_SUCCESS : EXIT_FAILURE; 281 | } 282 | 283 | argc -= optind; 284 | argv += optind; 285 | 286 | if (argc > 0) { 287 | in = fopen(argv[0], "r"); 288 | if (!in) { 289 | fprintf(stderr,"Unable to open input file \"%s\": %s\n", 290 | argv[0], strerror(errno)); 291 | return EXIT_FAILURE; } } 292 | 293 | if (!man_metadata.date) { 294 | if (in == stdin || stat(argv[0], &st) == -1) { 295 | ttm = time(NULL); 296 | tm = localtime(&ttm); } 297 | else 298 | tm = localtime(&st.st_mtime); 299 | strftime(datebuf, sizeof(datebuf), "%B %d, %Y", tm); 300 | man_metadata.date = datebuf; 301 | } 302 | 303 | if (in == stdin && !man_metadata.title) { 304 | fprintf(stderr, "When reading from stdin the title should be " 305 | "specified is expected\n"); 306 | return EXIT_FAILURE; } 307 | 308 | if (!man_metadata.title) { 309 | tmp = strrchr(argv[0], '/'); 310 | man_metadata.title = strrchr(argv[0], '/'); 311 | if (!tmp) 312 | tmp = argv[0]; 313 | else 314 | tmp++; 315 | man_metadata.title = tmp; 316 | tmp = strrchr(man_metadata.title, '.'); 317 | if (tmp) 318 | *tmp = '\0'; } 319 | 320 | /* Ensure the title is uppercase */ 321 | for (i = 0; i < strlen(man_metadata.title); i++) 322 | man_metadata.title[i] = toupper(man_metadata.title[i]); 323 | 324 | /* reading everything */ 325 | ib = bufnew(READ_UNIT); 326 | bufgrow(ib, READ_UNIT); 327 | while ((ret = fread(ib->data + ib->size, 1, 328 | ib->asize - ib->size, in)) > 0) { 329 | ib->size += ret; 330 | bufgrow(ib, ib->size + READ_UNIT); } 331 | if (in != stdin) fclose(in); 332 | 333 | to_man.opaque = &man_metadata; 334 | /* performing markdown to man */ 335 | ob = bufnew(OUTPUT_UNIT); 336 | markdown(ob, ib, &to_man); 337 | 338 | /* writing the result to stdout */ 339 | ret = fwrite(ob->data, 1, ob->size, stdout); 340 | if (ret < ob->size) 341 | fprintf(stderr, "Warning: only %zu output byte written, " 342 | "out of %zu\n", 343 | ret, 344 | ob->size); 345 | 346 | /* cleanup */ 347 | bufrelease(ib); 348 | bufrelease(ob); 349 | return EXIT_SUCCESS; } 350 | 351 | /* vim: set filetype=c: */ 352 | -------------------------------------------------------------------------------- /ConsoleRenderer.cpp: -------------------------------------------------------------------------------- 1 | extern "C" 2 | { 3 | #include "libsoldout/markdown.h" 4 | } 5 | 6 | #include <stack> 7 | #include <iostream> 8 | #include <string> 9 | #include <cstring> 10 | 11 | #include "ColouredBuffer.hpp" 12 | #include "include/console-markdown/CMDMD.hpp" 13 | 14 | #include "include/console-markdown/CPP.hpp" 15 | #include "include/console-markdown/Pawn.hpp" 16 | 17 | #ifndef CONMD_WINDOWS 18 | #include <sys/ioctl.h> 19 | #include <unistd.h> 20 | #define sprintf_s sprintf 21 | #endif 22 | 23 | static int RenderMarkdown(struct buf * ob, char const * style, struct buf const * text) 24 | { 25 | if (text && text->size) 26 | { 27 | size_t 28 | len = strlen(style); 29 | bufput(ob, style, len); 30 | bufput(ob, text->data, text->size); 31 | BUFPUTSL(ob, "\x1B[0m"); 32 | return 1; 33 | } 34 | return 0; 35 | } 36 | 37 | struct console_data_s 38 | { 39 | size_t WindowWidth; 40 | size_t BufferWidth; 41 | int BulletPoint; 42 | }; 43 | 44 | /*static void 45 | rndr_blockquote(struct buf *ob, struct buf *text, void *opaque) 46 | { 47 | if (ob->size) 48 | { 49 | bufputc(ob, '\n'); 50 | } 51 | BUFPUTSL(ob, "<blockquote>\n"); 52 | if (text) 53 | { 54 | bufput(ob, text->data, text->size); 55 | } 56 | BUFPUTSL(ob, "</blockquote>\n"); 57 | }*/ 58 | 59 | static int 60 | rndr_codespan(struct buf *ob, struct buf *text, void *opaque) 61 | { 62 | if (text) 63 | { 64 | BUFPUTSL(ob, "\x1B[47;30m"); 65 | bufput(ob, text->data, text->size); 66 | BUFPUTSL(ob, "\x1B[0m"); 67 | } 68 | return 1; 69 | } 70 | 71 | static int 72 | rndr_double_emphasis(struct buf *ob, struct buf *text, char c, void *opaque) 73 | { 74 | // Double `*` or `_` is `__bold__`. 75 | #ifndef CONMD_WINDOWS 76 | return RenderMarkdown(ob, "\x1B[5;47;30m", text); 77 | #else 78 | return RenderMarkdown(ob, "\x1B[1m", text); 79 | #endif 80 | } 81 | 82 | static int 83 | rndr_emphasis(struct buf *ob, struct buf *text, char c, void *opaque) 84 | { 85 | // Single `*` or `_` is `*italic*`. 86 | #ifdef CONMD_WINDOWS 87 | return RenderMarkdown(ob, "\x1B[37;1m", text); 88 | #else 89 | return RenderMarkdown(ob, "\x1B[4m", text); 90 | #endif 91 | } 92 | 93 | void rndr_blockcode(struct buf *ob, struct buf *text, void *opaque) 94 | { 95 | // Just indent it all 4 spaces. 96 | if (text && text->size) 97 | { 98 | std::string 99 | code(text->data, text->size); 100 | 101 | std::string::size_type 102 | lastPos = 0, 103 | findPos; 104 | 105 | size_t 106 | windowWidth = reinterpret_cast<console_data_s *>(opaque)->WindowWidth; 107 | bool 108 | nl = windowWidth < reinterpret_cast<console_data_s *>(opaque)->BufferWidth; 109 | if (ob->size) 110 | { 111 | bufputc(ob, '\n'); 112 | } 113 | while (std::string::npos != (findPos = code.find("\n", lastPos))) 114 | { 115 | BUFPUTSL(ob, "\x1B[0m \x1B[47;30m"); 116 | bufput(ob, code.c_str() + lastPos, findPos - lastPos); 117 | if (findPos - lastPos < windowWidth - 4) 118 | { 119 | bufputcn(ob, ' ', windowWidth - 4 - findPos + lastPos); 120 | if (nl) 121 | { 122 | bufputc(ob, '\n'); 123 | } 124 | } 125 | else 126 | { 127 | bufputc(ob, '\n'); 128 | } 129 | lastPos = findPos + 1; 130 | } 131 | bufput(ob, code.c_str() + lastPos, code.length() - lastPos); 132 | BUFPUTSL(ob, "\x1B[0m\n"); 133 | } 134 | } 135 | 136 | void rndr_fencedcode(struct buf *ob, struct buf *text, char *name, size_t namelen, void *opaque) 137 | { 138 | if (!text || !text->size) 139 | { 140 | return; 141 | } 142 | if (namelen) 143 | { 144 | if (!strncmp(name, "cpp", 3)) 145 | { 146 | if (ob->size) 147 | { 148 | bufputc(ob, '\n'); 149 | } 150 | std::string 151 | out = cmdmd::CPP(std::string(text->data, text->size)); 152 | bufput(ob, out.c_str(), out.length()); 153 | bufputc(ob, '\n'); 154 | return; 155 | } 156 | else if (!strncmp(name, "pawn", 3)) 157 | { 158 | if (ob->size) 159 | { 160 | bufputc(ob, '\n'); 161 | } 162 | std::string 163 | out = cmdmd::Pawn(std::string(text->data, text->size)); 164 | bufput(ob, out.c_str(), out.length()); 165 | bufputc(ob, '\n'); 166 | return; 167 | } 168 | } 169 | rndr_blockcode(ob, text, opaque); 170 | } 171 | 172 | static void 173 | rndr_header(struct buf *ob, struct buf *text, int level, void *opaque) 174 | { 175 | if (text && text->size) 176 | { 177 | if (ob->size) 178 | { 179 | bufputc(ob, '\n'); 180 | } 181 | size_t 182 | width = reinterpret_cast<console_data_s *>(opaque)->WindowWidth, 183 | len = text->size; 184 | if (len > width - 2) 185 | { 186 | len = width - 2; 187 | } 188 | switch (level) 189 | { 190 | case 1: 191 | BUFPUTSL(ob, "\x1B[42;5;30m "); 192 | bufput(ob, text->data, text->size); 193 | BUFPUTSL(ob, " \x1B[0;49m\n\x1B[42;5;30m"); 194 | bufputcn(ob, '=', len + 2); 195 | BUFPUTSL(ob, "\x1B[0m\n\x1B[0m"); 196 | break; 197 | case 2: 198 | BUFPUTSL(ob, "\x1B[32;1m "); 199 | bufput(ob, text->data, text->size); 200 | BUFPUTSL(ob, "\n "); 201 | bufputcn(ob, '-', len); 202 | BUFPUTSL(ob, "\x1B[0m\n"); 203 | break; 204 | case 3: 205 | BUFPUTSL(ob, "\x1B[33;1m "); 206 | bufput(ob, text->data, text->size); 207 | BUFPUTSL(ob, "\x1B[0m\n"); 208 | break; 209 | case 4: 210 | BUFPUTSL(ob, "\x1B[31;1m "); 211 | bufput(ob, text->data, text->size); 212 | BUFPUTSL(ob, "\x1B[0m\n"); 213 | break; 214 | } 215 | } 216 | } 217 | 218 | /*static int 219 | rndr_link(struct buf *ob, struct buf *link, struct buf *title, struct buf *content, void *opaque) 220 | { 221 | BUFPUTSL(ob, "<a href=\""); 222 | if (link && link->size) 223 | { 224 | lus_attr_escape(ob, link->data, link->size); 225 | } 226 | if (title && title->size) 227 | { 228 | BUFPUTSL(ob, "\" title=\""); 229 | lus_attr_escape(ob, title->data, title->size); 230 | } 231 | BUFPUTSL(ob, "\">"); 232 | if (content && content->size) 233 | { 234 | bufput(ob, content->data, content->size); 235 | } 236 | BUFPUTSL(ob, "</a>"); 237 | return 1; 238 | }*/ 239 | 240 | static void 241 | rndr_list(struct buf *ob, struct buf *text, int flags, void *opaque) 242 | { 243 | if (text) 244 | { 245 | if (ob->size) 246 | { 247 | bufputc(ob, '\n'); 248 | } 249 | bufput(ob, text->data, text->size); 250 | reinterpret_cast<console_data_s *>(opaque)->BulletPoint = 0; 251 | } 252 | } 253 | 254 | static void 255 | rndr_listitem(struct buf *ob, struct buf *text, int flags, void *opaque) 256 | { 257 | char num[4]; 258 | if (text) 259 | { 260 | while (text->size && text->data[text->size - 1] == '\n') 261 | { 262 | text->size -= 1; 263 | } 264 | if (text->size) 265 | { 266 | if (flags & MKD_LIST_ORDERED) 267 | { 268 | BUFPUTSL(ob, " \x1B[35m"); 269 | sprintf_s(num, "%d", ++reinterpret_cast<console_data_s *>(opaque)->BulletPoint); 270 | bufput(ob, num, strlen(num)); 271 | BUFPUTSL(ob, ")\x1B[0m "); 272 | bufput(ob, text->data, text->size); 273 | bufputc(ob, '\n'); 274 | } 275 | else 276 | { 277 | BUFPUTSL(ob, " \x1B[35m*\x1B[0m "); 278 | bufput(ob, text->data, text->size); 279 | bufputc(ob, '\n'); 280 | } 281 | } 282 | } 283 | } 284 | 285 | static void 286 | rndr_normal_text(struct buf *ob, struct buf *text, void *opaque) 287 | { 288 | bufput(ob, text->data, text->size); 289 | } 290 | 291 | static void 292 | null_block(struct buf *ob, struct buf *text, void *opaque) 293 | { 294 | bufput(ob, text->data, text->size); 295 | } 296 | 297 | static int 298 | null_span(struct buf *ob, struct buf *text, void *opaque) 299 | { 300 | bufput(ob, text->data, text->size); 301 | return 1; 302 | } 303 | 304 | static void 305 | rndr_paragraph(struct buf *ob, struct buf *text, void *opaque) 306 | { 307 | if (ob->size) 308 | { 309 | bufputc(ob, '\n'); 310 | } 311 | if (text) 312 | { 313 | size_t 314 | width = reinterpret_cast<console_data_s *>(opaque)->WindowWidth, 315 | rem = width, 316 | len = text->size; 317 | bool 318 | nl = reinterpret_cast<console_data_s *>(opaque)->BufferWidth > width; 319 | char 320 | * start = text->data, 321 | * end; 322 | while ((end = (char *)memchr(start, '\n', len))) 323 | { 324 | size_t 325 | diff = end - start; 326 | len -= diff + 1; 327 | while (diff >= rem) 328 | { 329 | bufput(ob, start, rem); 330 | diff -= rem; 331 | start += rem; 332 | rem = width; 333 | if (nl) 334 | { 335 | bufputc(ob, '\n'); 336 | } 337 | } 338 | if (diff) 339 | { 340 | bufput(ob, start, diff); 341 | rem = width - diff; 342 | } 343 | start = end + 1; 344 | } 345 | if (len) 346 | { 347 | while (len >= rem) 348 | { 349 | bufput(ob, start, rem); 350 | len -= rem; 351 | start += rem; 352 | rem = width; 353 | if (nl) 354 | { 355 | bufputc(ob, '\n'); 356 | } 357 | } 358 | if (len) 359 | { 360 | bufput(ob, start, len); 361 | } 362 | } 363 | } 364 | bufputc(ob, '\n'); 365 | } 366 | 367 | /*static void 368 | rndr_raw_block(struct buf *ob, struct buf *text, void *opaque) 369 | { 370 | size_t org, sz; 371 | if (!text) 372 | { 373 | return; 374 | } 375 | sz = text->size; 376 | while (sz > 0 && text->data[sz - 1] == '\n') 377 | { 378 | sz -= 1; 379 | } 380 | org = 0; 381 | while (org < sz && text->data[org] == '\n') 382 | { 383 | org += 1; 384 | } 385 | if (org >= sz) 386 | { 387 | return; 388 | } 389 | if (ob->size) 390 | { 391 | bufputc(ob, '\n'); 392 | } 393 | bufput(ob, text->data + org, sz - org); 394 | bufputc(ob, '\n'); 395 | } 396 | 397 | static int 398 | rndr_raw_inline(struct buf *ob, struct buf *text, void *opaque) 399 | { 400 | bufput(ob, text->data, text->size); 401 | return 1; 402 | }*/ 403 | 404 | static int 405 | rndr_linebreak(struct buf *ob, void *opaque) 406 | { 407 | bufputc(ob, '\n'); 408 | return 1; 409 | } 410 | 411 | void rndr_hrule(struct buf *ob, void *opaque) 412 | { 413 | if (ob->size) 414 | { 415 | bufputc(ob, '\n'); 416 | } 417 | size_t 418 | len = reinterpret_cast<console_data_s *>(opaque)->WindowWidth; 419 | BUFPUTSL(ob, "\x1B[44;34m"); 420 | bufputcn(ob, ' ', len); 421 | BUFPUTSL(ob, "\x1B[0m\n"); 422 | } 423 | 424 | /* exported renderer structures */ 425 | std::string cmdmd::Render(std::string const & input) 426 | { 427 | return cmdmd::Render(input.c_str(), input.length()); 428 | } 429 | 430 | std::string cmdmd::Render(char const * input) 431 | { 432 | return cmdmd::Render(input, strlen(input)); 433 | } 434 | 435 | std::string cmdmd::Render(char const * input, size_t len) 436 | { 437 | static struct mkd_renderer 438 | consoleRenderer = 439 | { 440 | NULL, 441 | NULL, 442 | 443 | rndr_blockcode,//NULL, 444 | rndr_fencedcode, 445 | null_block,//NULL, //discount_blockquote, 446 | null_block,//NULL, //rndr_raw_block, 447 | rndr_header, 448 | rndr_hrule, 449 | rndr_list, 450 | rndr_listitem, 451 | rndr_paragraph, 452 | NULL, //discount_table, 453 | NULL, //discount_table_cell, 454 | NULL, //discount_table_row, 455 | 456 | NULL, 457 | rndr_codespan, 458 | rndr_double_emphasis, 459 | rndr_emphasis, 460 | NULL, //html_discount_image, 461 | rndr_linebreak, 462 | NULL, //discount_link, 463 | null_span,//NULL, //rndr_raw_inline, 464 | NULL, //rndr_triple_emphasis, 465 | 466 | NULL, 467 | NULL, 468 | 469 | 64, 470 | "*_", 471 | NULL 472 | }; 473 | buf 474 | * ob = bufnew(64), 475 | ib = { 476 | const_cast<char *>(input), 477 | len, 478 | 0, 479 | 0, 480 | 0xFFFF 481 | }; 482 | #ifdef CONMD_WINDOWS 483 | CONSOLE_SCREEN_BUFFER_INFO 484 | console; 485 | size_t 486 | windowWidth = 80, 487 | bufferWidth = 80; 488 | if (GetConsoleScreenBufferInfo(GetStdHandle(STD_OUTPUT_HANDLE), &console)) 489 | { 490 | // Use the console window size, not the line buffer size. 491 | windowWidth = console.srWindow.Right - console.srWindow.Left + 1; 492 | bufferWidth = console.dwSize.X; 493 | } 494 | console_data_s 495 | data = { windowWidth, bufferWidth, 0 }; 496 | #else 497 | struct winsize w; 498 | ioctl(STDOUT_FILENO, TIOCGWINSZ, &w); 499 | 500 | console_data_s 501 | data = { w.ws_col, w.ws_col, 0 }; 502 | #endif 503 | consoleRenderer.opaque = reinterpret_cast<void *>(&data); 504 | markdown(ob, &ib, &consoleRenderer); 505 | 506 | std::string 507 | ret(ob->data, ob->size); 508 | 509 | bufrelease(ob); 510 | 511 | return ret; 512 | } 513 | 514 | void cmdmd::Init() 515 | { 516 | ColouredBuffer<char>::StandardInstall(); 517 | ColouredBuffer<wchar_t>::StandardInstall(); 518 | } 519 | 520 | std::string 521 | cmdmd:: 522 | Literals:: 523 | operator "" _cmdmd(char const * s, size_t len) 524 | { 525 | return ::cmdmd::Render(s, len); 526 | } 527 | 528 | std::string 529 | cmdmd:: 530 | Literals:: 531 | operator "" _md(char const * s, size_t len) 532 | { 533 | return ::cmdmd::Render(s, len); 534 | } 535 | 536 | -------------------------------------------------------------------------------- /libsoldout/mkd2latex.c: -------------------------------------------------------------------------------- 1 | /* mkd2latex.c - LaTeX-formatted output from markdown text */ 2 | 3 | /* 4 | * Copyright (c) 2009, Baptiste Daroussin, Natacha Porté, and Michael Huang 5 | * 6 | * Permission to use, copy, modify, and distribute this software for any 7 | * purpose with or without fee is hereby granted, provided that the above 8 | * copyright notice and this permission notice appear in all copies. 9 | * 10 | * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES 11 | * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF 12 | * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR 13 | * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES 14 | * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN 15 | * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF 16 | * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. 17 | */ 18 | 19 | /* 20 | * Links require the hyperref package, and images require the graphicx 21 | * package. 22 | */ 23 | 24 | #include "markdown.h" 25 | 26 | #include <stdio.h> 27 | #include <stdlib.h> 28 | #include <errno.h> 29 | #include <string.h> 30 | 31 | #define READ_UNIT 1024 32 | #define OUTPUT_UNIT 64 33 | 34 | /********************* 35 | * ENTITY CONVERSION * 36 | *********************/ 37 | 38 | struct str_pair { 39 | const char *entity; 40 | const char *latex; }; 41 | 42 | static struct str_pair entity_latex[] = { 43 | { "Æ", "\\AE{}" }, 44 | { "Á", "\\'A" }, 45 | { "Â", "\\^A" }, 46 | { "À", "\\`A" }, 47 | { "Å", "\\AA{}" }, 48 | { "Ã", "\\~A" }, 49 | { "Ä", "\\\"A" }, 50 | { "Ç", "\\c{C}" }, 51 | // { "Ð", "\\DH{}" }, // not available in OT1 52 | { "É", "\\'E" }, 53 | { "Ê", "\\^E" }, 54 | { "È", "\\`E" }, 55 | { "Ë", "\\\"E" }, 56 | { "Í", "\\'I" }, 57 | { "Î", "\\^I" }, 58 | { "Ì", "\\`I" }, 59 | { "Ï", "\\\"I" }, 60 | { "Ñ", "\\~N" }, 61 | { "Ó", "\\'O" }, 62 | { "Ô", "\\^O" }, 63 | { "Ò", "\\`O" }, 64 | { "Ø", "\\O{}" }, 65 | { "Õ", "\\~O" }, 66 | { "Ö", "\\\"O" }, 67 | // { "Þ", "\\TH{}" }, // not available in OT1 68 | { "Ú", "\\'U" }, 69 | { "Û", "\\^U" }, 70 | { "Ù", "\\`U" }, 71 | { "Ü", "\\\"U" }, 72 | { "&Ygrave;", "\\`Y" }, 73 | { "á", "\\'a" }, 74 | { "â", "\\^a" }, 75 | // { "´", "\\textasciiacute{}" }, // requires textcomp 76 | { "æ", "\\ae{}" }, 77 | { "à", "\\`a" }, 78 | { "&", "\\&" }, 79 | { "'", "'" }, 80 | { "å", "\\aa{}" }, 81 | { "ã", "\\~a" }, 82 | { "ä", "\\\"a" }, 83 | // { "¦", "\\textbrokenbar{}" }, // requires textcomp 84 | { "ç", "\\c{c}" }, 85 | { "¸", "\\c{}" }, 86 | // { "¢", "\\textcent{}" }, // requires textcomp 87 | { "©", "\\copyright{}" }, 88 | // { "¤", "\\textcurrency{}" }, // requires textcomp 89 | // { "°", "\\textdegree{}" }, // requires textcomp 90 | // { "÷", "\\textdiv{}" }, // requires textcomp 91 | { "é", "\\'e" }, 92 | { "ê", "\\^e" }, 93 | { "è", "\\`e" }, 94 | // { "ð", "\\dh{}" }. // not available in OT1 95 | { "ë", "\\\"e" }, 96 | // { "½", "\\textonehalf{}" }, // requires textcomp 97 | // { "¼", "\\textonequarter{}" }, // requires textcomp 98 | // { "¾", "\\textthreequarter{}" }, // requires textcomp 99 | { ">", "$>$" }, 100 | { "í", "\\'\\i{}" }, 101 | { "î", "\\^\\i{}" }, 102 | { "¡", "\\textexclamdown{}" }, 103 | { "ì", "\\`\\i{}" }, 104 | { "¿", "\\" }, 105 | { "ï", "\\\"\\i{}" }, 106 | // { "«", "\\guillemotleft{}" }, // not available in OT1 107 | { "<", "$<$" }, 108 | // { "¯", "\\textasciimacaron{}" }, // requires textcomp 109 | // { "µ", "\\textmu{}"}, // requires textcomp 110 | { "·", "\\textperiodcentered{}" }, 111 | { " ", "~" }, 112 | // { "¬", "\\textlnot{}" }, // requires textcomp 113 | { "ñ", "\\~n" }, 114 | { "ó", "\\'o" }, 115 | { "ô", "\\^o" }, 116 | { "ò", "\\`o" }, 117 | { "ª", "\\textordfeminine{}" }, 118 | { "º", "\\textordmasculine{}" }, 119 | { "ø", "\\o{}" }, 120 | { "õ", "\\~o" }, 121 | { "ö", "\\\"o" }, 122 | { "¶", "\\P{}" }, 123 | // { "±", "\\textpm{}" }, // requires textcomp 124 | { "£", "\\textsterling{}" }, 125 | { """, "\"" }, 126 | // { "»", "\\guillemotright{}" }, // not available in OT1 127 | { "®", "\\textregistered{}" }, 128 | { "§", "\\S{}" }, 129 | { "", "\\-" }, 130 | // { "¹", "\\textonesuperior{}" }, // requires textcomp 131 | // { "²", "\\texttwosuperior{}" }, // requires textcomp 132 | // { "³", "\\textthreesuperior{}" }, // requires textcomp 133 | { "ß", "\\ss{}" }, 134 | // { "þ", "\\th{}" }, // not available in OT1 135 | // { "×", "\\texttimes{}" }, // requires textcomp 136 | { "ú", "\\'u" }, 137 | { "û", "\\^u" }, 138 | { "ù", "\\`u" }, 139 | // { "¨", "\\textasciidieresis{}" }, // requires textcomp 140 | { "ü", "\\\"u" }, 141 | { "ý", "\\'y" }, 142 | // { "¥", "\\textyen{}" }, // requires textcomp 143 | { "ÿ", "\\\"y" }, 144 | }; 145 | 146 | static int cmp_entity(const void *key, const void *element) { 147 | const struct str_pair *pair = element; 148 | const struct buf *entity = key; 149 | return bufcmps(entity, pair->entity); } 150 | 151 | static const char *entity2latex(const struct buf *entity) { 152 | const struct str_pair *pair; 153 | pair = bsearch(entity, entity_latex, 154 | sizeof entity_latex / sizeof *entity_latex, 155 | sizeof *entity_latex, 156 | &cmp_entity); 157 | return pair ? pair->latex : 0; } 158 | 159 | 160 | 161 | /****************************** 162 | * MARKDOWN TO LATEX RENDERER * 163 | ******************************/ 164 | 165 | static void 166 | latex_text_escape(struct buf *ob, char *src, size_t size) { 167 | size_t i = 0, org; 168 | while (i < size) { 169 | /* copying directly unescaped characters */ 170 | org = i; 171 | while (i < size && src[i] != '&' && src[i] != '%' 172 | && src[i] != '$' && src[i] != '#' && src[i] != '_' 173 | && src[i] != '{' && src[i] != '}' && src[i] != '~' 174 | && src[i] != '^' && src[i] != '\\' && src[i] != '<' 175 | && src[i] != '>') 176 | i += 1; 177 | if (i > org) bufput(ob, src + org, i - org); 178 | 179 | /* escaping */ 180 | if (i >= size) break; 181 | else if (src[i] == '&') BUFPUTSL(ob, "\\&"); 182 | else if (src[i] == '%') BUFPUTSL(ob, "\\%"); 183 | else if (src[i] == '$') BUFPUTSL(ob, "\\$"); 184 | else if (src[i] == '#') BUFPUTSL(ob, "\\#"); 185 | else if (src[i] == '_') BUFPUTSL(ob, "\\_"); 186 | else if (src[i] == '{') BUFPUTSL(ob, "\\{"); 187 | else if (src[i] == '}') BUFPUTSL(ob, "\\}"); 188 | else if (src[i] == '<') BUFPUTSL(ob, "$<$"); 189 | else if (src[i] == '>') BUFPUTSL(ob, "$<$"); 190 | else if (src[i] == '~') BUFPUTSL(ob, "\\textasciitilde{}"); 191 | else if (src[i] == '^') BUFPUTSL(ob, "\\textasciicircum{}"); 192 | else if (src[i] == '\\') BUFPUTSL(ob, "\\textbackslash{}"); 193 | i += 1; } } 194 | 195 | static void 196 | latex_prolog(struct buf *ob, void *opaque) { 197 | BUFPUTSL(ob, 198 | "\\documentclass{article}\n" 199 | "\\usepackage{hyperref}\n" 200 | "\\usepackage{graphicx}\n" 201 | "\\begin{document}\n"); } 202 | 203 | static void 204 | latex_epilog(struct buf *ob, void *opaque) { 205 | BUFPUTSL(ob, "\n\\end{document}\n"); } 206 | 207 | static int 208 | latex_autolink(struct buf *ob, struct buf *link, enum mkd_autolink type, 209 | void *opaque) { 210 | if (!link || !link->size) return 0; 211 | BUFPUTSL(ob, "\\href{"); 212 | if (type == MKDA_IMPLICIT_EMAIL) BUFPUTSL(ob, "mailto:"); 213 | bufput(ob, link->data, link->size); 214 | BUFPUTSL(ob, "}{"); 215 | if (type == MKDA_EXPLICIT_EMAIL && link->size > 7) 216 | latex_text_escape(ob, link->data + 7, link->size - 7); 217 | else latex_text_escape(ob, link->data, link->size); 218 | BUFPUTSL(ob, "}"); 219 | return 1; } 220 | 221 | static int 222 | latex_link(struct buf *ob, struct buf *link, struct buf *title, 223 | struct buf *content, void *opaque) { 224 | BUFPUTSL(ob, "\\href{"); 225 | if (link && link->size) bufput(ob, link->data, link->size); 226 | BUFPUTSL(ob, "}{"); 227 | if (content && content->size) 228 | bufput(ob, content->data, content->size); 229 | BUFPUTSL(ob, "}"); 230 | return 1; } 231 | 232 | static int 233 | latex_image(struct buf *ob, struct buf *link, struct buf *title, 234 | struct buf *alt, void *opaque) { 235 | if (!link || !link->size) return 0; 236 | BUFPUTSL(ob, "\\includegraphics{"); 237 | bufput(ob, link->data, link->size); 238 | BUFPUTSL(ob, "}"); 239 | return 1; } 240 | 241 | static void 242 | latex_blockcode(struct buf *ob, struct buf *text, void *opaque) { 243 | if (ob->size) bufputc(ob, '\n'); 244 | BUFPUTSL(ob, "\\begin{verbatim}\n"); 245 | if (text) bufput(ob, text->data, text->size); 246 | BUFPUTSL(ob, "\\end{verbatim}\n"); } 247 | 248 | static void 249 | latex_blockquote(struct buf *ob, struct buf *text, void *opaque) { 250 | if (ob->size) bufputc(ob, '\n'); 251 | BUFPUTSL(ob, "\\begin{quote}\n"); 252 | if (text) bufput(ob, text->data, text->size); 253 | BUFPUTSL(ob, "\\end{quote}\n"); } 254 | 255 | static int 256 | latex_codespan(struct buf *ob, struct buf *text, void *opaque) { 257 | BUFPUTSL(ob, "\\texttt{"); 258 | if (text) latex_text_escape(ob, text->data, text->size); 259 | BUFPUTSL(ob, "}"); 260 | return 1; } 261 | 262 | static void 263 | latex_header(struct buf *ob, struct buf *text, int level, void *opaque) { 264 | if (ob->size) bufputc(ob, '\n'); 265 | switch(level) { 266 | case 1: 267 | BUFPUTSL(ob,"\\section{"); 268 | break; 269 | case 2: 270 | BUFPUTSL(ob, "\\subsection{"); 271 | break; 272 | case 3: 273 | BUFPUTSL(ob, "\\subsubsection{"); 274 | break; 275 | default: 276 | fprintf(stderr, "Warning: ignoring header level %d\n", 277 | level); 278 | } 279 | if (text) bufput(ob, text->data, text->size); 280 | if (level >= 1 && level <= 3) BUFPUTSL(ob, "}\n"); 281 | } 282 | 283 | static int 284 | latex_double_emphasis(struct buf *ob, struct buf *text, char c, void *opaque) { 285 | if (!text || !text->size) return 0; 286 | BUFPUTSL(ob, "\\textbf{"); 287 | bufput(ob, text->data, text->size); 288 | BUFPUTSL(ob, "}"); 289 | return 1; } 290 | 291 | static int 292 | latex_emphasis(struct buf *ob, struct buf *text, char c, void *opaque) { 293 | if (!text || !text->size) return 0; 294 | BUFPUTSL(ob, "\\emph{"); 295 | if (text) bufput(ob, text->data, text->size); 296 | BUFPUTSL(ob, "}"); 297 | return 1; } 298 | 299 | static int 300 | latex_linebreak(struct buf *ob, void *opaque) { 301 | BUFPUTSL(ob, "\\\\"); 302 | return 1; } 303 | 304 | static void 305 | latex_paragraph(struct buf *ob, struct buf *text, void *opaque) { 306 | if (ob->size) bufputc(ob, '\n'); 307 | if (text) bufput(ob, text->data, text->size); 308 | BUFPUTSL(ob, "\n"); } 309 | 310 | static void 311 | latex_list(struct buf *ob, struct buf *text, int flags, void *opaque) { 312 | if (ob->size) bufputc(ob, '\n'); 313 | if (flags & MKD_LIST_ORDERED) 314 | BUFPUTSL(ob, "\\begin{enumerate}\n"); 315 | else 316 | BUFPUTSL(ob, "\\begin{itemize}\n"); 317 | if (text) bufput(ob, text->data, text->size); 318 | if (flags & MKD_LIST_ORDERED) 319 | BUFPUTSL(ob, "\\end{enumerate}\n"); 320 | else 321 | BUFPUTSL(ob, "\\end{itemize}\n"); } 322 | 323 | static void 324 | latex_listitem(struct buf *ob, struct buf *text, int flags, void *opaque) { 325 | BUFPUTSL(ob, "\\item "); 326 | if (text) { 327 | while (text->size && text->data[text->size - 1] == '\n') 328 | text->size -= 1; 329 | bufput(ob, text->data, text->size); } 330 | BUFPUTSL(ob, "\n"); } 331 | 332 | static void 333 | latex_hrule(struct buf *ob, void *opaque) { 334 | if (ob->size) bufputc(ob, '\n'); 335 | BUFPUTSL(ob, "\\hrule"); } 336 | 337 | static int 338 | latex_triple_emphasis(struct buf *ob, struct buf *text, char c, void *opaque) { 339 | if (!text || !text->size) return 0; 340 | BUFPUTSL(ob, "\\textbf{\\emph{"); 341 | bufput(ob, text->data, text->size); 342 | BUFPUTSL(ob, "}}"); 343 | return 1; } 344 | 345 | static void 346 | latex_entity(struct buf *ob, struct buf *entity, void *opaque) { 347 | const char *rendered = entity2latex(entity); 348 | if (rendered) 349 | bufputs(ob, rendered); 350 | else { 351 | BUFPUTSL(ob, "\\texttt{"); 352 | bufput(ob, entity->data, entity->size); 353 | BUFPUTSL(ob, "}"); } } 354 | 355 | static void 356 | latex_normal_text(struct buf *ob, struct buf *text, void *opaque) { 357 | if (text) latex_text_escape(ob, text->data, text->size); } 358 | 359 | 360 | /* renderer structure */ 361 | static struct mkd_renderer to_latex = { 362 | /* document-level callbacks */ 363 | latex_prolog, 364 | latex_epilog, 365 | 366 | /* block-level callbacks */ 367 | latex_blockcode, 368 | NULL, 369 | latex_blockquote, 370 | latex_blockcode, 371 | latex_header, 372 | latex_hrule, 373 | latex_list, 374 | latex_listitem, 375 | latex_paragraph, 376 | NULL, 377 | NULL, 378 | NULL, 379 | 380 | /* span-level callbacks */ 381 | latex_autolink, 382 | latex_codespan, 383 | latex_double_emphasis, 384 | latex_emphasis, 385 | latex_image, 386 | latex_linebreak, 387 | latex_link, 388 | latex_codespan, 389 | latex_triple_emphasis, 390 | 391 | /* low-level callbacks */ 392 | latex_entity, 393 | latex_normal_text, 394 | 395 | /* renderer data */ 396 | 64, 397 | "*_", 398 | NULL }; 399 | 400 | 401 | 402 | /***************** 403 | * MAIN FUNCTION * 404 | *****************/ 405 | 406 | /* main • main function, interfacing STDIO with the parser */ 407 | int 408 | main(int argc, char **argv) { 409 | struct buf *ib, *ob; 410 | size_t ret; 411 | FILE *in = stdin; 412 | 413 | /* opening the file if given from the command line */ 414 | if (argc > 1) { 415 | in = fopen(argv[1], "r"); 416 | if (!in) { 417 | fprintf(stderr,"Unable to open input file \"%s\": %s\n", 418 | argv[1], strerror(errno)); 419 | return 1; } } 420 | 421 | /* reading everything */ 422 | ib = bufnew(READ_UNIT); 423 | bufgrow(ib, READ_UNIT); 424 | while ((ret = fread(ib->data + ib->size, 1, 425 | ib->asize - ib->size, in)) > 0) { 426 | ib->size += ret; 427 | bufgrow(ib, ib->size + READ_UNIT); } 428 | if (in != stdin) fclose(in); 429 | 430 | /* performing markdown to LaTeX */ 431 | ob = bufnew(OUTPUT_UNIT); 432 | markdown(ob, ib, &to_latex); 433 | 434 | /* writing the result to stdout */ 435 | ret = fwrite(ob->data, 1, ob->size, stdout); 436 | if (ret < ob->size) 437 | fprintf(stderr, "Warning: only %zu output byte written, " 438 | "out of %zu\n", 439 | ret, 440 | ob->size); 441 | 442 | /* cleanup */ 443 | bufrelease(ib); 444 | bufrelease(ob); 445 | return 0; } 446 | 447 | /* vim: set filetype=c: */ 448 | -------------------------------------------------------------------------------- /libsoldout/renderers.c: -------------------------------------------------------------------------------- 1 | /* renderers.h - example markdown renderers */ 2 | 3 | /* 4 | * Copyright (c) 2009, Natacha Porté 5 | * 6 | * Permission to use, copy, modify, and distribute this software for any 7 | * purpose with or without fee is hereby granted, provided that the above 8 | * copyright notice and this permission notice appear in all copies. 9 | * 10 | * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES 11 | * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF 12 | * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR 13 | * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES 14 | * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN 15 | * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF 16 | * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. 17 | */ 18 | 19 | #include "renderers.h" 20 | 21 | #include <strings.h> 22 | 23 | 24 | /***************************** 25 | * EXPORTED HELPER FUNCTIONS * 26 | *****************************/ 27 | 28 | /* lus_attr_escape • copy the buffer entity-escaping '<', '>', '&' and '"' */ 29 | void 30 | lus_attr_escape(struct buf *ob, const char *src, size_t size) { 31 | size_t i = 0, org; 32 | while (i < size) { 33 | /* copying directly unescaped characters */ 34 | org = i; 35 | while (i < size && src[i] != '<' && src[i] != '>' 36 | && src[i] != '&' && src[i] != '"') 37 | i += 1; 38 | if (i > org) bufput(ob, src + org, i - org); 39 | 40 | /* escaping */ 41 | if (i >= size) break; 42 | else if (src[i] == '<') BUFPUTSL(ob, "<"); 43 | else if (src[i] == '>') BUFPUTSL(ob, ">"); 44 | else if (src[i] == '&') BUFPUTSL(ob, "&"); 45 | else if (src[i] == '"') BUFPUTSL(ob, """); 46 | i += 1; } } 47 | 48 | 49 | /* lus_body_escape • copy the buffer entity-escaping '<', '>' and '&' */ 50 | void 51 | lus_body_escape(struct buf *ob, const char *src, size_t size) { 52 | size_t i = 0, org; 53 | while (i < size) { 54 | /* copying directly unescaped characters */ 55 | org = i; 56 | while (i < size && src[i] != '<' && src[i] != '>' 57 | && src[i] != '&') 58 | i += 1; 59 | if (i > org) bufput(ob, src + org, i - org); 60 | 61 | /* escaping */ 62 | if (i >= size) break; 63 | else if (src[i] == '<') BUFPUTSL(ob, "<"); 64 | else if (src[i] == '>') BUFPUTSL(ob, ">"); 65 | else if (src[i] == '&') BUFPUTSL(ob, "&"); 66 | i += 1; } } 67 | 68 | 69 | 70 | /******************** 71 | * GENERIC RENDERER * 72 | ********************/ 73 | 74 | static int 75 | rndr_autolink(struct buf *ob, struct buf *link, enum mkd_autolink type, 76 | void *opaque) { 77 | if (!link || !link->size) return 0; 78 | BUFPUTSL(ob, "<a href=\""); 79 | if (type == MKDA_IMPLICIT_EMAIL) BUFPUTSL(ob, "mailto:"); 80 | lus_attr_escape(ob, link->data, link->size); 81 | BUFPUTSL(ob, "\">"); 82 | if (type == MKDA_EXPLICIT_EMAIL && link->size > 7) 83 | lus_body_escape(ob, link->data + 7, link->size - 7); 84 | else lus_body_escape(ob, link->data, link->size); 85 | BUFPUTSL(ob, "</a>"); 86 | return 1; } 87 | 88 | static void 89 | rndr_blockcode(struct buf *ob, struct buf *text, void *opaque) { 90 | if (ob->size) bufputc(ob, '\n'); 91 | BUFPUTSL(ob, "<pre><code>"); 92 | if (text) lus_body_escape(ob, text->data, text->size); 93 | BUFPUTSL(ob, "</code></pre>\n"); } 94 | 95 | static void 96 | rndr_blockquote(struct buf *ob, struct buf *text, void *opaque) { 97 | if (ob->size) bufputc(ob, '\n'); 98 | BUFPUTSL(ob, "<blockquote>\n"); 99 | if (text) bufput(ob, text->data, text->size); 100 | BUFPUTSL(ob, "</blockquote>\n"); } 101 | 102 | static int 103 | rndr_codespan(struct buf *ob, struct buf *text, void *opaque) { 104 | BUFPUTSL(ob, "<code>"); 105 | if (text) lus_body_escape(ob, text->data, text->size); 106 | BUFPUTSL(ob, "</code>"); 107 | return 1; } 108 | 109 | static int 110 | rndr_double_emphasis(struct buf *ob, struct buf *text, char c, void *opaque) { 111 | if (!text || !text->size) return 0; 112 | BUFPUTSL(ob, "<strong>"); 113 | bufput(ob, text->data, text->size); 114 | BUFPUTSL(ob, "</strong>"); 115 | return 1; } 116 | 117 | static int 118 | rndr_emphasis(struct buf *ob, struct buf *text, char c, void *opaque) { 119 | if (!text || !text->size) return 0; 120 | BUFPUTSL(ob, "<em>"); 121 | if (text) bufput(ob, text->data, text->size); 122 | BUFPUTSL(ob, "</em>"); 123 | return 1; } 124 | 125 | static void 126 | rndr_header(struct buf *ob, struct buf *text, int level, void *opaque) { 127 | if (ob->size) bufputc(ob, '\n'); 128 | bufprintf(ob, "<h%d>", level); 129 | if (text) bufput(ob, text->data, text->size); 130 | bufprintf(ob, "</h%d>\n", level); } 131 | 132 | static int 133 | rndr_link(struct buf *ob, struct buf *link, struct buf *title, 134 | struct buf *content, void *opaque) { 135 | BUFPUTSL(ob, "<a href=\""); 136 | if (link && link->size) lus_attr_escape(ob, link->data, link->size); 137 | if (title && title->size) { 138 | BUFPUTSL(ob, "\" title=\""); 139 | lus_attr_escape(ob, title->data, title->size); } 140 | BUFPUTSL(ob, "\">"); 141 | if (content && content->size) bufput(ob, content->data, content->size); 142 | BUFPUTSL(ob, "</a>"); 143 | return 1; } 144 | 145 | static void 146 | rndr_list(struct buf *ob, struct buf *text, int flags, void *opaque) { 147 | if (ob->size) bufputc(ob, '\n'); 148 | bufput(ob, (flags & MKD_LIST_ORDERED) ? "<ol>\n" : "<ul>\n", 5); 149 | if (text) bufput(ob, text->data, text->size); 150 | bufput(ob, (flags & MKD_LIST_ORDERED) ? "</ol>\n" : "</ul>\n", 6); } 151 | 152 | static void 153 | rndr_listitem(struct buf *ob, struct buf *text, int flags, void *opaque) { 154 | BUFPUTSL(ob, "<li>"); 155 | if (text) { 156 | while (text->size && text->data[text->size - 1] == '\n') 157 | text->size -= 1; 158 | bufput(ob, text->data, text->size); } 159 | BUFPUTSL(ob, "</li>\n"); } 160 | 161 | static void 162 | rndr_normal_text(struct buf *ob, struct buf *text, void *opaque) { 163 | if (text) lus_body_escape(ob, text->data, text->size); } 164 | 165 | static void 166 | rndr_paragraph(struct buf *ob, struct buf *text, void *opaque) { 167 | if (ob->size) bufputc(ob, '\n'); 168 | BUFPUTSL(ob, "<p>"); 169 | if (text) bufput(ob, text->data, text->size); 170 | BUFPUTSL(ob, "</p>\n"); } 171 | 172 | static void 173 | rndr_raw_block(struct buf *ob, struct buf *text, void *opaque) { 174 | size_t org, sz; 175 | if (!text) return; 176 | sz = text->size; 177 | while (sz > 0 && text->data[sz - 1] == '\n') sz -= 1; 178 | org = 0; 179 | while (org < sz && text->data[org] == '\n') org += 1; 180 | if (org >= sz) return; 181 | if (ob->size) bufputc(ob, '\n'); 182 | bufput(ob, text->data + org, sz - org); 183 | bufputc(ob, '\n'); } 184 | 185 | static int 186 | rndr_raw_inline(struct buf *ob, struct buf *text, void *opaque) { 187 | bufput(ob, text->data, text->size); 188 | return 1; } 189 | 190 | static int 191 | rndr_triple_emphasis(struct buf *ob, struct buf *text, char c, void *opaque) { 192 | if (!text || !text->size) return 0; 193 | BUFPUTSL(ob, "<strong><em>"); 194 | bufput(ob, text->data, text->size); 195 | BUFPUTSL(ob, "</em></strong>"); 196 | return 1; } 197 | 198 | 199 | 200 | /******************* 201 | * HTML 4 RENDERER * 202 | *******************/ 203 | 204 | static void 205 | html_hrule(struct buf *ob, void *opaque) { 206 | if (ob->size) bufputc(ob, '\n'); 207 | BUFPUTSL(ob, "<hr>\n"); } 208 | 209 | static int 210 | html_image(struct buf *ob, struct buf *link, struct buf *title, 211 | struct buf *alt, void *opaque) { 212 | if (!link || !link->size) return 0; 213 | BUFPUTSL(ob, "<img src=\""); 214 | lus_attr_escape(ob, link->data, link->size); 215 | BUFPUTSL(ob, "\" alt=\""); 216 | if (alt && alt->size) 217 | lus_attr_escape(ob, alt->data, alt->size); 218 | if (title && title->size) { 219 | BUFPUTSL(ob, "\" title=\""); 220 | lus_attr_escape(ob, title->data, title->size); } 221 | BUFPUTSL(ob, "\">"); 222 | return 1; } 223 | 224 | static int 225 | html_linebreak(struct buf *ob, void *opaque) { 226 | BUFPUTSL(ob, "<br>\n"); 227 | return 1; } 228 | 229 | 230 | /* exported renderer structure */ 231 | const struct mkd_renderer mkd_html = { 232 | NULL, 233 | NULL, 234 | 235 | rndr_blockcode, 236 | NULL, 237 | rndr_blockquote, 238 | rndr_raw_block, 239 | rndr_header, 240 | html_hrule, 241 | rndr_list, 242 | rndr_listitem, 243 | rndr_paragraph, 244 | NULL, 245 | NULL, 246 | NULL, 247 | 248 | rndr_autolink, 249 | rndr_codespan, 250 | rndr_double_emphasis, 251 | rndr_emphasis, 252 | html_image, 253 | html_linebreak, 254 | rndr_link, 255 | rndr_raw_inline, 256 | rndr_triple_emphasis, 257 | 258 | NULL, 259 | rndr_normal_text, 260 | 261 | 64, 262 | "*_", 263 | NULL }; 264 | 265 | 266 | 267 | /********************** 268 | * XHTML 1.0 RENDERER * 269 | **********************/ 270 | 271 | static void 272 | xhtml_hrule(struct buf *ob, void *opaque) { 273 | if (ob->size) bufputc(ob, '\n'); 274 | BUFPUTSL(ob, "<hr />\n"); } 275 | 276 | static int 277 | xhtml_image(struct buf *ob, struct buf *link, struct buf *title, 278 | struct buf *alt, void *opaque) { 279 | if (!link || !link->size) return 0; 280 | BUFPUTSL(ob, "<img src=\""); 281 | lus_attr_escape(ob, link->data, link->size); 282 | BUFPUTSL(ob, "\" alt=\""); 283 | if (alt && alt->size) 284 | lus_attr_escape(ob, alt->data, alt->size); 285 | if (title && title->size) { 286 | BUFPUTSL(ob, "\" title=\""); 287 | lus_attr_escape(ob, title->data, title->size); } 288 | BUFPUTSL(ob, "\" />"); 289 | return 1; } 290 | 291 | static int 292 | xhtml_linebreak(struct buf *ob, void *opaque) { 293 | BUFPUTSL(ob, "<br />\n"); 294 | return 1; } 295 | 296 | 297 | /* exported renderer structure */ 298 | const struct mkd_renderer mkd_xhtml = { 299 | NULL, 300 | NULL, 301 | 302 | rndr_blockcode, 303 | NULL, 304 | rndr_blockquote, 305 | rndr_raw_block, 306 | rndr_header, 307 | xhtml_hrule, 308 | rndr_list, 309 | rndr_listitem, 310 | rndr_paragraph, 311 | NULL, 312 | NULL, 313 | NULL, 314 | 315 | rndr_autolink, 316 | rndr_codespan, 317 | rndr_double_emphasis, 318 | rndr_emphasis, 319 | xhtml_image, 320 | xhtml_linebreak, 321 | rndr_link, 322 | rndr_raw_inline, 323 | rndr_triple_emphasis, 324 | 325 | NULL, 326 | rndr_normal_text, 327 | 328 | 64, 329 | "*_", 330 | NULL }; 331 | 332 | 333 | 334 | /********************** 335 | * DISCOUNT RENDERERS * 336 | **********************/ 337 | 338 | static int 339 | print_link_wxh(struct buf *ob, struct buf *link) { 340 | size_t eq, ex, end; 341 | if (link->size < 1) return 0; 342 | eq = link->size - 1; 343 | while (eq > 0 && (link->data[eq - 1] != ' ' || link->data[eq] != '=')) 344 | eq -= 1; 345 | if (!eq) return 0; 346 | ex = eq + 1; 347 | while (ex < link->size 348 | && link->data[ex] >= '0' && link->data[ex] <= '9') 349 | ex += 1; 350 | if (ex >= link->size || ex == eq + 1 || link->data[ex] != 'x') return 0; 351 | end = ex + 1; 352 | while (end < link->size 353 | && link->data[end] >= '0' && link->data[end] <= '9') 354 | end += 1; 355 | if (end == ex + 1) return 0; 356 | /* everything is fine, proceeding to actual printing */ 357 | lus_attr_escape(ob, link->data, eq - 1); 358 | BUFPUTSL(ob, "\" width="); 359 | bufput(ob, link->data + eq + 1, ex - eq - 1); 360 | BUFPUTSL(ob, " height="); 361 | bufput(ob, link->data + ex + 1, end - ex - 1); 362 | return 1; } 363 | 364 | static int 365 | discount_image(struct buf *ob, struct buf *link, struct buf *title, 366 | struct buf *alt, int xhtml) { 367 | if (!link || !link->size) return 0; 368 | BUFPUTSL(ob, "<img src=\""); 369 | if (!print_link_wxh(ob, link)) { 370 | lus_attr_escape(ob, link->data, link->size); 371 | bufputc(ob, '"'); } 372 | BUFPUTSL(ob, " alt=\""); 373 | if (alt && alt->size) 374 | lus_attr_escape(ob, alt->data, alt->size); 375 | if (title && title->size) { 376 | BUFPUTSL(ob, "\" title=\""); 377 | lus_attr_escape(ob, title->data, title->size); } 378 | bufputs(ob, xhtml ? "\" />" : "\">"); 379 | return 1; } 380 | 381 | static int 382 | html_discount_image(struct buf *ob, struct buf *link, struct buf *title, 383 | struct buf *alt, void *opaque) { 384 | return discount_image(ob, link, title, alt, 0); } 385 | 386 | static int 387 | xhtml_discount_image(struct buf *ob, struct buf *link, struct buf *title, 388 | struct buf *alt, void *opaque) { 389 | return discount_image(ob, link, title, alt, 1); } 390 | 391 | static int 392 | discount_link(struct buf *ob, struct buf *link, struct buf *title, 393 | struct buf *content, void *opaque) { 394 | if (!link) return rndr_link(ob, link, title, content, opaque); 395 | else if (link->size > 5 && !strncasecmp(link->data, "abbr:", 5)) { 396 | BUFPUTSL(ob, "<abbr title=\""); 397 | lus_attr_escape(ob, link->data + 5, link->size - 5); 398 | BUFPUTSL(ob, "\">"); 399 | bufput(ob, content->data, content->size); 400 | BUFPUTSL(ob, "</abbr>"); 401 | return 1; } 402 | else if (link->size > 6 && !strncasecmp(link->data, "class:", 6)) { 403 | BUFPUTSL(ob, "<span class=\""); 404 | lus_attr_escape(ob, link->data + 6, link->size - 6); 405 | BUFPUTSL(ob, "\">"); 406 | bufput(ob, content->data, content->size); 407 | BUFPUTSL(ob, "</span>"); 408 | return 1; } 409 | else if (link->size > 3 && !strncasecmp(link->data, "id:", 3)) { 410 | BUFPUTSL(ob, "<span id=\""); 411 | lus_attr_escape(ob, link->data + 3, link->size - 3); 412 | BUFPUTSL(ob, "\">"); 413 | bufput(ob, content->data, content->size); 414 | BUFPUTSL(ob, "</span>"); 415 | return 1; } 416 | else if (link->size > 4 && !strncasecmp(link->data, "raw:", 4)) { 417 | bufput(ob, link->data + 4, link->size - 4); 418 | return 1; } 419 | return rndr_link(ob, link, title, content, opaque); } 420 | 421 | static void 422 | discount_blockquote(struct buf *ob, struct buf *text, void *opaque) { 423 | size_t i = 5, size = text->size; 424 | char *data = text->data; 425 | if (text->size < 5 || strncasecmp(text->data, "<p>%", 4)) { 426 | rndr_blockquote(ob, text, opaque); 427 | return; } 428 | while (i < size && data[i] != '\n' && data[i] != '%') 429 | i += 1; 430 | if (i >= size || data[i] != '%') { 431 | rndr_blockquote(ob, text, opaque); 432 | return; } 433 | BUFPUTSL(ob, "<div class=\""); 434 | bufput(ob, text->data + 4, i - 4); 435 | BUFPUTSL(ob, "\"><p>"); 436 | i += 1; 437 | if (i + 4 >= text->size && !strncasecmp(text->data + i, "</p>", 4)) { 438 | size_t old_i = i; 439 | i += 4; 440 | while (i + 3 < text->size 441 | && (data[i] != '<' || data[i + 1] != 'p' || data[i + 2] != '>')) 442 | i += 1; 443 | if (i + 3 >= text->size) i = old_i; } 444 | bufput(ob, text->data + i, text->size - i); 445 | BUFPUTSL(ob, "</div>\n"); } 446 | 447 | static void 448 | discount_table(struct buf *ob, struct buf *head_row, struct buf *rows, 449 | void *opaque) { 450 | if (ob->size) bufputc(ob, '\n'); 451 | BUFPUTSL(ob, "<table>\n"); 452 | if (head_row) { 453 | BUFPUTSL(ob, "<thead>\n"); 454 | bufput(ob, head_row->data, head_row->size); 455 | BUFPUTSL(ob, "</thead>\n<tbody>\n"); } 456 | if (rows) 457 | bufput(ob, rows->data, rows->size); 458 | if (head_row) 459 | BUFPUTSL(ob, "</tbody>\n"); 460 | BUFPUTSL(ob, "</table>\n"); } 461 | 462 | static void 463 | discount_table_row(struct buf *ob, struct buf *cells, int flags, void *opaque){ 464 | (void)flags; 465 | BUFPUTSL(ob, " <tr>\n"); 466 | if (cells) bufput(ob, cells->data, cells->size); 467 | BUFPUTSL(ob, " </tr>\n"); } 468 | 469 | static void 470 | discount_table_cell(struct buf *ob, struct buf *text, int flags, void *opaque){ 471 | if (flags & MKD_CELL_HEAD) 472 | BUFPUTSL(ob, " <th"); 473 | else 474 | BUFPUTSL(ob, " <td"); 475 | switch (flags & MKD_CELL_ALIGN_MASK) { 476 | case MKD_CELL_ALIGN_LEFT: 477 | BUFPUTSL(ob, " align=\"left\""); 478 | break; 479 | case MKD_CELL_ALIGN_RIGHT: 480 | BUFPUTSL(ob, " align=\"right\""); 481 | break; 482 | case MKD_CELL_ALIGN_CENTER: 483 | BUFPUTSL(ob, " align=\"center\""); 484 | break; } 485 | bufputc(ob, '>'); 486 | if (text) bufput(ob, text->data, text->size); 487 | if (flags & MKD_CELL_HEAD) 488 | BUFPUTSL(ob, "</th>\n"); 489 | else 490 | BUFPUTSL(ob, "</td>\n"); } 491 | 492 | /* exported renderer structures */ 493 | const struct mkd_renderer discount_html = { 494 | NULL, 495 | NULL, 496 | 497 | rndr_blockcode, 498 | NULL, 499 | discount_blockquote, 500 | rndr_raw_block, 501 | rndr_header, 502 | html_hrule, 503 | rndr_list, 504 | rndr_listitem, 505 | rndr_paragraph, 506 | discount_table, 507 | discount_table_cell, 508 | discount_table_row, 509 | 510 | rndr_autolink, 511 | rndr_codespan, 512 | rndr_double_emphasis, 513 | rndr_emphasis, 514 | html_discount_image, 515 | html_linebreak, 516 | discount_link, 517 | rndr_raw_inline, 518 | rndr_triple_emphasis, 519 | 520 | NULL, 521 | rndr_normal_text, 522 | 523 | 64, 524 | "*_", 525 | NULL }; 526 | const struct mkd_renderer discount_xhtml = { 527 | NULL, 528 | NULL, 529 | 530 | rndr_blockcode, 531 | NULL, 532 | discount_blockquote, 533 | rndr_raw_block, 534 | rndr_header, 535 | xhtml_hrule, 536 | rndr_list, 537 | rndr_listitem, 538 | rndr_paragraph, 539 | discount_table, 540 | discount_table_cell, 541 | discount_table_row, 542 | 543 | rndr_autolink, 544 | rndr_codespan, 545 | rndr_double_emphasis, 546 | rndr_emphasis, 547 | xhtml_discount_image, 548 | xhtml_linebreak, 549 | discount_link, 550 | rndr_raw_inline, 551 | rndr_triple_emphasis, 552 | 553 | NULL, 554 | rndr_normal_text, 555 | 556 | 64, 557 | "*_", 558 | NULL }; 559 | 560 | 561 | /**************************** 562 | * NATACHA'S OWN EXTENSIONS * 563 | ****************************/ 564 | 565 | static void 566 | nat_span(struct buf *ob, struct buf *text, char *tag) { 567 | bufprintf(ob, "<%s>", tag); 568 | bufput(ob, text->data, text->size); 569 | bufprintf(ob, "</%s>", tag); } 570 | 571 | static int 572 | nat_emphasis(struct buf *ob, struct buf *text, char c, void *opaque) { 573 | if (!text || !text->size || c == '+' || c == '-') return 0; 574 | if (c == '|') nat_span(ob, text, "span"); 575 | else nat_span(ob, text, "em"); 576 | return 1; } 577 | 578 | static int 579 | nat_double_emphasis(struct buf *ob, struct buf *text, char c, void *opaque) { 580 | if (!text || !text->size || c == '|') return 0; 581 | if (c == '+') nat_span(ob, text, "ins"); 582 | else if (c == '-') nat_span(ob, text, "del"); 583 | else nat_span(ob, text, "strong"); 584 | return 1; } 585 | 586 | static int 587 | nat_triple_emphasis(struct buf *ob, struct buf *text, char c, void *opaque) { 588 | if (!text || !text->size || c == '+' || c == '-' || c == '|') return 0; 589 | BUFPUTSL(ob, "<strong><em>"); 590 | bufput(ob, text->data, text->size); 591 | BUFPUTSL(ob, "</em></strong>"); 592 | return 1; } 593 | 594 | static void 595 | nat_header(struct buf *ob, struct buf *text, int level, void *opaque) { 596 | size_t i = 0; 597 | if (ob->size) bufputc(ob, '\n'); 598 | while (i < text->size && (text->data[i] == '-' || text->data[i] == '_' 599 | || text->data[i] == '.' || text->data[i] == ':' 600 | || (text->data[i] >= 'a' && text->data[i] <= 'z') 601 | || (text->data[i] >= 'A' && text->data[i] <= 'Z') 602 | || (text->data[i] >= '0' && text->data[i] <= '0'))) 603 | i += 1; 604 | bufprintf(ob, "<h%d", level); 605 | if (i < text->size && text->data[i] == '#') { 606 | bufprintf(ob, " id=\"%.*s\">", (int)i, text->data); 607 | i += 1; } 608 | else { 609 | bufputc(ob, '>'); 610 | i = 0; } 611 | bufput(ob, text->data + i, text->size - i); 612 | bufprintf(ob, "</h%d>\n", level); } 613 | 614 | static void 615 | nat_paragraph(struct buf *ob, struct buf *text, void *opaque) { 616 | size_t i = 0; 617 | if (ob->size) bufputc(ob, '\n'); 618 | BUFPUTSL(ob, "<p"); 619 | if (text && text->size && text->data[0] == '(') { 620 | i = 1; 621 | while (i < text->size && (text->data[i] == ' ' 622 | /* this seems to be a bit more restrictive than */ 623 | /* what is allowed for class names */ 624 | || (text->data[i] >= 'a' && text->data[i] <= 'z') 625 | || (text->data[i] >= 'A' && text->data[i] <= 'Z') 626 | || (text->data[i] >= '0' && text->data[i] <= '0'))) 627 | i += 1; 628 | if (i < text->size && text->data[i] == ')') { 629 | bufprintf(ob, " class=\"%.*s\"", 630 | (int)(i - 1), text->data + 1); 631 | i += 1; } 632 | else i = 0; } 633 | bufputc(ob, '>'); 634 | if (text) bufput(ob, text->data + i, text->size - i); 635 | BUFPUTSL(ob, "</p>\n"); } 636 | 637 | 638 | /* exported renderer structures */ 639 | const struct mkd_renderer nat_html = { 640 | NULL, 641 | NULL, 642 | 643 | rndr_blockcode, 644 | NULL, 645 | discount_blockquote, 646 | rndr_raw_block, 647 | nat_header, 648 | html_hrule, 649 | rndr_list, 650 | rndr_listitem, 651 | nat_paragraph, 652 | NULL, 653 | NULL, 654 | NULL, 655 | 656 | rndr_autolink, 657 | rndr_codespan, 658 | nat_double_emphasis, 659 | nat_emphasis, 660 | html_discount_image, 661 | html_linebreak, 662 | discount_link, 663 | rndr_raw_inline, 664 | nat_triple_emphasis, 665 | 666 | NULL, 667 | rndr_normal_text, 668 | 669 | 64, 670 | "*_-+|", 671 | NULL }; 672 | const struct mkd_renderer nat_xhtml = { 673 | NULL, 674 | NULL, 675 | 676 | rndr_blockcode, 677 | NULL, 678 | discount_blockquote, 679 | rndr_raw_block, 680 | nat_header, 681 | xhtml_hrule, 682 | rndr_list, 683 | rndr_listitem, 684 | nat_paragraph, 685 | NULL, 686 | NULL, 687 | NULL, 688 | 689 | rndr_autolink, 690 | rndr_codespan, 691 | nat_double_emphasis, 692 | nat_emphasis, 693 | xhtml_discount_image, 694 | xhtml_linebreak, 695 | discount_link, 696 | rndr_raw_inline, 697 | nat_triple_emphasis, 698 | 699 | NULL, 700 | rndr_normal_text, 701 | 702 | 64, 703 | "*_-+|", 704 | NULL }; 705 | -------------------------------------------------------------------------------- /console-colour.c: -------------------------------------------------------------------------------- 1 | #include "console-colour.h" 2 | 3 | struct console_colour_state_s 4 | gConsoleStreamState = { 5 | 0x80, 6 | STATE_NONE 7 | }; 8 | 9 | #ifdef CONMD_WINDOWS 10 | 11 | #include <stdio.h> 12 | #include <limits.h> 13 | 14 | static int OutputC(wchar_t c, struct console_colour_stream_s * const stream) 15 | { 16 | return stream->Call->OutputC(stream->Call->Data, c, stream); 17 | } 18 | 19 | static int OutputA(char const * c, int len, struct console_colour_stream_s * const stream) 20 | { 21 | return stream->Call->OutputA(stream->Call->Data, c, len, stream); 22 | } 23 | 24 | static int OutputW(wchar_t const * c, int len, struct console_colour_stream_s * const stream) 25 | { 26 | return stream->Call->OutputW(stream->Call->Data, c, len, stream); 27 | } 28 | 29 | static void Colour(struct console_colour_stream_s * const stream) 30 | { 31 | //if (!stream->State->Coloured) 32 | //{ 33 | // // Colouring is disabled. Just do nothing. 34 | //} 35 | //else 36 | if (stream->State->DefaultStyle & COMMON_LVB_REVERSE_VIDEO) 37 | { 38 | stream->Call->OutputColour(stream->Call->Data, stream->State->CurrentStyle >> 4 & 0x0F | stream->State->CurrentStyle << 4 & 0xF0, stream); 39 | } 40 | else 41 | { 42 | stream->Call->OutputColour(stream->Call->Data, stream->State->CurrentStyle, stream); 43 | } 44 | } 45 | 46 | void Backout(struct console_colour_stream_s * const stream) 47 | { 48 | // Back out of the buffering. 49 | char backout[32] = ""; 50 | // Run the state machine. 51 | switch (stream->State->State) 52 | { 53 | default: 54 | // Skip might be an issue - we are in the middle of a multi-byte character... 55 | return; 56 | case STATE_ESC: 57 | sprintf_s(backout, sizeof (backout), "\x1B"); 58 | break; 59 | case STATE_START: 60 | sprintf_s(backout, sizeof (backout), "\x1B["); 61 | break; 62 | case STATE_A00: 63 | sprintf_s(backout, sizeof (backout), "\x1B[%d", stream->State->Attr0); 64 | break; 65 | case STATE_S0: 66 | sprintf_s(backout, sizeof (backout), "\x1B[%d;", stream->State->Attr0); 67 | break; 68 | case STATE_A10: 69 | sprintf_s(backout, sizeof (backout), "\x1B[%d;%d", stream->State->Attr0, stream->State->Attr1); 70 | break; 71 | case STATE_S1: 72 | sprintf_s(backout, sizeof (backout), "\x1B[%d;%d;", stream->State->Attr0, stream->State->Attr1); 73 | break; 74 | case STATE_A20: 75 | sprintf_s(backout, sizeof (backout), "\x1B[%d;%d;%d", stream->State->Attr0, stream->State->Attr1, stream->State->Attr2); 76 | break; 77 | case STATE_S2: 78 | sprintf_s(backout, sizeof (backout), "\x1B[%d;%d;%d;", stream->State->Attr0, stream->State->Attr1, stream->State->Attr2); 79 | break; 80 | case STATE_A30: 81 | sprintf_s(backout, sizeof (backout), "\x1B[%d;%d;%d;%d", stream->State->Attr0, stream->State->Attr1, stream->State->Attr2, stream->State->Attr3); 82 | break; 83 | case STATE_S3: 84 | sprintf_s(backout, sizeof (backout), "\x1B[%d;%d;%d;%d;", stream->State->Attr0, stream->State->Attr1, stream->State->Attr2, stream->State->Attr3); 85 | break; 86 | case STATE_A40: 87 | sprintf_s(backout, sizeof (backout), "\x1B[%d;%d;%d;%d;%d", stream->State->Attr0, stream->State->Attr1, stream->State->Attr2, stream->State->Attr3, stream->State->Attr4); 88 | break; 89 | } 90 | if (backout[0]) 91 | { 92 | Colour(stream); 93 | OutputA(backout, (int)strlen(backout), stream); 94 | } 95 | stream->State->State = STATE_NONE; 96 | stream->State->Attr0 = stream->State->Attr1 = stream->State->Attr2 = stream->State->Attr3 = stream->State->Attr4 = 0; 97 | } 98 | 99 | static const WORD 100 | FOREGROUND = FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE | FOREGROUND_INTENSITY, 101 | BACKGROUND = BACKGROUND_RED | BACKGROUND_GREEN | BACKGROUND_BLUE | BACKGROUND_INTENSITY; 102 | 103 | static WORD NToColour(unsigned char attr) 104 | { 105 | switch (attr) 106 | { 107 | // Original colours. 108 | case 0: 109 | // Black. 110 | return 0; 111 | case 1: 112 | // Red. 113 | return FOREGROUND_RED; 114 | case 2: 115 | // Green. 116 | return FOREGROUND_GREEN; 117 | case 3: 118 | // Yellow. 119 | return FOREGROUND_RED | FOREGROUND_GREEN; 120 | case 4: 121 | // Blue. 122 | return FOREGROUND_BLUE; 123 | case 5: 124 | // Magenta. 125 | return FOREGROUND_RED | FOREGROUND_BLUE; 126 | case 6: 127 | // Cyan. 128 | return FOREGROUND_GREEN | FOREGROUND_BLUE; 129 | case 7: 130 | // White. 131 | return FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE; 132 | case 8 + 0: 133 | // Black. 134 | return FOREGROUND_INTENSITY; 135 | case 8 + 1: 136 | // Red. 137 | return FOREGROUND_RED | FOREGROUND_INTENSITY; 138 | case 8 + 2: 139 | // Green. 140 | return FOREGROUND_GREEN | FOREGROUND_INTENSITY; 141 | case 8 + 3: 142 | // Yellow. 143 | return FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_INTENSITY; 144 | case 8 + 4: 145 | // Blue. 146 | return FOREGROUND_BLUE | FOREGROUND_INTENSITY; 147 | case 8 + 5: 148 | // Magenta. 149 | return FOREGROUND_RED | FOREGROUND_BLUE | FOREGROUND_INTENSITY; 150 | case 8 + 6: 151 | // Cyan. 152 | return FOREGROUND_GREEN | FOREGROUND_BLUE | FOREGROUND_INTENSITY; 153 | case 8 + 7: 154 | // White. 155 | return FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE | FOREGROUND_INTENSITY; 156 | } 157 | return 0; 158 | } 159 | 160 | static unsigned char Shuffle6Colour(unsigned char colour) 161 | { 162 | // The search results are better using a different search order. 163 | const unsigned char 164 | ORDER[16] = { 0, 15, 1, 2, 4, 3, 6, 5, 9, 10, 12, 11, 14, 13, 8, 7 }; 165 | return ORDER[colour]; 166 | } 167 | 168 | static WORD Make6Colour(int r, int g, int b, struct console_colour_stream_s* const stream) 169 | { 170 | // There's no easy way to map 16 colours to 3 components. So we need full colour space mapping. 171 | // I didn't want to have to write that... I KNOW there are better mapping functions based on 172 | // what colours are eyes are better at perceiving, but I'm only mapping to 16 so it hardly 173 | // matters. 174 | // 175 | // I did a LOT of experimenting with orders to get this as the best order I could find. 176 | const int 177 | CONSOLE_COLOURS[16][3] = { 178 | {0, 0, 0}, 179 | {5, 5, 5}, 180 | {2, 0, 0}, 181 | {0, 2, 0}, 182 | {0, 0, 2}, 183 | {2, 2, 0}, 184 | {0, 2, 2}, 185 | {2, 0, 2}, 186 | {4, 0, 0}, 187 | {0, 4, 0}, 188 | {0, 0, 4}, 189 | {4, 4, 0}, 190 | {0, 4, 4}, 191 | {4, 0, 4}, 192 | {2, 2, 2}, 193 | {4, 4, 4}, 194 | }; 195 | 196 | int 197 | dist = INT_MAX; 198 | unsigned char 199 | found = 0; 200 | for (unsigned char i = 0; i != 16; ++i) 201 | { 202 | int 203 | tr = r - CONSOLE_COLOURS[i][0], 204 | tg = g - CONSOLE_COLOURS[i][1], 205 | tb = b - CONSOLE_COLOURS[i][2], 206 | cur = (tr * tr) + (tg * tg) + (tb * tb); 207 | if (cur < dist) 208 | { 209 | dist = cur; 210 | found = i; 211 | } 212 | } 213 | 214 | return NToColour(Shuffle6Colour(found)); 215 | } 216 | 217 | static WORD Make24Colour(int r, int g, int b, struct console_colour_stream_s* const stream) 218 | { 219 | // Use the reduced 6-value colour space mapping, since we've tweaked that a lot already. 220 | r = r + 25 / 51; // Rounding division, to get the component from 0-5 (inclusive). 221 | g = g + 25 / 51; // Rounding division, to get the component from 0-5 (inclusive). 222 | b = b + 25 / 51; // Rounding division, to get the component from 0-5 (inclusive). 223 | return Make6Colour(r, g, b, stream); 224 | } 225 | 226 | static WORD Make256Colour(unsigned char attr, struct console_colour_stream_s * const stream) 227 | { 228 | // Greyscales. Find the nearest. 229 | if (attr >= 250) 230 | return FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE | FOREGROUND_INTENSITY; 231 | else if (attr >= 244) 232 | return FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE; 233 | else if (attr >= 238) 234 | return FOREGROUND_INTENSITY; 235 | else if (attr >= 232) 236 | return 0; 237 | else if (attr >= 16) 238 | { 239 | // 216 colours. Extract the RGB from the colour. Then scale them to 8-bit. 240 | attr -= 16; 241 | unsigned char 242 | r = attr / 36, 243 | g = attr % 36 / 6, 244 | b = attr % 6; 245 | return Make6Colour(r, g, b, stream); 246 | } 247 | 248 | return NToColour(attr); 249 | } 250 | 251 | static WORD GetColour(unsigned char attr, struct console_colour_stream_s* const stream) 252 | { 253 | WORD 254 | current = stream->State->CurrentStyle; 255 | switch (attr) 256 | { 257 | case 1: 258 | return current | FOREGROUND_INTENSITY; 259 | case 2: 260 | return current & ~FOREGROUND_INTENSITY; 261 | case 3: 262 | case 7: 263 | // Inverse. Store the fact that it is inverted somewhere, and only apply it at the very 264 | // last possible moment. 265 | stream->State->DefaultStyle |= COMMON_LVB_REVERSE_VIDEO; 266 | break; 267 | case 23: 268 | case 27: 269 | stream->State->DefaultStyle &= ~COMMON_LVB_REVERSE_VIDEO; 270 | break; 271 | case 4: 272 | case 21: 273 | return current | COMMON_LVB_UNDERSCORE; 274 | case 22: 275 | return current & ~FOREGROUND_INTENSITY; 276 | case 24: 277 | return current & ~COMMON_LVB_UNDERSCORE; 278 | case 39: 279 | return current & ~FOREGROUND | stream->State->DefaultStyle & FOREGROUND; 280 | case 49: 281 | return current & ~BACKGROUND | stream->State->DefaultStyle & BACKGROUND; 282 | case 30: 283 | // Black. 284 | return current & ~FOREGROUND; 285 | case 31: 286 | // Red. 287 | return current & ~FOREGROUND | FOREGROUND_RED; 288 | case 32: 289 | // Green. 290 | return current & ~FOREGROUND | FOREGROUND_GREEN; 291 | case 33: 292 | // Yellow. 293 | return current & ~FOREGROUND | FOREGROUND_RED | FOREGROUND_GREEN; 294 | case 34: 295 | // Blue. 296 | return current & ~FOREGROUND | FOREGROUND_BLUE; 297 | case 35: 298 | // Magenta. 299 | return current & ~FOREGROUND | FOREGROUND_RED | FOREGROUND_BLUE; 300 | case 36: 301 | // Cyan. 302 | return current & ~FOREGROUND | FOREGROUND_GREEN | FOREGROUND_BLUE; 303 | case 37: 304 | // White. 305 | return current & ~FOREGROUND | FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE; 306 | case 40: 307 | // Black. 308 | return current & ~BACKGROUND; 309 | case 41: 310 | // Red. 311 | return current & ~BACKGROUND | BACKGROUND_RED; 312 | case 42: 313 | // Green. 314 | return current & ~BACKGROUND | BACKGROUND_GREEN; 315 | case 43: 316 | // Yellow. 317 | return current & ~BACKGROUND | BACKGROUND_RED | BACKGROUND_GREEN; 318 | case 44: 319 | // Blue. 320 | return current & ~BACKGROUND | BACKGROUND_BLUE; 321 | case 45: 322 | // Magenta. 323 | return current & ~BACKGROUND | BACKGROUND_RED | BACKGROUND_BLUE; 324 | case 46: 325 | // Cyan. 326 | return current & ~BACKGROUND | BACKGROUND_GREEN | BACKGROUND_BLUE; 327 | case 47: 328 | // White. 329 | return current & ~BACKGROUND | BACKGROUND_RED | BACKGROUND_GREEN | BACKGROUND_BLUE; 330 | case 60 + 30: 331 | // Black. 332 | return current & ~FOREGROUND | FOREGROUND_INTENSITY; 333 | case 60 + 31: 334 | // Red. 335 | return current & ~FOREGROUND | FOREGROUND_RED | FOREGROUND_INTENSITY; 336 | case 60 + 32: 337 | // Green. 338 | return current & ~FOREGROUND | FOREGROUND_GREEN | FOREGROUND_INTENSITY; 339 | case 60 + 33: 340 | // Yellow. 341 | return current & ~FOREGROUND | FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_INTENSITY; 342 | case 60 + 34: 343 | // Blue. 344 | return current & ~FOREGROUND | FOREGROUND_BLUE | FOREGROUND_INTENSITY; 345 | case 60 + 35: 346 | // Magenta. 347 | return current & ~FOREGROUND | FOREGROUND_RED | FOREGROUND_BLUE | FOREGROUND_INTENSITY; 348 | case 60 + 36: 349 | // Cyan. 350 | return current & ~FOREGROUND | FOREGROUND_GREEN | FOREGROUND_BLUE | FOREGROUND_INTENSITY; 351 | case 60 + 37: 352 | // White. 353 | return current & ~FOREGROUND | FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE | FOREGROUND_INTENSITY; 354 | case 60 + 40: 355 | // Black. 356 | return current & ~BACKGROUND | BACKGROUND_INTENSITY; 357 | case 60 + 41: 358 | // Red. 359 | return current & ~BACKGROUND | BACKGROUND_RED | BACKGROUND_INTENSITY; 360 | case 60 + 42: 361 | // Green. 362 | return current & ~BACKGROUND | BACKGROUND_GREEN | BACKGROUND_INTENSITY; 363 | case 60 + 43: 364 | // Yellow. 365 | return current & ~BACKGROUND | BACKGROUND_RED | BACKGROUND_GREEN | BACKGROUND_INTENSITY; 366 | case 60 + 44: 367 | // Blue. 368 | return current & ~BACKGROUND | BACKGROUND_BLUE | BACKGROUND_INTENSITY; 369 | case 60 + 45: 370 | // Magenta. 371 | return current & ~BACKGROUND | BACKGROUND_RED | BACKGROUND_BLUE | BACKGROUND_INTENSITY; 372 | case 60 + 46: 373 | // Cyan. 374 | return current & ~BACKGROUND | BACKGROUND_GREEN | BACKGROUND_BLUE | BACKGROUND_INTENSITY; 375 | case 60 + 47: 376 | // White. 377 | return current & ~BACKGROUND | BACKGROUND_RED | BACKGROUND_GREEN | BACKGROUND_BLUE | BACKGROUND_INTENSITY; 378 | } 379 | return current; 380 | } 381 | 382 | static void GetColours(struct console_colour_stream_s * const stream) 383 | { 384 | switch (stream->State->Attr0) 385 | { 386 | case 0: 387 | // Reset. 388 | stream->State->DefaultStyle &= ~COMMON_LVB_REVERSE_VIDEO; 389 | stream->State->CurrentStyle = stream->State->DefaultStyle; 390 | break; 391 | case 38: 392 | // Special foreground colour. 393 | // TODO: Find the closest working colour from the 16 available. 394 | switch (stream->State->Attr1) 395 | { 396 | case 5: 397 | // 256 colours. 398 | stream->State->CurrentStyle = stream->State->CurrentStyle & ~FOREGROUND | Make256Colour(stream->State->Attr2, stream); 399 | break; 400 | case 2: 401 | // 24-bit colours. 402 | stream->State->CurrentStyle = stream->State->CurrentStyle & ~FOREGROUND | Make24Colour(stream->State->Attr2, stream->State->Attr3, stream->State->Attr4, stream); 403 | break; 404 | } 405 | break; 406 | case 48: 407 | // Special background colour. 408 | switch (stream->State->Attr1) 409 | { 410 | case 5: 411 | // 256 colours. 412 | stream->State->CurrentStyle = stream->State->CurrentStyle & ~BACKGROUND | Make256Colour(stream->State->Attr2, stream) << 4; 413 | break; 414 | case 2: 415 | // 24-bit colours. 416 | stream->State->CurrentStyle = stream->State->CurrentStyle & ~BACKGROUND | Make24Colour(stream->State->Attr2, stream->State->Attr3, stream->State->Attr4, stream) << 4; 417 | break; 418 | } 419 | break; 420 | default: 421 | // Normal code. 422 | stream->State->CurrentStyle = GetColour(stream->State->Attr0, stream); 423 | stream->State->CurrentStyle = GetColour(stream->State->Attr1, stream); 424 | stream->State->CurrentStyle = GetColour(stream->State->Attr2, stream); 425 | stream->State->CurrentStyle = GetColour(stream->State->Attr3, stream); 426 | stream->State->CurrentStyle = GetColour(stream->State->Attr4, stream); 427 | break; 428 | } 429 | } 430 | 431 | static int RunStateMachine(wchar_t c, struct console_colour_stream_s * const stream) 432 | { 433 | // Run the stream machine. 434 | switch (stream->State->State) 435 | { 436 | case STATE_NONE: 437 | //if (stream->State->Coloured) 438 | { 439 | switch (c) 440 | { 441 | case 29: // '\x1D', Group separator. 442 | return /*stream->State->*/OutputC(10, stream); // `\n` 443 | case 31: // '\x1F', Unit separator. 444 | return /*stream->State->*/OutputC(10, stream); // `\n` 445 | case 30: // '\x1E', Record separator. 446 | return 0; 447 | } 448 | } 449 | if (c == '\x1B') 450 | { 451 | stream->State->State = STATE_ESC; 452 | return 0; 453 | } 454 | else if ((c & stream->State->UnicodeMask)) 455 | { 456 | // Start of a multi-byte character. Skip all checks on the next one. 457 | stream->State->State = STATE_SKIP; 458 | } 459 | break; 460 | case STATE_SKIP: 461 | if (!(c & stream->State->UnicodeMask)) 462 | { 463 | stream->State->State = STATE_NONE; 464 | } 465 | break; 466 | case STATE_ESC: 467 | if (c == '[') 468 | { 469 | stream->State->State = STATE_START; 470 | } 471 | else 472 | { 473 | Backout(stream); 474 | break; 475 | } 476 | return 0; 477 | case STATE_START: 478 | if ('0' <= c && c <= '9') 479 | { 480 | stream->State->Attr0 = (unsigned char)(c - '0'); 481 | stream->State->State = STATE_A00; 482 | } 483 | else 484 | { 485 | Backout(stream); 486 | break; 487 | } 488 | return 0; 489 | case STATE_A00: 490 | if ('0' <= c && c <= '9') 491 | { 492 | stream->State->Attr0 = stream->State->Attr0 * 10 + (unsigned char)(c - '0'); 493 | } 494 | else if (c == ';') 495 | stream->State->State = STATE_S0; 496 | //else if (c == ':') 497 | // stream->State->State = STATE_T416; 498 | else if (c == 'm') 499 | { 500 | stream->State->State = STATE_DONE; 501 | break; 502 | } 503 | else 504 | { 505 | Backout(stream); 506 | break; 507 | } 508 | return 0; 509 | case STATE_S0: 510 | if ('0' <= c && c <= '9') 511 | { 512 | stream->State->Attr1 = (unsigned char)(c - '0'); 513 | stream->State->State = STATE_A10; 514 | } 515 | else if (c == 'm') 516 | { 517 | stream->State->State = STATE_DONE; 518 | break; 519 | } 520 | else 521 | { 522 | Backout(stream); 523 | break; 524 | } 525 | return 0; 526 | case STATE_A10: 527 | if ('0' <= c && c <= '9') 528 | { 529 | stream->State->Attr1 = stream->State->Attr1 * 10 + (unsigned char)(c - '0'); 530 | } 531 | else if (c == ';') 532 | stream->State->State = STATE_S1; 533 | else if (c == 'm') 534 | { 535 | stream->State->State = STATE_DONE; 536 | break; 537 | } 538 | else 539 | { 540 | Backout(stream); 541 | break; 542 | } 543 | return 0; 544 | case STATE_S1: 545 | if ('0' <= c && c <= '9') 546 | { 547 | stream->State->Attr2 = (unsigned char)(c - '0'); 548 | stream->State->State = STATE_A20; 549 | } 550 | else if (c == 'm') 551 | { 552 | stream->State->State = STATE_DONE; 553 | break; 554 | } 555 | else 556 | { 557 | Backout(stream); 558 | break; 559 | } 560 | return 0; 561 | case STATE_A20: 562 | if ('0' <= c && c <= '9') 563 | { 564 | stream->State->Attr2 = stream->State->Attr2 * 10 + (unsigned char)(c - '0'); 565 | } 566 | else if (c == ';') 567 | stream->State->State = STATE_S2; 568 | else if (c == 'm') 569 | { 570 | stream->State->State = STATE_DONE; 571 | break; 572 | } 573 | else 574 | { 575 | Backout(stream); 576 | break; 577 | } 578 | return 0; 579 | case STATE_S2: 580 | if ('0' <= c && c <= '9') 581 | { 582 | stream->State->Attr3 = (unsigned char)(c - '0'); 583 | stream->State->State = STATE_A30; 584 | } 585 | else if (c == 'm') 586 | { 587 | stream->State->State = STATE_DONE; 588 | break; 589 | } 590 | else 591 | { 592 | Backout(stream); 593 | break; 594 | } 595 | return 0; 596 | case STATE_A30: 597 | if ('0' <= c && c <= '9') 598 | { 599 | stream->State->Attr3 = stream->State->Attr3 * 10 + (unsigned char)(c - '0'); 600 | } 601 | else if (c == ';') 602 | stream->State->State = STATE_S3; 603 | else if (c == 'm') 604 | { 605 | stream->State->State = STATE_DONE; 606 | break; 607 | } 608 | else 609 | { 610 | Backout(stream); 611 | break; 612 | } 613 | return 0; 614 | case STATE_S3: 615 | if ('0' <= c && c <= '9') 616 | { 617 | stream->State->Attr4 = (unsigned char)(c - '0'); 618 | stream->State->State = STATE_A40; 619 | } 620 | else if (c == 'm') 621 | { 622 | stream->State->State = STATE_DONE; 623 | break; 624 | } 625 | else 626 | { 627 | Backout(stream); 628 | break; 629 | } 630 | return 0; 631 | case STATE_A40: 632 | if ('0' <= c && c <= '9') 633 | { 634 | stream->State->Attr4 = stream->State->Attr4 * 10 + (unsigned char)(c - '0'); 635 | } 636 | else if (c == 'm') 637 | { 638 | stream->State->State = STATE_DONE; 639 | break; 640 | } 641 | else 642 | { 643 | Backout(stream); 644 | break; 645 | } 646 | return 0; 647 | } 648 | if (stream->State->State == STATE_DONE) 649 | { 650 | GetColours(stream); 651 | stream->State->State = STATE_NONE; 652 | stream->State->Attr0 = stream->State->Attr1 = stream->State->Attr2 = stream->State->Attr3 = stream->State->Attr4 = 0; 653 | return 0; 654 | } 655 | Colour(stream); 656 | return /*stream->State->*/OutputC(c, stream); 657 | } 658 | 659 | #if defined CONMD_HOOKS 660 | #define MAYBE_REMOVE_HOOKS() RemoveStreamHooks() 661 | #define MAYBE_READD_HOOKS() ReaddStreamHooks() 662 | #else 663 | #define MAYBE_REMOVE_HOOKS() ((void)0) 664 | #define MAYBE_READD_HOOKS() ((void)0) 665 | #endif 666 | 667 | int WriteColouredA(char const * s, int n, struct console_colour_stream_s* const stream) 668 | { 669 | MAYBE_REMOVE_HOOKS(); 670 | char const 671 | * cur, 672 | * start = s, 673 | * end = s + n; 674 | // Just in case. 675 | if (n == 0 || s == NULL) 676 | { 677 | MAYBE_READD_HOOKS(); 678 | return 0; 679 | } 680 | int 681 | ret = 0; 682 | for (; ; ) 683 | { 684 | while (*s && !(stream->State->State == STATE_NONE || stream->State->State == STATE_SKIP)) 685 | { 686 | RunStateMachine(*s++, stream); 687 | } 688 | char const * 689 | esc = s; 690 | //if (stream->State->Coloured) 691 | { 692 | for (cur = s; cur != end; ++cur) 693 | { 694 | switch (*cur) 695 | { 696 | case '\x1B': // ESC. 697 | stream->State->State = STATE_ESC; 698 | esc = cur; 699 | goto WriteColoured_loop_done; 700 | case '\x1D': // Group separator. 701 | stream->State->State = STATE_EXTRA_NL; 702 | esc = cur; 703 | goto WriteColoured_loop_done; 704 | case '\x1F': // Unit separator. 705 | stream->State->State = STATE_EXTRA_2NL; 706 | esc = cur; 707 | goto WriteColoured_loop_done; 708 | case '\x1E': // Record separator. 709 | stream->State->State = STATE_NONE; 710 | esc = cur; 711 | goto WriteColoured_loop_done; 712 | } 713 | } 714 | WriteColoured_loop_done: 715 | if (cur == end) 716 | { 717 | break; 718 | } 719 | } 720 | //else 721 | //{ 722 | // esc = strchr(s, '\x1B'); 723 | // if (esc == NULL) 724 | // break; 725 | // stream->State->State = STATE_ESC; 726 | //} 727 | // Write the bit before the escape sequence. 728 | if (esc != s) 729 | { 730 | // Restore the correct colours. 731 | Colour(stream); 732 | int 733 | wrote = OutputA(s, (int)(esc - s), stream); 734 | ret += wrote; 735 | if (wrote != esc - s) 736 | { 737 | MAYBE_READD_HOOKS(); 738 | return (int)(s - start + wrote); 739 | } 740 | } 741 | s = esc + 1; 742 | switch (stream->State->State) 743 | { 744 | case STATE_EXTRA_2NL: 745 | OutputC(10, stream); 746 | case STATE_EXTRA_NL: 747 | OutputC(10, stream); 748 | break; 749 | /*case STATE_EXTRA_SPACE: 750 | OutputA(' '); 751 | break;*/ 752 | default: 753 | continue; 754 | } 755 | stream->State->State = STATE_NONE; 756 | } 757 | // Restore the correct colours. 758 | Colour(stream); 759 | int 760 | wrote = OutputA(s, (int)(end - s), stream); 761 | MAYBE_READD_HOOKS(); 762 | return (int)(s - start + wrote); 763 | } 764 | 765 | int WriteColouredW(wchar_t const * s, int n, struct console_colour_stream_s* const stream) 766 | { 767 | MAYBE_REMOVE_HOOKS(); 768 | wchar_t const 769 | * cur, 770 | * start = s, 771 | * end = s + n; 772 | // Just in case. 773 | if (n == 0 || s == NULL) 774 | { 775 | MAYBE_READD_HOOKS(); 776 | return 0; 777 | } 778 | int 779 | ret = 0; 780 | for (; ; ) 781 | { 782 | while (*s && !(stream->State->State == STATE_NONE || stream->State->State == STATE_SKIP)) 783 | { 784 | RunStateMachine(*s++, stream); 785 | } 786 | wchar_t const * 787 | esc = s; 788 | //if (stream->State->Coloured) 789 | { 790 | for (cur = s; cur != end; ++cur) 791 | { 792 | switch (*cur) 793 | { 794 | case '\x1B': // ESC. 795 | stream->State->State = STATE_ESC; 796 | esc = cur; 797 | goto WriteColoured_loop_done; 798 | case '\x1D': // Group separator. 799 | stream->State->State = STATE_EXTRA_NL; 800 | esc = cur; 801 | goto WriteColoured_loop_done; 802 | case '\x1F': // Unit separator. 803 | stream->State->State = STATE_EXTRA_2NL; 804 | esc = cur; 805 | goto WriteColoured_loop_done; 806 | case '\x1E': // Record separator. 807 | stream->State->State = STATE_NONE; 808 | esc = cur; 809 | goto WriteColoured_loop_done; 810 | } 811 | } 812 | WriteColoured_loop_done: 813 | if (cur == end) 814 | { 815 | break; 816 | } 817 | } 818 | //else 819 | //{ 820 | // esc = wcschr(s, '\x1B'); 821 | // if (esc == NULL) 822 | // break; 823 | // stream->State->State = STATE_ESC; 824 | //} 825 | // Write the bit before the escape sequence. 826 | if (esc != s) 827 | { 828 | // Restore the correct colours. 829 | Colour(stream); 830 | int 831 | wrote = OutputW(s, (int)(esc - s), stream); 832 | ret += wrote; 833 | if (wrote != esc - s) 834 | { 835 | MAYBE_READD_HOOKS(); 836 | return (int)(s - start + wrote); 837 | } 838 | } 839 | s = esc + 1; 840 | switch (stream->State->State) 841 | { 842 | case STATE_EXTRA_2NL: 843 | OutputC(10, stream); 844 | case STATE_EXTRA_NL: 845 | OutputC(10, stream); 846 | break; 847 | /*case STATE_EXTRA_SPACE: 848 | OutputW(' '); 849 | break;*/ 850 | default: 851 | continue; 852 | } 853 | stream->State->State = STATE_NONE; 854 | } 855 | // Restore the correct colours. 856 | Colour(stream); 857 | int 858 | wrote = OutputW(s, (int)(end - s), stream); 859 | MAYBE_READD_HOOKS(); 860 | return (int)(s - start + wrote); 861 | } 862 | 863 | #endif 864 | 865 | --------------------------------------------------------------------------------