/
config.go
111 lines (88 loc) · 2.22 KB
/
config.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
// Copyright 2017 Mathew Robinson <chasinglogic@gmail.com>. All rights reserved.
// Use of this source code is governed by the GPLv3 license that can be found in
// the LICENSE file.
package config
import (
"encoding/json"
"fmt"
"io/ioutil"
"os"
"path/filepath"
"strings"
)
func Init() {
if checkForOldConfig() {
upgradeConfig()
return
}
loadConfig()
}
type Config struct {
// CurrentProfileName is the currently loaded profile.
CurrentProfileName string `yaml:"current_profile"`
}
var global Config
func CurrentProfile() string {
profile := filepath.Join(ProfileDir(), global.CurrentProfileName)
if profile == ProfileDir() {
files, err := ioutil.ReadDir(ProfileDir())
if err != nil {
fmt.Println("ERROR: Unable to load profiles:", err)
os.Exit(1)
}
if len(files) == 0 {
fmt.Println("ERROR: No dfm profiles found")
os.Exit(1)
}
for _, file := range files {
if strings.HasPrefix(file.Name(), ".") {
continue
}
SetCurrentProfile(file.Name())
return filepath.Join(ProfileDir(), file.Name())
}
}
return profile
}
func GetProfileByName(name string) string {
return filepath.Join(ProfileDir(), name)
}
func AddProfile(profile string) error {
return os.Mkdir(GetProfileByName(profile), os.ModePerm)
}
func AvailableProfiles() []string {
files, err := ioutil.ReadDir(ProfileDir())
if err != nil {
fmt.Println("ERROR: Unable to read config dir:", err)
}
var profiles []string
for _, f := range files {
if strings.HasPrefix(f.Name(), ".") {
continue
}
profiles = append(profiles, f.Name())
}
return profiles
}
func Dir() string {
return GetDefaultConfigDir()
}
// SetCurrentProfile will save the config.json
func SetCurrentProfile(profile string) error {
global.CurrentProfileName = profile
jsn, merr := json.Marshal(global)
if merr != nil {
fmt.Println(merr)
return merr
}
return ioutil.WriteFile(configFile(), jsn, 0644)
}
// ProfileDir will return the config.Dir joined with profiles.
func ProfileDir() string {
return filepath.Join(GetDefaultConfigDir(), "profiles")
}
// ProfileName returns the name of the profile from it's full path
func ProfileName(path string) string {
split := strings.Split(path, string(filepath.Separator))
return split[len(split)-1]
}