&& (streq (color_names[color2]->name, "none") \
|| streq (color_names[color2]->name, "default")) \
+#define ALLOC_COMPLETE_PART_LINE 8
+
#define COLOR_SEP_CHAR '/'
#define DEBUG_FILE "debug.txt"
-#define VERSION "0.56"
+#define VERSION "0.57"
typedef enum { false, true } bool;
static void process_args (unsigned int, char **, bool *, const struct color **, const char **, FILE **);
static void process_file_arg (const char *, const char **, FILE **);
static void read_print_stream (bool, const struct color **, const char *, FILE *);
+static void merge_print_line (bool, const struct color **, const char *, const char *, FILE *);
+static void complete_part_line (const char *, char **, FILE *);
+static bool get_next_char (char *, const char **, FILE *, bool *);
+static void save_char (char, char **, unsigned long *, size_t *);
static void find_color_entries (struct color_name **, const struct color **);
static void find_color_entry (const struct color_name *, unsigned int, const struct color **);
static void print_line (bool, const struct color **, const char * const, unsigned int);
int
main (int argc, char **argv)
{
- unsigned int arg_cnt = 0;
+ unsigned int arg_cnt;
bool bold = false;
{
unsigned int i;
for (i = 0; i < stacked_vars; i++)
- if (vars_list[i])
- free (vars_list[i]);
-
+ free (vars_list[i]);
free_null (vars_list);
}
}
assert (*file);
}
-#define MERGE_PRINT_LINE(part_line, line, flags, check_eof) do { \
- char *current_line, *merged_line = NULL; \
- if (part_line) \
- { \
- merged_line = str_concat (part_line, line); \
- free_null (part_line); \
- } \
- current_line = merged_line ? merged_line : (char *)line; \
- if (!check_eof || *current_line != '\0') \
- print_line (bold, colors, current_line, flags); \
- free (merged_line); \
-} while (false)
-
static void
read_print_stream (bool bold, const struct color **colors, const char *file, FILE *stream)
{
- char buf[BUF_SIZE + 1], *part_line = NULL;
+ char buf[BUF_SIZE + 1];
unsigned int flags = 0;
while (!feof (stream))
vfprintf_fail (formats[FMT_FILE], file, "unrecognized line ending");
p = eol + SKIP_LINE_ENDINGS (flags);
*eol = '\0';
- MERGE_PRINT_LINE (part_line, line, flags, false);
+ print_line (bold, colors, line, flags);
line = p;
}
- if (feof (stream)) {
- MERGE_PRINT_LINE (part_line, line, 0, true);
- }
- else if (*line != '\0')
+ if (feof (stream))
{
- if (!clean && !clean_all) /* efficiency */
+ if (*line != '\0')
print_line (bold, colors, line, 0);
- else if (!part_line)
- part_line = xstrdup (line);
+ }
+ else if (*line != '\0')
+ {
+ char *p;
+ if ((clean || clean_all) && (p = strrchr (line, '\033')))
+ merge_print_line (bold, colors, line, p, stream);
else
- {
- char *merged_line = str_concat (part_line, line);
- free (part_line);
- part_line = merged_line;
- }
+ print_line (bold, colors, line, 0);
}
}
}
+static void
+merge_print_line (bool bold, const struct color **colors, const char *line, const char *p, FILE *stream)
+{
+ char *buf = NULL;
+ char *merged_part_line = NULL;
+ const char *part_line;
+
+ complete_part_line (p + 1, &buf, stream);
+
+ if (buf)
+ part_line = merged_part_line = str_concat (line, buf);
+ else
+ part_line = line;
+ free (buf);
+
+#ifdef TEST_MERGE_PART_LINE
+ printf ("%s", part_line);
+ free (merged_part_line);
+ exit (EXIT_SUCCESS);
+#else
+ print_line (bold, colors, part_line, 0);
+ free (merged_part_line);
+#endif
+}
+
+static void
+complete_part_line (const char *p, char **buf, FILE *stream)
+{
+ bool got_next_char = false, read_from_stream;
+ char ch;
+ unsigned long i = 0;
+ size_t size;
+
+ if (get_next_char (&ch, &p, stream, &read_from_stream))
+ {
+ if (ch == '[')
+ {
+ if (read_from_stream)
+ save_char (ch, buf, &i, &size);
+ }
+ else
+ {
+ if (read_from_stream)
+ ungetc ((int)ch, stream);
+ return; /* cancel */
+ }
+ }
+ else
+ return; /* cancel */
+
+ while (get_next_char (&ch, &p, stream, &read_from_stream))
+ {
+ if (isdigit (ch) || ch == ';')
+ {
+ if (read_from_stream)
+ save_char (ch, buf, &i, &size);
+ }
+ else /* read next character */
+ {
+ got_next_char = true;
+ break;
+ }
+ }
+
+ if (got_next_char)
+ {
+ if (ch == 'm')
+ {
+ if (read_from_stream)
+ save_char (ch, buf, &i, &size);
+ }
+ else
+ {
+ if (read_from_stream)
+ ungetc ((int)ch, stream);
+ return; /* cancel */
+ }
+ }
+ else
+ return; /* cancel */
+}
+
+static bool
+get_next_char (char *ch, const char **p, FILE *stream, bool *read_from_stream)
+{
+ if (**p == '\0')
+ {
+ int c;
+ if ((c = fgetc (stream)) != EOF)
+ {
+ *ch = (char)c;
+ *read_from_stream = true;
+ return true;
+ }
+ else
+ {
+ *read_from_stream = false;
+ return false;
+ }
+ }
+ else
+ {
+ *ch = **p;
+ (*p)++;
+ *read_from_stream = false;
+ return true;
+ }
+}
+
+static void
+save_char (char ch, char **buf, unsigned long *i, size_t *size)
+{
+ if (!*buf)
+ {
+ *size = ALLOC_COMPLETE_PART_LINE;
+ *buf = xmalloc (*size);
+ }
+ /* +1: effective occupied size of buffer */
+ else if ((*i + 1) == *size)
+ {
+ *size *= 2;
+ *buf = xrealloc (*buf, *size);
+ }
+ (*buf)[*i] = ch;
+ (*buf)[*i + 1] = '\0';
+ (*i)++;
+}
+
static void
find_color_entries (struct color_name **color_names, const struct color **colors)
{
projects / colorize.git / blobdiff