Add DB objects

admin/key.go

@@ -0,0 +1,69 @@
+package main
+
+import (
+	"crypto/sha512"
+)
+
+type Key struct {
+	id          int64
+	id_exercice int64
+	Type        string
+	Value       [64]byte
+}
+
+func (e Exercice) GetKeys() ([]Key, error) {
+	if rows, err := DBQuery("SELECT id_key, type, value FROM exercice_keys WHERE id_exercice = ?", e.id); err != nil {
+		return nil, err
+	} else {
+		defer rows.Close()
+
+		var keys = make([]Key, 0)
+		for rows.Next() {
+			var k Key
+			k.id_exercice = e.id
+			if err := rows.Scan(&k.id, &k.id_exercice, &k.Type, &k.Value); err != nil {
+				return nil, err
+			}
+			keys = append(keys, k)
+		}
+		if err := rows.Err(); err != nil {
+			return nil, err
+		}
+
+		return keys, nil
+	}
+}
+
+func (e Exercice) AddRawKey(name string, raw_value string) (Key, error) {
+	return e.AddKey(name, sha512.Sum512([]byte(raw_value)))
+}
+
+func (e Exercice) AddKey(name string, value [64]byte) (Key, error) {
+	if res, err := DBExec("INSERT INTO exercice_keys (id_exercice, type, value) VALUES (?, ?, ?)", e.id, name, value); err != nil {
+		return Key{}, err
+	} else if kid, err := res.LastInsertId(); err != nil {
+		return Key{}, err
+	} else {
+		return Key{kid, e.id, name, value}, nil
+	}
+}
+
+func (k Key) Update() (int64, error) {
+	if res, err := DBExec("UPDATE exercice_keys SET id_exercice = ?, type = ?, value = ? WHERE id_key = ?", k.id_exercice, k.Type, k.Value, k.id); err != nil {
+		return 0, err
+	} else if nb, err := res.RowsAffected(); err != nil {
+		return 0, err
+	} else {
+		return nb, err
+	}
+}
+
+func (k Key) Delete() (int64, error) {
+	if res, err := DBExec("DELETE FROM exercice_keys WHERE id_key = ?", k.id); err != nil {
+		return 0, err
+	} else if nb, err := res.RowsAffected(); err != nil {
+		return 0, err
+	} else {
+		return nb, err
+	}
+}