[daemon] Use database to check history read permissions

1 files changed, 16 insertions, 6 deletions

diff --git a/storage/storage.go b/storage/storage.go
index 28f2564..510b587 100644
--- a/storage/storage.go
+++ b/storage/storage.go
@@ -109,16 +109,26 @@ func (db *Database) SetPermission(perm core.PermissionData) error {
 }
 
 func (db *Database) GetPermission(user, channel string) (core.PermissionData, error) {
-	var perm core.PermissionData
+	var read, write int
 
 	err := db.QueryRow(
-		"SELECT (read, write) FROM permissions WHERE user = ? AND channel = ?", user, channel).
-		Scan(&perm.Read, &perm.Write)
+		"SELECT read, write FROM permissions WHERE user = ? AND channel = ?", user, channel).
+		Scan(&read, &write)
 	if err != nil {
 		return core.PermissionData{}, err
 	}
 
-	perm.User = user
-	perm.Channel = channel
-	return perm, nil
+	return core.PermissionData{
+		User:    user,
+		Channel: channel,
+		Read:    itob(read),
+		Write:   itob(write),
+	}, nil
+}
+
+func itob(v int) bool {
+	if v == 1 {
+		return true
+	}
+	return false
 }