UserGroupModule.java

/**
 * ***************************************************************************
 * Copyright (c) 2010 Qcadoo Limited
 * Project: Qcadoo Framework
 * Version: 1.4
 *
 * This file is part of Qcadoo.
 *
 * Qcadoo is free software; you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published
 * by the Free Software Foundation; either version 3 of the License,
 * or (at your option) any later version.
 *
 * 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 Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 * ***************************************************************************
 */
package com.qcadoo.security.internal.module;

import com.qcadoo.model.api.DataDefinition;
import com.qcadoo.model.api.DataDefinitionService;
import com.qcadoo.model.api.Entity;
import com.qcadoo.model.api.search.SearchRestrictions;
import com.qcadoo.plugin.api.Module;

public class UserGroupModule extends Module {

    private final String name;

    private final String identifier;

    private final String roles;

    private final DataDefinitionService dataDefinitionService;

    public UserGroupModule(final String name, final String identifier, final String roles,
            final DataDefinitionService dataDefinitionService) {
        super();

        this.name = name;
        this.identifier = identifier;
        this.roles = roles;
        this.dataDefinitionService = dataDefinitionService;
    }

    @Override
    public void multiTenantEnable() {
        if (dataDefinitionService.get("qcadooSecurity", "group").find().add(SearchRestrictions.eq("identifier", identifier))
                .list().getTotalNumberOfEntities() > 0) {
            return;
        }

        Entity entity = dataDefinitionService.get("qcadooSecurity", "group").create();
        entity.setField("name", name);
        entity.setField("identifier", identifier);
        DataDefinition roleDD = dataDefinitionService.get("qcadooSecurity", "role");
        entity.setField("roles", roleDD.find().add(SearchRestrictions.in("identifier", (Object[]) roles.split(","))).list()
                .getEntities());
        dataDefinitionService.get("qcadooSecurity", "group").save(entity);
    }
}