nat/main.go

130 lines
2.5 KiB
Go
Raw Normal View History

package main
import (
"database/sql"
2016-06-11 03:33:29 +00:00
"fmt"
"html"
2016-06-11 01:22:19 +00:00
"io"
"io/ioutil"
2016-06-19 04:36:00 +00:00
"log"
"net/http"
2016-06-19 00:17:35 +00:00
2016-06-19 04:36:00 +00:00
"github.com/gorilla/mux"
2016-06-19 00:17:35 +00:00
"github.com/dchest/uniuri"
"github.com/ewhal/pygments"
_ "github.com/mattn/go-sqlite3"
)
2016-06-11 01:22:19 +00:00
const (
2016-06-19 04:36:00 +00:00
ADDRESS = "http://localhost:9900"
LENGTH = 6
TEXT = "$ <command> | curl -F 'p=<-' " + ADDRESS + "\n"
PORT = ":9900"
2016-06-11 01:22:19 +00:00
)
2016-06-10 14:29:22 +00:00
2016-06-11 03:33:29 +00:00
func check(err error) {
if err != nil {
fmt.Println(err)
}
}
func generateName() string {
2016-06-11 02:31:03 +00:00
s := uniuri.NewLen(LENGTH)
db, err := sql.Open("sqlite3", "./database.db")
check(err)
query, err := db.Query("select id from pastebin")
for query.Next() {
var id string
err := query.Scan(&id)
if err != nil {
}
if id == s {
generateName()
}
2016-06-10 14:29:22 +00:00
}
db.Close()
2016-06-10 14:29:22 +00:00
return s
}
2016-06-11 02:31:03 +00:00
func save(raw []byte) string {
paste := raw[86 : len(raw)-46]
s := generateName()
db, err := sql.Open("sqlite3", "./database.db")
check(err)
2016-06-19 04:36:00 +00:00
stmt, err := db.Prepare("INSERT INTO pastebin(id, hash, data, delkey) values(?,?,?,?)")
_, err = stmt.Exec(s, html.EscapeString(string(paste)))
check(err)
db.Close()
2016-06-11 01:22:19 +00:00
return s
}
2016-06-19 04:36:00 +00:00
func saveHandler(w http.ResponseWriter, r *http.Request) {
2016-06-10 14:29:22 +00:00
switch r.Method {
case "POST":
2016-06-11 02:22:22 +00:00
buf, err := ioutil.ReadAll(r.Body)
2016-06-11 03:33:29 +00:00
if err != nil {
http.Error(w, err.Error(), 500)
}
name := save(buf)
2016-06-19 04:36:00 +00:00
io.WriteString(w, ADDRESS+"/p/"+name+"\n")
}
}
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)
2016-06-10 14:29:22 +00:00
}
2016-06-19 04:36:00 +00:00
}
func pasteHandler(w http.ResponseWriter, r *http.Request) {
vars := mux.Vars(r)
paste := vars["pasteId"]
s := getPaste(paste)
io.WriteString(w, s)
}
func main() {
2016-06-19 04:36:00 +00:00
router := mux.NewRouter().StrictSlash(true)
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)
2016-06-11 03:33:29 +00:00
if err != nil {
2016-06-19 04:36:00 +00:00
log.Fatal(err)
2016-06-11 03:33:29 +00:00
}
}