73 lines
1.3 KiB
Go
73 lines
1.3 KiB
Go
package utils
|
|
|
|
import (
|
|
"encoding/json"
|
|
"os"
|
|
|
|
"github.com/sandertv/gophertunnel/minecraft/auth"
|
|
"github.com/sandertv/gophertunnel/minecraft/realms"
|
|
"github.com/sirupsen/logrus"
|
|
"golang.org/x/oauth2"
|
|
)
|
|
|
|
const TOKEN_FILE = "token.json"
|
|
|
|
var G_token_src oauth2.TokenSource
|
|
|
|
func GetTokenSource() oauth2.TokenSource {
|
|
if G_token_src != nil {
|
|
return G_token_src
|
|
}
|
|
token := get_token()
|
|
G_token_src = auth.RefreshTokenSource(&token)
|
|
new_token, err := G_token_src.Token()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
if !token.Valid() {
|
|
logrus.Info("Refreshed token")
|
|
write_token(new_token)
|
|
}
|
|
|
|
return G_token_src
|
|
}
|
|
|
|
var G_realms_api *realms.Client
|
|
|
|
func GetRealmsApi() *realms.Client {
|
|
if G_realms_api == nil {
|
|
G_realms_api = realms.NewClient(GetTokenSource())
|
|
}
|
|
return G_realms_api
|
|
}
|
|
|
|
func write_token(token *oauth2.Token) {
|
|
buf, err := json.Marshal(token)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
os.WriteFile(TOKEN_FILE, buf, 0o755)
|
|
}
|
|
|
|
func get_token() oauth2.Token {
|
|
var token oauth2.Token
|
|
if _, err := os.Stat(TOKEN_FILE); err == nil {
|
|
f, err := os.Open(TOKEN_FILE)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer f.Close()
|
|
if err := json.NewDecoder(f).Decode(&token); err != nil {
|
|
panic(err)
|
|
}
|
|
} else {
|
|
_token, err := auth.RequestLiveToken()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
write_token(_token)
|
|
token = *_token
|
|
}
|
|
return token
|
|
}
|