111 lines
2.4 KiB
Go
111 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/Luzifer/rconfig"
|
|
gorilla "github.com/gorilla/mux"
|
|
log "github.com/sirupsen/logrus"
|
|
"golang.org/x/oauth2"
|
|
)
|
|
|
|
var cfg = struct {
|
|
SettingsFile string `default:"settings.yml" flag:"settings-file" description:"Path to settings file"`
|
|
}{}
|
|
|
|
func main() {
|
|
rconfig.Parse(&cfg)
|
|
settingsUpdater()
|
|
|
|
loadUser()
|
|
|
|
log.Infof("User: %+v", user)
|
|
|
|
mux := gorilla.NewRouter()
|
|
mux.HandleFunc("/login", handleTwitchLogin)
|
|
mux.HandleFunc("/callback", handleTwitchCallback)
|
|
mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
|
w.Write([]byte("Hi!"))
|
|
})
|
|
|
|
go func() {
|
|
handleSaves()
|
|
c := time.Tick(20 * time.Second)
|
|
for range c {
|
|
handleSaves()
|
|
}
|
|
}()
|
|
|
|
log.Info("Starting webserver...")
|
|
log.Fatal(http.ListenAndServe(":8083", mux))
|
|
}
|
|
|
|
func handleSaves() {
|
|
if user == nil {
|
|
return
|
|
}
|
|
|
|
getUser(user)
|
|
|
|
if user.TwitchUser != nil {
|
|
user.TwitchUser.SaveFiles()
|
|
}
|
|
|
|
getChannel(user)
|
|
if user.TwitchChannel != nil {
|
|
user.TwitchChannel.SaveFiles()
|
|
}
|
|
|
|
getFollows(user)
|
|
if user.TwitchFollowers != nil {
|
|
user.TwitchFollowers.SaveFiles()
|
|
}
|
|
|
|
getSubs(user)
|
|
if user.TwitchSubscriptions != nil {
|
|
user.TwitchSubscriptions.SaveFiles()
|
|
}
|
|
|
|
getStreams(user)
|
|
if user.TwitchStream != nil {
|
|
user.TwitchStream.SaveFiles()
|
|
}
|
|
}
|
|
|
|
func handleTwitchLogin(w http.ResponseWriter, r *http.Request) {
|
|
url := twitchOauthConfig.AuthCodeURL(settings.VerificationToken, oauth2.AccessTypeOffline)
|
|
http.Redirect(w, r, url, http.StatusTemporaryRedirect)
|
|
}
|
|
|
|
func handleTwitchCallback(w http.ResponseWriter, r *http.Request) {
|
|
twitchAuthToToken(r.FormValue("state"), r.FormValue("code"), w, r)
|
|
http.Redirect(w, r, "/", http.StatusTemporaryRedirect)
|
|
}
|
|
|
|
func twitchAuthToToken(state string, code string, w http.ResponseWriter, r *http.Request) {
|
|
if state != settings.VerificationToken {
|
|
log.Fatal("invalid oauth state")
|
|
}
|
|
|
|
timeoutCtx, cancel := context.WithTimeout(oauth2.NoContext, 5*time.Second)
|
|
defer cancel()
|
|
|
|
token, err := twitchOauthConfig.Exchange(timeoutCtx, code, oauth2.AccessTypeOffline)
|
|
if err != nil {
|
|
http.Redirect(w, r, "/", http.StatusTemporaryRedirect)
|
|
log.WithError(err).Error("code exchange failed")
|
|
return
|
|
}
|
|
u, err := createUser(token)
|
|
if err != nil {
|
|
log.WithError(err).Error("Unable to get user")
|
|
return
|
|
}
|
|
u.Token = token
|
|
user = u
|
|
log.Infof("User: %+v, User: %+v", user, user.TwitchUser)
|
|
saveUser()
|
|
}
|