82 lines
1.7 KiB
Go
82 lines
1.7 KiB
Go
package main
|
|
|
|
import (
|
|
"log"
|
|
"net/http"
|
|
"strconv"
|
|
|
|
abcex "g.arns.lt/zordsdavini/abcex/v3"
|
|
"github.com/foolin/goview/supports/ginview"
|
|
"github.com/gin-contrib/sessions"
|
|
"github.com/gin-contrib/sessions/cookie"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func main() {
|
|
db, err := connectDB()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer db.Close()
|
|
|
|
r := gin.Default()
|
|
|
|
r.HTMLRender = ginview.Default()
|
|
|
|
store := cookie.NewStore([]byte("secret"))
|
|
r.Use(sessions.Sessions("mysession", store))
|
|
|
|
r.StaticFile("/favicon.ico", "./assets/favicon.ico")
|
|
|
|
r.GET("/", func(c *gin.Context) {
|
|
c.HTML(http.StatusOK, "index", gin.H{})
|
|
})
|
|
|
|
r.POST("/", func(c *gin.Context) {
|
|
shortUrl := getShortUrl(db, c.PostForm("url"))
|
|
if len(shortUrl) > 0 {
|
|
shortUrl = "https://" + c.Request.Host + "/" + shortUrl
|
|
}
|
|
c.HTML(http.StatusOK, "index", gin.H{
|
|
"shortUrl": shortUrl,
|
|
})
|
|
})
|
|
|
|
r.POST("/abcex", func(c *gin.Context) {
|
|
value, err := strconv.Atoi(c.PostForm("encode_abcex"))
|
|
if err != nil {
|
|
c.HTML(http.StatusOK, "index", gin.H{
|
|
"abcex_err": true,
|
|
})
|
|
} else {
|
|
c.HTML(http.StatusOK, "index", gin.H{
|
|
"abcex": abcex.Encode(int64(value)),
|
|
})
|
|
}
|
|
})
|
|
|
|
r.POST("/de-abcex", func(c *gin.Context) {
|
|
value := c.PostForm("decode_abcex")
|
|
if err != nil {
|
|
c.HTML(http.StatusOK, "index", gin.H{
|
|
"de_abcex_err": true,
|
|
})
|
|
} else {
|
|
c.HTML(http.StatusOK, "index", gin.H{
|
|
"de_abcex": abcex.Decode(value),
|
|
})
|
|
}
|
|
})
|
|
|
|
r.GET("/:short_url", func(c *gin.Context) {
|
|
redirectUrl := getRedirectUrl(db, c.Param("short_url"))
|
|
|
|
if len(redirectUrl) == 0 {
|
|
c.Redirect(http.StatusMovedPermanently, "/")
|
|
}
|
|
|
|
c.Redirect(http.StatusMovedPermanently, redirectUrl)
|
|
})
|
|
|
|
r.Run()
|
|
}
|