Chaim Lev-Ari
5ebb03cb4e
* feat(settings): introduce device mapping service * feat(containers): hide devices field when setting is on * feat(containers): prevent passing of devices when not allowed * feat(stacks): prevent non admin from device mapping * feat(stacks): disallow swarm stack creation for user * refactor(settings): replace disableDeviceMapping with allow * fix(stacks): remove check for disable device mappings from swarm * feat(settings): rename field to disable * feat(settings): supply default value for disableDeviceMapping * feat(container): check for endpoint admin
106 lines
2.3 KiB
Go
106 lines
2.3 KiB
Go
package migrator
|
|
|
|
import (
|
|
"github.com/portainer/portainer/api"
|
|
)
|
|
|
|
func (m *Migrator) updateSettingsToDBVersion23() error {
|
|
legacySettings, err := m.settingsService.Settings()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
legacySettings.AllowDeviceMappingForRegularUsers = true
|
|
|
|
return m.settingsService.UpdateSettings(legacySettings)
|
|
}
|
|
|
|
func (m *Migrator) updateTagsToDBVersion23() error {
|
|
tags, err := m.tagService.Tags()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, tag := range tags {
|
|
tag.EndpointGroups = make(map[portainer.EndpointGroupID]bool)
|
|
tag.Endpoints = make(map[portainer.EndpointID]bool)
|
|
err = m.tagService.UpdateTag(tag.ID, &tag)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (m *Migrator) updateEndpointsAndEndpointGroupsToDBVersion23() error {
|
|
tags, err := m.tagService.Tags()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
tagsNameMap := make(map[string]portainer.Tag)
|
|
for _, tag := range tags {
|
|
tagsNameMap[tag.Name] = tag
|
|
}
|
|
|
|
endpoints, err := m.endpointService.Endpoints()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, endpoint := range endpoints {
|
|
endpointTags := make([]portainer.TagID, 0)
|
|
for _, tagName := range endpoint.Tags {
|
|
tag, ok := tagsNameMap[tagName]
|
|
if ok {
|
|
endpointTags = append(endpointTags, tag.ID)
|
|
tag.Endpoints[endpoint.ID] = true
|
|
}
|
|
}
|
|
endpoint.TagIDs = endpointTags
|
|
err = m.endpointService.UpdateEndpoint(endpoint.ID, &endpoint)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
relation := &portainer.EndpointRelation{
|
|
EndpointID: endpoint.ID,
|
|
EdgeStacks: map[portainer.EdgeStackID]bool{},
|
|
}
|
|
|
|
err = m.endpointRelationService.CreateEndpointRelation(relation)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
endpointGroups, err := m.endpointGroupService.EndpointGroups()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, endpointGroup := range endpointGroups {
|
|
endpointGroupTags := make([]portainer.TagID, 0)
|
|
for _, tagName := range endpointGroup.Tags {
|
|
tag, ok := tagsNameMap[tagName]
|
|
if ok {
|
|
endpointGroupTags = append(endpointGroupTags, tag.ID)
|
|
tag.EndpointGroups[endpointGroup.ID] = true
|
|
}
|
|
}
|
|
endpointGroup.TagIDs = endpointGroupTags
|
|
err = m.endpointGroupService.UpdateEndpointGroup(endpointGroup.ID, &endpointGroup)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
for _, tag := range tagsNameMap {
|
|
err = m.tagService.UpdateTag(tag.ID, &tag)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|