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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
|
/*
macli - Unofficial CLI-Based MyAnimeList Client
Copyright © 2022-2024 Vidhu Kant Sharma <vidhukant@vidhukant.com>
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public Lice
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package auth
import (
mt "vidhukant.com/maltoken"
"errors"
"fmt"
"github.com/spf13/viper"
"github.com/zalando/go-keyring"
"os"
"os/exec"
"os/user"
"runtime"
)
// set with ldflags, allows for direct login without setting up client id
var VendoredClientId string
var serviceName string = "macli"
var userName string
func init() {
// get user and set username
currentUser, err := user.Current()
if err != nil {
fmt.Println("Error getting current user's info", err.Error())
os.Exit(1)
}
userName = currentUser.Username
mt.SuccessHTML = fmt.Sprintf(HTMLTemplate, "Login successful.", "Thanks for using macli. You may close this tab now.")
mt.BadRequestHTML = fmt.Sprintf(HTMLTemplate, "Invalid request.", "Required query parameters are missing.")
mt.ErrHTML = fmt.Sprintf(HTMLTemplate, "An error occoured.", "%s")
}
// asks for all the details
func Login(tk, clientId string, storeClientId bool) {
// check if an auth token already exists
var existingToken string
if NoSysKeyring {
existingToken = viper.GetString("auth.token")
} else {
/* if there is an error with keyring, askClientId would handle it
* can safely ignore error here */
existingToken, _ = keyring.Get(serviceName, userName)
}
if existingToken != "" {
if !confirmInput("Already logged in. Log in again? [Y/n] ", true) {
fmt.Println("Login process aborted")
os.Exit(0)
}
}
if clientId == "" {
if VendoredClientId == "" {
clientId = askClientId(storeClientId)
} else {
clientId = VendoredClientId
}
} else {
validateClientId(clientId)
if storeClientId {
setClientId(clientId)
}
}
if tk != "" {
setToken(tk)
fmt.Println("\x1b[32mYou have successfully logged into macli.\x1b[0m")
fmt.Println("\x1b[32mYou can close the web browser tab now.\x1b[0m")
return
}
challenge, link := mt.GetChallengeLink(clientId)
openInBrowser(link)
res, err := mt.Listen(clientId, challenge, 8000)
if err != nil {
fmt.Printf("\x1b[1;31mAn error occoured:\x1b[0m %s\nExiting...\n", err.Error())
os.Exit(1)
}
setToken(res["access_token"].(string))
}
func openInBrowser(url string) {
fmt.Println("Attempting to launch \033[36m" + url + "\033[0m in your default web browser. If it doesn't launch please manually copy-paste the link.")
var err error
switch runtime.GOOS {
case "linux":
err = exec.Command("xdg-open", url).Start()
case "windows":
err = exec.Command("rundll32", "url.dll,FileProtocolHandler", url).Start()
case "darwin":
err = exec.Command("open", url).Start()
default:
err = errors.New("<failed to detect platform>")
}
if err != nil {
fmt.Println("There was an error while launching your browser.")
fmt.Println("Please manually copy and paste the above URL into your web browser.")
fmt.Println(err)
}
}
func Logout() {
existingToken, _ := keyring.Get(serviceName, userName)
deleteToken()
if existingToken != "" {
fmt.Println("Deleted user credentials.")
}
// only ask to delete Client ID if it actually exists
existingClientId, _ := getClientId()
if existingClientId != "" && confirmInput("Delete your Client ID? [y/N] ", false) {
deleteClientId()
fmt.Println("Deleted Client ID.")
}
}
|
