150 lines
3.7 KiB
Go
150 lines
3.7 KiB
Go
package web
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/url"
|
|
"os"
|
|
"path"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/foolin/goview"
|
|
"github.com/foolin/goview/supports/ginview"
|
|
"github.com/gin-contrib/static"
|
|
"github.com/gin-gonic/contrib/sessions"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/kofalt/go-memoize"
|
|
|
|
"subscribe-bot/config"
|
|
"subscribe-bot/osuapi"
|
|
)
|
|
|
|
const (
|
|
USER_KEY = "user"
|
|
)
|
|
|
|
var (
|
|
cache = memoize.NewMemoizer(90*time.Second, 10*time.Minute)
|
|
)
|
|
|
|
func RunWeb(config *config.Config, api *osuapi.Osuapi) {
|
|
hc := http.Client{
|
|
Timeout: 10 * time.Second,
|
|
}
|
|
|
|
if !config.Debug {
|
|
gin.SetMode(gin.ReleaseMode)
|
|
}
|
|
|
|
r := gin.Default()
|
|
r.Use(gin.Recovery())
|
|
r.Use(static.Serve("/static", static.LocalFile("web/static", false)))
|
|
r.Use(sessions.Sessions("mysession", sessions.NewCookieStore([]byte(config.Web.SessionSecret))))
|
|
|
|
r.HTMLRender = ginview.New(goview.Config{
|
|
Root: "web/templates",
|
|
DisableCache: config.Debug,
|
|
})
|
|
|
|
r.GET("/login", func(c *gin.Context) {
|
|
url := url.URL{
|
|
Scheme: "https",
|
|
Host: "osu.ppy.sh",
|
|
Path: "/oauth/authorize",
|
|
}
|
|
q := url.Query()
|
|
q.Set("client_id", config.Oauth.ClientId)
|
|
q.Set("redirect_uri", config.Web.ServedAt+"/login/callback")
|
|
q.Set("response_type", "code")
|
|
q.Set("scope", "identify public")
|
|
q.Set("state", "urmom")
|
|
url.RawQuery = q.Encode()
|
|
fmt.Println("redirecting to", url.String())
|
|
c.Redirect(http.StatusTemporaryRedirect, url.String())
|
|
})
|
|
|
|
r.GET("/login/callback", func(c *gin.Context) {
|
|
receivedCode := c.Query("code")
|
|
|
|
bodyQuery := url.Values{}
|
|
bodyQuery.Set("client_id", config.Oauth.ClientId)
|
|
bodyQuery.Set("client_secret", config.Oauth.ClientSecret)
|
|
bodyQuery.Set("code", receivedCode)
|
|
bodyQuery.Set("grant_type", "authorization_code")
|
|
bodyQuery.Set("redirect_uri", config.Web.ServedAt+"/login/callback")
|
|
body := strings.NewReader(bodyQuery.Encode())
|
|
resp, _ := hc.Post("https://osu.ppy.sh/oauth/token", "application/x-www-form-urlencoded", body)
|
|
respBody, _ := ioutil.ReadAll(resp.Body)
|
|
type OsuToken struct {
|
|
TokenType string `json:"token_type"`
|
|
ExpiresIn int `json:"expires_in"`
|
|
AccessToken string `json:"access_token"`
|
|
RefreshToken string `json:"refresh_token"`
|
|
}
|
|
var token OsuToken
|
|
_ = json.Unmarshal(respBody, &token)
|
|
fmt.Println("TOKEN", token)
|
|
|
|
session := sessions.Default(c)
|
|
session.Set("access_token", token.AccessToken)
|
|
session.Save()
|
|
|
|
c.Redirect(http.StatusTemporaryRedirect, "/")
|
|
})
|
|
|
|
r.GET("/", func(c *gin.Context) {
|
|
session := sessions.Default(c)
|
|
var accessToken string
|
|
loggedIn := false
|
|
accessTokenI := session.Get("access_token")
|
|
if accessTokenI != nil {
|
|
accessToken = accessTokenI.(string)
|
|
if len(accessToken) > 0 {
|
|
loggedIn = true
|
|
}
|
|
}
|
|
|
|
beatmapSets := getRepos(config, api)
|
|
|
|
// render with master
|
|
c.HTML(http.StatusOK, "index.html", gin.H{
|
|
"LoggedIn": loggedIn,
|
|
"Beatmapsets": beatmapSets,
|
|
})
|
|
})
|
|
|
|
addr := fmt.Sprintf("%s:%d", config.Web.Host, config.Web.Port)
|
|
r.Run(addr)
|
|
}
|
|
|
|
func getRepos(config *config.Config, api *osuapi.Osuapi) []osuapi.Beatmapset {
|
|
expensive := func() (interface{}, error) {
|
|
repos := make([]osuapi.Beatmapset, 0)
|
|
reposDir := config.Repos
|
|
users, _ := ioutil.ReadDir(reposDir)
|
|
|
|
for _, user := range users {
|
|
userDir := path.Join(reposDir, user.Name())
|
|
var maps []os.FileInfo
|
|
maps, _ = ioutil.ReadDir(userDir)
|
|
|
|
for _, mapId := range maps {
|
|
mapDir := path.Join(userDir, mapId.Name())
|
|
fmt.Println(mapDir)
|
|
|
|
id, _ := strconv.Atoi(mapId.Name())
|
|
bs, _ := api.GetBeatmapSet(id)
|
|
repos = append(repos, bs)
|
|
}
|
|
}
|
|
|
|
return repos, nil
|
|
}
|
|
|
|
result, _, _ := cache.Memoize("key1", expensive)
|
|
return result.([]osuapi.Beatmapset)
|
|
}
|