Init

storage/user.go

@@ -0,0 +1,137 @@
+package storage
+
+import (
+	"bytes"
+	"encoding/json"
+
+	"github.com/boltdb/bolt"
+)
+
+type Server struct {
+	Address  string
+	Nick     string
+	Username string
+	Realname string
+}
+
+type Channel struct {
+	Server string   `json:"server"`
+	Name   string   `json:"name"`
+	Users  []string `json:"users"`
+}
+
+type User struct {
+	UUID string
+}
+
+func NewUser(uuid string) User {
+	user := User{
+		UUID: uuid,
+	}
+
+	go db.Update(func(tx *bolt.Tx) error {
+		b := tx.Bucket([]byte("Users"))
+		data, _ := json.Marshal(user)
+
+		b.Put([]byte(uuid), data)
+
+		return nil
+	})
+
+	return user
+}
+
+func LoadUsers() []User {
+	var users []User
+
+	db.View(func(tx *bolt.Tx) error {
+		b := tx.Bucket([]byte("Users"))
+
+		b.ForEach(func(k, v []byte) error {
+			users = append(users, User{string(k)})
+
+			return nil
+		})
+
+		return nil
+	})
+
+	return users
+}
+
+func (u User) GetServers() []Server {
+	var servers []Server
+
+	db.View(func(tx *bolt.Tx) error {
+		c := tx.Bucket([]byte("Servers")).Cursor()
+		prefix := []byte(u.UUID)
+
+		for k, v := c.Seek(prefix); bytes.HasPrefix(k, prefix); k, v = c.Next() {
+			var server Server
+			json.Unmarshal(v, &server)
+			servers = append(servers, server)
+		}
+
+		return nil
+	})
+
+	return servers
+}
+
+func (u User) GetChannels() []Channel {
+	var channels []Channel
+
+	db.View(func(tx *bolt.Tx) error {
+		c := tx.Bucket([]byte("Channels")).Cursor()
+
+		prefix := []byte(u.UUID)
+
+		for k, v := c.Seek(prefix); bytes.HasPrefix(k, prefix); k, v = c.Next() {
+			var channel Channel
+			json.Unmarshal(v, &channel)
+			channels = append(channels, channel)
+		}
+
+		return nil
+	})
+
+	return channels
+}
+
+func (u User) AddServer(server Server) {
+	db.Update(func(tx *bolt.Tx) error {
+		b := tx.Bucket([]byte("Servers"))
+		data, _ := json.Marshal(server)
+
+		b.Put([]byte(u.UUID+":"+server.Address), data)
+
+		return nil
+	})
+}
+
+func (u User) AddChannel(channel Channel) {
+	db.Update(func(tx *bolt.Tx) error {
+		b := tx.Bucket([]byte("Channels"))
+		data, _ := json.Marshal(channel)
+
+		b.Put([]byte(u.UUID+":"+channel.Server+":"+channel.Name), data)
+
+		return nil
+	})
+}
+
+func (u User) RemoveServer(address string) {
+	db.Update(func(tx *bolt.Tx) error {
+		tx.Bucket([]byte("Channels")).Delete([]byte(u.UUID + ":" + address))
+
+		return nil
+	})
+}
+
+func (u User) RemoveChannel(server, channel string) {
+	db.Update(func(tx *bolt.Tx) error {
+		tx.Bucket([]byte("Servers")).Delete([]byte(u.UUID + ":" + server + ":" + channel))
+
+		return nil
+	})
+}