From d9ccc888ecf284fc969a7e42b4883b85fb46136f Mon Sep 17 00:00:00 2001
From: Eliot Whalan <ewhal@pantsu.cat>
Date: Sun, 19 Jun 2016 16:16:08 +1000
Subject: [PATCH] Add json output

---
 main.go | 85 +++++++++++++++++++++++++++++++++++++++++++++------------
 1 file changed, 68 insertions(+), 17 deletions(-)

diff --git a/main.go b/main.go
index 0d8d59b..403094f 100644
--- a/main.go
+++ b/main.go
@@ -1,7 +1,10 @@
 package main
 
 import (
+	"crypto/sha1"
 	"database/sql"
+	"encoding/base64"
+	"encoding/json"
 	"fmt"
 	"html"
 	"io"
@@ -9,10 +12,9 @@ import (
 	"log"
 	"net/http"
 
-	"github.com/gorilla/mux"
-
 	"github.com/dchest/uniuri"
 	"github.com/ewhal/pygments"
+	"github.com/gorilla/mux"
 	_ "github.com/mattn/go-sqlite3"
 )
 
@@ -23,6 +25,14 @@ const (
 	PORT    = ":9900"
 )
 
+type Response struct {
+	ID     string `json:"id"`
+	HASH   string `json:"hash"`
+	URL    string `json:"url"`
+	SIZE   int    `json:"size"`
+	DELKEY string `json:"delkey"`
+}
+
 func check(err error) {
 	if err != nil {
 		fmt.Println(err)
@@ -50,29 +60,69 @@ func generateName() string {
 	return s
 
 }
-func save(raw []byte) string {
-	paste := raw[86 : len(raw)-46]
-
-	s := generateName()
-	db, err := sql.Open("sqlite3", "./database.db")
-	check(err)
-	stmt, err := db.Prepare("INSERT INTO pastebin(id, hash, data, delkey) values(?,?,?,?)")
-	_, err = stmt.Exec(s, html.EscapeString(string(paste)))
-	check(err)
-	db.Close()
-
-	return s
+func hash(paste []byte) string {
+	hasher := sha1.New()
+	hasher.Write(paste)
+	sha := base64.URLEncoding.EncodeToString(hasher.Sum(nil))
+	return sha
 }
 
+func save(raw []byte) []string {
+	p := raw[86 : len(raw)-46]
+
+	db, err := sql.Open("sqlite3", "./database.db")
+	check(err)
+
+	sha := hash(p)
+	id := generateName()
+	url := ADDRESS + "/p/" + id
+	delKey := uniuri.NewLen(40)
+	paste := html.EscapeString(string(p))
+
+	stmt, err := db.Prepare("INSERT INTO pastebin(id, hash, data, delkey) values(?,?,?,?)")
+	check(err)
+	_, err = stmt.Exec(id, sha, paste, delKey)
+	check(err)
+	db.Close()
+	return []string{id, sha, url, paste, delKey}
+
+}
+
+func delHandler(w http.ResponseWriter, r *http.Request) {
+}
 func saveHandler(w http.ResponseWriter, r *http.Request) {
+	vars := mux.Vars(r)
+	output := vars["output"]
 	switch r.Method {
 	case "POST":
 		buf, err := ioutil.ReadAll(r.Body)
 		if err != nil {
 			http.Error(w, err.Error(), 500)
 		}
-		name := save(buf)
-		io.WriteString(w, ADDRESS+"/p/"+name+"\n")
+
+		values := save(buf)
+
+		switch output {
+		case "json":
+			b := &Response{
+				ID:     values[0],
+				HASH:   values[1],
+				URL:    values[2],
+				SIZE:   len(values[3]),
+				DELKEY: values[4],
+			}
+
+			w.Header().Set("Content-Type", "application/json")
+			err := json.NewEncoder(w).Encode(b)
+
+			if err != nil {
+				http.Error(w, err.Error(), http.StatusInternalServerError)
+				return
+			}
+
+		default:
+			io.WriteString(w, values[2]+"\n")
+		}
 	}
 
 }
@@ -120,7 +170,8 @@ func main() {
 	router.HandleFunc("/p/{pasteId}", pasteHandler)
 	router.HandleFunc("/p/{pasteId}/{lang}", langHandler)
 	router.HandleFunc("/save", saveHandler)
-	router.HandleFunc("/save", saveHandler)
+	router.HandleFunc("/save/{output}", saveHandler)
+	router.HandleFunc("/del/{pasteId}/{delKey}", delHandler)
 	err := http.ListenAndServe(PORT, router)
 	if err != nil {
 		log.Fatal(err)