ohUrlShortener/controller/admin_controller.go

// Copyright (c) [2022] [巴拉迪维 BaratSemet]
// [ohUrlShortener] is licensed under Mulan PSL v2.
// You can use this software according to the terms and conditions of the Mulan PSL v2.
// You may obtain a copy of Mulan PSL v2 at:
// 				 http://license.coscl.org.cn/MulanPSL2
// THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
// See the Mulan PSL v2 for more details.

package controller

import (
	"fmt"
	"net/http"
	"ohurlshortener/core"
	"ohurlshortener/service"
	"ohurlshortener/utils"
	"strconv"
	"strings"

	"github.com/dchest/captcha"
	"github.com/gin-gonic/gin"
)

const (
	DEFAULT_PAGE_NUM  = 1
	DEFAULT_PAGE_SIZE = 20
)

func LoginPage(c *gin.Context) {
	c.HTML(http.StatusOK, "login.html", gin.H{
		"title": "登录 - ohUrlShortener",
	})
}

func DoLogin(c *gin.Context) {
	account := c.PostForm("account")
	password := c.PostForm("password")
	captchaText := c.PostForm("captcha-text")
	captchaId := c.PostForm("captcha-id")

	if utils.EemptyString(account) || utils.EemptyString(password) || len(account) < 5 || len(password) < 8 {
		c.HTML(http.StatusOK, "login.html", gin.H{
			"title": "错误 - ohUrlShortener",
			"error": "用户名或密码格式错误！",
		})
		return
	}

	if utils.EemptyString(captchaText) || utils.EemptyString(captchaId) || len(captchaText) < 6 {
		c.HTML(http.StatusOK, "login.html", gin.H{
			"title": "错误 - ohUrlShortener",
			"error": "验证码格式错误!",
		})
		return
	}

	//验证码有效性验证
	if !captcha.VerifyString(captchaId, captchaText) {
		c.HTML(http.StatusOK, "login.html", gin.H{
			"title": "错误 - ohUrlShortener",
			"error": "验证码错误，请刷新页面再重新尝试！",
		})
		return
	}

	//用户名密码有效性验证
	loginUser, err := service.Login(account, password)
	if err != nil || loginUser.IsEmpty() {
		c.HTML(http.StatusOK, "login.html", gin.H{
			"title": "错误 - ohUrlShortener",
			"error": err.Error(),
		})
		return
	}

	//Write Cookie to browser
	cValue, err := AdminCookieValue(loginUser)
	if err != nil {
		c.HTML(http.StatusOK, "login.html", gin.H{
			"title": "错误 - ohUrlShortener",
			"error": "内部错误，请联系管理员",
		})
		return
	}
	c.SetCookie("ohUrlShortenerAdmin", loginUser.Account, 3600, "/", "", true, true)
	c.SetCookie("ohUrlShortenerCookie", cValue, 3600, "/", "", true, true)
	c.Redirect(http.StatusFound, "/admin/dashboard")
}

func DoLogout(c *gin.Context) {
	c.SetCookie("ohUrlShortenerAdmin", "", -1, "/", "", true, true)
	c.SetCookie("ohUrlShortenerCookie", "", -1, "/", "", true, true)
	c.Redirect(http.StatusFound, "/login")
}

func ServeCaptchaImage(c *gin.Context) {
	captcha.Server(200, 45).ServeHTTP(c.Writer, c.Request)
}

func RequestCaptchaImage(c *gin.Context) {
	imageId := captcha.New()
	c.JSON(http.StatusOK, core.ResultJsonSuccessWithData(imageId))
}

func ChangeState(c *gin.Context) {
	destUrl := c.PostForm("dest_url")
	enable := c.PostForm("enable")

	if utils.EemptyString(destUrl) {
		c.JSON(http.StatusBadRequest, core.ResultJsonError("目标链接不存在！"))
		return
	}

	destEnable, err := strconv.ParseBool(enable)
	if err != nil {
		c.JSON(http.StatusBadRequest, core.ResultJsonError("参数不合法！"))
		return
	}

	result, er := service.ChangeState(destUrl, destEnable)
	if er != nil {
		c.JSON(http.StatusInternalServerError, core.ResultJsonError(er.Error()))
		return
	}

	c.JSON(http.StatusOK, core.ResultJsonSuccessWithData(result))
}

