break up handlers

This commit is contained in:
Eliot Whalan 2016-06-19 14:36:00 +10:00
parent 3686f74bcb
commit fe4b307c1c
No known key found for this signature in database
GPG Key ID: C0A42175139840D6
1 changed files with 53 additions and 33 deletions

86
main.go
View File

@ -6,15 +6,18 @@ import (
"html" "html"
"io" "io"
"io/ioutil" "io/ioutil"
"log"
"net/http" "net/http"
"github.com/gorilla/mux"
"github.com/dchest/uniuri" "github.com/dchest/uniuri"
"github.com/ewhal/pygments" "github.com/ewhal/pygments"
_ "github.com/mattn/go-sqlite3" _ "github.com/mattn/go-sqlite3"
) )
const ( const (
ADDRESS = "https://p.pantsu.cat/" ADDRESS = "http://localhost:9900"
LENGTH = 6 LENGTH = 6
TEXT = "$ <command> | curl -F 'p=<-' " + ADDRESS + "\n" TEXT = "$ <command> | curl -F 'p=<-' " + ADDRESS + "\n"
PORT = ":9900" PORT = ":9900"
@ -53,7 +56,7 @@ func save(raw []byte) string {
s := generateName() s := generateName()
db, err := sql.Open("sqlite3", "./database.db") db, err := sql.Open("sqlite3", "./database.db")
check(err) check(err)
stmt, err := db.Prepare("INSERT INTO pastebin(id, data) values(?,?)") stmt, err := db.Prepare("INSERT INTO pastebin(id, hash, data, delkey) values(?,?,?,?)")
_, err = stmt.Exec(s, html.EscapeString(string(paste))) _, err = stmt.Exec(s, html.EscapeString(string(paste)))
check(err) check(err)
db.Close() db.Close()
@ -61,49 +64,66 @@ func save(raw []byte) string {
return s return s
} }
func pasteHandler(w http.ResponseWriter, r *http.Request) { func saveHandler(w http.ResponseWriter, r *http.Request) {
switch r.Method { switch r.Method {
case "GET":
param1 := html.EscapeString(r.URL.Query().Get("p"))
param2 := html.EscapeString(r.URL.Query().Get("lang"))
db, err := sql.Open("sqlite3", "./database.db")
var s string
err = db.QueryRow("select data from pastebin where id=?", param1).Scan(&s)
db.Close()
check(err)
if param1 != "" {
if err == sql.ErrNoRows {
io.WriteString(w, "Error invalid paste")
}
if param2 != "" {
highlight := pygments.Highlight(html.UnescapeString(s), param2, "html", "full, style=autumn,linenos=True, lineanchors=True,anchorlinenos=True,", "utf-8")
io.WriteString(w, highlight)
} else {
io.WriteString(w, html.UnescapeString(s))
}
} else {
io.WriteString(w, TEXT)
}
case "POST": case "POST":
buf, err := ioutil.ReadAll(r.Body) buf, err := ioutil.ReadAll(r.Body)
if err != nil { if err != nil {
http.Error(w, err.Error(), 500) http.Error(w, err.Error(), 500)
} }
name := save(buf) name := save(buf)
io.WriteString(w, ADDRESS+"?p="+name+"\n") io.WriteString(w, ADDRESS+"/p/"+name+"\n")
case "DELETE":
// Remove the record.
} }
}
func rootHandler(w http.ResponseWriter, r *http.Request) {
io.WriteString(w, TEXT)
}
func langHandler(w http.ResponseWriter, r *http.Request) {
vars := mux.Vars(r)
paste := vars["pasteId"]
lang := vars["lang"]
s := getPaste(paste)
highlight := pygments.Highlight(html.UnescapeString(s), lang, "html", "full, style=autumn,linenos=True, lineanchors=True,anchorlinenos=True,", "utf-8")
io.WriteString(w, highlight)
}
func getPaste(paste string) string {
param1 := html.EscapeString(paste)
db, err := sql.Open("sqlite3", "./database.db")
var s string
err = db.QueryRow("select data from pastebin where id=?", param1).Scan(&s)
db.Close()
check(err)
if err == sql.ErrNoRows {
return "Error invalid paste"
} else {
return html.UnescapeString(s)
}
}
func pasteHandler(w http.ResponseWriter, r *http.Request) {
vars := mux.Vars(r)
paste := vars["pasteId"]
s := getPaste(paste)
io.WriteString(w, s)
} }
func main() { func main() {
http.HandleFunc("/", pasteHandler) router := mux.NewRouter().StrictSlash(true)
err := http.ListenAndServe(PORT, nil) router.HandleFunc("/", rootHandler)
router.HandleFunc("/p/{pasteId}", pasteHandler)
router.HandleFunc("/p/{pasteId}/{lang}", langHandler)
router.HandleFunc("/save", saveHandler)
router.HandleFunc("/save", saveHandler)
err := http.ListenAndServe(PORT, router)
if err != nil { if err != nil {
panic(err) log.Fatal(err)
} }
} }