SelectionDataRuleRemovalLink.java

package org.geoserver.web.security;

import java.io.IOException;
import java.util.List;

import org.apache.wicket.Component;
import org.apache.wicket.ajax.AjaxRequestTarget;
import org.apache.wicket.ajax.markup.html.AjaxLink;
import org.apache.wicket.model.StringResourceModel;
import org.geoserver.catalog.CascadeDeleteVisitor;
import org.geoserver.catalog.Catalog;
import org.geoserver.security.impl.DataAccessRule;
import org.geoserver.security.impl.DataAccessRuleDAO;
import org.geoserver.web.GeoServerApplication;
import org.geoserver.web.wicket.GeoServerDialog;
import org.geoserver.web.wicket.GeoServerTablePanel;
import org.geoserver.web.wicket.ParamResourceModel;

public class SelectionDataRuleRemovalLink extends AjaxLink {
    
    GeoServerTablePanel<DataAccessRule> rules;
    GeoServerDialog dialog;

    public SelectionDataRuleRemovalLink(String id, GeoServerTablePanel<DataAccessRule> rules, GeoServerDialog dialog) {
        super(id);
        this.rules = rules;
        this.dialog = dialog;
    }

    @Override
    public void onClick(AjaxRequestTarget target) {
        final List<DataAccessRule> selection = rules.getSelection();
        if(selection.size() == 0)
            return;
        
        dialog.setTitle(new ParamResourceModel("confirmRemoval", this));
        
     // if there is something to cancel, let's warn the user about what
        // could go wrong, and if the user accepts, let's delete what's needed
        dialog.showOkCancel(target, new GeoServerDialog.DialogDelegate() {
            protected Component getContents(String id) {
                // show a confirmation panel for all the objects we have to remove
                return new ConfirmRemovalDataAccessRulePanel(id, selection) {
                    @Override
                    protected StringResourceModel canRemove(DataAccessRule data) {
                        return SelectionDataRuleRemovalLink.this.canRemove(data);
                    }
                };
            }
            
            protected boolean onSubmit(AjaxRequestTarget target, Component contents) {
                // cascade delete the whole selection
                Catalog catalog = GeoServerApplication.get().getCatalog();
                CascadeDeleteVisitor visitor = new CascadeDeleteVisitor(catalog);
                DataAccessRuleDAO dao = DataAccessRuleDAO.get();
                for (DataAccessRule rule : selection) {
                    dao.removeRule(rule);   
                }
                try {
                    dao.storeRules();
                } catch (IOException e) {
                    e.printStackTrace();
                }
                
                // the deletion will have changed what we see in the page
                // so better clear out the selection
                rules.clearSelection();
                return true;
            }
            
            @Override
            public void onClose(AjaxRequestTarget target) {
                // if the selection has been cleared out it's sign a deletion
                // occurred, so refresh the table
                if(rules.getSelection().size() == 0) {
                    setEnabled(false);
                    target.addComponent(SelectionDataRuleRemovalLink.this);
                    target.addComponent(rules);
                }
            }
            
        });
        
    }
    
    protected StringResourceModel canRemove(DataAccessRule data) {
        return null;
    }

}