wandb

1
// Modified from: https://go.dev/src/cmd/go/internal/auth/netrc.go
2

3
// Copyright 2019 The Go Authors. All rights reserved.
4
// Use of this source code is governed by a BSD-style
5
// license that can be found in the LICENSE file.
6

7
package auth
8

9
import (
10
	"fmt"
11
	"os"
12
	"path/filepath"
13
	"runtime"
14
	"strings"
15
	// "sync"
16
)
17

18
type netrcLine struct {
19
	Machine  string
20
	Login    string
21
	Password string
22
}
23

24
var (
25
	// netrcOnce sync.Once
26
	netrc []netrcLine
27
	// netrcErr  error
28
)
29

30
func parseNetrc(data string) []netrcLine {
31
	// See https://www.gnu.org/software/inetutils/manual/html_node/The-_002enetrc-file.html
32
	// for documentation on the .netrc format.
33
	var nrc []netrcLine
34
	var l netrcLine
35
	inMacro := false
36
	for _, line := range strings.Split(data, "\n") {
37
		if inMacro {
38
			if line == "" {
39
				inMacro = false
40
			}
41
			continue
42
		}
43

44
		f := strings.Fields(line)
45
		i := 0
46
		for ; i < len(f)-1; i += 2 {
47
			// Reset at each "machine" token.
48
			// “The auto-login process searches the .netrc file for a machine token
49
			// that matches […]. Once a match is made, the subsequent .netrc tokens
50
			// are processed, stopping when the end of file is reached or another
51
			// machine or a default token is encountered.”
52
			switch f[i] {
53
			case "machine":
54
				l = netrcLine{Machine: f[i+1]}
55
			case "default":
56
				break
57
			case "login":
58
				l.Login = f[i+1]
59
			case "password":
60
				l.Password = f[i+1]
61
			case "macdef":
62
				// “A macro is defined with the specified name; its contents begin with
63
				// the next .netrc line and continue until a null line (consecutive
64
				// new-line characters) is encountered.”
65
				inMacro = true
66
			}
67
			if l.Machine != "" && l.Login != "" && l.Password != "" {
68
				nrc = append(nrc, l)
69
				l = netrcLine{}
70
			}
71
		}
72

73
		if i < len(f) && f[i] == "default" {
74
			// “There can be only one default token, and it must be after all machine tokens.”
75
			break
76
		}
77
	}
78

79
	return nrc
80
}
81

82
func netrcPath() (string, error) {
83
	if env := os.Getenv("NETRC"); env != "" {
84
		return env, nil
85
	}
86
	dir, err := os.UserHomeDir()
87
	if err != nil {
88
		return "", err
89
	}
90
	base := ".netrc"
91
	if runtime.GOOS == "windows" {
92
		base = "_netrc"
93
	}
94
	return filepath.Join(dir, base), nil
95
}
96

97
func ReadNetrc() ([]netrcLine, error) {
98
	path, err := netrcPath()
99
	if err != nil {
100
		// netrcErr = err
101
		return []netrcLine{}, err
102
	}
103

104
	data, err := os.ReadFile(path)
105
	if err != nil {
106
		// if !os.IsNotExist(err) {
107
		// 	netrcErr = err
108
		// }
109
		return []netrcLine{}, err
110
	}
111

112
	netrc = parseNetrc(string(data))
113
	return netrc, nil
114
}
115

116
func GetNetrcLogin(machine string) (string, string, error) {
117
	netrcLines, err := ReadNetrc()
118
	if err != nil {
119
		return "", "", err
120
	}
121
	for _, l := range netrcLines {
122
		if l.Machine == machine {
123
			return l.Login, l.Password, nil
124
		}
125
	}
126
	return "", "", fmt.Errorf("no entry for %s in %s", machine, netrcLines)
127
}
128