2020-03-29 08:48:43 +00:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
2021-08-10 20:34:25 +00:00
|
|
|
"context"
|
2020-03-29 08:48:43 +00:00
|
|
|
"database/sql"
|
2020-05-14 10:20:05 +00:00
|
|
|
"fmt"
|
2020-03-29 08:48:43 +00:00
|
|
|
"log"
|
|
|
|
"math/rand"
|
|
|
|
"net/http"
|
2021-08-10 20:34:25 +00:00
|
|
|
"os"
|
2020-05-14 10:20:05 +00:00
|
|
|
"strconv"
|
2020-11-15 15:52:03 +00:00
|
|
|
"sync"
|
2020-03-29 08:48:43 +00:00
|
|
|
"time"
|
2020-11-15 15:52:03 +00:00
|
|
|
|
2021-08-11 08:09:22 +00:00
|
|
|
gsd "git.jlel.se/jlelse/go-shutdowner"
|
2021-08-10 20:34:25 +00:00
|
|
|
"github.com/go-chi/chi/v5"
|
|
|
|
"github.com/go-chi/chi/v5/middleware"
|
2020-11-15 15:52:03 +00:00
|
|
|
_ "github.com/mattn/go-sqlite3"
|
|
|
|
"github.com/spf13/viper"
|
|
|
|
)
|
|
|
|
|
2021-08-11 08:09:22 +00:00
|
|
|
type app struct {
|
|
|
|
config *config
|
|
|
|
database *sql.DB
|
|
|
|
write sync.Mutex
|
|
|
|
shutdown gsd.Shutdowner
|
|
|
|
}
|
|
|
|
|
|
|
|
type config struct {
|
|
|
|
Port int `mapstructure:"port"`
|
|
|
|
DBPath string `mapstructure:"dbPath"`
|
|
|
|
Password string `mapstructure:"password"`
|
|
|
|
ShortUrl string `mapstructure:"shortUrl"`
|
|
|
|
DefaultUrl string `mapstructure:"defaultUrl"`
|
|
|
|
}
|
2020-03-29 08:48:43 +00:00
|
|
|
|
2021-08-11 08:09:22 +00:00
|
|
|
func (a *app) initRouter() (router *chi.Mux) {
|
|
|
|
router = chi.NewMux()
|
|
|
|
router.Use(middleware.GetHead)
|
|
|
|
router.Group(func(r chi.Router) {
|
|
|
|
r.Use(a.loginMiddleware)
|
2021-08-10 20:34:25 +00:00
|
|
|
r.Get("/s", shortenFormHandler)
|
2021-08-11 08:09:22 +00:00
|
|
|
r.Post("/s", a.shortenHandler)
|
2021-08-10 20:34:25 +00:00
|
|
|
r.Get("/t", shortenTextFormHandler)
|
2021-08-11 08:09:22 +00:00
|
|
|
r.Post("/t", a.shortenTextHandler)
|
2021-08-10 20:34:25 +00:00
|
|
|
r.Get("/u", updateFormHandler)
|
|
|
|
r.Get("/ut", updateTextFormHandler)
|
2021-08-11 08:09:22 +00:00
|
|
|
r.Post("/u", a.updateHandler)
|
2021-08-10 20:34:25 +00:00
|
|
|
r.Get("/d", deleteFormHandler)
|
2021-08-11 08:09:22 +00:00
|
|
|
r.Post("/d", a.deleteHandler)
|
|
|
|
r.Get("/l", a.listHandler)
|
2021-08-10 20:34:25 +00:00
|
|
|
})
|
2021-08-11 08:09:22 +00:00
|
|
|
router.Get("/{slug}", a.shortenedURLHandler)
|
|
|
|
router.Get("/", a.defaultURLRedirectHandler)
|
|
|
|
return
|
2020-11-15 15:52:03 +00:00
|
|
|
}
|
2020-03-29 08:48:43 +00:00
|
|
|
|
|
|
|
func main() {
|
|
|
|
rand.Seed(time.Now().UTC().UnixNano())
|
|
|
|
|
|
|
|
viper.SetDefault("dbPath", "data/goshort.db")
|
2020-05-14 10:20:05 +00:00
|
|
|
viper.SetDefault("port", 8080)
|
2020-03-29 08:48:43 +00:00
|
|
|
|
|
|
|
viper.SetConfigName("config")
|
|
|
|
viper.AddConfigPath("./config")
|
|
|
|
viper.AddConfigPath(".")
|
|
|
|
_ = viper.ReadInConfig()
|
|
|
|
|
|
|
|
if !viper.IsSet("password") {
|
|
|
|
log.Fatal("No password (password) is configured.")
|
2021-08-11 08:09:22 +00:00
|
|
|
return
|
2020-03-29 08:48:43 +00:00
|
|
|
}
|
|
|
|
if !viper.IsSet("shortUrl") {
|
|
|
|
log.Fatal("No short URL (shortUrl) is configured.")
|
2021-08-11 08:09:22 +00:00
|
|
|
return
|
2020-03-29 08:48:43 +00:00
|
|
|
}
|
|
|
|
if !viper.IsSet("defaultUrl") {
|
|
|
|
log.Fatal("No default URL (defaultUrl) is configured.")
|
2021-08-11 08:09:22 +00:00
|
|
|
return
|
2020-03-29 08:48:43 +00:00
|
|
|
}
|
|
|
|
|
2021-08-11 08:09:22 +00:00
|
|
|
app := &app{}
|
|
|
|
|
|
|
|
app.config = &config{}
|
|
|
|
err := viper.Unmarshal(app.config)
|
2020-03-29 08:48:43 +00:00
|
|
|
if err != nil {
|
2021-08-11 08:09:22 +00:00
|
|
|
log.Fatal("Failed to unmarshal config:", err.Error())
|
|
|
|
return
|
2020-03-29 08:48:43 +00:00
|
|
|
}
|
|
|
|
|
2021-08-11 08:09:22 +00:00
|
|
|
err = app.openDatabase()
|
|
|
|
if err != nil {
|
|
|
|
log.Println("Error opening database:", err.Error())
|
|
|
|
app.shutdown.ShutdownAndWait()
|
|
|
|
os.Exit(1)
|
|
|
|
return
|
|
|
|
}
|
2020-03-29 08:48:43 +00:00
|
|
|
|
2021-08-10 20:34:25 +00:00
|
|
|
httpServer := &http.Server{
|
2021-08-11 08:09:22 +00:00
|
|
|
Addr: ":" + strconv.Itoa(app.config.Port),
|
|
|
|
Handler: app.initRouter(),
|
2021-08-10 20:34:25 +00:00
|
|
|
ReadTimeout: 5 * time.Minute,
|
|
|
|
WriteTimeout: 5 * time.Minute,
|
|
|
|
}
|
2021-08-11 08:09:22 +00:00
|
|
|
app.shutdown.Add(func() {
|
2021-08-10 20:34:25 +00:00
|
|
|
toc, c := context.WithTimeout(context.Background(), 5*time.Second)
|
|
|
|
defer c()
|
|
|
|
if err := httpServer.Shutdown(toc); err != nil {
|
|
|
|
log.Println("Error on server shutdown:", err.Error())
|
|
|
|
}
|
|
|
|
log.Println("Stopped server")
|
|
|
|
})
|
|
|
|
go func() {
|
|
|
|
fmt.Println("Listening to " + httpServer.Addr)
|
|
|
|
if err := httpServer.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
|
|
|
log.Println("Failed to start HTTP server:", err.Error())
|
|
|
|
}
|
|
|
|
}()
|
|
|
|
|
2021-08-11 08:09:22 +00:00
|
|
|
app.shutdown.Wait()
|
2020-03-29 08:48:43 +00:00
|
|
|
}
|
|
|
|
|
2021-08-11 08:09:22 +00:00
|
|
|
func (a *app) loginMiddleware(next http.Handler) http.Handler {
|
2020-04-02 09:28:35 +00:00
|
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
2021-08-11 08:09:22 +00:00
|
|
|
if !a.checkPassword(w, r) {
|
2020-04-02 09:28:35 +00:00
|
|
|
return
|
|
|
|
}
|
|
|
|
next.ServeHTTP(w, r)
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2020-11-15 15:52:03 +00:00
|
|
|
func shortenFormHandler(w http.ResponseWriter, r *http.Request) {
|
2021-08-10 20:34:25 +00:00
|
|
|
if err := generateURLForm(w, "Shorten URL", "s", [][]string{{"url", r.FormValue("url")}, {"slug", r.FormValue("slug")}}); err != nil {
|
2020-05-14 10:20:05 +00:00
|
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
2020-03-29 08:48:43 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-11-15 15:52:03 +00:00
|
|
|
func updateFormHandler(w http.ResponseWriter, r *http.Request) {
|
2021-08-10 20:34:25 +00:00
|
|
|
if err := generateURLForm(w, "Update short link", "u", [][]string{{"slug", r.FormValue("slug")}, {"type", "url"}, {"new", r.FormValue("new")}}); err != nil {
|
2020-04-01 17:31:49 +00:00
|
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-11-15 15:52:03 +00:00
|
|
|
func updateTextFormHandler(w http.ResponseWriter, r *http.Request) {
|
2021-08-10 20:34:25 +00:00
|
|
|
if err := generateTextForm(w, "Update text", "u", [][]string{{"slug", r.FormValue("slug")}, {"type", "text"}}, [][]string{{"new", r.FormValue("new")}}); err != nil {
|
2020-04-01 17:31:49 +00:00
|
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-11-15 15:52:03 +00:00
|
|
|
func deleteFormHandler(w http.ResponseWriter, r *http.Request) {
|
2021-08-10 20:34:25 +00:00
|
|
|
if err := generateURLForm(w, "Delete short link", "d", [][]string{{"slug", r.FormValue("slug")}}); err != nil {
|
2020-04-01 17:31:49 +00:00
|
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-11-15 15:52:03 +00:00
|
|
|
func shortenTextFormHandler(w http.ResponseWriter, r *http.Request) {
|
2021-08-10 20:34:25 +00:00
|
|
|
if err := generateTextForm(w, "Save text", "t", [][]string{{"slug", r.FormValue("slug")}}, [][]string{{"text", r.FormValue("text")}}); err != nil {
|
2020-05-14 10:20:05 +00:00
|
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
2020-04-01 17:31:49 +00:00
|
|
|
}
|
2020-05-14 10:20:05 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func generateURLForm(w http.ResponseWriter, title string, url string, fields [][]string) error {
|
2020-11-15 15:52:03 +00:00
|
|
|
return urlFormTemplate.Execute(w, map[string]interface{}{
|
|
|
|
"Title": title,
|
|
|
|
"URL": url,
|
|
|
|
"Fields": fields,
|
2020-04-01 17:31:49 +00:00
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2020-05-14 10:20:05 +00:00
|
|
|
func generateTextForm(w http.ResponseWriter, title string, url string, fields [][]string, textAreas [][]string) error {
|
2020-11-15 15:52:03 +00:00
|
|
|
return textFormTemplate.Execute(w, map[string]interface{}{
|
|
|
|
"Title": title,
|
|
|
|
"URL": url,
|
|
|
|
"Fields": fields,
|
|
|
|
"TextAreas": textAreas,
|
2020-05-14 10:20:05 +00:00
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2021-08-11 08:09:22 +00:00
|
|
|
func (a *app) shortenHandler(w http.ResponseWriter, r *http.Request) {
|
2020-11-15 15:52:03 +00:00
|
|
|
writeShortenedURL := func(w http.ResponseWriter, slug string) {
|
2021-08-11 08:09:22 +00:00
|
|
|
_, _ = w.Write([]byte(a.config.ShortUrl + "/" + slug))
|
2020-03-29 08:48:43 +00:00
|
|
|
}
|
|
|
|
|
2021-08-10 20:34:25 +00:00
|
|
|
requestURL := r.FormValue("url")
|
2020-11-15 15:52:03 +00:00
|
|
|
if requestURL == "" {
|
2020-03-29 08:48:43 +00:00
|
|
|
http.Error(w, "url parameter not set", http.StatusBadRequest)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2021-08-10 20:34:25 +00:00
|
|
|
slug := r.FormValue("slug")
|
2020-03-29 11:32:28 +00:00
|
|
|
manualSlug := false
|
2020-03-29 09:38:47 +00:00
|
|
|
if slug == "" {
|
2021-08-11 08:09:22 +00:00
|
|
|
_ = a.database.QueryRow("SELECT slug FROM redirect WHERE url = ?", requestURL).Scan(&slug)
|
2020-03-29 11:32:28 +00:00
|
|
|
} else {
|
|
|
|
manualSlug = true
|
2020-03-29 08:48:43 +00:00
|
|
|
}
|
|
|
|
|
2020-03-29 09:38:47 +00:00
|
|
|
if slug != "" {
|
2021-08-11 08:09:22 +00:00
|
|
|
if e, _ := a.slugExists(slug); e {
|
2020-03-29 11:32:28 +00:00
|
|
|
if manualSlug {
|
|
|
|
http.Error(w, "slug already in use", http.StatusBadRequest)
|
|
|
|
return
|
|
|
|
}
|
2020-11-15 15:52:03 +00:00
|
|
|
writeShortenedURL(w, slug)
|
2020-03-29 08:48:43 +00:00
|
|
|
return
|
|
|
|
}
|
2020-03-29 09:38:47 +00:00
|
|
|
} else {
|
2021-08-10 20:34:25 +00:00
|
|
|
exists := true
|
|
|
|
var err error
|
|
|
|
for exists {
|
2020-03-29 09:38:47 +00:00
|
|
|
slug = generateSlug()
|
2021-08-11 08:09:22 +00:00
|
|
|
exists, err = a.slugExists(slug)
|
2020-03-29 09:38:47 +00:00
|
|
|
if err != nil {
|
|
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
2020-03-29 08:48:43 +00:00
|
|
|
}
|
|
|
|
|
2021-08-11 08:09:22 +00:00
|
|
|
if err := a.insertRedirect(slug, requestURL, typUrl); err != nil {
|
2020-03-29 08:48:43 +00:00
|
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
w.WriteHeader(http.StatusCreated)
|
2020-11-15 15:52:03 +00:00
|
|
|
writeShortenedURL(w, slug)
|
2020-03-29 08:48:43 +00:00
|
|
|
}
|
|
|
|
|
2021-08-11 08:09:22 +00:00
|
|
|
func (a *app) shortenTextHandler(w http.ResponseWriter, r *http.Request) {
|
2020-11-15 15:52:03 +00:00
|
|
|
writeShortenedURL := func(w http.ResponseWriter, slug string) {
|
2021-08-11 08:09:22 +00:00
|
|
|
_, _ = w.Write([]byte(a.config.ShortUrl + "/" + slug))
|
2020-05-14 10:20:05 +00:00
|
|
|
}
|
|
|
|
|
2021-08-10 20:34:25 +00:00
|
|
|
requestText := r.FormValue("text")
|
2020-05-14 10:20:05 +00:00
|
|
|
if requestText == "" {
|
|
|
|
http.Error(w, "text parameter not set", http.StatusBadRequest)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2021-08-10 20:34:25 +00:00
|
|
|
slug := r.FormValue("slug")
|
2020-05-14 10:20:05 +00:00
|
|
|
manualSlug := false
|
|
|
|
if slug == "" {
|
2021-08-11 08:09:22 +00:00
|
|
|
_ = a.database.QueryRow("SELECT slug FROM redirect WHERE url = ? and type = 'text'", requestText).Scan(&slug)
|
2020-05-14 10:20:05 +00:00
|
|
|
} else {
|
|
|
|
manualSlug = true
|
|
|
|
}
|
|
|
|
|
|
|
|
if slug != "" {
|
2021-08-11 08:09:22 +00:00
|
|
|
if e, _ := a.slugExists(slug); e {
|
2020-05-14 10:20:05 +00:00
|
|
|
if manualSlug {
|
|
|
|
http.Error(w, "slug already in use", http.StatusBadRequest)
|
|
|
|
return
|
|
|
|
}
|
2020-11-15 15:52:03 +00:00
|
|
|
writeShortenedURL(w, slug)
|
2020-05-14 10:20:05 +00:00
|
|
|
return
|
|
|
|
}
|
|
|
|
} else {
|
2021-08-10 20:34:25 +00:00
|
|
|
exists := true
|
|
|
|
var err error
|
|
|
|
for exists {
|
2020-05-14 10:20:05 +00:00
|
|
|
slug = generateSlug()
|
2021-08-11 08:09:22 +00:00
|
|
|
exists, err = a.slugExists(slug)
|
2020-05-14 10:20:05 +00:00
|
|
|
if err != nil {
|
|
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-08-11 08:09:22 +00:00
|
|
|
if err := a.insertRedirect(slug, requestText, typText); err != nil {
|
2020-05-14 10:20:05 +00:00
|
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
w.WriteHeader(http.StatusCreated)
|
2020-11-15 15:52:03 +00:00
|
|
|
writeShortenedURL(w, slug)
|
2020-05-14 10:20:05 +00:00
|
|
|
}
|
|
|
|
|
2021-08-11 08:09:22 +00:00
|
|
|
func (a *app) updateHandler(w http.ResponseWriter, r *http.Request) {
|
2021-08-10 20:34:25 +00:00
|
|
|
slug := r.FormValue("slug")
|
2020-03-30 07:47:21 +00:00
|
|
|
if slug == "" {
|
|
|
|
http.Error(w, "Specify the slug to update", http.StatusBadRequest)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2021-08-10 20:34:25 +00:00
|
|
|
newURL := r.FormValue("new")
|
2020-11-15 15:52:03 +00:00
|
|
|
if newURL == "" {
|
2020-03-30 07:47:21 +00:00
|
|
|
http.Error(w, "Specify the new URL", http.StatusBadRequest)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2021-08-10 20:34:25 +00:00
|
|
|
typeString := r.FormValue("type")
|
2020-05-14 10:20:05 +00:00
|
|
|
if typeString == "" {
|
|
|
|
typeString = "url"
|
|
|
|
}
|
|
|
|
|
2021-08-11 08:09:22 +00:00
|
|
|
if e, err := a.slugExists(slug); err != nil || !e {
|
2020-11-15 16:05:09 +00:00
|
|
|
http.NotFound(w, r)
|
2020-03-30 07:47:21 +00:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2021-08-11 08:09:22 +00:00
|
|
|
a.write.Lock()
|
|
|
|
if _, err := a.database.Exec("UPDATE redirect SET url = ?, type = ? WHERE slug = ?", newURL, typeString, slug); err != nil {
|
|
|
|
a.write.Unlock()
|
2020-03-30 07:47:21 +00:00
|
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
|
|
return
|
|
|
|
}
|
2021-08-11 08:09:22 +00:00
|
|
|
a.write.Unlock()
|
2020-03-30 07:47:21 +00:00
|
|
|
|
|
|
|
w.WriteHeader(http.StatusAccepted)
|
|
|
|
_, _ = w.Write([]byte("Slug updated"))
|
|
|
|
}
|
|
|
|
|
2021-08-11 08:09:22 +00:00
|
|
|
func (a *app) deleteHandler(w http.ResponseWriter, r *http.Request) {
|
2021-08-10 20:34:25 +00:00
|
|
|
slug := r.FormValue("slug")
|
2020-03-29 10:47:35 +00:00
|
|
|
if slug == "" {
|
|
|
|
http.Error(w, "Specify the slug to delete", http.StatusBadRequest)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2021-08-11 08:09:22 +00:00
|
|
|
if e, err := a.slugExists(slug); !e || err != nil {
|
2020-11-15 16:05:09 +00:00
|
|
|
http.NotFound(w, r)
|
2020-03-29 10:47:35 +00:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2021-08-11 08:09:22 +00:00
|
|
|
if err := a.deleteSlug(slug); err != nil {
|
2020-03-29 10:47:35 +00:00
|
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
w.WriteHeader(http.StatusAccepted)
|
|
|
|
_, _ = w.Write([]byte("Slug deleted"))
|
|
|
|
}
|
|
|
|
|
2021-08-11 08:09:22 +00:00
|
|
|
func (a *app) listHandler(w http.ResponseWriter, r *http.Request) {
|
2020-04-02 09:28:35 +00:00
|
|
|
type row struct {
|
|
|
|
Slug string
|
2020-11-15 15:52:03 +00:00
|
|
|
URL string
|
2020-04-02 09:28:35 +00:00
|
|
|
Hits int
|
|
|
|
}
|
|
|
|
var list []row
|
2021-08-11 08:09:22 +00:00
|
|
|
rows, err := a.database.Query("SELECT slug, url, hits FROM redirect")
|
2020-04-02 09:28:35 +00:00
|
|
|
if err != nil {
|
|
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
for rows.Next() {
|
|
|
|
var r row
|
2020-11-15 15:52:03 +00:00
|
|
|
err = rows.Scan(&r.Slug, &r.URL, &r.Hits)
|
2020-04-02 09:28:35 +00:00
|
|
|
if err != nil {
|
|
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
list = append(list, r)
|
|
|
|
}
|
2020-05-14 10:20:05 +00:00
|
|
|
err = listTemplate.Execute(w, &list)
|
2020-04-02 09:28:35 +00:00
|
|
|
if err != nil {
|
|
|
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-08-11 08:09:22 +00:00
|
|
|
func (a *app) checkPassword(w http.ResponseWriter, r *http.Request) bool {
|
|
|
|
// Check basic auth
|
|
|
|
if _, pass, ok := r.BasicAuth(); ok && pass == a.config.Password {
|
2020-03-29 11:08:47 +00:00
|
|
|
return true
|
|
|
|
}
|
2021-08-11 08:09:22 +00:00
|
|
|
// Check query or form param
|
|
|
|
if r.FormValue("password") == a.config.Password {
|
|
|
|
return true
|
2020-03-29 11:08:47 +00:00
|
|
|
}
|
2021-08-11 08:09:22 +00:00
|
|
|
// Require password
|
|
|
|
w.Header().Set("WWW-Authenticate", `Basic realm="Please enter a password!"`)
|
|
|
|
http.Error(w, "Not authenticated", http.StatusUnauthorized)
|
|
|
|
return false
|
2020-03-29 11:08:47 +00:00
|
|
|
}
|
|
|
|
|
2020-03-29 08:48:43 +00:00
|
|
|
func generateSlug() string {
|
|
|
|
var chars = []rune("0123456789abcdefghijklmnopqrstuvwxyz")
|
|
|
|
s := make([]rune, 6)
|
|
|
|
for i := range s {
|
|
|
|
s[i] = chars[rand.Intn(len(chars))]
|
|
|
|
}
|
|
|
|
return string(s)
|
|
|
|
}
|
|
|
|
|
2021-08-11 08:09:22 +00:00
|
|
|
func (a *app) shortenedURLHandler(w http.ResponseWriter, r *http.Request) {
|
2020-11-15 15:52:03 +00:00
|
|
|
slug := chi.URLParam(r, "slug")
|
2020-03-29 08:48:43 +00:00
|
|
|
|
2020-11-15 15:52:03 +00:00
|
|
|
var redirectURL, typeString string
|
2021-08-11 08:09:22 +00:00
|
|
|
err := a.database.QueryRow("SELECT url, type FROM redirect WHERE slug = ?", slug).Scan(&redirectURL, &typeString)
|
2020-03-29 08:48:43 +00:00
|
|
|
if err != nil {
|
|
|
|
http.NotFound(w, r)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2021-08-11 08:09:22 +00:00
|
|
|
a.increaseHits(slug)
|
2020-03-29 08:48:43 +00:00
|
|
|
|
2021-08-11 08:09:22 +00:00
|
|
|
switch typeString {
|
|
|
|
case typText:
|
2020-11-15 15:52:03 +00:00
|
|
|
_, _ = w.Write([]byte(redirectURL))
|
2021-08-11 08:09:22 +00:00
|
|
|
default:
|
2020-11-15 15:52:03 +00:00
|
|
|
http.Redirect(w, r, redirectURL, http.StatusTemporaryRedirect)
|
2020-05-14 10:20:05 +00:00
|
|
|
}
|
2020-03-29 08:48:43 +00:00
|
|
|
}
|
|
|
|
|
2021-08-11 08:09:22 +00:00
|
|
|
func (a *app) defaultURLRedirectHandler(w http.ResponseWriter, r *http.Request) {
|
|
|
|
http.Redirect(w, r, a.config.DefaultUrl, http.StatusTemporaryRedirect)
|
2020-03-29 08:48:43 +00:00
|
|
|
}
|