Skip to content

Commit

Permalink
home: fix migration, imp code
Browse files Browse the repository at this point in the history
  • Loading branch information
@ainar-g
ainar-g committed Mar 24, 2021
1 parent e10a3fa commit 93b025a
Show file tree
Hide file tree
Showing 2 changed files with 69 additions and 94 deletions.
157 changes: 66 additions & 91 deletions internal/home/upgrade.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -26,19 +26,19 @@ type (
// Performs necessary upgrade operations if needed
func upgradeConfig() error {
// read a config file into an interface map, so we can manipulate values without losing any
diskConfig := yobj{}
diskConf := yobj{}
body, err := readConfigFile()
if err != nil {
return err
}

err = yaml.Unmarshal(body, &diskConfig)
err = yaml.Unmarshal(body, &diskConf)
if err != nil {
log.Printf("Couldn't parse config file: %s", err)
return err
}

schemaVersionInterface, ok := diskConfig["schema_version"]
schemaVersionInterface, ok := diskConf["schema_version"]
log.Tracef("got schema version %v", schemaVersionInterface)
if !ok {
// no schema version, set it to 0
Expand All @@ -57,84 +57,59 @@ func upgradeConfig() error {
return nil
}

return upgradeConfigSchema(schemaVersion, &diskConfig)
return upgradeConfigSchema(schemaVersion, diskConf)
}

// upgradeFunc is a function that upgrades a config and returns an error.
type upgradeFunc = func(diskConf yobj) (err error)

// Upgrade from oldVersion to newVersion
func upgradeConfigSchema(oldVersion int, diskConfig *yobj) error {
switch oldVersion {
case 0:
err := upgradeSchema0to1(diskConfig)
if err != nil {
return err
}
fallthrough
case 1:
err := upgradeSchema1to2(diskConfig)
if err != nil {
return err
}
fallthrough
case 2:
err := upgradeSchema2to3(diskConfig)
if err != nil {
return err
}
fallthrough
case 3:
err := upgradeSchema3to4(diskConfig)
if err != nil {
return err
}
fallthrough
case 4:
err := upgradeSchema4to5(diskConfig)
if err != nil {
return err
}
fallthrough
case 5:
err := upgradeSchema5to6(diskConfig)
if err != nil {
return err
}
fallthrough
case 6:
err := upgradeSchema6to7(diskConfig)
if err != nil {
return err
}
case 7:
err := upgradeSchema7to8(diskConfig)
if err != nil {
return err
func upgradeConfigSchema(oldVersion int, diskConf yobj) (err error) {
upgrades := []upgradeFunc{
upgradeSchema0to1,
upgradeSchema1to2,
upgradeSchema2to3,
upgradeSchema3to4,
upgradeSchema4to5,
upgradeSchema5to6,
upgradeSchema6to7,
upgradeSchema7to8,
}

n := 0
for i, u := range upgrades {
if i >= oldVersion {
err = u(diskConf)
if err != nil {
return err
}

n++
}
default:
err := fmt.Errorf("configuration file contains unknown schema_version, abort")
log.Println(err)
return err
}

configFile := config.getConfigFilename()
body, err := yaml.Marshal(diskConfig)
if n == 0 {
return fmt.Errorf("unknown configuration schema version %d", oldVersion)
}

body, err := yaml.Marshal(diskConf)
if err != nil {
log.Printf("Couldn't generate YAML file: %s", err)
return err
return fmt.Errorf("generating new config: %w", err)
}

config.fileData = body
err = file.SafeWrite(configFile, body)
confFile := config.getConfigFilename()
err = file.SafeWrite(confFile, body)
if err != nil {
log.Printf("Couldn't save YAML config: %s", err)
return err
return fmt.Errorf("saving new config: %w", err)
}

return nil
}

// The first schema upgrade:
// No more "dnsfilter.txt", filters are now kept in data/filters/
func upgradeSchema0to1(diskConfig *yobj) error {
func upgradeSchema0to1(diskConf yobj) error {
log.Printf("%s(): called", funcName())

dnsFilterPath := filepath.Join(Context.workDir, "dnsfilter.txt")
Expand All @@ -147,15 +122,15 @@ func upgradeSchema0to1(diskConfig *yobj) error {
}
}

(*diskConfig)["schema_version"] = 1
diskConf["schema_version"] = 1

return nil
}

// Second schema upgrade:
// coredns is now dns in config
// delete 'Corefile', since we don't use that anymore
func upgradeSchema1to2(diskConfig *yobj) error {
func upgradeSchema1to2(diskConf yobj) error {
log.Printf("%s(): called", funcName())

coreFilePath := filepath.Join(Context.workDir, "Corefile")
Expand All @@ -168,22 +143,22 @@ func upgradeSchema1to2(diskConfig *yobj) error {
}
}

if _, ok := (*diskConfig)["dns"]; !ok {
(*diskConfig)["dns"] = (*diskConfig)["coredns"]
delete((*diskConfig), "coredns")
if _, ok := diskConf["dns"]; !ok {
diskConf["dns"] = diskConf["coredns"]
delete(diskConf, "coredns")
}
(*diskConfig)["schema_version"] = 2
diskConf["schema_version"] = 2

return nil
}

// Third schema upgrade:
// Bootstrap DNS becomes an array
func upgradeSchema2to3(diskConfig *yobj) error {
func upgradeSchema2to3(diskConf yobj) error {
log.Printf("%s(): called", funcName())

// Let's read dns configuration from diskConfig
dnsConfig, ok := (*diskConfig)["dns"]
// Let's read dns configuration from diskConf
dnsConfig, ok := diskConf["dns"]
if !ok {
return fmt.Errorf("no DNS configuration in config file")
}
Expand All @@ -208,21 +183,21 @@ func upgradeSchema2to3(diskConfig *yobj) error {

newBootstrapConfig := []string{fmt.Sprint(bootstrapDNS)}
newDNSConfig["bootstrap_dns"] = newBootstrapConfig
(*diskConfig)["dns"] = newDNSConfig
diskConf["dns"] = newDNSConfig

// Bump schema version
(*diskConfig)["schema_version"] = 3
diskConf["schema_version"] = 3

return nil
}

// Add use_global_blocked_services=true setting for existing "clients" array
func upgradeSchema3to4(diskConfig *yobj) error {
func upgradeSchema3to4(diskConf yobj) error {
log.Printf("%s(): called", funcName())

(*diskConfig)["schema_version"] = 4
diskConf["schema_version"] = 4

clients, ok := (*diskConfig)["clients"]
clients, ok := diskConf["clients"]
if !ok {
return nil
}
Expand Down Expand Up @@ -253,12 +228,12 @@ func upgradeSchema3to4(diskConfig *yobj) error {
// - name: "..."
// password: "..."
// ...
func upgradeSchema4to5(diskConfig *yobj) error {
func upgradeSchema4to5(diskConf yobj) error {
log.Printf("%s(): called", funcName())

(*diskConfig)["schema_version"] = 5
diskConf["schema_version"] = 5

name, ok := (*diskConfig)["auth_name"]
name, ok := diskConf["auth_name"]
if !ok {
return nil
}
Expand All @@ -268,7 +243,7 @@ func upgradeSchema4to5(diskConfig *yobj) error {
return nil
}

pass, ok := (*diskConfig)["auth_pass"]
pass, ok := diskConf["auth_pass"]
if !ok {
return nil
}
Expand All @@ -292,7 +267,7 @@ func upgradeSchema4to5(diskConfig *yobj) error {
PasswordHash: string(hash),
}
users := []User{u}
(*diskConfig)["users"] = users
diskConf["users"] = users
return nil
}

Expand All @@ -308,12 +283,12 @@ func upgradeSchema4to5(diskConfig *yobj) error {
// ids:
// - 127.0.0.1
// - ...
func upgradeSchema5to6(diskConfig *yobj) error {
func upgradeSchema5to6(diskConf yobj) error {
log.Printf("%s(): called", funcName())

(*diskConfig)["schema_version"] = 6
diskConf["schema_version"] = 6

clients, ok := (*diskConfig)["clients"]
clients, ok := diskConf["clients"]
if !ok {
return nil
}
Expand Down Expand Up @@ -378,12 +353,12 @@ func upgradeSchema5to6(diskConfig *yobj) error {
// dhcpv4:
// gateway_ip: 192.168.56.1
// ...
func upgradeSchema6to7(diskConfig *yobj) error {
func upgradeSchema6to7(diskConf yobj) error {
log.Printf("Upgrade yaml: 6 to 7")

(*diskConfig)["schema_version"] = 7
diskConf["schema_version"] = 7

dhcpVal, ok := (*diskConfig)["dhcp"]
dhcpVal, ok := diskConf["dhcp"]
if !ok {
return nil
}
Expand Down Expand Up @@ -462,12 +437,12 @@ func upgradeSchema6to7(diskConfig *yobj) error {
// 'bind_hosts':
// - '127.0.0.1'
//
func upgradeSchema7to8(diskConfig *yobj) (err error) {
func upgradeSchema7to8(diskConf yobj) (err error) {
log.Printf("Upgrade yaml: 7 to 8")

(*diskConfig)["schema_version"] = 8
diskConf["schema_version"] = 8

dnsVal, ok := (*diskConfig)["dns"]
dnsVal, ok := diskConf["dns"]
if !ok {
return nil
}
Expand Down
6 changes: 3 additions & 3 deletions internal/home/upgrade_test.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@ import (
func TestUpgradeSchema1to2(t *testing.T) {
diskConf := testDiskConf(1)

err := upgradeSchema1to2(&diskConf)
err := upgradeSchema1to2(diskConf)
require.Nil(t, err)

require.Equal(t, diskConf["schema_version"], 2)
Expand All @@ -35,7 +35,7 @@ func TestUpgradeSchema1to2(t *testing.T) {
func TestUpgradeSchema2to3(t *testing.T) {
diskConf := testDiskConf(2)

err := upgradeSchema2to3(&diskConf)
err := upgradeSchema2to3(diskConf)
require.Nil(t, err)

require.Equal(t, diskConf["schema_version"], 3)
Expand Down Expand Up @@ -73,7 +73,7 @@ func TestUpgradeSchema7to8(t *testing.T) {
"schema_version": 7,
}

err := upgradeSchema7to8(&oldConf)
err := upgradeSchema7to8(oldConf)
require.Nil(t, err)

require.Equal(t, oldConf["schema_version"], 8)
Expand Down

0 comments on commit 93b025a

Please sign in to comment.