Initial backend that collects entries

3 files changed, 102 insertions, 0 deletions

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..54438e8
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,2 @@
+stats
+*.db
diff --git a/entry.go b/entry.go
new file mode 100644
index 0000000..544a8c5
--- /dev/null
+++ b/entry.go
@@ -0,0 +1,62 @@
+package main
+
+import (
+	"github.com/jmoiron/sqlx"
+	_ "github.com/mattn/go-sqlite3"
+	"log"
+	"os"
+	"time"
+)
+
+const schema = `
+CREATE TABLE IF NOT EXISTS entries (
+	id INTEGER PRIMARY KEY,
+	location TEXT NOT NULL,
+	referrer TEXT,
+	started_at DATETIME NOT NULL,
+	time_spent INTEGER NOT NULL,
+	scrolled_to REAL NOT NULL
+);`
+
+type Entry struct {
+	Id         int       `db:"id"`
+	Location   string    `json:"location" db:"location"`
+	Referrer   string    `json:"referrer" db:"referrer"`
+	StartedAt  time.Time `json:"startedAt" db:"started_at"`
+	TimeSpent  int       `json:"timeSpent" db:"time_spent"`
+	ScrolledTo float32   `json:"scrolledTo" db:"scrolled_to"`
+}
+
+var db *sqlx.DB
+
+func InitEntries() {
+	file := os.Getenv("STATSDB")
+	if file == "" {
+		log.Println("Defaulting to STATSDB=./stats.db")
+		file = "./stats.db"
+	}
+
+	var err error
+	db, err = sqlx.Connect("sqlite3", file)
+	if err != nil {
+		log.Fatalln(err)
+	}
+
+	_, err = db.Exec(schema)
+	if err != nil {
+		log.Fatalln("Could not initialize schema")
+		log.Fatalln(err)
+	}
+}
+
+func CloseEntries() {
+	db.Close()
+}
+
+func InsertEntry(entry *Entry) error {
+	const query = `
+	INSERT INTO entries (location, referrer, started_at, time_spent, scrolled_to)
+	VALUES (:location, :referrer, :started_at, :time_spent, :scrolled_to);`
+	_, err := db.NamedExec(query, entry)
+	return err
+}
diff --git a/main.go b/main.go
new file mode 100644
index 0000000..1b5dd5b
--- /dev/null
+++ b/main.go
@@ -0,0 +1,38 @@
+package main
+
+import (
+	"encoding/json"
+	"github.com/gorilla/mux"
+	"log"
+	"net/http"
+)
+
+func handleEntry(w http.ResponseWriter, r *http.Request) {
+	var entry Entry
+
+	err := json.NewDecoder(r.Body).Decode(&entry)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusBadRequest)
+		return
+	}
+
+	err = InsertEntry(&entry)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+}
+
+func handleRequests() {
+	router := mux.NewRouter()
+	router.HandleFunc("/entry", handleEntry).Methods("POST")
+
+	log.Fatal(http.ListenAndServe(":8080", router))
+}
+
+func main() {
+	log.Println("Starting up")
+	InitEntries()
+	defer CloseEntries()
+	handleRequests()
+}