From 245ab6f11d7026e8e8218752839659bc04fdc102 Mon Sep 17 00:00:00 2001
From: rtz12 <koenig@fagott.pw>
Date: Tue, 11 Oct 2016 21:04:33 +0200
Subject: NSA implementiert


diff --git a/nsa.go b/nsa.go
new file mode 100644
index 0000000..827b4e6
--- /dev/null
+++ b/nsa.go
@@ -0,0 +1,129 @@
+package nsa
+
+import (
+	"database/sql"
+	"encoding/json"
+	"fmt"
+	"os"
+	"strings"
+)
+
+var database *sql.DB
+var eventmap = make(map[string]int)
+
+type Logger struct {
+	ip string
+}
+
+func NewLogger(ip string) *Logger {
+	return &Logger{
+		ip: ip,
+	}
+}
+
+func (l *Logger) getEventID(event string) int {
+	evID, ok := eventmap[event]
+	if !ok {
+		logErr("Event %s does not exist.", event)
+		return 0
+	}
+	return evID
+}
+
+func (l *Logger) getJSON(data interface{}) string {
+	jsonStr, err := json.Marshal(data)
+	if err != nil {
+		logErr("%v", err)
+		return ""
+	}
+	return string(jsonStr)
+}
+
+func (l *Logger) insertEvent(id int, data string) {
+	_, err := database.Exec(
+		"INSERT INTO nsa.events (type, data, client_ip) VALUES ($1, $2, $3);",
+		id, data, l.ip)
+	if err != nil {
+		logErr("%v", err)
+	}
+}
+
+func (l *Logger) Log(event string, data interface{}) {
+	go func() {
+		id := l.getEventID(event)
+		if id == 0 {
+			return
+		}
+		d := l.getJSON(data)
+		if d == "" {
+			return
+		}
+		l.insertEvent(id, d)
+	}()
+}
+
+func logErr(msg string, params ...interface{}) {
+	fmt.Fprintf(
+		os.Stderr,
+		"NSA: Error: %v\n",
+		fmt.Sprintf(msg, params...))
+}
+
+func Init(db *sql.DB) error {
+	err := db.Ping()
+	if err != nil {
+		return err
+	}
+	database = db
+	return nil
+}
+
+func SetEvents(events []string) error {
+	rows, err := database.Query("SELECT id, name FROM nsa.event_types;")
+	if err != nil {
+		return err
+	}
+	defer rows.Close()
+	for rows.Next() {
+		var id int
+		var name string
+		rows.Scan(&id, &name)
+		eventmap[name] = id
+	}
+	newEvents := []string{}
+	params := ""
+	paramNum := 1
+	for _, event := range events {
+		e := strings.ToUpper(event)
+		if _, ok := eventmap[event]; ok {
+			continue
+		}
+		newEvents = append(newEvents, e)
+		params += fmt.Sprintf("($%d),", paramNum)
+		paramNum++
+	}
+	if len(newEvents) > 0 {
+		params = params[0 : len(params)-1]
+		values := []interface{}{}
+		for _, event := range newEvents {
+			values = append(values, event)
+		}
+		rows, err = database.Query(
+			fmt.Sprintf(
+				"INSERT INTO nsa.event_types (name) VALUES %s RETURNING id;",
+				params),
+			values...)
+		if err != nil {
+			return err
+		}
+		defer rows.Close()
+		i := 0
+		for rows.Next() {
+			var id int
+			rows.Scan(&id)
+			eventmap[newEvents[i]] = id
+			i++
+		}
+	}
+	return nil
+}
-- 
cgit v0.10.1