summaryrefslogtreecommitdiffstats
path: root/libsmartcols/samples/colors.c
blob: 0e9ab663f414513074e058bccd9c7deb7b72a913 (plain)
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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
/*
 * Copyright (C) 2010-2014 Karel Zak <kzak@redhat.com>
 *
 * This file may be redistributed under the terms of the
 * GNU Lesser General Public License.
 */
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <dirent.h>
#include <getopt.h>

#include "c.h"
#include "nls.h"
#include "strutils.h"
#include "xalloc.h"

#include "libsmartcols.h"


enum { COL_NAME, COL_FOO, COL_BAR };

/* add columns to the @tb */
static void setup_columns(struct libscols_table *tb)
{
	if (!scols_table_new_column(tb, "NAME", 0, 0))
		goto fail;
	if (!scols_table_new_column(tb, "BAR", 0, 0))
		goto fail;
	if (!scols_table_new_column(tb, "FOO", 0, 0))
		goto fail;
	return;
fail:
	scols_unref_table(tb);
	err(EXIT_FAILURE, "failed to create output columns");
}

static struct libscols_line *add_line(struct libscols_table *tb, const char *name, const char *data)
{
	struct libscols_line *ln = scols_table_new_line(tb, NULL);
	if (!ln)
		err(EXIT_FAILURE, "failed to create output line");

	if (scols_line_set_data(ln, COL_NAME, name))
		goto fail;
	if (scols_line_set_data(ln, COL_FOO, data))
		goto fail;
	if (scols_line_set_data(ln, COL_BAR, data))
		goto fail;
	return ln;
fail:
	scols_unref_table(tb);
	err(EXIT_FAILURE, "failed to create output line");
}

int main(int argc, char *argv[])
{
	struct libscols_table *tb;
	struct libscols_column *cl;
	struct libscols_line *ln;
	struct libscols_cell *ce;
	int c;

	static const struct option longopts[] = {
		{ "maxout", 0, NULL, 'm' },
		{ "width",  1, NULL, 'w' },
		{ "help",   1, NULL, 'h' },

		{ NULL, 0, NULL, 0 },
	};

	setlocale(LC_ALL, "");	/* just to have enable UTF8 chars */

	scols_init_debug(0);

	tb = scols_new_table();
	if (!tb)
		err(EXIT_FAILURE, "failed to create output table");

	while((c = getopt_long(argc, argv, "hmw:", longopts, NULL)) != -1) {
		switch(c) {
		case 'h':
			printf("%s [--help | --maxout | --width <num>]\n", program_invocation_short_name);
			break;
		case 'm':
			scols_table_enable_maxout(tb, TRUE);
			break;
		case 'w':
			scols_table_set_termforce(tb, SCOLS_TERMFORCE_ALWAYS);
			scols_table_set_termwidth(tb, strtou32_or_err(optarg, "failed to parse terminal width"));
			break;
		}
	}

	scols_table_enable_colors(tb, isatty(STDOUT_FILENO));
	setup_columns(tb);
	add_line(tb, "AAA", "bla bla bla");
	add_line(tb, "BB", "b");
	add_line(tb, "CCCC", "fooo");
	add_line(tb, "D",   "baaar");
	add_line(tb, "EE", "eee");

	cl = scols_table_get_column(tb, 1);
	scols_column_set_color(cl, "red");		/* red column */

	cl = scols_table_get_column(tb, 2);
	scols_column_set_color(cl, "reverse");		/* reverse column */

	ln = scols_table_get_line(tb, 0);
	scols_line_set_color(ln, "\033[37;41m");	/* line with red bg */
	ce = scols_line_get_cell(ln, 0);
	scols_cell_set_color(ce, "\033[37;45m");	/* cell with purple bg */

	ln = scols_table_get_line(tb, 3);
	scols_line_set_color(ln, "\033[37;41m");	/* line with red bg */
	ce = scols_line_get_cell(ln, 2);
	scols_cell_set_color(ce, "\033[37;44m");	/* cell with blue bg */

	scols_print_table(tb);
	scols_unref_table(tb);
	return EXIT_SUCCESS;
}