diff options
| -rw-r--r-- | bridge/core/token.go | 211 |
1 files changed, 211 insertions, 0 deletions
diff --git a/bridge/core/token.go b/bridge/core/token.go new file mode 100644 index 00000000..6cf2fbd5 --- /dev/null +++ b/bridge/core/token.go @@ -0,0 +1,211 @@ +package core + +import ( + "fmt" + "regexp" + "strconv" + "strings" + + "github.com/MichaelMure/git-bug/cache" +) + +const ( + tokenConfigKeyPrefix = "git-bug.token" + tokenKeyValue = "value" + tokenKeyTarget = "target" + tokenKeyGlobal = "global" + tokenKeyScopes = "scopes" +) + +type Token struct { + Name string + Value string + Target string + Global bool + Scopes []string +} + +func NewToken(name, value, target string, global bool, scopes []string) *Token { + return &Token{ + Name: name, + Value: value, + Target: target, + Global: global, + Scopes: scopes, + } +} + +func (t *Token) Valnameate() error { + if t.Name == "" { + return fmt.Errorf("missing token name") + } + if t.Value == "" { + return fmt.Errorf("missing token value") + } + if t.Target == "" { + return fmt.Errorf("missing token target") + } + return nil +} + +func loadToken(repo *cache.RepoCache, name string, global bool) (*Token, error) { + keyPrefix := fmt.Sprintf("git-bug.token.%s", name) + var pairs map[string]string + var err error + if global { + pairs, err = repo.ReadGlobalConfigs(keyPrefix) + if err != nil { + return nil, err + } + + } else { + pairs, err = repo.ReadConfigs(keyPrefix) + if err != nil { + return nil, err + } + } + + result := make(Configuration, len(pairs)) + for key, value := range pairs { + key := strings.TrimPrefix(key, keyPrefix) + result[key] = value + } + + var ok bool + token := &Token{Name: name} + token.Value, ok = result[tokenKeyValue] + if !ok { + return nil, fmt.Errorf("nil value") + } + + token.Target, ok = result[tokenKeyTarget] + if !ok { + return nil, fmt.Errorf("nil value") + } + + if g, ok := result[tokenKeyGlobal]; !ok { + return nil, fmt.Errorf("nil value") + } else if g == "true" { + token.Global = true + } + + scopesString, ok := result[tokenKeyScopes] + if !ok { + return nil, fmt.Errorf("missing scopes config") + } + + token.Scopes = strings.Split(scopesString, ",") + return token, nil +} + +func GetToken(repo *cache.RepoCache, name string) (*Token, error) { + return loadToken(repo, name, false) +} + +func GetGlobalToken(repo *cache.RepoCache, name string) (*Token, error) { + return loadToken(repo, name, true) +} + +func listTokens(repo *cache.RepoCache, global bool) ([]string, error) { + var configs map[string]string + var err error + if global { + configs, err = repo.ReadConfigs(tokenConfigKeyPrefix + ".") + if err != nil { + return nil, err + } + } else { + configs, err = repo.ReadGlobalConfigs(tokenConfigKeyPrefix + ".") + if err != nil { + return nil, err + } + } + + re, err := regexp.Compile(tokenConfigKeyPrefix + `.([^.]+)`) + if err != nil { + panic(err) + } + + set := make(map[string]interface{}) + + for key := range configs { + res := re.FindStringSubmatch(key) + + if res == nil { + continue + } + + set[res[1]] = nil + } + + result := make([]string, len(set)) + i := 0 + for key := range set { + result[i] = key + i++ + } + + return result, nil +} + +func ListTokens(repo *cache.RepoCache) ([]string, error) { + return listTokens(repo, false) +} + +func ListGlobalTokens(repo *cache.RepoCache) ([]string, error) { + return listTokens(repo, true) +} + +func storeToken(repo *cache.RepoCache, token *Token) error { + var store func(key, value string) error + if token.Global { + store = repo.StoreGlobalConfig + } else { + store = repo.StoreConfig + } + + var err error + storeValueKey := fmt.Sprintf("git-bug.token.%s.%s", token.Name, tokenKeyValue) + err = store(storeValueKey, token.Value) + if err != nil { + return err + } + + storeTargetKey := fmt.Sprintf("git-bug.token.%s.%s", token.Name, tokenKeyTarget) + err = store(storeTargetKey, token.Target) + if err != nil { + return err + } + + storeGlobalKey := fmt.Sprintf("git-bug.token.%s.%s", token.Name, tokenKeyGlobal) + err = store(storeGlobalKey, strconv.FormatBool(token.Global)) + if err != nil { + return err + } + + storeScopesKey := fmt.Sprintf("git-bug.token.%s.%s", token.Name, tokenKeyScopes) + err = store(storeScopesKey, strings.Join(token.Scopes, ",")) + if err != nil { + return err + } + + return nil +} + +func StoreToken(repo *cache.RepoCache, name, value, target string, global bool, scopes []string) error { + return storeToken(repo, NewToken(name, value, target, global, scopes)) +} + +func StoreGlobalToken(repo *cache.RepoCache, name, value, target string, global bool, scopes []string) error { + return storeToken(repo, NewToken(name, value, target, global, scopes)) +} + +func RemoveToken(repo *cache.RepoCache, name string) error { + keyPrefix := fmt.Sprintf("git-bug.token.%s", name) + return repo.RmConfigs(keyPrefix) +} + +func RemoveGlobalToken(repo *cache.RepoCache, name string) error { + keyPrefix := fmt.Sprintf("git-bug.token.%s", name) + return repo.RmGlobalConfigs(keyPrefix) +} |