Add logger command - also migrate from old config to newer config

server-config.go

@@ -0,0 +1,218 @@
+/*
+ * Minio Cloud Storage, (C) 2015 Minio, Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package main
+
+import (
+	"errors"
+	"os"
+	"os/user"
+	"path/filepath"
+
+	"github.com/minio/minio-xl/pkg/probe"
+	"github.com/minio/minio-xl/pkg/quick"
+)
+
+// configV1
+type configV1 struct {
+	Version         string `json:"version"`
+	AccessKeyID     string `json:"accessKeyId"`
+	SecretAccessKey string `json:"secretAccessKey"`
+}
+
+// configV2
+type configV2 struct {
+	Version     string `json:"version"`
+	Credentials struct {
+		AccessKeyID     string `json:"accessKeyId"`
+		SecretAccessKey string `json:"secretAccessKey"`
+	} `json:"credentials"`
+	MongoLogger struct {
+		Addr       string `json:"addr"`
+		DB         string `json:"db"`
+		Collection string `json:"collection"`
+	} `json:"mongoLogger"`
+	SyslogLogger struct {
+		Network string `json:"network"`
+		Addr    string `json:"addr"`
+	} `json:"syslogLogger"`
+	FileLogger struct {
+		Filename string `json:"filename"`
+	} `json:"fileLogger"`
+}
+
+// getConfigPath get users config path
+func getConfigPath() (string, *probe.Error) {
+	if customConfigPath != "" {
+		return customConfigPath, nil
+	}
+	u, err := user.Current()
+	if err != nil {
+		return "", probe.NewError(err)
+	}
+	configPath := filepath.Join(u.HomeDir, ".minio")
+	return configPath, nil
+}
+
+// createConfigPath create users config path
+func createConfigPath() *probe.Error {
+	configPath, err := getConfigPath()
+	if err != nil {
+		return err.Trace()
+	}
+	if err := os.MkdirAll(configPath, 0700); err != nil {
+		return probe.NewError(err)
+	}
+	return nil
+}
+
+// isAuthConfigFileExists is auth config file exists?
+func isConfigFileExists() bool {
+	if _, err := os.Stat(mustGetConfigFile()); err != nil {
+		if os.IsNotExist(err) {
+			return false
+		}
+		panic(err)
+	}
+	return true
+}
+
+// mustGetConfigFile always get users config file, if not panic
+func mustGetConfigFile() string {
+	configFile, err := getConfigFile()
+	if err != nil {
+		panic(err)
+	}
+	return configFile
+}
+
+// getConfigFile get users config file
+func getConfigFile() (string, *probe.Error) {
+	configPath, err := getConfigPath()
+	if err != nil {
+		return "", err.Trace()
+	}
+	return filepath.Join(configPath, "config.json"), nil
+}
+
+// configPath for custom config path only for testing purposes
+var customConfigPath string
+
+// saveConfig save config
+func saveConfig(a *configV2) *probe.Error {
+	configFile, err := getConfigFile()
+	if err != nil {
+		return err.Trace()
+	}
+	qc, err := quick.New(a)
+	if err != nil {
+		return err.Trace()
+	}
+	if err := qc.Save(configFile); err != nil {
+		return err.Trace()
+	}
+	return nil
+}
+
+// loadConfigV2 load config
+func loadConfigV2() (*configV2, *probe.Error) {
+	configFile, err := getConfigFile()
+	if err != nil {
+		return nil, err.Trace()
+	}
+	if _, err := os.Stat(configFile); err != nil {
+		return nil, probe.NewError(err)
+	}
+	a := &configV2{}
+	a.Version = "2"
+	qc, err := quick.New(a)
+	if err != nil {
+		return nil, err.Trace()
+	}
+	if err := qc.Load(configFile); err != nil {
+		return nil, err.Trace()
+	}
+	return qc.Data().(*configV2), nil
+}
+
+// loadConfigV1 load config
+func loadConfigV1() (*configV1, *probe.Error) {
+	configPath, err := getConfigPath()
+	if err != nil {
+		return nil, err.Trace()
+	}
+	configFile := filepath.Join(configPath, "fsUsers.json")
+	if _, err := os.Stat(configFile); err != nil {
+		return nil, probe.NewError(err)
+	}
+	a := &configV1{}
+	a.Version = "1"
+	qc, err := quick.New(a)
+	if err != nil {
+		return nil, err.Trace()
+	}
+	if err := qc.Load(configFile); err != nil {
+		return nil, err.Trace()
+	}
+	return qc.Data().(*configV1), nil
+}
+
+func newConfigV2() *configV2 {
+	config := &configV2{}
+	config.Version = "2"
+	config.Credentials.AccessKeyID = ""
+	config.Credentials.SecretAccessKey = ""
+	config.MongoLogger.Addr = ""
+	config.MongoLogger.DB = ""
+	config.MongoLogger.Collection = ""
+	config.SyslogLogger.Network = ""
+	config.SyslogLogger.Addr = ""
+	config.FileLogger.Filename = ""
+	return config
+}
+
+func migrateConfig() {
+	migrateV1ToV2()
+}
+
+func migrateV1ToV2() {
+	cv1, err := loadConfigV1()
+	if err != nil {
+		if os.IsNotExist(err.ToGoError()) {
+			return
+		}
+	}
+	fatalIf(err.Trace(), "Unable to load config version ‘1’.", nil)
+
+	if cv1.Version != "1" {
+		fatalIf(probe.NewError(errors.New("")), "Invalid version loaded ‘"+cv1.Version+"’.", nil)
+	}
+
+	cv2 := newConfigV2()
+	cv2.Credentials.AccessKeyID = cv1.AccessKeyID
+	cv2.Credentials.SecretAccessKey = cv1.SecretAccessKey
+	err = saveConfig(cv2)
+	fatalIf(err.Trace(), "Unable to save config version ‘2’.", nil)
+
+	Println("Migration from version ‘1’ to ‘2’ completed successfully.")
+
+	/// Purge old fsUsers.json file
+	configPath, err := getConfigPath()
+	fatalIf(err.Trace(), "Unable to retrieve config path.", nil)
+
+	configFile := filepath.Join(configPath, "fsUsers.json")
+	os.RemoveAll(configFile)
+}