mirror of https://github.com/portainer/portainer
125 lines
3.1 KiB
Go
125 lines
3.1 KiB
Go
package migrator
|
|
|
|
import (
|
|
portainer "github.com/portainer/portainer/api"
|
|
"github.com/portainer/portainer/api/bolt/errors"
|
|
)
|
|
|
|
func (m *Migrator) migrateDBVersionTo32() error {
|
|
err := m.updateRegistriesToDB32()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = m.updateDockerhubToDB32()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (m *Migrator) updateRegistriesToDB32() error {
|
|
registries, err := m.registryService.Registries()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
endpoints, err := m.endpointService.Endpoints()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, registry := range registries {
|
|
|
|
registry.RegistryAccesses = portainer.RegistryAccesses{}
|
|
|
|
for _, endpoint := range endpoints {
|
|
|
|
filteredUserAccessPolicies := portainer.UserAccessPolicies{}
|
|
for userId, registryPolicy := range registry.UserAccessPolicies {
|
|
if _, found := endpoint.UserAccessPolicies[userId]; found {
|
|
filteredUserAccessPolicies[userId] = registryPolicy
|
|
}
|
|
}
|
|
|
|
filteredTeamAccessPolicies := portainer.TeamAccessPolicies{}
|
|
for teamId, registryPolicy := range registry.TeamAccessPolicies {
|
|
if _, found := endpoint.TeamAccessPolicies[teamId]; found {
|
|
filteredTeamAccessPolicies[teamId] = registryPolicy
|
|
}
|
|
}
|
|
|
|
registry.RegistryAccesses[endpoint.ID] = portainer.RegistryAccessPolicies{
|
|
UserAccessPolicies: filteredUserAccessPolicies,
|
|
TeamAccessPolicies: filteredTeamAccessPolicies,
|
|
Namespaces: []string{},
|
|
}
|
|
}
|
|
m.registryService.UpdateRegistry(registry.ID, ®istry)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (m *Migrator) updateDockerhubToDB32() error {
|
|
dockerhub, err := m.dockerhubService.DockerHub()
|
|
if err == errors.ErrObjectNotFound {
|
|
return nil
|
|
} else if err != nil {
|
|
return err
|
|
}
|
|
|
|
if !dockerhub.Authentication {
|
|
return nil
|
|
}
|
|
|
|
registry := &portainer.Registry{
|
|
Type: portainer.DockerHubRegistry,
|
|
Name: "Dockerhub (authenticated - migrated)",
|
|
URL: "docker.io",
|
|
Authentication: true,
|
|
Username: dockerhub.Username,
|
|
Password: dockerhub.Password,
|
|
RegistryAccesses: portainer.RegistryAccesses{},
|
|
}
|
|
|
|
endpoints, err := m.endpointService.Endpoints()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, endpoint := range endpoints {
|
|
|
|
if endpoint.Type != portainer.KubernetesLocalEnvironment &&
|
|
endpoint.Type != portainer.AgentOnKubernetesEnvironment &&
|
|
endpoint.Type != portainer.EdgeAgentOnKubernetesEnvironment {
|
|
|
|
userAccessPolicies := portainer.UserAccessPolicies{}
|
|
for userId := range endpoint.UserAccessPolicies {
|
|
if _, found := endpoint.UserAccessPolicies[userId]; found {
|
|
userAccessPolicies[userId] = portainer.AccessPolicy{
|
|
RoleID: 0,
|
|
}
|
|
}
|
|
}
|
|
|
|
teamAccessPolicies := portainer.TeamAccessPolicies{}
|
|
for teamId := range endpoint.TeamAccessPolicies {
|
|
if _, found := endpoint.TeamAccessPolicies[teamId]; found {
|
|
teamAccessPolicies[teamId] = portainer.AccessPolicy{
|
|
RoleID: 0,
|
|
}
|
|
}
|
|
}
|
|
|
|
registry.RegistryAccesses[endpoint.ID] = portainer.RegistryAccessPolicies{
|
|
UserAccessPolicies: userAccessPolicies,
|
|
TeamAccessPolicies: teamAccessPolicies,
|
|
Namespaces: []string{},
|
|
}
|
|
}
|
|
}
|
|
|
|
return m.registryService.CreateRegistry(registry)
|
|
}
|