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	}
22
23	url := query.Get("url")
24	if len(url) == 0 {
25		http.Error(writer, "400 Bad Request: URL parameter is required", 400)
26		return
27	}
28
29	name := query.Get("name")
30	if len(name) == 0 {
31		name = uniuri.NewLen(4)
32	}
33
34	log.Println("Saving \"" + url + "\" mapped to \"" + name + "\"")
35
36	err := m.database.Update(func(txn *badger.Txn) error {
37		return txn.Set([]byte(name), []byte(url))
38	})
39	if err != nil {
40		log.Fatal(err)
41	}
42
43	http.Error(writer, fmt.Sprint("URL mapped to "+name), 200)
44}