src/main/java/com/gitblit/wicket/pages/TreePage.java - gitblit - Git at Google

 /*
  * 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.io.OutputStream;
 import java.util.List;

 import org.apache.wicket.request.handler.resource.ResourceStreamRequestHandler;
 import org.apache.wicket.request.mapper.parameter.PageParameters;
 import org.apache.wicket.markup.html.basic.Label;
 import org.apache.wicket.markup.html.link.BookmarkablePageLink;
 import org.apache.wicket.markup.html.link.ExternalLink;
 import org.apache.wicket.markup.html.link.Link;
 import org.apache.wicket.markup.html.panel.Fragment;
 import org.apache.wicket.markup.repeater.Item;
 import org.apache.wicket.markup.repeater.data.DataView;
 import org.apache.wicket.markup.repeater.data.ListDataProvider;
 import org.apache.wicket.util.resource.AbstractResourceStreamWriter;
 import org.apache.wicket.util.resource.IResourceStream;
 import org.eclipse.jgit.lib.FileMode;
 import org.eclipse.jgit.lib.Repository;
 import org.eclipse.jgit.revwalk.RevCommit;

 import com.gitblit.models.PathModel;
 import com.gitblit.models.SubmoduleModel;
 import com.gitblit.models.UserModel;
 import com.gitblit.servlet.RawServlet;
 import com.gitblit.utils.ByteFormat;
 import com.gitblit.utils.GitBlitRequestUtils;
 import com.gitblit.utils.JGitUtils;
 import com.gitblit.wicket.CacheControl;
 import com.gitblit.wicket.GitBlitWebSession;
 import com.gitblit.wicket.CacheControl.LastModified;
 import com.gitblit.wicket.WicketUtils;
 import com.gitblit.wicket.panels.CommitHeaderPanel;
 import com.gitblit.wicket.panels.CompressedDownloadsPanel;
 import com.gitblit.wicket.panels.LinkPanel;
 import com.gitblit.wicket.panels.PathBreadcrumbsPanel;

 @CacheControl(LastModified.BOOT)
 public class TreePage extends RepositoryPage {

 	private static final long serialVersionUID = 1L;

 	public TreePage(PageParameters params) {
 		super(params);

 		final String path = WicketUtils.getPath(params);

 		Repository r = getRepository();
 		RevCommit commit = getCommit();
 		List<PathModel> paths = JGitUtils.getFilesInPath2(r, path, commit);

 		// tree page links
 		add(new BookmarkablePageLink<Void>("historyLink", HistoryPage.class,
 				WicketUtils.newPathParameter(repositoryName, objectId, path)));
 		add(new CompressedDownloadsPanel("compressedLinks", GitBlitRequestUtils.getRelativePathPrefixToContextRoot(),
 				repositoryName, objectId, path));

 		add(new CommitHeaderPanel("commitHeader", repositoryName, commit));

 		// breadcrumbs
 		add(new PathBreadcrumbsPanel("breadcrumbs", repositoryName, path, objectId));
 		if (path != null && path.trim().length() > 0) {
 			// add .. parent path entry
 			String parentPath = null;
 			if (path.lastIndexOf('/') > -1) {
 				parentPath = path.substring(0, path.lastIndexOf('/'));
 			}
 			PathModel model = new PathModel("..", parentPath, null, 0, FileMode.TREE.getBits(), null, objectId);
 			model.isParentPath = true;
 			paths.add(0, model);
 		}

 		final String id = getBestCommitId(commit);

 		final ByteFormat byteFormat = new ByteFormat();
 		final String baseUrl = WicketUtils.getGitblitURL(getRequest());

 		// changed paths list
 		ListDataProvider<PathModel> pathsDp = new ListDataProvider<PathModel>(paths);
 		DataView<PathModel> pathsView = new DataView<PathModel>("changedPath", pathsDp) {
 			private static final long serialVersionUID = 1L;
 			int counter;

 			@Override
 			public void populateItem(final Item<PathModel> item) {
 				final PathModel entry = item.getModelObject();

 				item.add(new Label("pathPermissions", JGitUtils.getPermissionsFromMode(entry.mode)));
 				item.add(WicketUtils.setHtmlTooltip(new Label("filestore", ""), getString("gb.filestore"))
 						.setVisible(entry.isFilestoreItem()));

 				if (entry.isParentPath) {
 					// parent .. path
 					item.add(WicketUtils.newBlankImage("pathIcon"));
 					item.add(new Label("pathSize", ""));
 					item.add(new LinkPanel("pathName", null, entry.name, TreePage.class,
 							WicketUtils.newPathParameter(repositoryName, id, entry.path)));
 					item.add(new Label("pathLinks", ""));
 				} else {
 					if (entry.isTree()) {
 						// folder/tree link
 						item.add(WicketUtils.newImage("pathIcon", "folder_16x16.png"));
 						item.add(new Label("pathSize", ""));
 						item.add(new LinkPanel("pathName", "list", entry.name, TreePage.class,
 								WicketUtils.newPathParameter(repositoryName, id, entry.path)));

 						// links
 						Fragment links = new Fragment("pathLinks", "treeLinks", TreePage.this);
 						links.add(new BookmarkablePageLink<Void>("tree", TreePage.class,
 								WicketUtils.newPathParameter(repositoryName, id, entry.path)));
 						links.add(new BookmarkablePageLink<Void>("history", HistoryPage.class,
 								WicketUtils.newPathParameter(repositoryName, id, entry.path)));
 						links.add(new CompressedDownloadsPanel("compressedLinks", baseUrl, repositoryName, objectId,
 								entry.path));

 						item.add(links);
 					} else if (entry.isSubmodule()) {
 						// submodule
 						String submoduleId = entry.objectId;
 						String submodulePath;
 						boolean hasSubmodule = false;
 						SubmoduleModel submodule = getSubmodule(entry.path);
 						submodulePath = submodule.gitblitPath;
 						hasSubmodule = submodule.hasSubmodule;

 						item.add(WicketUtils.newImage("pathIcon", "git-orange-16x16.png"));
 						item.add(new Label("pathSize", ""));
 						item.add(new LinkPanel("pathName", "list", entry.name + " @ " + getShortObjectId(submoduleId),
 								TreePage.class, WicketUtils.newPathParameter(submodulePath, submoduleId, ""))
 										.setEnabled(hasSubmodule));

 						Fragment links = new Fragment("pathLinks", "submoduleLinks", TreePage.this);
 						links.add(new BookmarkablePageLink<Void>("view", SummaryPage.class,
 								WicketUtils.newRepositoryParameter(submodulePath)).setEnabled(hasSubmodule));
 						links.add(new BookmarkablePageLink<Void>("tree", TreePage.class,
 								WicketUtils.newPathParameter(submodulePath, submoduleId, "")).setEnabled(hasSubmodule));
 						links.add(new BookmarkablePageLink<Void>("history", HistoryPage.class,
 								WicketUtils.newPathParameter(repositoryName, id, entry.path)));
 						links.add(
 								new CompressedDownloadsPanel("compressedLinks", baseUrl, submodulePath, submoduleId, "")
 										.setEnabled(hasSubmodule));
 						item.add(links);
 					} else {
 						// blob link
 						String displayPath = entry.name;
 						String path = entry.path;
 						if (entry.isSymlink()) {
 							path = JGitUtils.getStringContent(getRepository(), getCommit().getTree(), path);
 							displayPath = entry.name + " -> " + path;
 						}
 						item.add(WicketUtils.getFileImage("pathIcon", entry.name));
 						item.add(new Label("pathSize", byteFormat.format(entry.size)));

 						// links
 						Fragment links = new Fragment("pathLinks", "blobLinks", TreePage.this);

 						if (entry.isFilestoreItem()) {
 							item.add(new LinkPanel("pathName", "list", displayPath, new Link<Object>("link", null) {

 								private static final long serialVersionUID = 1L;

 								@Override
 								public void onClick() {

 									IResourceStream resourceStream = new AbstractResourceStreamWriter() {

 										private static final long serialVersionUID = 1L;

 										@Override
 										public void write(OutputStream output) {
 											UserModel user = GitBlitWebSession.get().getUser();
 											user = user == null ? UserModel.ANONYMOUS : user;

 											app().filestore().downloadBlob(entry.getFilestoreOid(), user,
 													getRepositoryModel(), output);
 										}
 									};

 									ResourceStreamRequestHandler resourceStreamRequestHandler = new ResourceStreamRequestHandler(
 											resourceStream, entry.path);
 									getRequestCycle().scheduleRequestHandlerAfterCurrent(resourceStreamRequestHandler);

 								}
 							}));

 							links.add(new Link<Object>("view", null) {

 								private static final long serialVersionUID = 1L;

 								@Override
 								public void onClick() {

 									IResourceStream resourceStream = new AbstractResourceStreamWriter() {

 										private static final long serialVersionUID = 1L;

 										@Override
 										public void write(OutputStream output) {
 											UserModel user = GitBlitWebSession.get().getUser();
 											user = user == null ? UserModel.ANONYMOUS : user;

 											app().filestore().downloadBlob(entry.getFilestoreOid(), user,
 													getRepositoryModel(), output);
 										}
 									};

 									ResourceStreamRequestHandler resourceStreamRequestHandler = new ResourceStreamRequestHandler(
 											resourceStream, entry.path);
 									getRequestCycle().scheduleRequestHandlerAfterCurrent(resourceStreamRequestHandler);
 								}
 							});

 							links.add(new Link<Object>("raw", null) {

 								private static final long serialVersionUID = 1L;

 								@Override
 								public void onClick() {

 									IResourceStream resourceStream = new AbstractResourceStreamWriter() {

 										private static final long serialVersionUID = 1L;

 										@Override
 										public void write(OutputStream output) {
 											UserModel user = GitBlitWebSession.get().getUser();
 											user = user == null ? UserModel.ANONYMOUS : user;

 											app().filestore().downloadBlob(entry.getFilestoreOid(), user,
 													getRepositoryModel(), output);
 										}
 									};

 									ResourceStreamRequestHandler resourceStreamRequestHandler = new ResourceStreamRequestHandler(
 											resourceStream, entry.path);
 									getRequestCycle().scheduleRequestHandlerAfterCurrent(resourceStreamRequestHandler);
 								}
 							});

 						} else {
 							item.add(new LinkPanel("pathName", "list", displayPath, BlobPage.class,
 									WicketUtils.newPathParameter(repositoryName, id, path)));

 							links.add(new BookmarkablePageLink<Void>("view", BlobPage.class,
 									WicketUtils.newPathParameter(repositoryName, id, path)));
 							String rawUrl = RawServlet.asLink(getContextUrl(), repositoryName, id, path);
 							links.add(new ExternalLink("raw", rawUrl));
 						}

 						links.add(new BookmarkablePageLink<Void>("blame", BlamePage.class,
 								WicketUtils.newPathParameter(repositoryName, id, path)));
 						links.add(new BookmarkablePageLink<Void>("history", HistoryPage.class,
 								WicketUtils.newPathParameter(repositoryName, id, path)));
 						item.add(links);
 					}
 				}
 				WicketUtils.setAlternatingBackground(item, counter);
 				counter++;
 			}
 		};
 		add(pathsView);
 	}

 	@Override
 	protected String getPageName() {
 		return getString("gb.tree");
 	}

 	@Override
 	protected boolean isCommitPage() {
 		return true;
 	}

 }
	/*
	* 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.io.OutputStream;
	import java.util.List;

	import org.apache.wicket.request.handler.resource.ResourceStreamRequestHandler;
	import org.apache.wicket.request.mapper.parameter.PageParameters;
	import org.apache.wicket.markup.html.basic.Label;
	import org.apache.wicket.markup.html.link.BookmarkablePageLink;
	import org.apache.wicket.markup.html.link.ExternalLink;
	import org.apache.wicket.markup.html.link.Link;
	import org.apache.wicket.markup.html.panel.Fragment;
	import org.apache.wicket.markup.repeater.Item;
	import org.apache.wicket.markup.repeater.data.DataView;
	import org.apache.wicket.markup.repeater.data.ListDataProvider;
	import org.apache.wicket.util.resource.AbstractResourceStreamWriter;
	import org.apache.wicket.util.resource.IResourceStream;
	import org.eclipse.jgit.lib.FileMode;
	import org.eclipse.jgit.lib.Repository;
	import org.eclipse.jgit.revwalk.RevCommit;

	import com.gitblit.models.PathModel;
	import com.gitblit.models.SubmoduleModel;
	import com.gitblit.models.UserModel;
	import com.gitblit.servlet.RawServlet;
	import com.gitblit.utils.ByteFormat;
	import com.gitblit.utils.GitBlitRequestUtils;
	import com.gitblit.utils.JGitUtils;
	import com.gitblit.wicket.CacheControl;
	import com.gitblit.wicket.GitBlitWebSession;
	import com.gitblit.wicket.CacheControl.LastModified;
	import com.gitblit.wicket.WicketUtils;
	import com.gitblit.wicket.panels.CommitHeaderPanel;
	import com.gitblit.wicket.panels.CompressedDownloadsPanel;
	import com.gitblit.wicket.panels.LinkPanel;
	import com.gitblit.wicket.panels.PathBreadcrumbsPanel;

	@CacheControl(LastModified.BOOT)
	public class TreePage extends RepositoryPage {

	private static final long serialVersionUID = 1L;

	public TreePage(PageParameters params) {
	super(params);

	final String path = WicketUtils.getPath(params);

	Repository r = getRepository();
	RevCommit commit = getCommit();
	List<PathModel> paths = JGitUtils.getFilesInPath2(r, path, commit);

	// tree page links
	add(new BookmarkablePageLink<Void>("historyLink", HistoryPage.class,
	WicketUtils.newPathParameter(repositoryName, objectId, path)));
	add(new CompressedDownloadsPanel("compressedLinks", GitBlitRequestUtils.getRelativePathPrefixToContextRoot(),
	repositoryName, objectId, path));

	add(new CommitHeaderPanel("commitHeader", repositoryName, commit));

	// breadcrumbs
	add(new PathBreadcrumbsPanel("breadcrumbs", repositoryName, path, objectId));
	if (path != null && path.trim().length() > 0) {
	// add .. parent path entry
	String parentPath = null;
	if (path.lastIndexOf('/') > -1) {
	parentPath = path.substring(0, path.lastIndexOf('/'));
	}
	PathModel model = new PathModel("..", parentPath, null, 0, FileMode.TREE.getBits(), null, objectId);
	model.isParentPath = true;
	paths.add(0, model);
	}

	final String id = getBestCommitId(commit);

	final ByteFormat byteFormat = new ByteFormat();
	final String baseUrl = WicketUtils.getGitblitURL(getRequest());

	// changed paths list
	ListDataProvider<PathModel> pathsDp = new ListDataProvider<PathModel>(paths);
	DataView<PathModel> pathsView = new DataView<PathModel>("changedPath", pathsDp) {
	private static final long serialVersionUID = 1L;
	int counter;

	@Override
	public void populateItem(final Item<PathModel> item) {
	final PathModel entry = item.getModelObject();

	item.add(new Label("pathPermissions", JGitUtils.getPermissionsFromMode(entry.mode)));
	item.add(WicketUtils.setHtmlTooltip(new Label("filestore", ""), getString("gb.filestore"))
	.setVisible(entry.isFilestoreItem()));

	if (entry.isParentPath) {
	// parent .. path
	item.add(WicketUtils.newBlankImage("pathIcon"));
	item.add(new Label("pathSize", ""));
	item.add(new LinkPanel("pathName", null, entry.name, TreePage.class,
	WicketUtils.newPathParameter(repositoryName, id, entry.path)));
	item.add(new Label("pathLinks", ""));
	} else {
	if (entry.isTree()) {
	// folder/tree link
	item.add(WicketUtils.newImage("pathIcon", "folder_16x16.png"));
	item.add(new Label("pathSize", ""));
	item.add(new LinkPanel("pathName", "list", entry.name, TreePage.class,
	WicketUtils.newPathParameter(repositoryName, id, entry.path)));

	// links
	Fragment links = new Fragment("pathLinks", "treeLinks", TreePage.this);
	links.add(new BookmarkablePageLink<Void>("tree", TreePage.class,
	WicketUtils.newPathParameter(repositoryName, id, entry.path)));
	links.add(new BookmarkablePageLink<Void>("history", HistoryPage.class,
	WicketUtils.newPathParameter(repositoryName, id, entry.path)));
	links.add(new CompressedDownloadsPanel("compressedLinks", baseUrl, repositoryName, objectId,
	entry.path));

	item.add(links);
	} else if (entry.isSubmodule()) {
	// submodule
	String submoduleId = entry.objectId;
	String submodulePath;
	boolean hasSubmodule = false;
	SubmoduleModel submodule = getSubmodule(entry.path);
	submodulePath = submodule.gitblitPath;
	hasSubmodule = submodule.hasSubmodule;

	item.add(WicketUtils.newImage("pathIcon", "git-orange-16x16.png"));
	item.add(new Label("pathSize", ""));
	item.add(new LinkPanel("pathName", "list", entry.name + " @ " + getShortObjectId(submoduleId),
	TreePage.class, WicketUtils.newPathParameter(submodulePath, submoduleId, ""))
	.setEnabled(hasSubmodule));

	Fragment links = new Fragment("pathLinks", "submoduleLinks", TreePage.this);
	links.add(new BookmarkablePageLink<Void>("view", SummaryPage.class,
	WicketUtils.newRepositoryParameter(submodulePath)).setEnabled(hasSubmodule));
	links.add(new BookmarkablePageLink<Void>("tree", TreePage.class,
	WicketUtils.newPathParameter(submodulePath, submoduleId, "")).setEnabled(hasSubmodule));
	links.add(new BookmarkablePageLink<Void>("history", HistoryPage.class,
	WicketUtils.newPathParameter(repositoryName, id, entry.path)));
	links.add(
	new CompressedDownloadsPanel("compressedLinks", baseUrl, submodulePath, submoduleId, "")
	.setEnabled(hasSubmodule));
	item.add(links);
	} else {
	// blob link
	String displayPath = entry.name;
	String path = entry.path;
	if (entry.isSymlink()) {
	path = JGitUtils.getStringContent(getRepository(), getCommit().getTree(), path);
	displayPath = entry.name + " -> " + path;
	}
	item.add(WicketUtils.getFileImage("pathIcon", entry.name));
	item.add(new Label("pathSize", byteFormat.format(entry.size)));

	// links
	Fragment links = new Fragment("pathLinks", "blobLinks", TreePage.this);

	if (entry.isFilestoreItem()) {
	item.add(new LinkPanel("pathName", "list", displayPath, new Link<Object>("link", null) {

	private static final long serialVersionUID = 1L;

	@Override
	public void onClick() {

	IResourceStream resourceStream = new AbstractResourceStreamWriter() {

	private static final long serialVersionUID = 1L;

	@Override
	public void write(OutputStream output) {
	UserModel user = GitBlitWebSession.get().getUser();
	user = user == null ? UserModel.ANONYMOUS : user;

	app().filestore().downloadBlob(entry.getFilestoreOid(), user,
	getRepositoryModel(), output);
	}
	};

	ResourceStreamRequestHandler resourceStreamRequestHandler = new ResourceStreamRequestHandler(
	resourceStream, entry.path);
	getRequestCycle().scheduleRequestHandlerAfterCurrent(resourceStreamRequestHandler);

	}
	}));

	links.add(new Link<Object>("view", null) {

	private static final long serialVersionUID = 1L;

	@Override
	public void onClick() {

	IResourceStream resourceStream = new AbstractResourceStreamWriter() {

	private static final long serialVersionUID = 1L;

	@Override
	public void write(OutputStream output) {
	UserModel user = GitBlitWebSession.get().getUser();
	user = user == null ? UserModel.ANONYMOUS : user;

	app().filestore().downloadBlob(entry.getFilestoreOid(), user,
	getRepositoryModel(), output);
	}
	};

	ResourceStreamRequestHandler resourceStreamRequestHandler = new ResourceStreamRequestHandler(
	resourceStream, entry.path);
	getRequestCycle().scheduleRequestHandlerAfterCurrent(resourceStreamRequestHandler);
	}
	});

	links.add(new Link<Object>("raw", null) {

	private static final long serialVersionUID = 1L;

	@Override
	public void onClick() {

	IResourceStream resourceStream = new AbstractResourceStreamWriter() {

	private static final long serialVersionUID = 1L;

	@Override
	public void write(OutputStream output) {
	UserModel user = GitBlitWebSession.get().getUser();
	user = user == null ? UserModel.ANONYMOUS : user;

	app().filestore().downloadBlob(entry.getFilestoreOid(), user,
	getRepositoryModel(), output);
	}
	};

	ResourceStreamRequestHandler resourceStreamRequestHandler = new ResourceStreamRequestHandler(
	resourceStream, entry.path);
	getRequestCycle().scheduleRequestHandlerAfterCurrent(resourceStreamRequestHandler);
	}
	});

	} else {
	item.add(new LinkPanel("pathName", "list", displayPath, BlobPage.class,
	WicketUtils.newPathParameter(repositoryName, id, path)));

	links.add(new BookmarkablePageLink<Void>("view", BlobPage.class,
	WicketUtils.newPathParameter(repositoryName, id, path)));
	String rawUrl = RawServlet.asLink(getContextUrl(), repositoryName, id, path);
	links.add(new ExternalLink("raw", rawUrl));
	}

	links.add(new BookmarkablePageLink<Void>("blame", BlamePage.class,
	WicketUtils.newPathParameter(repositoryName, id, path)));
	links.add(new BookmarkablePageLink<Void>("history", HistoryPage.class,
	WicketUtils.newPathParameter(repositoryName, id, path)));
	item.add(links);
	}
	}
	WicketUtils.setAlternatingBackground(item, counter);
	counter++;
	}
	};
	add(pathsView);
	}

	@Override
	protected String getPageName() {
	return getString("gb.tree");
	}

	@Override
	protected boolean isCommitPage() {
	return true;
	}

	}