From bb61143fe249283bc19008de17fd6a18be3b72e9 Mon Sep 17 00:00:00 2001 From: Amolith Date: Mon, 24 Jan 2022 05:23:16 -0500 Subject: [PATCH] Refactor and improve formatting --- create.go | 46 +++++++++++++++++++++++----------------------- 1 file changed, 23 insertions(+), 23 deletions(-) diff --git a/create.go b/create.go index e46ad4b6d3637df2a516a8c6bbf432041bc72d94..4a8327bcee0f8a778a7169a3071200286b3e529a 100644 --- a/create.go +++ b/create.go @@ -16,29 +16,29 @@ func (m *model) create(writer http.ResponseWriter, request *http.Request) { token := request.Header.Get("Authorization") token = strings.TrimPrefix(token, "Bearer ") - if token == m.AccessToken { - url := query.Get("url") - if len(url) == 0 { - http.Error(writer, "400 Bad Request: URL parameter is required", 400) - } else { - name := query.Get("name") - if len(name) == 0 { - name = uniuri.NewLen(4) - } - - log.Println("Saving \"" + url + "\" mapped to \"" + name + "\"") - - if len(name) != 0 && len(url) != 0 { - err := m.database.Update(func(txn *badger.Txn) error { - return txn.Set([]byte(name), []byte(url)) - }) - if err != nil { - log.Fatal(err) - } - http.Error(writer, fmt.Sprint("URL mapped to "+name), 200) - } - } - } else { + if token != m.AccessToken { http.Error(writer, "401 Unauthorized: You do not have permission to create shortlinks", 403) } + + url := query.Get("url") + if len(url) == 0 { + http.Error(writer, "400 Bad Request: URL parameter is required", 400) + return + } + + name := query.Get("name") + if len(name) == 0 { + name = uniuri.NewLen(4) + } + + log.Println("Saving \"" + url + "\" mapped to \"" + name + "\"") + + err := m.database.Update(func(txn *badger.Txn) error { + return txn.Set([]byte(name), []byte(url)) + }) + if err != nil { + log.Fatal(err) + } + + http.Error(writer, fmt.Sprint("URL mapped to "+name), 200) }