nat/main.go

101 lines
1.7 KiB
Go

package main
import (
"fmt"
"io"
"io/ioutil"
"net/http"
"os"
"github.com/dchest/uniuri"
"github.com/ewhal/pygments"
)
const (
DIRECTORY = "/tmp/"
ADDRESS = "http://localhost:8080/"
LENGTH = 4
TEXT = "$ <command> | curl -F 'p=<-' lang='python'" + ADDRESS + "\n"
PORT = ":8080"
)
func check(err error) {
if err != nil {
fmt.Println(err)
}
}
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(raw []byte) string {
paste := raw[85 : len(raw)-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":
param1 := r.URL.Query().Get("p")
param2 := r.URL.Query().Get("lang")
if param1 != "" {
d := DIRECTORY + param1
s, err := ioutil.ReadFile(d)
if err != nil {
http.Error(w, err.Error(), 500)
}
if param2 != "" {
highlight := pygments.Highlight(string(s), param2, "html", "full, style=autumn,linenos=True, lineanchors=True,anchorlinenos=True,", "utf-8")
io.WriteString(w, string(highlight))
} else {
io.WriteString(w, string(s))
}
} else {
io.WriteString(w, TEXT)
}
case "POST":
buf, err := ioutil.ReadAll(r.Body)
if err != nil {
http.Error(w, err.Error(), 500)
}
io.WriteString(w, ADDRESS+"?p="+save(buf)+"\n")
case "DELETE":
// Remove the record.
}
}
func main() {
http.HandleFunc("/", pasteHandler)
err := http.ListenAndServe(PORT, nil)
if err != nil {
panic(err)
}
}