▍ humdrum codex / dots v1.0.0
1.2 KB raw
 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
package auth

import (
	"errors"
	"os"
	"path/filepath"
	"strings"

	"github.com/zalando/go-keyring"
)

const service = "dots-cli"
const account = "token"

// ErrNoToken means no token is stored anywhere.
var ErrNoToken = errors.New("no token stored")

func tokenDir() string {
	if d := os.Getenv("DOTS_TOKEN_DIR"); d != "" {
		return d
	}
	home, _ := os.UserHomeDir()
	return filepath.Join(home, ".config", "dots")
}

func saveTokenFile(token string) error {
	dir := tokenDir()
	if err := os.MkdirAll(dir, 0o700); err != nil {
		return err
	}
	return os.WriteFile(filepath.Join(dir, "token"), []byte(token+"\n"), 0o600)
}

func loadTokenFile() (string, error) {
	b, err := os.ReadFile(filepath.Join(tokenDir(), "token"))
	if err != nil {
		return "", ErrNoToken
	}
	return strings.TrimSpace(string(b)), nil
}

// SaveToken prefers the OS keychain, falling back to a 0600 file.
func SaveToken(token string) error {
	if err := keyring.Set(service, account, token); err == nil {
		return nil
	}
	return saveTokenFile(token)
}

// LoadToken checks the keychain first, then the fallback file.
func LoadToken() (string, error) {
	if t, err := keyring.Get(service, account); err == nil && t != "" {
		return t, nil
	}
	return loadTokenFile()
}