James Moger
2012-06-07 ae9e157ef4e6a3708489725d4436cc15d273308f
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
/*
 * Copyright 2011 gitblit.com.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.gitblit.wicket.pages;
 
import java.util.HashMap;
import java.util.Map;
 
import org.apache.wicket.Component;
import org.apache.wicket.PageParameters;
import org.apache.wicket.markup.html.WebPage;
import org.apache.wicket.markup.html.basic.Label;
import org.eclipse.jgit.lib.Repository;
import org.eclipse.jgit.revwalk.RevCommit;
 
import com.gitblit.GitBlit;
import com.gitblit.Keys;
import com.gitblit.utils.JGitUtils;
import com.gitblit.utils.StringUtils;
import com.gitblit.wicket.WicketUtils;
 
public class RawPage extends WebPage {
 
    public RawPage(PageParameters params) {
        super(params);
 
        if (!params.containsKey("r")) {
            error(getString("gb.repositoryNotSpecified"));
            redirectToInterceptPage(new RepositoriesPage());
        }
        final String repositoryName = WicketUtils.getRepositoryName(params);
        final String objectId = WicketUtils.getObject(params);
        final String blobPath = WicketUtils.getPath(params);
        String [] encodings = GitBlit.getEncodings();
        
        Repository r = GitBlit.self().getRepository(repositoryName);
        if (r == null) {
            error(getString("gb.canNotLoadRepository") + " " + repositoryName);
            redirectToInterceptPage(new RepositoriesPage());
            return;
        }
 
        if (StringUtils.isEmpty(blobPath)) {
            // objectid referenced raw view
            Label blobLabel = new Label("rawText", JGitUtils.getStringContent(r, objectId, encodings));
            WicketUtils.setCssClass(blobLabel, "plainprint");
            add(blobLabel);
        } else {
            // standard raw blob view
            RevCommit commit = JGitUtils.getCommit(r, objectId);
 
            String extension = null;
            if (blobPath.lastIndexOf('.') > -1) {
                extension = blobPath.substring(blobPath.lastIndexOf('.') + 1);
            }
 
            // Map the extensions to types
            Map<String, Integer> map = new HashMap<String, Integer>();
            for (String ext : GitBlit.getStrings(Keys.web.imageExtensions)) {
                map.put(ext.toLowerCase(), 2);
            }
            for (String ext : GitBlit.getStrings(Keys.web.binaryExtensions)) {
                map.put(ext.toLowerCase(), 3);
            }
 
            if (extension != null) {
                int type = 0;
                if (map.containsKey(extension)) {
                    type = map.get(extension);
                }
                Component c = null;
                switch (type) {
                case 2:
                    // TODO image blobs
                    c = new Label("rawText", "Image File");
                    break;
                case 3:
                    // TODO binary blobs
                    c = new Label("rawText", "Binary File");
                    break;
                default:
                    // plain text
                    c = new Label("rawText", JGitUtils.getStringContent(r, commit.getTree(),
                            blobPath, encodings));
                    WicketUtils.setCssClass(c, "plainprint");
                }
                add(c);
            } else {
                // plain text
                Label blobLabel = new Label("rawText", JGitUtils.getStringContent(r,
                        commit.getTree(), blobPath, encodings));
                WicketUtils.setCssClass(blobLabel, "plainprint");
                add(blobLabel);
            }
        }
        r.close();
    }
}