mirror of https://github.com/portainer/portainer
90 lines
2.7 KiB
Go
90 lines
2.7 KiB
Go
package migrator
|
|
|
|
import portainer "github.com/portainer/portainer/api"
|
|
|
|
func (m *Migrator) updateResourceControlsToDBVersion22() error {
|
|
legacyResourceControls, err := m.resourceControlService.ResourceControls()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, resourceControl := range legacyResourceControls {
|
|
resourceControl.AdministratorsOnly = false
|
|
|
|
err := m.resourceControlService.UpdateResourceControl(resourceControl.ID, &resourceControl)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (m *Migrator) updateUsersAndRolesToDBVersion22() error {
|
|
legacyUsers, err := m.userService.Users()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
settings, err := m.settingsService.Settings()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, user := range legacyUsers {
|
|
user.PortainerAuthorizations = portainer.DefaultPortainerAuthorizations()
|
|
err = m.userService.UpdateUser(user.ID, &user)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
endpointAdministratorRole, err := m.roleService.Role(portainer.RoleID(1))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
endpointAdministratorRole.Priority = 1
|
|
endpointAdministratorRole.Authorizations = portainer.DefaultEndpointAuthorizationsForEndpointAdministratorRole()
|
|
|
|
err = m.roleService.UpdateRole(endpointAdministratorRole.ID, endpointAdministratorRole)
|
|
|
|
helpDeskRole, err := m.roleService.Role(portainer.RoleID(2))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
helpDeskRole.Priority = 2
|
|
helpDeskRole.Authorizations = portainer.DefaultEndpointAuthorizationsForHelpDeskRole(settings.AllowVolumeBrowserForRegularUsers)
|
|
|
|
err = m.roleService.UpdateRole(helpDeskRole.ID, helpDeskRole)
|
|
|
|
standardUserRole, err := m.roleService.Role(portainer.RoleID(3))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
standardUserRole.Priority = 3
|
|
standardUserRole.Authorizations = portainer.DefaultEndpointAuthorizationsForStandardUserRole(settings.AllowVolumeBrowserForRegularUsers)
|
|
|
|
err = m.roleService.UpdateRole(standardUserRole.ID, standardUserRole)
|
|
|
|
readOnlyUserRole, err := m.roleService.Role(portainer.RoleID(4))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
readOnlyUserRole.Priority = 4
|
|
readOnlyUserRole.Authorizations = portainer.DefaultEndpointAuthorizationsForReadOnlyUserRole(settings.AllowVolumeBrowserForRegularUsers)
|
|
|
|
err = m.roleService.UpdateRole(readOnlyUserRole.ID, readOnlyUserRole)
|
|
|
|
authorizationServiceParameters := &portainer.AuthorizationServiceParameters{
|
|
EndpointService: m.endpointService,
|
|
EndpointGroupService: m.endpointGroupService,
|
|
RegistryService: m.registryService,
|
|
RoleService: m.roleService,
|
|
TeamMembershipService: m.teamMembershipService,
|
|
UserService: m.userService,
|
|
}
|
|
|
|
authorizationService := portainer.NewAuthorizationService(authorizationServiceParameters)
|
|
return authorizationService.UpdateUsersAuthorizations()
|
|
}
|