summaryrefslogtreecommitdiffstats
path: root/kernel/tests/lib/tst_ansi_color.c
blob: 1c29268f2e7313e63c8c4e9c3d98a84063f4e252 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * Copyright (c) 2017 Petr Vorel <pvorel@suse.cz>
 */

#include <unistd.h>
#include <stdlib.h>
#include <string.h>

#include "tst_res_flags.h"
#include "tst_ansi_color.h"

char* tst_ttype2color(int ttype)
{
	switch (TTYPE_RESULT(ttype)) {
	case TPASS:
		return ANSI_COLOR_GREEN;
	break;
	case TFAIL:
		return ANSI_COLOR_RED;
	break;
	case TBROK:
		return ANSI_COLOR_RED;
	break;
	case TCONF:
		return ANSI_COLOR_YELLOW;
	break;
	case TWARN:
		return ANSI_COLOR_MAGENTA;
	break;
	case TINFO:
		return ANSI_COLOR_BLUE;
	break;
	default:
		return "";
	}
}

int tst_color_enabled(int fd)
{
	static int color;

	if (color)
		return color - 1;

	char *env = getenv("LTP_COLORIZE_OUTPUT");

	if (env) {
		if (!strcmp(env, "n") || !strcmp(env, "0"))
			color = 1;

		if (!strcmp(env, "y") || !strcmp(env, "1"))
			color = 2;

		return color - 1;
	}

	if (isatty(fd) == 0)
		color = 1;
	else
		color = 2;

	return color - 1;
}