vodkas

commit dbac86c18fa40bfcd599541cee0ce8ddd0af7a05
Author: Vetle Haflan <vetle@haflan.dev>
Date:   Mon, 30 Sep 2019 22:01:16 +0200

Init with vodkas.go

Diffstat:
A
.gitignore
 | 
2
++
A
vodkas.go
 | 
119
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

2 files changed, 121 insertions(+), 0 deletions(-)
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1,2 @@
+.idea/
+vodkas
diff --git a/vodkas.go b/vodkas.go
@@ -0,0 +1,119 @@
+//  Seems to work: `go build -ldflags "-linkmode external -extldflags -static"`
+package main
+
+import (
+        //"crypto/rand"
+        "flag"
+        "fmt"
+        "github.com/gorilla/mux"
+        "io/ioutil"
+        "log"
+        "net/http"
+        "strings"
+        "sync"
+)
+
+
+/**************** Handler Functions ****************/
+
+const InfoMessage = `Usage: 
+- POUR:     curl vetle.vodka[/<requested-shot-code>] -d <data>
+- SHOT:     curl vetle.vodka/<shot-code>
+
+A shot code is an at-the-moment unique ID that's linked to the dumped data.
+As soon as a specific shot has been accessed both the link and the contents 
+are removed completely.
+`
+
+type SyncedStorage struct{
+        links map[string]string
+        sync.Mutex
+}
+
+var storage SyncedStorage
+
+func (storage *SyncedStorage) push(code string, contents string) {
+        storage.Lock()
+        defer storage.Unlock()
+        storage.links[code] = contents
+}
+
+func (storage *SyncedStorage) pop(code string) (contents string, found bool) {
+        storage.Lock()
+        defer storage.Unlock()
+        contents, found = storage.links[code]
+        delete(storage.links, code)
+        return
+}
+
+// TODO: Separate GETs and POSTs
+func MainHandler(res http.ResponseWriter, r *http.Request) {
+        b, err := ioutil.ReadAll(r.Body)
+        if err != nil {
+                res.WriteHeader(http.StatusInternalServerError)
+                return
+        }
+        storage.push("sorandom", string(b))
+        textOnly := r.Header.Get("Simple") != ""    // Should be able to force simple
+        textOnly = textOnly || strings.Contains(r.Header.Get("User-Agent"), "curl")
+        var responseText string
+        if textOnly {
+                responseText = InfoMessage
+        } else {
+                responseText = "This will be replaced by a template"
+        }
+        if _, err = res.Write([]byte(responseText)); err != nil {
+                log.Panicln("Error when trying to write response body")
+        }
+        return
+}
+
+// func PourHandler(res http.ResponseWriter, r *http.Request) {
+//}
+
+// TODO: Generate random code if no request given (https://flaviocopes.com/go-random/)
+func ShotHandler(res http.ResponseWriter, r *http.Request) {
+        code := mux.Vars(r)["shotCode"]
+        contents, found := storage.pop(code)
+        // GET requests only
+        /*if !found {
+                res.WriteHeader(http.StatusNotFound)
+                if _, err := res.Write([]byte(fmt.Sprint("404 no shot here\n"))); err != nil {
+                        log.Panicln("Error when trying to write response")
+                }
+                return*/
+        if found {
+                // For POST requests to specific code, this should actually return 'link taken' or something
+                if _, err := res.Write([]byte(contents)); err != nil {
+                        log.Panicln("Error when trying to write response body")
+                }
+        } else {
+                b, err := ioutil.ReadAll(r.Body)
+                if err != nil {
+                        res.WriteHeader(http.StatusInternalServerError)
+                        log.Panicln("Error when trying to read body")
+                }
+                contents = string(b)
+                storage.push(code, contents)
+                if _, err := res.Write([]byte(fmt.Sprint("Contents stored in given link\n"))); err != nil {
+                        log.Panicln("Error when trying to write response")
+                }
+        }
+        fmt.Printf("Request from client: %v\n", r.Header.Get("User-Agent"))
+        return
+}
+
+/**************** Main ****************/
+func main(){
+        storage.links = make(map[string]string)
+        port := flag.Int("p", 8080, "Port")
+        flag.Parse()
+        defer fmt.Println("Server shutting down")
+        fmt.Println("Server started listening at port", *port)
+        router := mux.NewRouter()
+        router.HandleFunc("/", MainHandler)
+        router.HandleFunc("/{shotCode}", ShotHandler)
+        http.Handle("/", router)
+
+        log.Fatal(http.ListenAndServe(fmt.Sprintf(":%v", *port), nil))
+}