com.gitblit.wicket.panels.BranchesPanel.java Source code

Java tutorial

Introduction

Here is the source code for com.gitblit.wicket.panels.BranchesPanel.java

Source

/*
 * 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.panels;

import java.io.IOException;
import java.text.MessageFormat;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

import org.apache.wicket.request.http.handler.RedirectRequestHandler;
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.model.StringResourceModel;
import org.eclipse.jgit.lib.Ref;
import org.eclipse.jgit.lib.Repository;

import com.gitblit.Constants;
import com.gitblit.models.RefModel;
import com.gitblit.models.RepositoryModel;
import com.gitblit.models.UserModel;
import com.gitblit.servlet.RawServlet;
import com.gitblit.servlet.SyndicationServlet;
import com.gitblit.utils.CommitCache;
import com.gitblit.utils.GitBlitRequestUtils;
import com.gitblit.utils.JGitUtils;
import com.gitblit.utils.RefLogUtils;
import com.gitblit.utils.StringUtils;
import com.gitblit.wicket.GitBlitWebSession;
import com.gitblit.wicket.WicketUtils;
import com.gitblit.wicket.pages.BranchesPage;
import com.gitblit.wicket.pages.CommitPage;
import com.gitblit.wicket.pages.GitSearchPage;
import com.gitblit.wicket.pages.LogPage;
import com.gitblit.wicket.pages.MetricsPage;
import com.gitblit.wicket.pages.TreePage;

public class BranchesPanel extends BasePanel {

    private static final long serialVersionUID = 1L;

    private final boolean hasBranches;

    public BranchesPanel(String wicketId, final RepositoryModel model, Repository r, final int maxCount,
            final boolean showAdmin) {
        super(wicketId);

        // branches
        List<RefModel> branches = new ArrayList<RefModel>();
        UserModel user = GitBlitWebSession.get().getUser();
        if (user == null) {
            user = UserModel.ANONYMOUS;
        }

        List<RefModel> localBranches = JGitUtils.getLocalBranches(r, false, -1);
        for (RefModel refModel : localBranches) {
            if (user.canView(model, refModel.reference.getName())) {
                branches.add(refModel);
            }
        }
        if (model.showRemoteBranches) {
            List<RefModel> remoteBranches = JGitUtils.getRemoteBranches(r, false, -1);
            for (RefModel refModel : remoteBranches) {
                if (user.canView(model, refModel.reference.getName())) {
                    branches.add(refModel);
                }
            }
        }
        Collections.sort(branches);
        Collections.reverse(branches);
        if (maxCount > 0 && branches.size() > maxCount) {
            branches = new ArrayList<RefModel>(branches.subList(0, maxCount));
        }

        if (maxCount > 0) {
            // summary page
            // show branches page link
            add(new LinkPanel("branches", "title", new StringResourceModel("gb.branches", this, null),
                    BranchesPage.class, WicketUtils.newRepositoryParameter(model.name)));
        } else {
            // branches page
            add(new Label("branches", new StringResourceModel("gb.branches", this, null)));
        }

        // only allow delete if we have multiple branches
        final boolean showDelete = showAdmin && branches.size() > 1;

        ListDataProvider<RefModel> branchesDp = new ListDataProvider<RefModel>(branches);
        DataView<RefModel> branchesView = new DataView<RefModel>("branch", branchesDp) {
            private static final long serialVersionUID = 1L;
            int counter;

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

                PageParameters shortUniqRef = WicketUtils.newObjectParameter(model.name,
                        Repository.shortenRefName(entry.getName()));

                item.add(WicketUtils.createDateLabel("branchDate", entry.getDate(), getTimeZone(), getTimeUtils()));

                item.add(new LinkPanel("branchName", "list name", StringUtils.trimString(entry.displayName, 28),
                        LogPage.class, shortUniqRef));

                String author = entry.getAuthorIdent().getName();
                LinkPanel authorLink = new LinkPanel("branchAuthor", "list", author, GitSearchPage.class,
                        WicketUtils.newSearchParameter(model.name, entry.getName(), author,
                                Constants.SearchType.AUTHOR));
                setPersonSearchTooltip(authorLink, author, Constants.SearchType.AUTHOR);
                item.add(authorLink);

                // short message
                String shortMessage = entry.getShortMessage();
                String trimmedMessage = StringUtils.trimString(shortMessage, Constants.LEN_SHORTLOG);
                LinkPanel shortlog = new LinkPanel("branchLog", "list subject", trimmedMessage, CommitPage.class,
                        shortUniqRef);
                if (!shortMessage.equals(trimmedMessage)) {
                    shortlog.setTooltip(shortMessage);
                }
                item.add(shortlog);

                if (maxCount <= 0) {
                    Fragment fragment = new Fragment("branchLinks",
                            showDelete ? "branchPageAdminLinks" : "branchPageLinks", BranchesPanel.this);
                    fragment.add(new BookmarkablePageLink<Void>("log", LogPage.class, shortUniqRef));
                    fragment.add(new BookmarkablePageLink<Void>("tree", TreePage.class, shortUniqRef));
                    String rawUrl = RawServlet.asLink(getContextUrl(), model.name,
                            Repository.shortenRefName(entry.getName()), null);
                    fragment.add(new ExternalLink("raw", rawUrl));
                    fragment.add(new BookmarkablePageLink<Void>("metrics", MetricsPage.class, shortUniqRef));
                    fragment.add(new ExternalLink("syndication",
                            SyndicationServlet.asLink(GitBlitRequestUtils.getRelativePathPrefixToContextRoot(),
                                    model.name, Repository.shortenRefName(entry.getName()), 0)));
                    if (showDelete) {
                        fragment.add(createDeleteBranchLink(model, entry));
                    }
                    item.add(fragment);
                } else {
                    Fragment fragment = new Fragment("branchLinks", "branchPanelLinks", BranchesPanel.this);
                    fragment.add(new BookmarkablePageLink<Void>("log", LogPage.class, shortUniqRef));
                    fragment.add(new BookmarkablePageLink<Void>("tree", TreePage.class, shortUniqRef));
                    String rawUrl = RawServlet.asLink(getContextUrl(), model.name,
                            Repository.shortenRefName(entry.getName()), null);
                    fragment.add(new ExternalLink("raw", rawUrl));
                    item.add(fragment);
                }
                WicketUtils.setAlternatingBackground(item, counter);
                counter++;
            }
        };
        add(branchesView);
        if (branches.size() < maxCount || maxCount <= 0) {
            add(new Label("allBranches", "").setVisible(false));
        } else {
            add(new LinkPanel("allBranches", "link", new StringResourceModel("gb.allBranches", this, null),
                    BranchesPage.class, WicketUtils.newRepositoryParameter(model.name)));
        }
        // We always have 1 branch
        hasBranches = (branches.size() > 1)
                || ((branches.size() == 1) && !branches.get(0).displayName.equalsIgnoreCase("master"));
    }

    public BranchesPanel hideIfEmpty() {
        setVisible(hasBranches);
        return this;
    }

    private Link<Void> createDeleteBranchLink(final RepositoryModel repositoryModel, final RefModel entry) {
        Link<Void> deleteLink = new Link<Void>("deleteBranch") {
            private static final long serialVersionUID = 1L;

            @Override
            public void onClick() {
                Repository r = app().repositories().getRepository(repositoryModel.name);
                if (r == null) {
                    if (app().repositories().isCollectingGarbage(repositoryModel.name)) {
                        error(MessageFormat.format(getString("gb.busyCollectingGarbage"), repositoryModel.name));
                    } else {
                        error(MessageFormat.format("Failed to find repository {0}", repositoryModel.name));
                    }
                    return;
                }
                final String branch = entry.getName();
                Ref ref = null;
                try {
                    ref = r.findRef(branch);
                    if (ref == null && !branch.startsWith(Constants.R_HEADS)) {
                        ref = r.findRef(Constants.R_HEADS + branch);
                    }
                } catch (IOException e) {
                }
                if (ref != null) {
                    boolean success = JGitUtils.deleteBranchRef(r, ref.getName());
                    if (success) {
                        // clear commit cache
                        CommitCache.instance().clear(repositoryModel.name, branch);

                        // optionally update reflog
                        if (RefLogUtils.hasRefLogBranch(r)) {
                            UserModel user = GitBlitWebSession.get().getUser();
                            RefLogUtils.deleteRef(user, r, ref);
                        }
                    }

                    if (success) {
                        info(MessageFormat.format("Branch \"{0}\" deleted", branch));
                    } else {
                        error(MessageFormat.format("Failed to delete branch \"{0}\"", branch));
                    }
                }
                r.close();

                // redirect to the owning page
                PageParameters params = WicketUtils.newRepositoryParameter(repositoryModel.name);
                String absoluteUrl = GitBlitRequestUtils.toAbsoluteUrl(getPage().getClass(), params);
                getRequestCycle().scheduleRequestHandlerAfterCurrent(new RedirectRequestHandler(absoluteUrl));
            }
        };

        deleteLink.add(new JavascriptEventConfirmation("click",
                MessageFormat.format("Delete branch \"{0}\"?", entry.displayName)));
        return deleteLink;
    }
}