init

2019-08-08 17:13:34 +08:00
parent e8e5966a8c
commit f0bf2d5fec
2885 changed files with 1195993 additions and 12 deletions
@@ -0,0 +1,96 @@
+package controllers
+
+import (
+	"anytunnel/at-admin/app/models"
+	"anytunnel/at-admin/app/utils"
+	"image/color"
+	"image/png"
+	"strings"
+
+	"github.com/afocus/captcha"
+	"github.com/astaxie/beego"
+)
+
+var (
+	cap = captcha.New()
+)
+
+func init() {
+	bs, _ := utils.NewEncrypt().Base64DecodeBytes(fontData)
+	cap.AddFontFromBytes(bs)
+}
+
+type LoginController struct {
+	BaseController
+}
+
+func (this *LoginController) Index() {
+	this.Layout = "layout/login.html"
+	this.TplName = "login/login.html"
+	this.Data["title"] = beego.AppConfig.String("sys.name") + "登录"
+	this.Render()
+}
+
+//login
+func (this *LoginController) Login() {
+	if this.isLogin() {
+		return
+	}
+	userModel := models.User{}
+	name := strings.TrimSpace(this.GetString("username"))
+	password := strings.TrimSpace(this.GetString("password"))
+	captcha := strings.TrimSpace(this.GetString("captcha"))
+	captchaSession := this.GetSession("captcha")
+	if captchaSession == nil || captcha == "" || captchaSession != strings.ToLower(captcha) {
+		this.SetSession("captcha", "")
+		this.jsonError("验证码错误!")
+	}
+	this.SetSession("captcha", "")
+	user, err := userModel.GetUserByName(name)
+	if err != nil {
+		this.jsonError(err)
+		return
+	}
+	if len(user) == 0 {
+		this.jsonError("账号错误!")
+	}
+	encrypt := new(utils.Encrypt)
+	password = userModel.EncodePassword(password)
+
+	if user["password"] != password {
+		this.jsonError("账号或密码错误!")
+	}
+	//加载权限列表
+
+	//保存 session
+	this.SetSession("author", user)
+	//保存 cookie
+	identify := encrypt.Md5Encode(this.Ctx.Request.UserAgent() + this.getClientIp() + password)
+	passportValue := encrypt.Base64Encode(name + "@" + identify)
+	passport := beego.AppConfig.String("author.passport")
+	//fmt.Println("set cookie " + passportValue)
+	this.Ctx.SetCookie(passport, passportValue, 3600)
+
+	this.jsonSuccess("登录成功", "", "/main/index.html")
+}
+
+//logout
+func (this *LoginController) Logout() {
+	passport := beego.AppConfig.String("author.passport")
+	this.Ctx.SetCookie(passport, "")
+	this.SetSession("author", "")
+	this.Redirect("/login/index.html", 302)
+	this.StopRun()
+}
+
+func (this *LoginController) Captcha() {
+	//bs, _ := ioutil.ReadFile("static/index/comic.ttf")
+	//ioutil.WriteFile("a.txt", utils.NewEncrypt().Base64EncodeBytes(bs), os.ModeAppend)
+	cap.SetSize(80, 28)
+	cap.SetDisturbance(captcha.NORMAL)
+	cap.SetFrontColor(color.RGBA{255, 255, 255, 255})
+	cap.SetBkgColor(color.RGBA{255, 100, 100, 100})
+	img, str := cap.Create(4, captcha.ALL)
+	this.SetSession("captcha", strings.ToLower(str))
+	png.Encode(this.Ctx.ResponseWriter, img)
+}