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

Introduction

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

Source

/*
 * Copyright 2014 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.util.ArrayList;
import java.util.Arrays;
import java.util.List;

import org.apache.wicket.ajax.AjaxRequestTarget;
import org.apache.wicket.ajax.markup.html.AjaxLink;
import org.apache.wicket.ajax.markup.html.form.AjaxButton;
import org.apache.wicket.markup.html.basic.Label;
import org.apache.wicket.markup.html.form.Form;
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.IModel;
import org.apache.wicket.model.Model;

import com.gitblit.Constants.AccessPermission;
import com.gitblit.models.UserModel;
import com.gitblit.transport.ssh.SshKey;
import com.gitblit.utils.StringUtils;
import com.gitblit.wicket.GitBlitWebSession;

/**
 * A panel that enumerates and manages SSH public keys using AJAX.
 *
 * @author James Moger
 *
 */
public class SshKeysPanel extends BasePanel {

    private static final long serialVersionUID = 1L;

    private final UserModel user;
    private final boolean canWriteKeys;

    public SshKeysPanel(String wicketId, UserModel user) {
        super(wicketId);

        this.user = user;
        this.canWriteKeys = app().keys().supportsWritingKeys(user);
    }

    @Override
    protected void onInitialize() {
        super.onInitialize();

        setOutputMarkupId(true);

        final List<SshKey> keys = new ArrayList<SshKey>(app().keys().getKeys(user.username));
        final ListDataProvider<SshKey> dp = new ListDataProvider<SshKey>(keys);
        final DataView<SshKey> keysView = new DataView<SshKey>("keys", dp) {
            private static final long serialVersionUID = 1L;

            @Override
            public void populateItem(final Item<SshKey> item) {
                final SshKey key = item.getModelObject();
                item.add(new Label("comment", key.getComment()));
                item.add(new Label("fingerprint", key.getFingerprint()));
                item.add(new Label("permission", key.getPermission().toString()));
                item.add(new Label("algorithm", key.getAlgorithm()));

                AjaxLink<Void> delete = new AjaxLink<Void>("delete") {

                    private static final long serialVersionUID = 1L;

                    @Override
                    public void onClick(AjaxRequestTarget target) {
                        if (app().keys().removeKey(user.username, key)) {
                            // reset the keys list
                            keys.clear();
                            keys.addAll(app().keys().getKeys(user.username));

                            // update the panel
                            target.add(SshKeysPanel.this);
                        }
                    }
                };
                if (!canWriteKeys) {
                    delete.setVisibilityAllowed(false);
                }
                item.add(delete);
            }
        };
        add(keysView);

        Form<Void> addKeyForm = new Form<Void>("addKeyForm");

        final IModel<String> keyData = Model.of("");
        addKeyForm.add(new TextAreaOption("addKeyData", getString("gb.key"), null, "span5", keyData));

        final IModel<AccessPermission> keyPermission = Model.of(AccessPermission.PUSH);
        addKeyForm.add(new ChoiceOption<AccessPermission>("addKeyPermission", getString("gb.permission"),
                getString("gb.sshKeyPermissionDescription"), keyPermission,
                Arrays.asList(AccessPermission.SSHPERMISSIONS)));

        final IModel<String> keyComment = Model.of("");
        addKeyForm.add(new TextOption("addKeyComment", getString("gb.comment"),
                getString("gb.sshKeyCommentDescription"), "span5", keyComment));

        addKeyForm.add(new AjaxButton("addKeyButton") {

            private static final long serialVersionUID = 1L;

            @Override
            protected void onSubmit(AjaxRequestTarget target) {

                UserModel user = GitBlitWebSession.get().getUser();
                String data = keyData.getObject();
                if (StringUtils.isEmpty(data)) {
                    // do not submit empty key
                    return;
                }

                SshKey key = new SshKey(data);
                try {
                    key.getPublicKey();
                } catch (Exception e) {
                    // failed to parse the key
                    return;
                }

                AccessPermission permission = keyPermission.getObject();
                key.setPermission(permission);

                String comment = keyComment.getObject();
                if (!StringUtils.isEmpty(comment)) {
                    key.setComment(comment);
                }

                if (app().keys().addKey(user.username, key)) {
                    // reset add key fields
                    keyData.setObject("");
                    keyPermission.setObject(AccessPermission.PUSH);
                    keyComment.setObject("");

                    // reset the keys list
                    keys.clear();
                    keys.addAll(app().keys().getKeys(user.username));

                    // update the panel
                    target.add(SshKeysPanel.this);
                }
            }
        });

        if (!canWriteKeys) {
            addKeyForm.setVisibilityAllowed(false);
        }

        add(addKeyForm);
    }
}