"fmt"
"io/ioutil"
"os"
- "path/filepath"
"strings"
"github.com/mattermost/mattermost-server/v5/model"
s := strings.Split(url, "/")
machine, fileId := s[2], s[6]
- netrcPath, ok := os.LookupEnv("NETRC")
- if !ok {
- netrcPath = filepath.Join(os.Getenv("HOME"), ".netrc")
- }
- data, err := ioutil.ReadFile(netrcPath)
- if err != nil {
- panic(err)
- }
- var login string
- var password string
- for _, line := range strings.Split(string(data), "\n") {
- if i := strings.Index(line, "#"); i >= 0 {
- line = line[:i]
- }
- f := strings.Fields(line)
- if len(f) >= 6 &&
- f[0] == "machine" && f[1] == machine &&
- f[2] == "login" && f[4] == "password" {
- login, password = f[3], f[5]
- break
- }
- }
+ login, password := findInNetrc(machine)
if login == "" || password == "" {
fmt.Fprintln(os.Stderr, "No credentials found for:", machine)
os.Exit(1)
fmt.Fprintln(os.Stderr, "Download? Ctrl-D/C")
os.Stdin.Read(make([]byte, 1))
}
- data, _ = Client.GetFile(fileId)
- if err = ioutil.WriteFile(filename, data, os.FileMode(0666)); err != nil {
+ data, _ := Client.GetFile(fileId)
+ if err := ioutil.WriteFile(filename, data, os.FileMode(0666)); err != nil {
panic(err)
}
}
--- /dev/null
+// Nearly all code is taken from src/cmd/go/internal/auth/netrc.go
+
+package main
+
+import (
+ "io/ioutil"
+ "log"
+ "os"
+ "path/filepath"
+ "strings"
+)
+
+func findInNetrc(host string) (string, string) {
+ netrcPath, ok := os.LookupEnv("NETRC")
+ if !ok {
+ homeDir, err := os.UserHomeDir()
+ if err != nil {
+ log.Fatalln(err)
+ }
+ netrcPath = filepath.Join(homeDir, ".netrc")
+ }
+ data, err := ioutil.ReadFile(netrcPath)
+ if err != nil {
+ if os.IsNotExist(err) {
+ return "", ""
+ }
+ log.Fatalln(err)
+ }
+ inMacro := false
+ var machine, login, password string
+ for _, line := range strings.Split(string(data), "\n") {
+ if inMacro {
+ if line == "" {
+ inMacro = false
+ }
+ continue
+ }
+ fields := strings.Fields(line)
+ i := 0
+ for ; i < len(fields)-1; i += 2 {
+ switch fields[i] {
+ case "machine":
+ machine = fields[i+1]
+ login = ""
+ password = ""
+ case "default":
+ break
+ case "login":
+ login = fields[i+1]
+ case "password":
+ password = fields[i+1]
+ case "macdef":
+ inMacro = true
+ }
+ if machine != "" && login != "" && password != "" {
+ if machine == host {
+ return login, password
+ }
+ machine, login, password = "", "", ""
+ }
+ }
+ if i < len(fields) && fields[i] == "default" {
+ break
+ }
+ }
+ return "", ""
+}