Init

storage/channel.go

@@ -0,0 +1,68 @@
+package storage
+
+import (
+	"sync"
+)
+
+type ChannelStore struct {
+	data map[string]map[string][]string
+	lock sync.Mutex
+}
+
+func NewChannelStore() *ChannelStore {
+	return &ChannelStore{
+		data: make(map[string]map[string][]string),
+	}
+}
+
+func (c *ChannelStore) GetUsers(server, channel string) []string {
+	c.lock.Lock()
+
+	users := make([]string, len(c.data[server][channel]))
+	copy(users, c.data[server][channel])
+
+	c.lock.Unlock()
+
+	return users
+}
+
+func (c *ChannelStore) SetUsers(users []string, server, channel string) {
+	c.lock.Lock()
+
+	if _, ok := c.data[server]; !ok {
+		c.data[server] = make(map[string][]string)
+	}
+
+	c.data[server][channel] = users
+
+	c.lock.Unlock()
+}
+
+func (c *ChannelStore) AddUser(user, server, channel string) {
+	c.lock.Lock()
+
+	if _, ok := c.data[server]; !ok {
+		c.data[server] = make(map[string][]string)
+	}
+
+	if users, ok := c.data[server][channel]; ok {
+		c.data[server][channel] = append(users, user)
+	} else {
+		c.data[server][channel] = []string{user}
+	}
+
+	c.lock.Unlock()
+}
+
+func (c *ChannelStore) RemoveUser(user, server, channel string) {
+	c.lock.Lock()
+	defer c.lock.Unlock()
+
+	for i, u := range c.data[server][channel] {
+		if u == user {
+			users := c.data[server][channel]
+			c.data[server][channel] = append(users[:i], users[i+1:]...)
+			return
+		}
+	}
+}

storage/init.go

@@ -0,0 +1,22 @@
+package storage
+
+import (
+	"github.com/boltdb/bolt"
+)
+
+var db *bolt.DB
+
+func init() {
+	db, _ = bolt.Open("data.db", 0600, nil)
+
+	db.Update(func(tx *bolt.Tx) error {
+		tx.CreateBucketIfNotExists([]byte("Users"))
+		tx.CreateBucketIfNotExists([]byte("Servers"))
+		tx.CreateBucketIfNotExists([]byte("Channels"))
+		return nil
+	})
+}
+
+func Cleanup() {
+	db.Close()
+}

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
+	})
+}