url_shortiner/main.go

package main

import (
	"log"
	"net/http"
	"strconv"

	"g.arns.lt/zordsdavini/abcex"
	"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()
}