From 30f4ff12755533a3f22d625c24e14a35f8c14535 Mon Sep 17 00:00:00 2001 From: Eliot Whalan Date: Sat, 25 Jun 2016 08:33:56 +1000 Subject: [PATCH] Add expiry checking --- main.go | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/main.go b/main.go index 233f13d..3b3f7e3 100644 --- a/main.go +++ b/main.go @@ -252,11 +252,18 @@ func highlight(s string, lang string) (string, error) { func getPaste(paste string, lang string) (string, string) { param1 := html.EscapeString(paste) db, err := sql.Open("mysql", DATABASE) - var title, s, expiry string + var title, s string + var expiry string err = db.QueryRow("select title, data, expiry from pastebin where id=?", param1).Scan(&title, &s, &expiry) - fmt.Println(expiry) - db.Close() check(err) + if expiry > time.Now().Format(time.RFC3339) { + stmt, err := db.Prepare("delete from pastebin where id=?") + check(err) + _, err = stmt.Exec(param1) + check(err) + return "Error invalid paste", "" + } + db.Close() if err == sql.ErrNoRows { return "Error invalid paste", ""