a
/
nat
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Document more code

This commit is contained in:
Eliot Whalan 2016-07-04 12:52:23 +10:00
parent 8a4629c4b4
commit b2839aacfa
No known key found for this signature in database
GPG Key ID: C0A42175139840D6
1 changed files with 15 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
pastebin.go
View File

@ -43,6 +43,10 @@ const (
DATABASE = USERNAME + ":" + PASS + "@/" + NAME + "?charset=utf8" DATABASE = USERNAME + ":" + PASS + "@/" + NAME + "?charset=utf8"
) )
// Template pages
var templates = template.Must(template.ParseFiles("assets/paste.html", "assets/index.html", "assets/clone.html"))
var syntax, _ = ioutil.ReadFile("assets/syntax.html")
// Response API struct // Response API struct
type Response struct { type Response struct {
ID string `json:"id"` ID string `json:"id"`
@ -73,11 +77,13 @@ func check(err error) {
// generateName uses uniuri to generate a random string that isn't in the // generateName uses uniuri to generate a random string that isn't in the
// database // database
func generateName() string { func generateName() string {
// use uniuri to generate random string
id := uniuri.NewLen(LENGTH) id := uniuri.NewLen(LENGTH)
db, err := sql.Open("mysql", DATABASE) db, err := sql.Open("mysql", DATABASE)
check(err) check(err)
defer db.Close() defer db.Close()
// query database if id exists and if it does call generateName again
query, err := db.Query("select id from pastebin where id=?", id) query, err := db.Query("select id from pastebin where id=?", id)
if err != sql.ErrNoRows { if err != sql.ErrNoRows {
for query.Next() { for query.Next() {
@ -132,8 +138,10 @@ func save(raw string, lang string, title string, expiry string) Response {
check(err) check(err)
defer db.Close() defer db.Close()
// hash paste data and query database to see if paste exists
sha := hash(raw) sha := hash(raw)
query, err := db.Query("select id, title, hash, data, delkey from pastebin where hash=?", sha) query, err := db.Query("select id, title, hash, data, delkey from pastebin where hash=?", sha)
if err != sql.ErrNoRows { if err != sql.ErrNoRows {
for query.Next() { for query.Next() {
var id, title, hash, paste, delkey string var id, title, hash, paste, delkey string
@ -162,6 +170,7 @@ func save(raw string, lang string, title string, expiry string) Response {
} }
_, err = stmt.Exec(id, html.EscapeString(title), sha, dataEscaped, delKey, expiryTime) _, err = stmt.Exec(id, html.EscapeString(title), sha, dataEscaped, delKey, expiryTime)
check(err) check(err)
return Response{id, title, sha, url, len(dataEscaped), delKey} return Response{id, title, sha, url, len(dataEscaped), delKey}
} }
@ -253,7 +262,8 @@ func highlight(s string, lang string) (string, error) {
} }
// getPaste // getPaste takes pasteid and language
// queries the database and returns paste data
func getPaste(paste string, lang string) (string, string) { func getPaste(paste string, lang string) (string, string) {
param1 := html.EscapeString(paste) param1 := html.EscapeString(paste)
db, err := sql.Open("mysql", DATABASE) db, err := sql.Open("mysql", DATABASE)
@ -282,9 +292,6 @@ func getPaste(paste string, lang string) (string, string) {
return html.UnescapeString(s), html.UnescapeString(title) return html.UnescapeString(s), html.UnescapeString(title)
} }
var templates = template.Must(template.ParseFiles("assets/paste.html", "assets/index.html", "assets/clone.html"))
var syntax, _ = ioutil.ReadFile("assets/syntax.html")
func pasteHandler(w http.ResponseWriter, r *http.Request) { func pasteHandler(w http.ResponseWriter, r *http.Request) {
vars := mux.Vars(r) vars := mux.Vars(r)
paste := vars["pasteId"] paste := vars["pasteId"]
@ -339,6 +346,8 @@ func downloadHandler(w http.ResponseWriter, r *http.Request) {
vars := mux.Vars(r) vars := mux.Vars(r)
paste := vars["pasteId"] paste := vars["pasteId"]
s, _ := getPaste(paste, "") s, _ := getPaste(paste, "")
// Set header to an attachment so browser will automatically download it
w.Header().Set("Content-Disposition", "attachment; filename="+paste) w.Header().Set("Content-Disposition", "attachment; filename="+paste)
w.Header().Set("Content-Type", r.Header.Get("Content-Type")) w.Header().Set("Content-Type", r.Header.Get("Content-Type"))
io.WriteString(w, s) io.WriteString(w, s)
@ -348,6 +357,7 @@ func rawHandler(w http.ResponseWriter, r *http.Request) {
vars := mux.Vars(r) vars := mux.Vars(r)
paste := vars["pasteId"] paste := vars["pasteId"]
s, _ := getPaste(paste, "") s, _ := getPaste(paste, "")
// simply write string to browser
io.WriteString(w, s) io.WriteString(w, s)
} }