minio/pkg/utils/config/config.go

/*
 * Mini Object 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 config

import (
	"encoding/json"
	"io"
	"os"
	"os/user"
	"path"
	"sync"
)

type Config struct {
	configPath string
	configFile string
	configLock *sync.RWMutex
	Users      map[string]User
}

type User struct {
	Name      string
	AccessKey string
	SecretKey string
}

// Initialize config directory and template config
func (c *Config) SetupConfig() error {
	u, err := user.Current()
	if err != nil {
		return err
	}

	confPath := path.Join(u.HomeDir, ".minio")
	if err := os.MkdirAll(confPath, os.ModeDir); err != nil {
		return err
	}

	c.configPath = confPath
	c.configFile = path.Join(c.configPath, "config.json")
	if _, err := os.Stat(c.configFile); os.IsNotExist(err) {
		_, err = os.Create(c.configFile)
		if err != nil {
			return err
		}
	}

	c.configLock = new(sync.RWMutex)
	return nil
}

// Get config file location
func (c *Config) GetConfigPath() string {
	return c.configPath
}

// Verify if user exists
func (c *Config) IsUserExists(username string) bool {
	for _, user := range c.Users {
		if user.Name == username {
			return true
		}
	}
	return false
}

// Get user based on accesskey
func (c *Config) GetKey(accessKey string) User {
	value, ok := c.Users[accessKey]
	if !ok {
		return User{}
	}
	return value
}

// Get user based on username
func (c *Config) GetUser(username string) User {
	for _, user := range c.Users {
		if user.Name == username {
			return user
		}
	}
	return User{}
}

// Add a new user into existing User list
func (c *Config) AddUser(user User) {
	var currentUsers map[string]User
	if len(c.Users) == 0 {
		currentUsers = make(map[string]User)
	} else {
		currentUsers = c.Users
	}
	currentUsers[user.AccessKey] = user
	c.Users = currentUsers
}

// Write encoded json in config file
func (c *Config) WriteConfig() error {
	c.configLock.Lock()
	defer c.configLock.Unlock()

	var file *os.File
	var err error

	file, err = os.OpenFile(c.configFile, os.O_WRONLY, 0666)
	defer file.Close()
	if err != nil {
		return err
	}

	encoder := json.NewEncoder(file)
	encoder.Encode(c.Users)
	return nil
}

// Read json config file and decode
func (c *Config) ReadConfig() error {
	c.configLock.RLock()
	defer c.configLock.RUnlock()

	var file *os.File
	var err error

	file, err = os.OpenFile(c.configFile, os.O_RDONLY, 0666)
	defer file.Close()
	if err != nil {
		return err
	}

	users := make(map[string]User)
	decoder := json.NewDecoder(file)
	err = decoder.Decode(&users)
	switch err {
	case io.EOF:
		return nil
	case nil:
		c.Users = users
		return nil
	default:
		return err
	}
}

/// helpers

// Load all users into memory
func Loadusers() map[string]User {
	c := Config{}
	c.SetupConfig()
	c.ReadConfig()
	return c.Users
}

// Load a given user based on accessKey
func Loadkey(accessKeyId string) User {
	c := Config{}
	c.SetupConfig()
	c.ReadConfig()
	return c.GetKey(accessKeyId)
}

// Load a given user based on username
func Loaduser(username string) User {
	c := Config{}
	c.SetupConfig()
	c.ReadConfig()
	return c.GetUser(username)
}