001 /* 002 * Sonar, open source software quality management tool. 003 * Copyright (C) 2008-2011 SonarSource 004 * mailto:contact AT sonarsource DOT com 005 * 006 * Sonar is free software; you can redistribute it and/or 007 * modify it under the terms of the GNU Lesser General Public 008 * License as published by the Free Software Foundation; either 009 * version 3 of the License, or (at your option) any later version. 010 * 011 * Sonar is distributed in the hope that it will be useful, 012 * but WITHOUT ANY WARRANTY; without even the implied warranty of 013 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 014 * Lesser General Public License for more details. 015 * 016 * You should have received a copy of the GNU Lesser General Public 017 * License along with Sonar; if not, write to the Free Software 018 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02 019 */ 020 package org.sonar.plugins.core.security; 021 022 import org.apache.commons.lang.StringUtils; 023 import org.sonar.api.database.DatabaseSession; 024 import org.sonar.api.security.GroupRole; 025 import org.sonar.api.security.UserRole; 026 027 import javax.persistence.Query; 028 import java.util.List; 029 030 /** 031 * @since 1.12 032 */ 033 public class RoleManager { 034 035 protected static final String DEFAULT_ROLE_PREFIX = "default-"; 036 private DatabaseSession session; 037 038 public RoleManager(DatabaseSession session) { 039 this.session = session; 040 } 041 042 public void affectDefaultRolesToResource(int resourceId) { 043 for (UserRole defaultRole : getDefaultUserRoles()) { 044 session.save(createResourceRoleFromDefault(defaultRole, resourceId)); 045 } 046 for (GroupRole defaultRole : getDefaultGroupRoles()) { 047 session.save(createResourceRoleFromDefault(defaultRole, resourceId)); 048 } 049 session.commit(); 050 } 051 052 public List<UserRole> getUserRoles(int resourceId) { 053 return session.getResults(UserRole.class, "resourceId", resourceId); 054 } 055 056 public List<GroupRole> getGroupRoles(int resourceId) { 057 return session.getResults(GroupRole.class, "resourceId", resourceId); 058 } 059 060 protected List<UserRole> getDefaultUserRoles() { 061 final Query query = session.createQuery("from " + UserRole.class.getSimpleName() + " ur where ur.resourceId is null and ur.role like '" + DEFAULT_ROLE_PREFIX + "%'"); 062 return query.getResultList(); 063 } 064 065 protected List<GroupRole> getDefaultGroupRoles() { 066 final Query query = session.createQuery("from " + GroupRole.class.getSimpleName() + " gr where gr.resourceId is null and gr.role like '" + DEFAULT_ROLE_PREFIX + "%'"); 067 return query.getResultList(); 068 } 069 070 protected UserRole createResourceRoleFromDefault(UserRole defaultUserRole, int resourceId) { 071 final UserRole result = new UserRole(); 072 result.setRole(convertDefaultRoleName(defaultUserRole.getRole())); 073 result.setResourceId(resourceId); 074 result.setUserId(defaultUserRole.getUserId()); 075 return result; 076 } 077 078 protected GroupRole createResourceRoleFromDefault(GroupRole defaultUserRole, int resourceId) { 079 final GroupRole result = new GroupRole(); 080 result.setRole(convertDefaultRoleName(defaultUserRole.getRole())); 081 result.setResourceId(resourceId); 082 result.setGroupId(defaultUserRole.getGroupId()); 083 return result; 084 } 085 086 protected static String convertDefaultRoleName(String defaultRoleName) { 087 return StringUtils.substringAfter(defaultRoleName, DEFAULT_ROLE_PREFIX); 088 } 089 }