func Permissions(permissionName string) martini.Handler {
return func(token oauth2.Tokens, w http.ResponseWriter, r *http.Request, c martini.Context) {
if token == nil || token.Expired() {
next := url.QueryEscape(r.URL.RequestURI())
http.Redirect(w, r, oauth2.PathLogin+"?next="+next, 302)
return
}
id, err := GetId(token.Access())
if err != nil {
log.Printf("Error getting player token id:", err.Error())
http.Redirect(w, r, "/error", 302)
return
}
user := ols.GetUserDAO().GetUserFB(id)
if user.LeagueId == 0 {
next := url.QueryEscape(r.URL.RequestURI())
http.Redirect(w, r, "/register?next="+next, 302)
}
// TODO - fix this
if !true {
http.Redirect(w, r, "/error", 302)
}
c.Map(user)
c.Next()
}
}
func getCaptains() []*DraftCaptain {
captains := ols.GetPlayersDAO().All()
sort.Sort(captains)
draftCaptains := []*DraftCaptain{}
var captains_sorted []*ols.Player = captains
for _, player := range captains_sorted {
team := ols.GetTeamsDAO().LoadPlayerByCaptain(player.Id)
if team.Captain == player.Id {
user := ols.GetUserDAO().GetUserLeague(player.Id)
draftCaptains = append(draftCaptains, &DraftCaptain{Player: *player, FacebookID: user.FacebookId, Points: team.Points, TeamName: team.Name})
}
}
return draftCaptains
}
var CaptainRequiredFunc = func() martini.Handler {
return func(token oauth2.Tokens, w http.ResponseWriter, r *http.Request, c martini.Context) {
if token == nil || token.Expired() {
next := url.QueryEscape(r.URL.RequestURI())
http.Redirect(w, r, oauth2.PathLogin+"?next="+next, 302)
return
}
id, err := GetId(token.Access())
if err != nil {
log.Printf("Error getting captain token id:", err.Error())
http.Redirect(w, r, "/error", 302)
return
}
user := ols.GetUserDAO().GetUserFB(id)
if user.LeagueId == 0 {
next := url.QueryEscape(r.URL.RequestURI())
http.Redirect(w, r, "/register?next="+next, 302)
return
}
// TODO - fix this
if true {
c.Map(user)
c.Next()
} else {
http.Redirect(w, r, "/captain", 401)
return
}
}
func main() {
file, err := os.OpenFile("log.txt", os.O_APPEND|os.O_CREATE|os.O_RDWR, 0666)
if err != nil {
panic(err)
}
log.SetOutput(file)
log.Println("Success?")
m := martini.Classic()
goriot.SetAPIKey(LeagueApiKey)
goriot.SetLongRateLimit(LongLeagueLimit, 10*time.Minute)
goriot.SetSmallRateLimit(ShortLeagueLimit, 10*time.Second)
// Setup middleware to be attached to the controllers on every call.
if Debug {
InitDebugMiddleware(m)
} else {
InitMiddleware(m)
}
teamHandler := func(renderer render.Render) {
teams := ols.GetTeamsDAO().All()
renderer.JSON(200, teams)
}
individualTeamHandler := func(params martini.Params, renderer render.Render) {
team := ols.GetTeamsDAO().Load(params["name"])
renderer.JSON(200, team)
}
m.Get("/admin", func(renderer render.Render) {
renderer.HTML(200, "admin", 1)
})
m.Get("/error", func(urls url.Values, renderer render.Render) {
renderer.HTML(200, "error", urls.Get("status"))
})
m.Get("/players/:name", func(params martini.Params, renderer render.Render) {
normalizedName := params["name"]
fmt.Println(normalizedName)
player := ols.GetPlayersDAO().LoadNormalizedIGN(normalizedName)
fmt.Println(player)
displayTeam := ols.GetTeamsDAO().LoadPlayerDisplay(player.Id)
fmt.Println(displayTeam)
renderer.HTML(200, "team", displayTeam)
})
m.Get("/results", func(renderer render.Render) {
players := ols.GetPlayersDAO().All()
allPlayerDisplay := []ols.PlayerDraftResult{}
for _, player := range players {
team := ols.GetTeamsDAO().LoadPlayer(player.Id)
playerDisplay := ols.PlayerDraftResult{Ign: player.Ign, Team: team.Name, NormalizedIgn: player.NormalizedIgn}
allPlayerDisplay = append(allPlayerDisplay, playerDisplay)
}
renderer.HTML(200, "drafted", allPlayerDisplay)
})
m.Get("/teams", teamHandler)
m.Get("/team/:name", individualTeamHandler)
m.Get("/", func(renderer render.Render) {
renderer.HTML(200, "main", 1)
})
m.Get("/register", LoginRequired, func(urls url.Values, renderer render.Render) {
renderer.HTML(200, "register", Register{Next: urls.Get("next")})
})
m.Get("/oauth2error", func(token oauth2.Tokens, renderer render.Render) {
renderer.JSON(200, token)
})
m.Get("/rankings", func(renderer render.Render) {
players := ols.GetPlayersDAO().All()
sort.Sort(players)
renderer.HTML(200, "rank", players)
})
//m.Get("/draft-socket/:drafter", sockets.JSON(Message{}), func(params martini.Params, receiver <-chan *Message, sender chan<- *Message, done <-chan bool, disconnect chan<- int, errorChannel <-chan error) {
//})
m.Get("/register/complete", LoginRequired, func(urls url.Values, renderer render.Render, token oauth2.Tokens, w http.ResponseWriter, r *http.Request) {
summonerName := urls.Get("summoner")
teamName := urls.Get("team")
if token.Expired() {
http.Redirect(w, r, "/error?status=InvalidFacebook", 302)
return
}
id, err := GetId(token.Access())
if err != nil {
renderer.Status(404)
return
}
normalizedSummonerName := goriot.NormalizeSummonerName(summonerName)[0]
player := ols.GetPlayersDAO().LoadNormalizedIGN(normalizedSummonerName)
if player.Id == 0 {
http.Redirect(w, r, "/error?status=No Player Found", 302)
return
}
user := ols.GetUserDAO().GetUserFB(id)
// User is registered registered
if user.LeagueId != 0 {
http.Redirect(w, r, "/error?status=Already Registered User", 302)
return
}
user = site.User{LeagueId: player.Id, FacebookId: id}
log.Println("User registered:", user)
team := ols.GetTeamsDAO().LoadPlayerByCaptain(player.Id)
newTeam := team
if team.Name != "" && teamName != "" {
newTeam.Name = teamName
ols.GetTeamsDAO().Update(team, newTeam)
}
ols.GetUserDAO().Save(user)
//next := urls.Get("next")
log.Println("[REGISTER]: ", teamName)
renderer.HTML(200, "register_complete", 1)
})
initFunnyRouter(m)
SocketRouter(m)
err = http.ListenAndServe(":6060", m) // Nginx needs to redirect here, so we don't need sudo priv to test.
if err != nil {
log.Println(err)
}
}