From fddc901332aa724c1f3b727b21b23c11330e9c19 Mon Sep 17 00:00:00 2001
From: Alex Yatskov <alex@foosoft.net>
Date: Thu, 26 Mar 2015 13:20:34 +0900
Subject: [PATCH] Adding clear handler

---
 server.go | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/server.go b/server.go
index b6ebaa2..bcefe22 100644
--- a/server.go
+++ b/server.go
@@ -25,6 +25,7 @@ package main
 import (
 	"database/sql"
 	"encoding/json"
+	"fmt"
 	_ "github.com/go-sql-driver/mysql"
 	"log"
 	"net/http"
@@ -238,6 +239,19 @@ func accessReview(rw http.ResponseWriter, req *http.Request) {
 	}
 }
 
+func clearHistory(rw http.ResponseWriter, req *http.Request) {
+	if _, err := db.Exec("DELETE FROM history"); err != nil {
+		log.Fatal(err)
+	}
+
+	if _, err := db.Exec("DELETE FROM historyGroups"); err != nil {
+		log.Fatal(err)
+	}
+
+	rw.Header().Set("Content-Type", "text/plain")
+	fmt.Fprint(rw, "History tables cleared")
+}
+
 func staticPath() (string, error) {
 	if len(os.Args) > 1 {
 		return os.Args[1], nil
@@ -263,6 +277,7 @@ func main() {
 	http.HandleFunc("/learn", addCategory)
 	http.HandleFunc("/forget", removeCategory)
 	http.HandleFunc("/access", accessReview)
+	http.HandleFunc("/clear", clearHistory)
 	http.Handle("/", http.FileServer(http.Dir(dir)))
 
 	log.Fatal(http.ListenAndServe(":8080", nil))