func (this *UserController) Signup() {
uid := this.GetSession("Uid")
if uid != nil {
this.Redirect("/", 302)
}
if this.Ctx.Input.Param("0") != "submit" {
this.Redirect("/user/login", 302)
}
user := models.User{
Username: this.GetString("username"),
Password: this.GetString("passkey"),
Name: this.GetString("name"),
College: this.GetString("college"),
Email: this.GetString("email"),
}
// All the fields verified, as well checked if username and email are unique
err := user.SignupVerify()
if err != nil {
flash := beego.NewFlash()
flash.Error(err.Error())
flash.Store(&this.Controller)
}
uid, done := user.Create()
if done {
this.SetSession("Uid", this.GetString("username"))
this.SetSession("id", uid)
this.Redirect("/", 302)
}
this.Redirect("/user/login", 302)
}
// To-do: Show programs solved by the user
func (this *UserController) Show() {
if this.Ctx.Input.Param("0") == "" {
this.Abort("404")
}
user := models.User{Username: this.Ctx.Input.Param("0")}
if user.GetByUsername() {
this.Data["title"] = user.Username
this.Data["userDetails"] = user
log := models.Problemlogs{Uid: user.Uid}
logs, count := log.GetByUid()
problems := make(map[int]models.Problem)
if count == 0 {
this.Data["solvedProblemsExist"] = false
} else {
this.Data["solvedProblemsExist"] = true
for index, element := range logs {
p := models.Problem{Pid: element.Pid}
p.GetByPid()
problems[index] = p
}
}
this.Data["solvedProblems"] = problems
this.Layout = "layout.tpl"
this.TplNames = "user/show.tpl"
this.LayoutSections = make(map[string]string)
this.LayoutSections["HtmlHead"] = ""
this.LayoutSections["Sidebar"] = ""
} else {
this.Abort("404")
}
}
// /contest/:name/saveproblem
func (this *ContestController) saveProblem() {
if !this.isAdmin() {
return
}
contestName := this.Ctx.Input.Param(":name")
id := this.GetSession("id")
user := models.User{}
user.Uid = id.(int)
points, _ := strconv.Atoi(this.GetString("points"))
//remove replace foe newlines
sampleInput := strings.Replace(this.GetString("sample_input"), "\r", "", -1)
sampleOutput := strings.Replace(this.GetString("sample_output"), "\r", "", -1)
problem := models.Problem{
Statement: this.GetString("statement"),
Description: this.GetString("description"),
Constraints: this.GetString("constraints"),
//Sample_input: this.GetString("sample_input"),
//Sample_output: this.GetString("sample_output"),
Difficulty: this.GetString("difficulty"),
Points: points,
Uid: id.(int),
}
problem.Sample_output = sampleOutput
problem.Sample_input = sampleInput
problem.Type = "contest" + contestName
pid, status := problem.Create()
if status {
id := strconv.Itoa(pid)
this.Redirect("/contest/"+contestName+"/"+id, 302)
}
// handle the failure
}
func (this *ProblemController) SaveTestCase() {
if !this.isLoggedIn() {
this.Redirect("/user/login", 302)
return
}
//Redirect if user doesnt hold editor rights
uid := this.GetSession("id")
user := models.User{Uid: uid.(int)}
if !user.IsEditor() {
this.Redirect("/", 302)
return
}
pid := this.Ctx.Input.Param(":id")
id, _ := strconv.Atoi(pid)
timeout, _ := strconv.Atoi(this.GetString("timeout"))
//remove string replace
input := strings.Replace(this.GetString("input"), "\r", "", -1)
output := strings.Replace(this.GetString("output"), "\r", "", -1)
testcase := models.Testcases{
Pid: id,
//Input: this.GetString("input"),
//Output: this.GetString("output"),
Timeout: timeout,
}
testcase.Input = input
testcase.Output = output
done := testcase.Create()
if done == true {
this.Redirect("/problem/"+pid, 302)
}
this.Redirect("/problem/"+pid+"/addtest", 302)
}
func (this *ProblemController) AddTestCase() {
if !this.isLoggedIn() {
this.Redirect("/user/login", 302)
return
}
//Redirect if user doesnt hold editor rights
uid := this.GetSession("id")
user := models.User{Uid: uid.(int)}
if !user.IsEditor() {
this.Redirect("/", 302)
return
}
pid := this.Ctx.Input.Param(":id")
id, _ := strconv.Atoi(pid)
problem := models.Problem{Pid: id}
problem.GetByPid()
this.Data["problem"] = problem
testcases := models.Testcases{Pid: id}
cases, _ := testcases.GetAllByPid()
this.Data["title"] = "Add Test Case"
this.Data["cases"] = cases
this.Layout = "layout.tpl"
this.TplNames = "problem/addtest.tpl"
this.LayoutSections = make(map[string]string)
this.LayoutSections["HtmlHead"] = ""
this.LayoutSections["Sidebar"] = ""
}
func (this *AdminController) ShowEditors() {
if !this.isAdmin() {
return
}
user := models.User{}
users := user.GetEditors()
bytes, _ := json.Marshal(users)
this.Data["json"] = string(bytes)
this.ServeJson()
}
// /admin/revokeEditor/:uid
func (this *AdminController) RevokeEditor() {
if !this.isAdmin() {
return
}
uid := this.Ctx.Input.Param("uid")
user := models.User{}
id, _ := strconv.Atoi(uid)
user.Uid = id
status := user.RevokeEditor()
this.Data["status"] = status
}
// /admin/search/name/:name
func (this *AdminController) SearchName() {
if !this.isAdmin() {
return
}
name := this.Ctx.Input.Param(":name")
user := models.User{}
user.Name = name
users, _ := user.SearchByName()
bytes, _ := json.Marshal(users)
this.Data["json"] = string(bytes)
this.ServeJson()
}
// Save Problem
// To-do: Clean info before save - Ambigous
// To-do: Check login and user previlages - Done
func (this *ProblemController) SaveProblem() {
if !this.isLoggedIn() {
this.Redirect("/user/login", 302)
return
}
//Redirect if user doesnt hold editor rights
id := this.GetSession("id")
user := models.User{}
user.Uid = id.(int)
if !user.IsEditor() {
this.Redirect("/", 302)
return
}
points, _ := strconv.Atoi(this.GetString("points"))
//remove replace foe newlines
sampleInput := strings.Replace(this.GetString("sample_input"), "\r", "", -1)
sampleOutput := strings.Replace(this.GetString("sample_output"), "\r", "", -1)
problem := models.Problem{
Statement: this.GetString("statement"),
Description: this.GetString("description"),
Constraints: this.GetString("constraints"),
//Sample_input: this.GetString("sample_input"),
//Sample_output: this.GetString("sample_output"),
Type: this.GetString("type"),
Difficulty: this.GetString("difficulty"),
Points: points,
Uid: id.(int),
}
problem.Sample_output = sampleOutput
problem.Sample_input = sampleInput
id, noerr := problem.Create()
pid := strconv.Itoa(id.(int))
if noerr == true {
this.Redirect("/problem/"+pid, 302)
}
this.Data["title"] = "Create Problem "
this.Layout = "layout.tpl"
this.TplNames = "problem/create.tpl"
this.LayoutSections = make(map[string]string)
this.LayoutSections["HtmlHead"] = ""
this.LayoutSections["Sidebar"] = ""
}
// /contest/:name/:pid
func (this *ContestController) GetProblem() {
pid := this.Ctx.Input.Param(":id")
contestName := this.Ctx.Input.Param(":name")
id, err := strconv.Atoi(pid)
if err != nil {
// Redirect to 404
this.Abort("404")
}
p := models.Problem{Pid: id}
p.GetById()
check := strings.Contains(p.Type, contestName)
if !check {
this.Redirect("/contest/"+contestName, 302)
return
}
log := models.Problemlogs{Pid: id}
logs, count := log.GetRecentByPid()
users := make(map[int]models.User)
if count == 0 {
this.Data["recentlySolvedUsersExist"] = false
} else {
this.Data["recentlySolvedUsersExist"] = true
for index, element := range logs {
u := models.User{Uid: element.Uid}
u.GetUserInfo()
users[index] = u
}
}
this.Data["title"] = p.Statement
this.Data["problem"] = p
this.Data["Author"] = contestName
this.Data["recentlySolvedUsers"] = users
// Handle problem log of a user
if this.isLoggedIn() {
problemLog := models.Problemlogs{}
problemLog.Pid = p.Pid
problemLog.Uid = p.Uid
if problemLog.GetByPidUid() {
this.Data["userScore"] = problemLog.Points
this.Data["solvedCount"] = problemLog.Solved
}
}
}
// Serves the Problem Page
// To-do: Show recently solved users and their language on sidebar
// To-do: Later, add least execution time log on sidebar
func (this *ProblemController) ProblemById() {
pid := this.Ctx.Input.Param(":id")
id, err := strconv.Atoi(pid)
if err != nil {
// Redirect to 404
this.Abort("404")
}
p := models.Problem{Pid: id}
p.GetById()
if strings.Contains(p.Type, "contest") {
this.Redirect("/", 302)
return
}
log := models.Problemlogs{Pid: id}
logs, count := log.GetRecentByPid()
users := make(map[int]models.User)
if count == 0 {
this.Data["recentlySolvedUsersExist"] = false
} else {
this.Data["recentlySolvedUsersExist"] = true
for index, element := range logs {
u := models.User{Uid: element.Uid}
u.GetUserInfo()
users[index] = u
}
}
//Author added
user := models.User{}
user.Uid = p.Uid
user.GetUserInfo()
this.Data["title"] = p.Statement
this.Data["problem"] = p
this.Data["Author"] = user.Username
this.Data["recentlySolvedUsers"] = users
// Handle problem log of a user
if this.isLoggedIn() {
problemLog := models.Problemlogs{}
problemLog.Pid = p.Pid
problemLog.Uid = p.Uid
if problemLog.GetByPidUid() {
this.Data["userScore"] = problemLog.Points
this.Data["solvedCount"] = problemLog.Solved
}
}
this.Layout = "layout.tpl"
this.TplNames = "problem/show.tpl"
this.LayoutSections = make(map[string]string)
this.LayoutSections["HtmlHead"] = "problem/submit_head.tpl"
this.LayoutSections["Sidebar"] = "sidebar/recently_solved_by.tpl"
}
// Create Page
func (this *ProblemController) Create() {
// If not logged redirect to login
if !this.isLoggedIn() {
this.Redirect("/user/login", 302)
return
}
//Redirect if user doesnt hold editor rights
id := this.GetSession("id")
user := models.User{}
user.Uid = id.(int)
if !user.IsEditor() {
this.Redirect("/", 302)
return
}
this.Data["title"] = "Create Problem "
this.Layout = "layout.tpl"
this.TplNames = "problem/create.tpl"
this.LayoutSections = make(map[string]string)
this.LayoutSections["HtmlHead"] = ""
this.LayoutSections["Sidebar"] = ""
}
func (this *UserController) Login() {
uid := this.GetSession("Uid")
if uid != nil {
this.Redirect("/", 302)
}
if this.Ctx.Input.Param("0") == "submit" {
user := models.User{
Username: this.GetString("username"),
Password: this.GetString("password"),
}
// Handle the flash messages
err := user.LoginVerify()
if err != nil {
flash := beego.NewFlash()
flash.Error(err.Error())
flash.Store(&this.Controller)
}
if user.Login() == true {
this.SetSession("Uid", this.GetString("username"))
user.GetUserInfo()
this.SetSession("id", user.Uid)
// store the user ID in the session
this.Redirect("/", 302)
}
//If login failed, flash a relevent message
}
this.Data["title"] = "Login"
this.Layout = "layout.tpl"
this.TplNames = "user/login.tpl"
this.LayoutSections = make(map[string]string)
this.LayoutSections["HtmlHead"] = ""
this.LayoutSections["Sidebar"] = ""
this.LayoutSections["ErrorHead"] = "errorHead.tpl"
}