From ca036b56c198deb9cac69b06ae78d0928ec49d8e Mon Sep 17 00:00:00 2001
From: Anthony Lapenna <lapenna.anthony@gmail.com>
Date: Tue, 28 Nov 2017 13:40:33 +0100
Subject: [PATCH] feat(database-migration): enable donation header when
 upgrading Portainer (#1450)

---
 api/bolt/migrate_dbversion6.go | 16 ++++++++++++++++
 api/bolt/migrator.go           |  8 ++++++++
 api/portainer.go               |  2 +-
 3 files changed, 25 insertions(+), 1 deletion(-)
 create mode 100644 api/bolt/migrate_dbversion6.go

diff --git a/api/bolt/migrate_dbversion6.go b/api/bolt/migrate_dbversion6.go
new file mode 100644
index 000000000..95d53af61
--- /dev/null
+++ b/api/bolt/migrate_dbversion6.go
@@ -0,0 +1,16 @@
+package bolt
+
+func (m *Migrator) updateSettingsToVersion7() error {
+	legacySettings, err := m.SettingsService.Settings()
+	if err != nil {
+		return err
+	}
+	legacySettings.DisplayDonationHeader = true
+
+	err = m.SettingsService.StoreSettings(legacySettings)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
diff --git a/api/bolt/migrator.go b/api/bolt/migrator.go
index 8a42cedc6..f74a29b34 100644
--- a/api/bolt/migrator.go
+++ b/api/bolt/migrator.go
@@ -81,6 +81,14 @@ func (m *Migrator) Migrate() error {
 		}
 	}
 
+	// https://github.com/portainer/portainer/issues/1449
+	if m.CurrentDBVersion < 7 {
+		err := m.updateSettingsToVersion7()
+		if err != nil {
+			return err
+		}
+	}
+
 	err := m.VersionService.StoreDBVersion(portainer.DBVersion)
 	if err != nil {
 		return err
diff --git a/api/portainer.go b/api/portainer.go
index 751043810..50fa9e289 100644
--- a/api/portainer.go
+++ b/api/portainer.go
@@ -392,7 +392,7 @@ const (
 	// APIVersion is the version number of the Portainer API.
 	APIVersion = "1.15.3"
 	// DBVersion is the version number of the Portainer database.
-	DBVersion = 6
+	DBVersion = 7
 	// DefaultTemplatesURL represents the default URL for the templates definitions.
 	DefaultTemplatesURL = "https://raw.githubusercontent.com/portainer/templates/master/templates.json"
 )