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.DataAccessRule;
import org.geoserver.security.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;
}
}