kuvia2/account.go

package main

import (
	"fmt"
	"github.com/flamego/captcha"
	"github.com/flamego/flamego"
	"github.com/flamego/session"
	"github.com/flamego/template"
	"github.com/rs/zerolog/log"
	"golang.org/x/crypto/bcrypt"
	"image"
	user2 "kuvia/pkg/user"
	"net/http"
	"regexp"
	"time"
)

type User struct {
	Id int
	Username string
	Password string
	Name *string
	AboutMe *string `db:"aboutme"`
	Location *string
	Webpage *string
	Image *string
}

func loginForm(t template.Template, data template.Data, s session.Session, captcha captcha.Captcha, c flamego.Context) {
	fmt.Println(s.Get("user_id"))
	if s.Get("user_id") != nil {
		c.Redirect("/", http.StatusTemporaryRedirect)
		return
	}
	data["CaptchaHTML"] = captcha.HTML()
	t.HTML(http.StatusOK, "login")
}

func registerForm(t template.Template, data template.Data, s session.Session, captcha captcha.Captcha, c flamego.Context) {
	if s.Get("user_id") != nil {
		c.Redirect("/", http.StatusTemporaryRedirect)
		return
	}
	data["CaptchaHTML"] = captcha.HTML()
	t.HTML(http.StatusOK, "register")
}

func register(t template.Template, data template.Data, s session.Session, captcha captcha.Captcha, c flamego.Context) {
	if s.Get("user_id") != nil {
		c.Redirect("/", http.StatusTemporaryRedirect)
		return
	}
	username := c.Request().FormValue("username")
	password := c.Request().FormValue("password")
	//validate username
	var re = regexp.MustCompile(`^[a-z0-9]{8,25}$`)

	if re.Match([]byte(username)) == false {
		data["msg"] = "Username just contains lower letters and numbers"
		data["CaptchaHTML"] = captcha.HTML()
		t.HTML(http.StatusOK, "register")
		return
	}

	if len(password) < 8 {
		data["msg"] = "Password too short"
		data["CaptchaHTML"] = captcha.HTML()
		t.HTML(http.StatusOK, "register")
		return
	}

	//ca := c.Param("captcha")
	if captcha.ValidText(c.Request().FormValue("captcha")) == false {
		data["msg"] = "Please enter valide captcha code"
		data["CaptchaHTML"] = captcha.HTML()
		t.HTML(http.StatusOK, "register")
		return
	}

	//Check username exists
	usernameIsFree, err := user2.CheckUsernameIsFree(sqlConnection, username)
	if err != nil || usernameIsFree == false {
		data["msg"] = "Username is taken, choose another one"
		data["CaptchaHTML"] = captcha.HTML()
		t.HTML(http.StatusOK, "register")
		return
	}

	id, err := user2.Register(sqlConnection, username, password)
	if err != nil {
		panic(err)
	}
	fmt.Println(id)


	c.ResponseWriter().Write([]byte("OK"))
}

func logout(s session.Session, c flamego.Context) {
	s.Flush()
	c.Redirect("/login", http.StatusTemporaryRedirect)
}

func login(t template.Template, data template.Data, s session.Session, captcha captcha.Captcha, c flamego.Context) {
	if s.Get("user_id") != nil {
		c.Redirect("/", http.StatusTemporaryRedirect)
		return
	}
	username := c.Request().FormValue("username")
	password := c.Request().FormValue("password")

	/*if captcha.ValidText(c.Request().FormValue("captcha")) == false {
		data["msg"] = "Please enter valide captcha code"
		data["CaptchaHTML"] = captcha.HTML()
		t.HTML(http.StatusOK, "login")
		return
	}*/

	user, err := user2.GetUserByUsername(sqlConnection, username)
	if err != nil {
		data["msg"] = err.Error()
		data["CaptchaHTML"] = captcha.HTML()
		t.HTML(http.StatusOK, "login")
		return
	}


	err = bcrypt.CompareHashAndPassword([]byte(user.Password), []byte(password))
	if err != nil {
		data["msg"] = "Wrong Password"
		data["CaptchaHTML"] = captcha.HTML()
		t.HTML(http.StatusOK, "login")
		return
	}

	s.Set("user_id", user.Id)

	c.Redirect("/", http.StatusTemporaryRedirect)
}

func profile(t template.Template, data template.Data, s session.Session, c flamego.Context) {
	if s.Get("user_id") == nil {
		c.Redirect("/", http.StatusTemporaryRedirect)
		return
	}

	user, err := user2.GetUserById(sqlConnection, s.Get("user_id").(int))
	if err != nil {
		panic(err)
	}
	data["user"] = user

	t.HTML(http.StatusOK, "settings_profile")
}

func profileUpdate(s session.Session, c flamego.Context)  {
	user, err := user2.GetUserById(sqlConnection, s.Get("user_id").(int))
	if err != nil {
		panic(err)
	}

	user.Name = c.Request().FormValue("name")
	user.AboutMe = c.Request().FormValue("aboutme")
	user.Location = c.Request().FormValue("location")
	user.Webpage = c.Request().FormValue("webpage")


	err = user2.UpdateUser(sqlConnection, user)
	if err != nil{
		panic(err)
	}
}

func profileImage(t template.Template, data template.Data, s session.Session, c flamego.Context) {
	if s.Get("user_id") == nil {
		c.Redirect("/", http.StatusTemporaryRedirect)
		return
	}

	user, err := user2.GetUserById(sqlConnection, s.Get("user_id").(int))
	if err != nil {
		panic(err)
	}
	data["user"] = user

	t.HTML(http.StatusOK, "settings_image")
}

func profileImageChange(s session.Session, c flamego.Context) {
	log.Debug().Msg("Change Profile Image")
	c.Request().ParseMultipartForm(9001)

	user, err := user2.GetUserById(sqlConnection, s.Get("user_id").(int))
	if err != nil {
		panic(err)
	}

	//Upload Croped Image
	file, _, err  := c.Request().FormFile("cropImage")
	if err != nil {
		panic(err)
	}
	defer file.Close()

	cropedImage, _ , err := image.Decode(file)
	if err != nil {
		panic(err)
	}
	path := fmt.Sprintf("/user/%v/profilepic/%v-%v", user.Id, time.Now().UnixNano(), "crop.png")
	uploadImageToS3(cropedImage, path)

	//Upload Original Image

	fileOrginal, fileHeaderOrginal, err  := c.Request().FormFile("orginal")
	if err != nil {
		panic(err)
	}
	defer fileOrginal.Close()

	orignalImage, _ , err := image.Decode(fileOrginal)
	if err != nil {
		panic(err)
	}
	pathOrignal := fmt.Sprintf("/user/%v/profilepic/%v-orginal-%v", user.Id, time.Now().UnixNano(), fileHeaderOrginal.Filename)
	uploadImageToS3(orignalImage, pathOrignal)

	user.Image = path
	err = user2.UpdateUser(sqlConnection, user)
	if err != nil {
		panic(err)
	}
}