2019-09-09 22:58:26 +00:00
|
|
|
package migrator
|
|
|
|
|
2019-10-08 03:18:32 +00:00
|
|
|
import (
|
|
|
|
"strings"
|
|
|
|
|
|
|
|
portainer "github.com/portainer/portainer/api"
|
|
|
|
)
|
2019-09-09 22:58:26 +00:00
|
|
|
|
|
|
|
func (m *Migrator) updateUsersToDBVersion20() error {
|
|
|
|
authorizationServiceParameters := &portainer.AuthorizationServiceParameters{
|
|
|
|
EndpointService: m.endpointService,
|
|
|
|
EndpointGroupService: m.endpointGroupService,
|
2019-10-07 02:42:01 +00:00
|
|
|
RegistryService: m.registryService,
|
2019-09-09 22:58:26 +00:00
|
|
|
RoleService: m.roleService,
|
|
|
|
TeamMembershipService: m.teamMembershipService,
|
|
|
|
UserService: m.userService,
|
|
|
|
}
|
|
|
|
|
|
|
|
authorizationService := portainer.NewAuthorizationService(authorizationServiceParameters)
|
2019-10-07 02:42:01 +00:00
|
|
|
return authorizationService.UpdateUsersAuthorizations()
|
2019-09-09 22:58:26 +00:00
|
|
|
}
|
2019-10-08 00:17:58 +00:00
|
|
|
|
|
|
|
func (m *Migrator) updateSettingsToDBVersion20() error {
|
|
|
|
legacySettings, err := m.settingsService.Settings()
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
legacySettings.AllowVolumeBrowserForRegularUsers = false
|
|
|
|
|
|
|
|
return m.settingsService.UpdateSettings(legacySettings)
|
|
|
|
}
|
2019-10-08 03:18:32 +00:00
|
|
|
|
|
|
|
func (m *Migrator) updateSchedulesToDBVersion20() error {
|
|
|
|
legacySchedules, err := m.scheduleService.Schedules()
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, schedule := range legacySchedules {
|
|
|
|
if schedule.JobType == portainer.ScriptExecutionJobType {
|
|
|
|
if schedule.CronExpression == "0 0 * * *" {
|
|
|
|
schedule.CronExpression = "0 * * * *"
|
|
|
|
} else if schedule.CronExpression == "0 0 0/2 * *" {
|
|
|
|
schedule.CronExpression = "0 */2 * * *"
|
|
|
|
} else if schedule.CronExpression == "0 0 0 * *" {
|
|
|
|
schedule.CronExpression = "0 0 * * *"
|
|
|
|
} else {
|
|
|
|
revisedCronExpression := strings.Split(schedule.CronExpression, " ")
|
|
|
|
if len(revisedCronExpression) == 5 {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
|
|
|
|
revisedCronExpression = revisedCronExpression[1:]
|
|
|
|
schedule.CronExpression = strings.Join(revisedCronExpression, " ")
|
|
|
|
}
|
|
|
|
|
|
|
|
err := m.scheduleService.UpdateSchedule(schedule.ID, &schedule)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|