add commit count options to git-shortlog
[git/jnareb-git.git] / wt-status.c
blob4b74e6858400108fe4967cfd371d20915d806379
1 #include "wt-status.h"
2 #include "color.h"
3 #include "cache.h"
4 #include "object.h"
5 #include "dir.h"
6 #include "commit.h"
7 #include "diff.h"
8 #include "revision.h"
9 #include "diffcore.h"
11 int wt_status_use_color = 0;
12 static char wt_status_colors[][COLOR_MAXLEN] = {
13 "", /* WT_STATUS_HEADER: normal */
14 "\033[32m", /* WT_STATUS_UPDATED: green */
15 "\033[31m", /* WT_STATUS_CHANGED: red */
16 "\033[31m", /* WT_STATUS_UNTRACKED: red */
19 static int parse_status_slot(const char *var, int offset)
21 if (!strcasecmp(var+offset, "header"))
22 return WT_STATUS_HEADER;
23 if (!strcasecmp(var+offset, "updated"))
24 return WT_STATUS_UPDATED;
25 if (!strcasecmp(var+offset, "changed"))
26 return WT_STATUS_CHANGED;
27 if (!strcasecmp(var+offset, "untracked"))
28 return WT_STATUS_UNTRACKED;
29 die("bad config variable '%s'", var);
32 static const char* color(int slot)
34 return wt_status_use_color ? wt_status_colors[slot] : "";
37 void wt_status_prepare(struct wt_status *s)
39 unsigned char sha1[20];
40 const char *head;
42 s->is_initial = get_sha1("HEAD", sha1) ? 1 : 0;
44 head = resolve_ref(git_path("HEAD"), sha1, 0);
45 s->branch = head ?
46 strdup(head + strlen(get_git_dir()) + 1) :
47 NULL;
49 s->reference = "HEAD";
50 s->amend = 0;
51 s->verbose = 0;
52 s->commitable = 0;
53 s->untracked = 0;
56 static void wt_status_print_header(const char *main, const char *sub)
58 const char *c = color(WT_STATUS_HEADER);
59 color_printf_ln(c, "# %s:", main);
60 color_printf_ln(c, "# (%s)", sub);
61 color_printf_ln(c, "#");
64 static void wt_status_print_trailer(void)
66 color_printf_ln(color(WT_STATUS_HEADER), "#");
69 static void wt_status_print_filepair(int t, struct diff_filepair *p)
71 const char *c = color(t);
72 color_printf(color(WT_STATUS_HEADER), "#\t");
73 switch (p->status) {
74 case DIFF_STATUS_ADDED:
75 color_printf(c, "new file: %s", p->one->path); break;
76 case DIFF_STATUS_COPIED:
77 color_printf(c, "copied: %s -> %s",
78 p->one->path, p->two->path);
79 break;
80 case DIFF_STATUS_DELETED:
81 color_printf(c, "deleted: %s", p->one->path); break;
82 case DIFF_STATUS_MODIFIED:
83 color_printf(c, "modified: %s", p->one->path); break;
84 case DIFF_STATUS_RENAMED:
85 color_printf(c, "renamed: %s -> %s",
86 p->one->path, p->two->path);
87 break;
88 case DIFF_STATUS_TYPE_CHANGED:
89 color_printf(c, "typechange: %s", p->one->path); break;
90 case DIFF_STATUS_UNKNOWN:
91 color_printf(c, "unknown: %s", p->one->path); break;
92 case DIFF_STATUS_UNMERGED:
93 color_printf(c, "unmerged: %s", p->one->path); break;
94 default:
95 die("bug: unhandled diff status %c", p->status);
97 printf("\n");
100 static void wt_status_print_updated_cb(struct diff_queue_struct *q,
101 struct diff_options *options,
102 void *data)
104 struct wt_status *s = data;
105 int shown_header = 0;
106 int i;
107 if (q->nr) {
109 for (i = 0; i < q->nr; i++) {
110 if (q->queue[i]->status == 'U')
111 continue;
112 if (!shown_header) {
113 wt_status_print_header("Updated but not checked in",
114 "will commit");
115 s->commitable = 1;
116 shown_header = 1;
118 wt_status_print_filepair(WT_STATUS_UPDATED, q->queue[i]);
120 if (shown_header)
121 wt_status_print_trailer();
124 static void wt_status_print_changed_cb(struct diff_queue_struct *q,
125 struct diff_options *options,
126 void *data)
128 int i;
129 if (q->nr)
130 wt_status_print_header("Changed but not updated",
131 "use git-update-index to mark for commit");
132 for (i = 0; i < q->nr; i++)
133 wt_status_print_filepair(WT_STATUS_CHANGED, q->queue[i]);
134 if (q->nr)
135 wt_status_print_trailer();
138 void wt_status_print_initial(struct wt_status *s)
140 int i;
141 read_cache();
142 if (active_nr) {
143 s->commitable = 1;
144 wt_status_print_header("Updated but not checked in",
145 "will commit");
147 for (i = 0; i < active_nr; i++) {
148 color_printf(color(WT_STATUS_HEADER), "#\t");
149 color_printf_ln(color(WT_STATUS_UPDATED), "new file: %s",
150 active_cache[i]->name);
152 if (active_nr)
153 wt_status_print_trailer();
156 static void wt_status_print_updated(struct wt_status *s)
158 struct rev_info rev;
159 const char *argv[] = { NULL, NULL, NULL };
160 argv[1] = s->reference;
161 init_revisions(&rev, NULL);
162 setup_revisions(2, argv, &rev, NULL);
163 rev.diffopt.output_format |= DIFF_FORMAT_CALLBACK;
164 rev.diffopt.format_callback = wt_status_print_updated_cb;
165 rev.diffopt.format_callback_data = s;
166 rev.diffopt.detect_rename = 1;
167 run_diff_index(&rev, 1);
170 static void wt_status_print_changed(struct wt_status *s)
172 struct rev_info rev;
173 const char *argv[] = { NULL, NULL };
174 init_revisions(&rev, "");
175 setup_revisions(1, argv, &rev, NULL);
176 rev.diffopt.output_format |= DIFF_FORMAT_CALLBACK;
177 rev.diffopt.format_callback = wt_status_print_changed_cb;
178 rev.diffopt.format_callback_data = s;
179 run_diff_files(&rev, 0);
182 static void wt_status_print_untracked(const struct wt_status *s)
184 struct dir_struct dir;
185 const char *x;
186 int i;
187 int shown_header = 0;
189 memset(&dir, 0, sizeof(dir));
191 dir.exclude_per_dir = ".gitignore";
192 if (!s->untracked) {
193 dir.show_other_directories = 1;
194 dir.hide_empty_directories = 1;
196 x = git_path("info/exclude");
197 if (file_exists(x))
198 add_excludes_from_file(&dir, x);
200 read_directory(&dir, ".", "", 0);
201 for(i = 0; i < dir.nr; i++) {
202 /* check for matching entry, which is unmerged; lifted from
203 * builtin-ls-files:show_other_files */
204 struct dir_entry *ent = dir.entries[i];
205 int pos = cache_name_pos(ent->name, ent->len);
206 struct cache_entry *ce;
207 if (0 <= pos)
208 die("bug in wt_status_print_untracked");
209 pos = -pos - 1;
210 if (pos < active_nr) {
211 ce = active_cache[pos];
212 if (ce_namelen(ce) == ent->len &&
213 !memcmp(ce->name, ent->name, ent->len))
214 continue;
216 if (!shown_header) {
217 wt_status_print_header("Untracked files",
218 "use \"git add\" to add to commit");
219 shown_header = 1;
221 color_printf(color(WT_STATUS_HEADER), "#\t");
222 color_printf_ln(color(WT_STATUS_UNTRACKED), "%.*s",
223 ent->len, ent->name);
227 static void wt_status_print_verbose(struct wt_status *s)
229 struct rev_info rev;
230 const char *argv[] = { NULL, NULL, NULL };
231 argv[1] = s->reference;
232 init_revisions(&rev, NULL);
233 setup_revisions(2, argv, &rev, NULL);
234 rev.diffopt.output_format |= DIFF_FORMAT_PATCH;
235 rev.diffopt.detect_rename = 1;
236 run_diff_index(&rev, 1);
239 void wt_status_print(struct wt_status *s)
241 if (s->branch && strcmp(s->branch, "refs/heads/master"))
242 color_printf_ln(color(WT_STATUS_HEADER),
243 "# On branch %s", s->branch);
245 if (s->is_initial) {
246 color_printf_ln(color(WT_STATUS_HEADER), "#");
247 color_printf_ln(color(WT_STATUS_HEADER), "# Initial commit");
248 color_printf_ln(color(WT_STATUS_HEADER), "#");
249 wt_status_print_initial(s);
251 else {
252 wt_status_print_updated(s);
253 discard_cache();
256 wt_status_print_changed(s);
257 wt_status_print_untracked(s);
259 if (s->verbose && !s->is_initial)
260 wt_status_print_verbose(s);
261 if (!s->commitable)
262 printf("%s\n", s->amend ? "# No changes" : "nothing to commit");
265 int git_status_config(const char *k, const char *v)
267 if (!strcmp(k, "status.color")) {
268 wt_status_use_color = git_config_colorbool(k, v);
269 return 0;
271 if (!strncmp(k, "status.color.", 13)) {
272 int slot = parse_status_slot(k, 13);
273 color_parse(v, k, wt_status_colors[slot]);
275 return git_default_config(k, v);