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
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
|
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#ifdef ERR
#include <stdarg.h>
#endif
#ifdef GIT
#include <git2.h>
#endif
enum face { EYES, NOSE, MOUTH };
void
l(const char *fmt, ...)
{
#ifdef ERR
va_list ap;
va_start(ap, fmt);
vfprintf(stderr, fmt, ap);
va_end(ap);
if (fmt[0] && fmt[strlen(fmt) - 1] == ':') {
fputc(' ', stderr);
perror(NULL);
} else {
fputc('\n', stderr);
}
#endif
}
#ifdef GIT
/**
* @brief open git repo if one is available at the current path
*
* @return a pointer to the git repo object
*/
git_repository
*init_git()
{
git_buf buf = GIT_BUF_INIT_CONST(NULL, 0);
git_repository *repo;
if (git_libgit2_init() < 0) {
l("Failed to initalize libgit2, proceeding without git functionality enabled.");
return NULL;
}
if (git_repository_discover(&buf, ".", 0, NULL) < 0) {
l("Failed to discover git repo: %s", git_error_last()->message);
return NULL;
}
if (git_repository_open(&repo, buf.ptr) < 0) {
l("Failed to open git repo: %s", git_error_last()->message);
git_buf_dispose(&buf);
return NULL;
}
/* get rid of object containing git repo path and return the repo */
git_buf_dispose(&buf);
return repo;
}
/**
* @brief check for any existing stashes in the provided git repo
*
* @param repo git repo to check for existing stashes
* @return 1 if stashes found 0 otherwise
*/
int
has_stashes(git_repository *repo)
{
git_reference *stash = NULL;
int e;
e = git_reference_lookup(&stash, repo, "refs/stash");
if (e == GIT_ENOTFOUND) {
git_error_clear();
return 0;
} else if (e < GIT_OK) {
l("Error looking up stash reference: %s", git_error_last()->message);
return 0;
} else {
e = 1;
}
git_reference_free(stash);
return e;
}
/**
* @brief check for any untracked changes in the current git repo
*
* @param repo git repository object
* @return 1 if any untracked changes have been found 0 otherwise
*/
int
has_untracked(git_repository *repo)
{
git_status_options opts = GIT_STATUS_OPTIONS_INIT;
git_status_list *list = NULL;
int r = 0;
opts.show = GIT_STATUS_SHOW_INDEX_AND_WORKDIR;
opts.flags = GIT_STATUS_OPT_INCLUDE_UNTRACKED |
GIT_STATUS_OPT_RENAMES_HEAD_TO_INDEX;
if (git_status_list_new(&list, repo, &opts) < 0) {
l("Error checking for untracked changes: %s", git_error_last()->message);
return 0;
}
/* if any changes are found return 1 */
if (git_status_list_entrycount(list) > 0) {
r = 1;
}
git_status_list_free(list);
return r;
}
int
has_staged(git_repository *repo)
{
git_status_entry entry;
git_status_list *list = NULL;
int i, c, r = 0;
if (git_status_list_new(&list, repo, NULL) < 0) {
l("Error checking for staged changes: %s", git_error_last()->message);
return 0;
}
/* if any staged changes are found return 1 */
if ((c = git_status_list_entrycount(list)) > 0) {
for (i = 0; i < c; i++) {
entry = *git_status_byindex(list, i);
if (entry.status & GIT_STATUS_INDEX_NEW
|| entry.status & GIT_STATUS_INDEX_DELETED
|| entry.status & GIT_STATUS_INDEX_MODIFIED) {
r = 1;
break;
}
}
}
git_status_list_free(list);
return r;
}
#endif
int
main(int argc, char *argv[])
{
int i, code = -1;
char face[] = { ':', 0, '|' };
/* print version information */
for (i = 1; i < argc; i++) {
if (strcmp(argv[i], "-v") == 0) {
printf("XD v%s\n", VERSION);
return 0;
}
}
#ifdef GIT
git_repository *repo;
int git_ok = 0;
if ((repo = init_git())) {
git_ok = 1;
}
if (git_ok) {
/* change the eyes depending on the current git repo's status */
if (has_stashes(repo)) {
face[EYES] = '8'; /* goggle eyes if we have some stashed changes */
} else if (git_repository_is_empty(repo)) {
face[EYES] = 'B'; /* sunglasses if we're in a new repo with no HEAD */
} else {
face[EYES] = ';'; /* wink when we're in a git repo */
}
/* change the nose depending on the current git repo's status */
if (has_staged(repo)) {
face[NOSE] = '*'; /* change to broken nose for staged changes */
} else if (has_untracked(repo)) {
face[NOSE] = '^'; /* add a little nose when there are untracked changes in the repo */
} else if (git_repository_head_detached(repo)) {
face[NOSE] = '-'; /* add a minus nose when the HEAD is detached */
}
}
#endif
/* get exit code from user args */
if (argv[1]) {
code = atoi(argv[1]);
}
/* change mouth based on exit code */
if (code >= 0) {
switch (code) {
case 0: face[MOUTH] = ')'; break; /* all good */
case 130: face[MOUTH] = 'O'; break; /* Ctrl-c pressed (SIGTERM) */
case 126: face[MOUTH] = 'P'; break; /* permission denied */
case 127: face[MOUTH] = '/'; break; /* command not found */
default: face[MOUTH] = '('; break; /* all other codes (usually the program saying it has failed) */
}
} else {
face[MOUTH] = '|'; /* no code info */
}
#ifdef GIT
git_repository_free(repo);
git_libgit2_shutdown();
#endif
printf("%c%c%c", face[EYES], face[NOSE], face[MOUTH]);
}
|