83 lines
1.3 KiB
Go
83 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"github.com/dchest/uniuri"
|
|
"io"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"os"
|
|
)
|
|
|
|
const (
|
|
directory = "/tmp/"
|
|
address = "http://localhost:8080/?"
|
|
length = 4
|
|
text = "$ <command> | curl -F 'paste=<-'" + address + "\n"
|
|
)
|
|
|
|
func check(e error) {
|
|
if e != nil {
|
|
panic(e)
|
|
}
|
|
}
|
|
|
|
func exists(location string) bool {
|
|
if _, err := os.Stat(location); err != nil {
|
|
if os.IsNotExist(err) {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
|
|
}
|
|
|
|
func generateName() string {
|
|
s := uniuri.NewLen(length)
|
|
file := exists(directory + s)
|
|
if file == true {
|
|
generateName()
|
|
}
|
|
|
|
return s
|
|
|
|
}
|
|
func save(buf []byte) string {
|
|
paste := buf[92 : len(buf)-46]
|
|
|
|
s := generateName()
|
|
location := directory + s
|
|
|
|
err := ioutil.WriteFile(location, paste, 0644)
|
|
check(err)
|
|
|
|
return s
|
|
}
|
|
|
|
func pasteHandler(w http.ResponseWriter, r *http.Request) {
|
|
switch r.Method {
|
|
case "GET":
|
|
param := r.URL.RawQuery
|
|
if param != "" {
|
|
d := directory + param
|
|
s, err := ioutil.ReadFile(d)
|
|
check(err)
|
|
io.WriteString(w, string(s))
|
|
} else {
|
|
io.WriteString(w, text)
|
|
}
|
|
case "POST":
|
|
buf, err := ioutil.ReadAll(r.Body)
|
|
check(err)
|
|
io.WriteString(w, address+save(buf)+"\n")
|
|
case "DELETE":
|
|
// Remove the record.
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
http.HandleFunc("/", pasteHandler)
|
|
err := http.ListenAndServe(":8080", nil)
|
|
check(err)
|
|
|
|
}
|