1 27bf601b 2021-03-12 op /*
  2 27bf601b 2021-03-12 op  * Copyright (c) 2021 Omar Polo <op@omarpolo.com>
  3 27bf601b 2021-03-12 op  *
  4 27bf601b 2021-03-12 op  * Permission to use, copy, modify, and distribute this software for any
  5 27bf601b 2021-03-12 op  * purpose with or without fee is hereby granted, provided that the above
  6 27bf601b 2021-03-12 op  * copyright notice and this permission notice appear in all copies.
  7 27bf601b 2021-03-12 op  *
  8 27bf601b 2021-03-12 op  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  9 27bf601b 2021-03-12 op  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 10 27bf601b 2021-03-12 op  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 11 27bf601b 2021-03-12 op  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 12 27bf601b 2021-03-12 op  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 13 27bf601b 2021-03-12 op  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 14 27bf601b 2021-03-12 op  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 15 27bf601b 2021-03-12 op  */
 16 27bf601b 2021-03-12 op 
 17 27bf601b 2021-03-12 op /*
 18 27bf601b 2021-03-12 op  * A streaming text/plain "parser."
 19 27bf601b 2021-03-12 op  */
 20 27bf601b 2021-03-12 op 
 21 27bf601b 2021-03-12 op #include "telescope.h"
 22 27bf601b 2021-03-12 op 
 23 27bf601b 2021-03-12 op #include <stdlib.h>
 24 27bf601b 2021-03-12 op #include <string.h>
 25 27bf601b 2021-03-12 op 
 26 27bf601b 2021-03-12 op static int	textplain_parse(struct parser*, const char*, size_t);
 27 27bf601b 2021-03-12 op static int	textplain_free(struct parser*);
 28 27bf601b 2021-03-12 op 
 29 27bf601b 2021-03-12 op static inline int
 30 27bf601b 2021-03-12 op emit_line(struct parser *p, enum line_type type, const char *line, size_t len)
 31 27bf601b 2021-03-12 op {
 32 27bf601b 2021-03-12 op 	struct line *l;
 33 27bf601b 2021-03-12 op 
 34 27bf601b 2021-03-12 op 	if ((l = calloc(1, sizeof(*l))) == NULL)
 35 27bf601b 2021-03-12 op 		return 0;
 36 27bf601b 2021-03-12 op 
 37 27bf601b 2021-03-12 op 	l->type = type;
 38 27bf601b 2021-03-12 op 
 39 27bf601b 2021-03-12 op 	if (len != 0) {
 40 27bf601b 2021-03-12 op 		if ((l->line = calloc(1, len+1)) == NULL) {
 41 27bf601b 2021-03-12 op 			free(l);
 42 27bf601b 2021-03-12 op 			return 0;
 43 27bf601b 2021-03-12 op 		}
 44 27bf601b 2021-03-12 op 
 45 27bf601b 2021-03-12 op 		memcpy(l->line, line, len);
 46 27bf601b 2021-03-12 op 	}
 47 27bf601b 2021-03-12 op 
 48 27bf601b 2021-03-12 op 	if (TAILQ_EMPTY(&p->head))
 49 27bf601b 2021-03-12 op 		TAILQ_INSERT_HEAD(&p->head, l, lines);
 50 27bf601b 2021-03-12 op 	else
 51 27bf601b 2021-03-12 op 		TAILQ_INSERT_TAIL(&p->head, l, lines);
 52 27bf601b 2021-03-12 op 
 53 27bf601b 2021-03-12 op 	return 1;
 54 27bf601b 2021-03-12 op }
 55 27bf601b 2021-03-12 op 
 56 27bf601b 2021-03-12 op void
 57 27bf601b 2021-03-12 op textplain_initparser(struct parser *p)
 58 27bf601b 2021-03-12 op {
 59 27bf601b 2021-03-12 op 	memset(p, 0, sizeof(*p));
 60 27bf601b 2021-03-12 op 
 61 fc43eadd 2021-03-12 op 	p->name = "text/plain";
 62 27bf601b 2021-03-12 op 	p->parse = &textplain_parse;
 63 27bf601b 2021-03-12 op 	p->free = &textplain_free;
 64 27bf601b 2021-03-12 op 
 65 27bf601b 2021-03-12 op 	emit_line(p, LINE_PRE_START, NULL, 0);
 66 27bf601b 2021-03-12 op }
 67 27bf601b 2021-03-12 op 
 68 27bf601b 2021-03-12 op static int
 69 27bf601b 2021-03-12 op textplain_parse(struct parser *p, const char *buf, size_t size)
 70 27bf601b 2021-03-12 op {
 71 27bf601b 2021-03-12 op 	const char	*b, *e;
 72 27bf601b 2021-03-12 op 	size_t		 len, l;
 73 27bf601b 2021-03-12 op 	int		 r;
 74 27bf601b 2021-03-12 op 
 75 27bf601b 2021-03-12 op 	if (p->len == 0) {
 76 27bf601b 2021-03-12 op 		b = buf;
 77 27bf601b 2021-03-12 op 		len = size;
 78 27bf601b 2021-03-12 op 	} else {
 79 3d70083d 2021-03-12 op 		if (!parser_append(p, buf, size))
 80 27bf601b 2021-03-12 op 			return 0;
 81 27bf601b 2021-03-12 op 		b = p->buf;
 82 27bf601b 2021-03-12 op 		len = p->len;
 83 27bf601b 2021-03-12 op 	}
 84 27bf601b 2021-03-12 op 
 85 27bf601b 2021-03-12 op 	while (len > 0) {
 86 27bf601b 2021-03-12 op 		if ((e = telescope_strnchr((char*)b, '\n', len)) == NULL)
 87 27bf601b 2021-03-12 op 			break;
 88 27bf601b 2021-03-12 op 		l = e - b;
 89 27bf601b 2021-03-12 op 
 90 27bf601b 2021-03-12 op 		r = emit_line(p, LINE_PRE_CONTENT, b, l);
 91 27bf601b 2021-03-12 op 		if (!r)
 92 27bf601b 2021-03-12 op 			return 0;
 93 27bf601b 2021-03-12 op 
 94 27bf601b 2021-03-12 op 		len -= l;
 95 27bf601b 2021-03-12 op 		b += l;
 96 27bf601b 2021-03-12 op 
 97 27bf601b 2021-03-12 op 		if (len > 0) {
 98 27bf601b 2021-03-12 op 			/* skip \n */
 99 27bf601b 2021-03-12 op 			len--;
100 27bf601b 2021-03-12 op 			b++;
101 27bf601b 2021-03-12 op 		}
102 27bf601b 2021-03-12 op 	}
103 27bf601b 2021-03-12 op 
104 3d70083d 2021-03-12 op 	return parser_set_buf(p, b, len);
105 27bf601b 2021-03-12 op }
106 27bf601b 2021-03-12 op 
107 27bf601b 2021-03-12 op static int
108 27bf601b 2021-03-12 op textplain_free(struct parser *p)
109 27bf601b 2021-03-12 op {
110 27bf601b 2021-03-12 op 	/* flush the buffer */
111 27bf601b 2021-03-12 op 	if (p->len != 0) {
112 27bf601b 2021-03-12 op 		if (!emit_line(p, LINE_PRE_CONTENT, p->buf, p->len))
113 27bf601b 2021-03-12 op 			return 0;
114 27bf601b 2021-03-12 op 	}
115 27bf601b 2021-03-12 op 
116 27bf601b 2021-03-12 op 	return emit_line(p, LINE_PRE_END, NULL, 0);
117 27bf601b 2021-03-12 op }