Project has been created
This commit is contained in:
commit
60a842e195
68
.gitignore
vendored
Normal file
68
.gitignore
vendored
Normal file
|
@ -0,0 +1,68 @@
|
||||||
|
# Created by .ignore support plugin (hsz.mobi)
|
||||||
|
### JetBrains template
|
||||||
|
# Covers JetBrains IDEs: IntelliJ, RubyMine, PhpStorm, AppCode, PyCharm, CLion, Android Studio and Webstorm
|
||||||
|
# Reference: https://intellij-support.jetbrains.com/hc/en-us/articles/206544839
|
||||||
|
|
||||||
|
# User-specific stuff:
|
||||||
|
.idea/**/workspace.xml
|
||||||
|
.idea/**/tasks.xml
|
||||||
|
.idea/dictionaries
|
||||||
|
|
||||||
|
# Sensitive or high-churn files:
|
||||||
|
.idea/**/dataSources/
|
||||||
|
.idea/**/dataSources.ids
|
||||||
|
.idea/**/dataSources.xml
|
||||||
|
.idea/**/dataSources.local.xml
|
||||||
|
.idea/**/sqlDataSources.xml
|
||||||
|
.idea/**/dynamic.xml
|
||||||
|
.idea/**/uiDesigner.xml
|
||||||
|
|
||||||
|
# Gradle:
|
||||||
|
.idea/**/gradle.xml
|
||||||
|
.idea/**/libraries
|
||||||
|
|
||||||
|
# Mongo Explorer plugin:
|
||||||
|
.idea/**/mongoSettings.xml
|
||||||
|
|
||||||
|
## File-based project format:
|
||||||
|
*.iws
|
||||||
|
|
||||||
|
## Plugin-specific files:
|
||||||
|
|
||||||
|
# IntelliJ
|
||||||
|
/out/
|
||||||
|
|
||||||
|
# mpeltonen/sbt-idea plugin
|
||||||
|
.idea_modules/
|
||||||
|
|
||||||
|
# JIRA plugin
|
||||||
|
atlassian-ide-plugin.xml
|
||||||
|
|
||||||
|
# Crashlytics plugin (for Android Studio and IntelliJ)
|
||||||
|
com_crashlytics_export_strings.xml
|
||||||
|
crashlytics.properties
|
||||||
|
crashlytics-build.properties
|
||||||
|
fabric.properties
|
||||||
|
### Go template
|
||||||
|
# Binaries for programs and plugins
|
||||||
|
*.exe
|
||||||
|
*.dll
|
||||||
|
*.so
|
||||||
|
*.dylib
|
||||||
|
|
||||||
|
# Test binary, build with `go test -c`
|
||||||
|
*.test
|
||||||
|
|
||||||
|
# Output of the go coverage tool, specifically when used with LiteIDE
|
||||||
|
*.out
|
||||||
|
|
||||||
|
# Project-local glide cache, RE: https://github.com/Masterminds/glide/issues/736
|
||||||
|
.glide/
|
||||||
|
.idea/
|
||||||
|
*.iml
|
||||||
|
|
||||||
|
vendor/
|
||||||
|
glide.lock
|
||||||
|
.DS_Store
|
||||||
|
dist/
|
||||||
|
debug
|
32
.vscode/launch.json
vendored
Normal file
32
.vscode/launch.json
vendored
Normal file
|
@ -0,0 +1,32 @@
|
||||||
|
{
|
||||||
|
"version": "0.2.0",
|
||||||
|
"configurations": [
|
||||||
|
{
|
||||||
|
"name": "Debug",
|
||||||
|
"type": "go",
|
||||||
|
"request": "launch",
|
||||||
|
"mode": "debug",
|
||||||
|
"remotePath": "",
|
||||||
|
"port": 2345,
|
||||||
|
"host": "127.0.0.1",
|
||||||
|
"program": "${workspaceRoot}/main.go",
|
||||||
|
"env": {},
|
||||||
|
"args": [],
|
||||||
|
"showLog": true
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "File Debug",
|
||||||
|
"type": "go",
|
||||||
|
"request": "launch",
|
||||||
|
"mode": "debug",
|
||||||
|
"remotePath": "",
|
||||||
|
"port": 2345,
|
||||||
|
"host": "127.0.0.1",
|
||||||
|
"program": "${fileDirname}",
|
||||||
|
"env": {},
|
||||||
|
"args": [],
|
||||||
|
"showLog": true
|
||||||
|
}
|
||||||
|
|
||||||
|
]
|
||||||
|
}
|
3
.vscode/settings.json
vendored
Normal file
3
.vscode/settings.json
vendored
Normal file
|
@ -0,0 +1,3 @@
|
||||||
|
// Place your settings in this file to overwrite default and user settings.
|
||||||
|
{
|
||||||
|
}
|
57
flags.go
Normal file
57
flags.go
Normal file
|
@ -0,0 +1,57 @@
|
||||||
|
package config
|
||||||
|
|
||||||
|
import "github.com/spf13/pflag"
|
||||||
|
|
||||||
|
// FlagValueSet is an interface that users can implement
|
||||||
|
// to bind a set of flags to viper.
|
||||||
|
type FlagValueSet interface {
|
||||||
|
VisitAll(fn func(FlagValue))
|
||||||
|
}
|
||||||
|
|
||||||
|
// FlagValue is an interface that users can implement
|
||||||
|
// to bind different flags to viper.
|
||||||
|
type FlagValue interface {
|
||||||
|
HasChanged() bool
|
||||||
|
Name() string
|
||||||
|
ValueString() string
|
||||||
|
ValueType() string
|
||||||
|
}
|
||||||
|
|
||||||
|
// pflagValueSet is a wrapper around *pflag.ValueSet
|
||||||
|
// that implements FlagValueSet.
|
||||||
|
type pflagValueSet struct {
|
||||||
|
flags *pflag.FlagSet
|
||||||
|
}
|
||||||
|
|
||||||
|
// VisitAll iterates over all *pflag.Flag inside the *pflag.FlagSet.
|
||||||
|
func (p pflagValueSet) VisitAll(fn func(flag FlagValue)) {
|
||||||
|
p.flags.VisitAll(func(flag *pflag.Flag) {
|
||||||
|
fn(pflagValue{flag})
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// pflagValue is a wrapper aroung *pflag.flag
|
||||||
|
// that implements FlagValue
|
||||||
|
type pflagValue struct {
|
||||||
|
flag *pflag.Flag
|
||||||
|
}
|
||||||
|
|
||||||
|
// HasChanges returns whether the flag has changes or not.
|
||||||
|
func (p pflagValue) HasChanged() bool {
|
||||||
|
return p.flag.Changed
|
||||||
|
}
|
||||||
|
|
||||||
|
// Name returns the name of the flag.
|
||||||
|
func (p pflagValue) Name() string {
|
||||||
|
return p.flag.Name
|
||||||
|
}
|
||||||
|
|
||||||
|
// ValueString returns the value of the flag as a string.
|
||||||
|
func (p pflagValue) ValueString() string {
|
||||||
|
return p.flag.Value.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
// ValueType returns the type of the flag as a string.
|
||||||
|
func (p pflagValue) ValueType() string {
|
||||||
|
return p.flag.Value.Type()
|
||||||
|
}
|
16
glide.yaml
Normal file
16
glide.yaml
Normal file
|
@ -0,0 +1,16 @@
|
||||||
|
package: git.loafle.net/commons_go/config
|
||||||
|
import:
|
||||||
|
- package: github.com/spf13/afero
|
||||||
|
- package: gopkg.in/yaml.v2
|
||||||
|
- package: github.com/hashicorp/hcl
|
||||||
|
- package: github.com/pelletier/go-toml
|
||||||
|
version: v1.0.0
|
||||||
|
- package: github.com/magiconair/properties
|
||||||
|
version: v1.7.3
|
||||||
|
- package: github.com/spf13/cast
|
||||||
|
version: v1.1.0
|
||||||
|
- package: github.com/spf13/pflag
|
||||||
|
version: v1.0.0
|
||||||
|
- package: github.com/mitchellh/mapstructure
|
||||||
|
- package: github.com/fsnotify/fsnotify
|
||||||
|
version: v1.4.2
|
254
util.go
Normal file
254
util.go
Normal file
|
@ -0,0 +1,254 @@
|
||||||
|
package config
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"runtime"
|
||||||
|
"strings"
|
||||||
|
"unicode"
|
||||||
|
|
||||||
|
"github.com/hashicorp/hcl"
|
||||||
|
"github.com/magiconair/properties"
|
||||||
|
toml "github.com/pelletier/go-toml"
|
||||||
|
"github.com/spf13/cast"
|
||||||
|
yaml "gopkg.in/yaml.v2"
|
||||||
|
)
|
||||||
|
|
||||||
|
// ConfigParseError denotes failing to parse configuration file.
|
||||||
|
type ConfigParseError struct {
|
||||||
|
err error
|
||||||
|
}
|
||||||
|
|
||||||
|
// Error returns the formatted configuration error.
|
||||||
|
func (pe ConfigParseError) Error() string {
|
||||||
|
return fmt.Sprintf("While parsing config: %s", pe.err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
func absPathify(inPath string) (string, error) {
|
||||||
|
if strings.HasPrefix(inPath, "$HOME") {
|
||||||
|
inPath = userHomeDir() + inPath[5:]
|
||||||
|
}
|
||||||
|
|
||||||
|
if strings.HasPrefix(inPath, "$") {
|
||||||
|
end := strings.Index(inPath, string(os.PathSeparator))
|
||||||
|
inPath = os.Getenv(inPath[1:end]) + inPath[end:]
|
||||||
|
}
|
||||||
|
|
||||||
|
if filepath.IsAbs(inPath) {
|
||||||
|
return filepath.Clean(inPath), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
p, err := filepath.Abs(inPath)
|
||||||
|
if err == nil {
|
||||||
|
return filepath.Clean(p), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
func userHomeDir() string {
|
||||||
|
if runtime.GOOS == "windows" {
|
||||||
|
home := os.Getenv("HOMEDRIVE") + os.Getenv("HOMEPATH")
|
||||||
|
if home == "" {
|
||||||
|
home = os.Getenv("USERPROFILE")
|
||||||
|
}
|
||||||
|
return home
|
||||||
|
}
|
||||||
|
return os.Getenv("HOME")
|
||||||
|
}
|
||||||
|
|
||||||
|
func stringInSlice(a string, list []string) bool {
|
||||||
|
for _, b := range list {
|
||||||
|
if b == a {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func exists(path string) (bool, error) {
|
||||||
|
_, err := _c.fs.Stat(path)
|
||||||
|
if err == nil {
|
||||||
|
return true, nil
|
||||||
|
}
|
||||||
|
if os.IsNotExist(err) {
|
||||||
|
return false, nil
|
||||||
|
}
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func unmarshallConfigReader(in io.Reader, c map[string]interface{}, configType string) error {
|
||||||
|
buf := new(bytes.Buffer)
|
||||||
|
buf.ReadFrom(in)
|
||||||
|
|
||||||
|
switch strings.ToLower(configType) {
|
||||||
|
case "yaml", "yml":
|
||||||
|
if err := yaml.Unmarshal(buf.Bytes(), &c); err != nil {
|
||||||
|
return ConfigParseError{err}
|
||||||
|
}
|
||||||
|
|
||||||
|
case "json":
|
||||||
|
if err := json.Unmarshal(buf.Bytes(), &c); err != nil {
|
||||||
|
return ConfigParseError{err}
|
||||||
|
}
|
||||||
|
|
||||||
|
case "hcl":
|
||||||
|
obj, err := hcl.Parse(string(buf.Bytes()))
|
||||||
|
if err != nil {
|
||||||
|
return ConfigParseError{err}
|
||||||
|
}
|
||||||
|
if err = hcl.DecodeObject(&c, obj); err != nil {
|
||||||
|
return ConfigParseError{err}
|
||||||
|
}
|
||||||
|
|
||||||
|
case "toml":
|
||||||
|
tree, err := toml.LoadReader(buf)
|
||||||
|
if err != nil {
|
||||||
|
return ConfigParseError{err}
|
||||||
|
}
|
||||||
|
tmap := tree.ToMap()
|
||||||
|
for k, v := range tmap {
|
||||||
|
c[k] = v
|
||||||
|
}
|
||||||
|
|
||||||
|
case "properties", "props", "prop":
|
||||||
|
var p *properties.Properties
|
||||||
|
var err error
|
||||||
|
if p, err = properties.Load(buf.Bytes(), properties.UTF8); err != nil {
|
||||||
|
return ConfigParseError{err}
|
||||||
|
}
|
||||||
|
for _, key := range p.Keys() {
|
||||||
|
value, _ := p.Get(key)
|
||||||
|
// recursively build nested maps
|
||||||
|
path := strings.Split(key, ".")
|
||||||
|
lastKey := strings.ToLower(path[len(path)-1])
|
||||||
|
deepestMap := deepSearch(c, path[0:len(path)-1])
|
||||||
|
// set innermost value
|
||||||
|
deepestMap[lastKey] = value
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
insensitiviseMap(c)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func insensitiviseMap(m map[string]interface{}) {
|
||||||
|
for key, val := range m {
|
||||||
|
switch val.(type) {
|
||||||
|
case map[interface{}]interface{}:
|
||||||
|
// nested map: cast and recursively insensitivise
|
||||||
|
val = cast.ToStringMap(val)
|
||||||
|
insensitiviseMap(val.(map[string]interface{}))
|
||||||
|
case map[string]interface{}:
|
||||||
|
// nested map: recursively insensitivise
|
||||||
|
insensitiviseMap(val.(map[string]interface{}))
|
||||||
|
}
|
||||||
|
|
||||||
|
lower := strings.ToLower(key)
|
||||||
|
if key != lower {
|
||||||
|
// remove old key (not lower-cased)
|
||||||
|
delete(m, key)
|
||||||
|
}
|
||||||
|
// update map
|
||||||
|
m[lower] = val
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func deepSearch(m map[string]interface{}, path []string) map[string]interface{} {
|
||||||
|
for _, k := range path {
|
||||||
|
m2, ok := m[k]
|
||||||
|
if !ok {
|
||||||
|
// intermediate key does not exist
|
||||||
|
// => create it and continue from there
|
||||||
|
m3 := make(map[string]interface{})
|
||||||
|
m[k] = m3
|
||||||
|
m = m3
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
m3, ok := m2.(map[string]interface{})
|
||||||
|
if !ok {
|
||||||
|
// intermediate key is a value
|
||||||
|
// => replace with a new map
|
||||||
|
m3 = make(map[string]interface{})
|
||||||
|
m[k] = m3
|
||||||
|
}
|
||||||
|
// continue search from here
|
||||||
|
m = m3
|
||||||
|
}
|
||||||
|
return m
|
||||||
|
}
|
||||||
|
|
||||||
|
func toCaseInsensitiveValue(value interface{}) interface{} {
|
||||||
|
switch v := value.(type) {
|
||||||
|
case map[interface{}]interface{}:
|
||||||
|
value = copyAndInsensitiviseMap(cast.ToStringMap(v))
|
||||||
|
case map[string]interface{}:
|
||||||
|
value = copyAndInsensitiviseMap(v)
|
||||||
|
}
|
||||||
|
|
||||||
|
return value
|
||||||
|
}
|
||||||
|
|
||||||
|
func copyAndInsensitiviseMap(m map[string]interface{}) map[string]interface{} {
|
||||||
|
nm := make(map[string]interface{})
|
||||||
|
|
||||||
|
for key, val := range m {
|
||||||
|
lkey := strings.ToLower(key)
|
||||||
|
switch v := val.(type) {
|
||||||
|
case map[interface{}]interface{}:
|
||||||
|
nm[lkey] = copyAndInsensitiviseMap(cast.ToStringMap(v))
|
||||||
|
case map[string]interface{}:
|
||||||
|
nm[lkey] = copyAndInsensitiviseMap(v)
|
||||||
|
default:
|
||||||
|
nm[lkey] = v
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nm
|
||||||
|
}
|
||||||
|
|
||||||
|
func safeMul(a, b uint) uint {
|
||||||
|
c := a * b
|
||||||
|
if a > 1 && b > 1 && c/b != a {
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
return c
|
||||||
|
}
|
||||||
|
|
||||||
|
func parseSizeInBytes(sizeStr string) uint {
|
||||||
|
sizeStr = strings.TrimSpace(sizeStr)
|
||||||
|
lastChar := len(sizeStr) - 1
|
||||||
|
multiplier := uint(1)
|
||||||
|
|
||||||
|
if lastChar > 0 {
|
||||||
|
if sizeStr[lastChar] == 'b' || sizeStr[lastChar] == 'B' {
|
||||||
|
if lastChar > 1 {
|
||||||
|
switch unicode.ToLower(rune(sizeStr[lastChar-1])) {
|
||||||
|
case 'k':
|
||||||
|
multiplier = 1 << 10
|
||||||
|
sizeStr = strings.TrimSpace(sizeStr[:lastChar-1])
|
||||||
|
case 'm':
|
||||||
|
multiplier = 1 << 20
|
||||||
|
sizeStr = strings.TrimSpace(sizeStr[:lastChar-1])
|
||||||
|
case 'g':
|
||||||
|
multiplier = 1 << 30
|
||||||
|
sizeStr = strings.TrimSpace(sizeStr[:lastChar-1])
|
||||||
|
default:
|
||||||
|
multiplier = 1
|
||||||
|
sizeStr = strings.TrimSpace(sizeStr[:lastChar])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
size := cast.ToInt(sizeStr)
|
||||||
|
if size < 0 {
|
||||||
|
size = 0
|
||||||
|
}
|
||||||
|
|
||||||
|
return safeMul(uint(size), multiplier)
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user