TomcatUserDatabaseComponent.java

/*
 * Jopr Management Platform
 * Copyright (C) 2005-2008 Red Hat, Inc.
 * All rights reserved.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License, version 2, as
 * published by the Free Software Foundation, and/or the GNU Lesser
 * General Public License, version 2.1, also as published by the Free
 * Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License and the GNU Lesser General Public License
 * for more details.
 *
 * You should have received a copy of the GNU General Public License
 * and the GNU Lesser General Public License along with this program;
 * if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 */

package org.jboss.on.plugins.tomcat;

import org.jboss.on.plugins.tomcat.helper.CreateResourceHelper;

import org.rhq.core.domain.configuration.Configuration;
import org.rhq.core.domain.configuration.PropertySimple;
import org.rhq.core.domain.resource.CreateResourceStatus;
import org.rhq.core.pluginapi.inventory.CreateChildResourceFacet;
import org.rhq.core.pluginapi.inventory.CreateResourceReport;
import org.rhq.plugins.jmx.MBeanResourceComponent;

/**
 * Manage a Tomcat User Database
 * 
 * @author Jay Shaughnessy
 */
public class TomcatUserDatabaseComponent extends MBeanResourceComponent<TomcatServerComponent<?>> implements
    CreateChildResourceFacet {

    public CreateResourceReport createResource(CreateResourceReport report) {
        String resourceTypeName = report.getResourceType().getName();
        String name = null;
        String objectName = null;
        String operation = null;
        try {
            PropertySimple newGroups = null;
            PropertySimple newRoles = null;
            if (TomcatGroupComponent.RESOURCE_TYPE_NAME.equals(resourceTypeName)) {
                name = report.getResourceConfiguration().getSimple("groupname").getStringValue();
                newRoles = report.getResourceConfiguration().getSimple(TomcatGroupComponent.CONFIG_ROLES);
                report.getResourceConfiguration().remove(TomcatGroupComponent.CONFIG_ROLES);
                objectName = String.format("Users:type=Group,groupname=\"%s\",database=UserDatabase", name);
                operation = "createGroup";
            } else if (TomcatRoleComponent.RESOURCE_TYPE_NAME.equals(resourceTypeName)) {
                name = report.getResourceConfiguration().getSimple("rolename").getStringValue();
                objectName = String.format("Users:type=Role,rolename=%s,database=UserDatabase", name);
                operation = "createRole";
            } else if (TomcatUserComponent.RESOURCE_TYPE_NAME.equals(resourceTypeName)) {
                name = report.getResourceConfiguration().getSimple("username").getStringValue();
                newRoles = report.getResourceConfiguration().getSimple(TomcatUserComponent.CONFIG_ROLES);
                newGroups = report.getResourceConfiguration().getSimple(TomcatUserComponent.CONFIG_GROUPS);
                report.getResourceConfiguration().remove(TomcatUserComponent.CONFIG_ROLES);
                report.getResourceConfiguration().remove(TomcatUserComponent.CONFIG_GROUPS);
                objectName = String.format("Users:type=User,username=\"%s\",database=UserDatabase", name);
                operation = "createUser";
            } else {
                throw new UnsupportedOperationException("Unsupported Resource type: " + resourceTypeName);
            }

            // IMPORTANT: The object name must be canonicalized so it matches the resource key that
            //            MBeanResourceDiscoveryComponent uses, which is the canonical object name.
            report.setResourceKey(CreateResourceHelper.getCanonicalName(objectName));
            CreateResourceHelper.setResourceName(report, name);
            this.invokeOperation(operation, report.getResourceConfiguration());

            if (TomcatGroupComponent.RESOURCE_TYPE_NAME.equals(resourceTypeName)) {
                report.getResourceConfiguration().put(newRoles);
                // FIXME: Add newRoles to the group
            } else if (TomcatUserComponent.RESOURCE_TYPE_NAME.equals(resourceTypeName)) {
                report.getResourceConfiguration().put(newGroups);
                report.getResourceConfiguration().put(newRoles);
                // FIXME: Add newRoles and newGroups to the user
            }
            // If all went well, persist the changes to the Tomcat user Database
            save();

            report.setStatus(CreateResourceStatus.SUCCESS);

        } catch (Exception e) {
            CreateResourceHelper.setErrorOnReport(report, e);
        }

        return report;
    }

    /** Persist local changes to the user database */
    void save() throws Exception {
        invokeOperation("save", new Configuration());
    }
}