1package main
 2
 3import (
 4	"fmt"
 5	"log"
 6	"net/http"
 7	"strings"
 8
 9	"github.com/dchest/uniuri"
10	"github.com/dgraph-io/badger/v3"
11)
12
13func (m *model) create(writer http.ResponseWriter, request *http.Request) {
14	query := request.URL.Query()
15
16	token := request.Header.Get("Authorization")
17	token = strings.TrimPrefix(token, "Bearer ")
18
19	if token != m.AccessToken {
20		http.Error(writer, "401 Unauthorized: You do not have permission to create shortlinks", 403)
21		return
22	}
23
24	url := query.Get("url")
25	if len(url) == 0 {
26		http.Error(writer, "400 Bad Request: URL parameter is required", 400)
27		return
28	}
29
30	name := query.Get("name")
31	if len(name) == 0 {
32		name = uniuri.NewLen(4)
33	}
34
35	log.Println("Saving \"" + url + "\" mapped to \"" + name + "\"")
36
37	err := m.database.Update(func(txn *badger.Txn) error {
38		return txn.Set([]byte(name), []byte(url))
39	})
40	if err != nil {
41		log.Fatal(err)
42	}
43
44	http.Error(writer, fmt.Sprint("URL mapped to "+name), 200)
45}