func GenerateShortUrl(c *gin.Context) {
	destUrl := c.PostForm("dest_url")
	memo := c.PostForm("memo")

	if utils.EemptyString(destUrl) {
		c.JSON(http.StatusBadRequest, core.ResultJsonError("目标链接不能为空！"))
		return
	}

	result, err := service.GenerateShortUrl(destUrl, memo)
	if err != nil {
		c.JSON(http.StatusInternalServerError, core.ResultJsonError(err.Error()))
		return
	}

	json := map[string]string{
		"short_url": fmt.Sprintf("%s%s", utils.AppConfig.UrlPrefix, result),
	}
	c.JSON(http.StatusOK, core.ResultJsonSuccessWithData(json))
}

func StatsPage(c *gin.Context) {
	url := c.DefaultQuery("url", "")
	strPage := c.DefaultQuery("page", strconv.Itoa(DEFAULT_PAGE_NUM))
	strSize := c.DefaultQuery("size", strconv.Itoa(DEFAULT_PAGE_SIZE))
	page, err := strconv.Atoi(strPage)
	if err != nil {
		page = DEFAULT_PAGE_NUM
	}
	size, err := strconv.Atoi(strSize)
	if err != nil {
		size = DEFAULT_PAGE_SIZE
	}
	urls, err := service.GetPagedUrlIpCountStats(strings.TrimSpace(url), page, size)
	c.HTML(http.StatusOK, "stats.html", gin.H{
		"title":       "数据统计 - ohUrlShortener",
		"current_url": c.Request.URL.Path,
		"error":       err,
		"shortUrls":   urls,
		"page":        page,
		"size":        size,
		"prefix":      utils.AppConfig.UrlPrefix,
		"first_page":  page == 1,
		"last_page":   len(urls) < size,
		"url":         strings.TrimSpace(url),
	})
}

func UrlsPage(c *gin.Context) {
	url := c.DefaultQuery("url", "")
	strPage := c.DefaultQuery("page", strconv.Itoa(DEFAULT_PAGE_NUM))
	strSize := c.DefaultQuery("size", strconv.Itoa(DEFAULT_PAGE_SIZE))
	page, err := strconv.Atoi(strPage)
	if err != nil {
		page = DEFAULT_PAGE_NUM
	}
	size, err := strconv.Atoi(strSize)
	if err != nil {
		size = DEFAULT_PAGE_SIZE
	}
	urls, err := service.GetPagesShortUrls(strings.TrimSpace(url), page, size)
	c.HTML(http.StatusOK, "urls.html", gin.H{
		"title":       "短链接列表 - ohUrlShortener",
		"current_url": c.Request.URL.Path,
		"error":       err,
		"shortUrls":   urls,
		"page":        page,
		"size":        size,
		"prefix":      utils.AppConfig.UrlPrefix,
		"first_page":  page == 1,
		"last_page":   len(urls) < size,
		"url":         strings.TrimSpace(url),
	})
}

func AccessLogsPage(c *gin.Context) {
	url := c.DefaultQuery("url", "")
	strPage := c.DefaultQuery("page", strconv.Itoa(DEFAULT_PAGE_NUM))
	strSize := c.DefaultQuery("size", strconv.Itoa(DEFAULT_PAGE_SIZE))
	page, err := strconv.Atoi(strPage)
	if err != nil {
		page = DEFAULT_PAGE_NUM
	}
	size, err := strconv.Atoi(strSize)
	if err != nil {
		size = DEFAULT_PAGE_SIZE
	}
	logs, err := service.GetPagedAccessLogs(strings.TrimSpace(url), page, size)
	c.HTML(http.StatusOK, "access_logs.html", gin.H{
		"title":       "访问日志查询 - ohUrlShortener",
		"current_url": c.Request.URL.Path,
		"error":       err,
		"logs":        logs,
		"page":        page,
		"size":        size,
		"prefix":      utils.AppConfig.UrlPrefix,
		"first_page":  page == 1,
		"last_page":   len(logs) < size,
		"url":         strings.TrimSpace(url),
	})
}

func DashbaordPage(c *gin.Context) {
	count, stats, err := service.GetSumOfUrlStats()
	if err != nil {
		c.HTML(http.StatusOK, "dashboard.html", gin.H{
			"title":       "仪表盘 - ohUrlShortener",
			"current_url": c.Request.URL.Path,
			"error":       err,
		})
		return
	}

	top25, er := service.GetTop25Url()
	if er != nil {
		c.HTML(http.StatusOK, "dashboard.html", gin.H{
			"title":       "仪表盘 - ohUrlShortener",
			"current_url": c.Request.URL.Path,
			"error":       er,
		})
		return
	}

	c.HTML(http.StatusOK, "dashboard.html", gin.H{
		"title":       "仪表盘 - ohUrlShortener",
		"current_url": c.Request.URL.Path,
		"error":       err,
		"total_count": count,
		"prefix":      utils.AppConfig.UrlPrefix,
		"stats":       stats,
		"top25":       top25,
	})
}