James Moger
2014-03-22 d5bccb1dd65f7f519780081edd77d00514039994
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
package com.gitblit.transport.ssh.commands;
 
import java.text.DateFormat;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
 
import org.kohsuke.args4j.Argument;
import org.kohsuke.args4j.Option;
 
import com.gitblit.utils.StringUtils;
 
public abstract class ListCommand<T> extends SshCommand {
 
    @Option(name = "--verbose", aliases = { "-v" }, usage = "verbose")
    protected boolean verbose;
 
    @Option(name = "--tabbed", aliases = { "-t" }, usage = "as tabbed output")
    private boolean tabbed;
 
    @Argument(index = 0, metaVar = "REGEX", usage = "regex filter expression")
    protected String regexFilter;
    
    private DateFormat df;
 
    protected abstract List<T> getItems();
    
    protected abstract boolean matches(T t);
    
    @Override
    public void run() {
        List<T> list = getItems();
        List<T> filtered;
        if (StringUtils.isEmpty(regexFilter)) {
            // no regex filter 
            filtered = list;
        } else {
            // regex filter the list
            filtered = new ArrayList<T>();
            for (T t : list) {
                if (matches(t)) {
                    filtered.add(t);
                }
            }
        }
 
        if (tabbed) {
            asTabbed(filtered);
        } else {
            asTable(filtered);
        }
    }
 
    protected abstract void asTable(List<T> list);
    
    protected abstract void asTabbed(List<T> list);
    
    protected void outTabbed(Object... values) {
        StringBuilder pattern = new StringBuilder();
        for (int i = 0; i < values.length; i++) {
            pattern.append("%s\t");
        }
        pattern.setLength(pattern.length() - 1);
        stdout.println(String.format(pattern.toString(), values));
    }
    
    protected String formatDate(Date date) {
        if (df == null) {
            df = new SimpleDateFormat("yyyy-MM-dd");
        }
        return df.format(date);
    }
}