Store auth info in a JWT token in a cookie
This commit is contained in:
Ken-Håvard Lieng
parent
3e0a1be6bc
commit
fb54d4966c
18 changed files with 499 additions and 331 deletions
|
|
@ -21,32 +21,28 @@ func (d directory) LetsEncrypt() string {
|
|||
return filepath.Join(d.Root(), "letsencrypt")
|
||||
}
|
||||
|
||||
func (d directory) Logs() string {
|
||||
return filepath.Join(d.Root(), "logs")
|
||||
}
|
||||
|
||||
func (d directory) Log(userID string) string {
|
||||
return filepath.Join(d.Logs(), userID+".log")
|
||||
}
|
||||
|
||||
func (d directory) Index(userID string) string {
|
||||
return filepath.Join(d.Logs(), userID+".idx")
|
||||
}
|
||||
|
||||
func (d directory) Users() string {
|
||||
return filepath.Join(d.Root(), "users")
|
||||
}
|
||||
|
||||
func (d directory) User(userID string) string {
|
||||
return filepath.Join(d.Users(), userID)
|
||||
func (d directory) User(username string) string {
|
||||
return filepath.Join(d.Users(), username)
|
||||
}
|
||||
|
||||
func (d directory) Certificate(userID string) string {
|
||||
return filepath.Join(d.User(userID), "cert.pem")
|
||||
func (d directory) Log(username string) string {
|
||||
return filepath.Join(d.User(username), "log")
|
||||
}
|
||||
|
||||
func (d directory) Key(userID string) string {
|
||||
return filepath.Join(d.User(userID), "key.pem")
|
||||
func (d directory) Index(username string) string {
|
||||
return filepath.Join(d.User(username), "index")
|
||||
}
|
||||
|
||||
func (d directory) Certificate(username string) string {
|
||||
return filepath.Join(d.User(username), "cert.pem")
|
||||
}
|
||||
|
||||
func (d directory) Key(username string) string {
|
||||
return filepath.Join(d.User(username), "key.pem")
|
||||
}
|
||||
|
||||
func (d directory) Config() string {
|
||||
|
|
@ -56,3 +52,7 @@ func (d directory) Config() string {
|
|||
func (d directory) Database() string {
|
||||
return filepath.Join(d.Root(), "dispatch.db")
|
||||
}
|
||||
|
||||
func (d directory) HMACKey() string {
|
||||
return filepath.Join(d.Root(), "hmac.key")
|
||||
}
|
||||
|
|
|
|||
|
|
@ -1,8 +1,8 @@
|
|||
package storage
|
||||
|
||||
import (
|
||||
"encoding/binary"
|
||||
"log"
|
||||
"os"
|
||||
|
||||
"github.com/khlieng/dispatch/Godeps/_workspace/src/github.com/boltdb/bolt"
|
||||
)
|
||||
|
|
@ -20,11 +20,6 @@ var (
|
|||
|
||||
func Initialize(dir string) {
|
||||
Path = directory(dir)
|
||||
|
||||
err := os.MkdirAll(Path.Logs(), 0700)
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
func Open() {
|
||||
|
|
@ -46,3 +41,13 @@ func Open() {
|
|||
func Close() {
|
||||
db.Close()
|
||||
}
|
||||
|
||||
func idToBytes(i uint64) []byte {
|
||||
b := make([]byte, 8)
|
||||
binary.BigEndian.PutUint64(b, i)
|
||||
return b
|
||||
}
|
||||
|
||||
func idFromBytes(b []byte) uint64 {
|
||||
return binary.BigEndian.Uint64(b)
|
||||
}
|
||||
|
|
|
|||
274
storage/user.go
274
storage/user.go
|
|
@ -6,14 +6,23 @@ import (
|
|||
"encoding/json"
|
||||
"log"
|
||||
"strconv"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/khlieng/dispatch/Godeps/_workspace/src/github.com/blevesearch/bleve"
|
||||
"github.com/khlieng/dispatch/Godeps/_workspace/src/github.com/boltdb/bolt"
|
||||
)
|
||||
|
||||
type User struct {
|
||||
ID uint64
|
||||
Username string
|
||||
|
||||
id []byte
|
||||
messageLog *bolt.DB
|
||||
messageIndex bleve.Index
|
||||
certificate *tls.Certificate
|
||||
lock sync.Mutex
|
||||
}
|
||||
|
||||
type Server struct {
|
||||
Name string `json:"name"`
|
||||
Host string `json:"host"`
|
||||
|
|
@ -32,39 +41,37 @@ type Channel struct {
|
|||
Topic string `json:"topic,omitempty"`
|
||||
}
|
||||
|
||||
type Message struct {
|
||||
ID uint64 `json:"id"`
|
||||
Server string `json:"server"`
|
||||
From string `json:"from"`
|
||||
To string `json:"to"`
|
||||
Content string `json:"content"`
|
||||
Time int64 `json:"time"`
|
||||
}
|
||||
func NewUser() *User {
|
||||
user := &User{}
|
||||
|
||||
type User struct {
|
||||
UUID string
|
||||
err := db.Update(func(tx *bolt.Tx) error {
|
||||
b := tx.Bucket(bucketUsers)
|
||||
|
||||
messageLog *bolt.DB
|
||||
messageIndex bleve.Index
|
||||
certificate *tls.Certificate
|
||||
lock sync.Mutex
|
||||
}
|
||||
var err error
|
||||
user.ID, err = b.NextSequence()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
func NewUser(uuid string) *User {
|
||||
user := &User{
|
||||
UUID: uuid,
|
||||
}
|
||||
user.Username = strconv.FormatUint(user.ID, 10)
|
||||
user.id = idToBytes(user.ID)
|
||||
|
||||
db.Update(func(tx *bolt.Tx) error {
|
||||
b := tx.Bucket([]byte("Users"))
|
||||
data, _ := json.Marshal(user)
|
||||
data, err := json.Marshal(user)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
b.Put([]byte(uuid), data)
|
||||
|
||||
return nil
|
||||
return b.Put(user.id, data)
|
||||
})
|
||||
|
||||
user.openMessageLog()
|
||||
if err != nil {
|
||||
return nil
|
||||
}
|
||||
|
||||
err = user.openMessageLog()
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
|
||||
return user
|
||||
}
|
||||
|
|
@ -73,14 +80,18 @@ func LoadUsers() []*User {
|
|||
var users []*User
|
||||
|
||||
db.View(func(tx *bolt.Tx) error {
|
||||
b := tx.Bucket([]byte("Users"))
|
||||
b := tx.Bucket(bucketUsers)
|
||||
|
||||
b.ForEach(func(k, v []byte) error {
|
||||
user := User{UUID: string(k)}
|
||||
user.openMessageLog()
|
||||
user.loadCertificate()
|
||||
b.ForEach(func(k, _ []byte) error {
|
||||
id := idFromBytes(k)
|
||||
user := &User{
|
||||
ID: id,
|
||||
Username: strconv.FormatUint(id, 10),
|
||||
id: make([]byte, 8),
|
||||
}
|
||||
copy(user.id, k)
|
||||
|
||||
users = append(users, &user)
|
||||
users = append(users, user)
|
||||
|
||||
return nil
|
||||
})
|
||||
|
|
@ -88,6 +99,11 @@ func LoadUsers() []*User {
|
|||
return nil
|
||||
})
|
||||
|
||||
for _, user := range users {
|
||||
user.openMessageLog()
|
||||
user.loadCertificate()
|
||||
}
|
||||
|
||||
return users
|
||||
}
|
||||
|
||||
|
|
@ -95,10 +111,9 @@ 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)
|
||||
c := tx.Bucket(bucketServers).Cursor()
|
||||
|
||||
for k, v := c.Seek(prefix); bytes.HasPrefix(k, prefix); k, v = c.Next() {
|
||||
for k, v := c.Seek(u.id); bytes.HasPrefix(k, u.id); k, v = c.Next() {
|
||||
var server Server
|
||||
json.Unmarshal(v, &server)
|
||||
servers = append(servers, server)
|
||||
|
|
@ -114,11 +129,9 @@ func (u *User) GetChannels() []Channel {
|
|||
var channels []Channel
|
||||
|
||||
db.View(func(tx *bolt.Tx) error {
|
||||
c := tx.Bucket([]byte("Channels")).Cursor()
|
||||
c := tx.Bucket(bucketChannels).Cursor()
|
||||
|
||||
prefix := []byte(u.UUID)
|
||||
|
||||
for k, v := c.Seek(prefix); bytes.HasPrefix(k, prefix); k, v = c.Next() {
|
||||
for k, v := c.Seek(u.id); bytes.HasPrefix(k, u.id); k, v = c.Next() {
|
||||
var channel Channel
|
||||
json.Unmarshal(v, &channel)
|
||||
channels = append(channels, channel)
|
||||
|
|
@ -132,10 +145,10 @@ func (u *User) GetChannels() []Channel {
|
|||
|
||||
func (u *User) AddServer(server Server) {
|
||||
db.Update(func(tx *bolt.Tx) error {
|
||||
b := tx.Bucket([]byte("Servers"))
|
||||
b := tx.Bucket(bucketServers)
|
||||
data, _ := json.Marshal(server)
|
||||
|
||||
b.Put([]byte(u.UUID+":"+server.Host), data)
|
||||
b.Put(u.serverID(server.Host), data)
|
||||
|
||||
return nil
|
||||
})
|
||||
|
|
@ -143,10 +156,10 @@ func (u *User) AddServer(server Server) {
|
|||
|
||||
func (u *User) AddChannel(channel Channel) {
|
||||
db.Update(func(tx *bolt.Tx) error {
|
||||
b := tx.Bucket([]byte("Channels"))
|
||||
b := tx.Bucket(bucketChannels)
|
||||
data, _ := json.Marshal(channel)
|
||||
|
||||
b.Put([]byte(u.UUID+":"+channel.Server+":"+channel.Name), data)
|
||||
b.Put(u.channelID(channel.Server, channel.Name), data)
|
||||
|
||||
return nil
|
||||
})
|
||||
|
|
@ -154,8 +167,8 @@ func (u *User) AddChannel(channel Channel) {
|
|||
|
||||
func (u *User) SetNick(nick, address string) {
|
||||
db.Update(func(tx *bolt.Tx) error {
|
||||
b := tx.Bucket([]byte("Servers"))
|
||||
id := []byte(u.UUID + ":" + address)
|
||||
b := tx.Bucket(bucketServers)
|
||||
id := u.serverID(address)
|
||||
var server Server
|
||||
|
||||
json.Unmarshal(b.Get(id), &server)
|
||||
|
|
@ -170,11 +183,10 @@ func (u *User) SetNick(nick, address string) {
|
|||
|
||||
func (u *User) RemoveServer(address string) {
|
||||
db.Update(func(tx *bolt.Tx) error {
|
||||
serverID := []byte(u.UUID + ":" + address)
|
||||
serverID := u.serverID(address)
|
||||
tx.Bucket(bucketServers).Delete(serverID)
|
||||
|
||||
tx.Bucket([]byte("Servers")).Delete(serverID)
|
||||
|
||||
b := tx.Bucket([]byte("Channels"))
|
||||
b := tx.Bucket(bucketChannels)
|
||||
c := b.Cursor()
|
||||
|
||||
for k, _ := c.Seek(serverID); bytes.HasPrefix(k, serverID); k, _ = c.Next() {
|
||||
|
|
@ -187,157 +199,31 @@ func (u *User) RemoveServer(address string) {
|
|||
|
||||
func (u *User) RemoveChannel(server, channel string) {
|
||||
db.Update(func(tx *bolt.Tx) error {
|
||||
tx.Bucket([]byte("Channels")).Delete([]byte(u.UUID + ":" + server + ":" + channel))
|
||||
b := tx.Bucket(bucketChannels)
|
||||
id := u.channelID(server, channel)
|
||||
|
||||
b.Delete(id)
|
||||
|
||||
return nil
|
||||
})
|
||||
}
|
||||
|
||||
func (u *User) LogMessage(server, from, to, content string) {
|
||||
bucketKey := server + ":" + to
|
||||
var id uint64
|
||||
var idStr string
|
||||
var message Message
|
||||
|
||||
u.messageLog.Update(func(tx *bolt.Tx) error {
|
||||
b, _ := tx.Bucket(bucketMessages).CreateBucketIfNotExists([]byte(bucketKey))
|
||||
id, _ = b.NextSequence()
|
||||
idStr = strconv.FormatUint(id, 10)
|
||||
|
||||
message = Message{
|
||||
ID: id,
|
||||
Content: content,
|
||||
Server: server,
|
||||
From: from,
|
||||
To: to,
|
||||
Time: time.Now().Unix(),
|
||||
}
|
||||
|
||||
data, _ := json.Marshal(message)
|
||||
b.Put([]byte(idStr), data)
|
||||
|
||||
return nil
|
||||
})
|
||||
|
||||
u.messageIndex.Index(bucketKey+":"+idStr, message)
|
||||
}
|
||||
|
||||
func (u *User) GetLastMessages(server, channel string, count int) ([]Message, error) {
|
||||
messages := make([]Message, count)
|
||||
|
||||
u.messageLog.View(func(tx *bolt.Tx) error {
|
||||
b := tx.Bucket(bucketMessages).Bucket([]byte(server + ":" + channel))
|
||||
if b == nil {
|
||||
return nil
|
||||
}
|
||||
|
||||
c := b.Cursor()
|
||||
|
||||
for k, v := c.Last(); count > 0 && k != nil; k, v = c.Prev() {
|
||||
count--
|
||||
json.Unmarshal(v, &messages[count])
|
||||
}
|
||||
|
||||
return nil
|
||||
})
|
||||
|
||||
if count < len(messages) {
|
||||
return messages[count:], nil
|
||||
} else {
|
||||
return nil, nil
|
||||
}
|
||||
}
|
||||
|
||||
func (u *User) GetMessages(server, channel string, count int, fromID uint64) ([]Message, error) {
|
||||
messages := make([]Message, count)
|
||||
|
||||
u.messageLog.View(func(tx *bolt.Tx) error {
|
||||
b := tx.Bucket(bucketMessages).Bucket([]byte(server + ":" + channel))
|
||||
if b == nil {
|
||||
return nil
|
||||
}
|
||||
|
||||
c := b.Cursor()
|
||||
c.Seek([]byte(strconv.FormatUint(fromID, 10)))
|
||||
|
||||
for k, v := c.Prev(); count > 0 && k != nil; k, v = c.Prev() {
|
||||
count--
|
||||
json.Unmarshal(v, &messages[count])
|
||||
}
|
||||
|
||||
return nil
|
||||
})
|
||||
|
||||
if count < len(messages) {
|
||||
return messages[count:], nil
|
||||
} else {
|
||||
return nil, nil
|
||||
}
|
||||
}
|
||||
|
||||
func (u *User) SearchMessages(server, channel, phrase string) ([]Message, error) {
|
||||
serverQuery := bleve.NewMatchQuery(server)
|
||||
serverQuery.SetField("server")
|
||||
channelQuery := bleve.NewMatchQuery(channel)
|
||||
channelQuery.SetField("to")
|
||||
contentQuery := bleve.NewMatchQuery(phrase)
|
||||
contentQuery.SetField("content")
|
||||
|
||||
query := bleve.NewBooleanQuery([]bleve.Query{serverQuery, channelQuery, contentQuery}, nil, nil)
|
||||
|
||||
search := bleve.NewSearchRequest(query)
|
||||
searchResults, err := u.messageIndex.Search(search)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
messages := []Message{}
|
||||
u.messageLog.View(func(tx *bolt.Tx) error {
|
||||
b := tx.Bucket(bucketMessages)
|
||||
|
||||
for _, hit := range searchResults.Hits {
|
||||
idx := strings.LastIndex(hit.ID, ":")
|
||||
bc := b.Bucket([]byte(hit.ID[:idx]))
|
||||
var message Message
|
||||
|
||||
json.Unmarshal(bc.Get([]byte(hit.ID[idx+1:])), &message)
|
||||
messages = append(messages, message)
|
||||
}
|
||||
|
||||
return nil
|
||||
})
|
||||
|
||||
return messages, nil
|
||||
}
|
||||
|
||||
func (u *User) Close() {
|
||||
u.messageLog.Close()
|
||||
u.messageIndex.Close()
|
||||
}
|
||||
|
||||
func (u *User) openMessageLog() {
|
||||
var err error
|
||||
|
||||
u.messageLog, err = bolt.Open(Path.Log(u.UUID), 0600, nil)
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
|
||||
u.messageLog.Update(func(tx *bolt.Tx) error {
|
||||
tx.CreateBucketIfNotExists(bucketMessages)
|
||||
|
||||
return nil
|
||||
})
|
||||
|
||||
indexPath := Path.Index(u.UUID)
|
||||
u.messageIndex, err = bleve.Open(indexPath)
|
||||
if err == bleve.ErrorIndexPathDoesNotExist {
|
||||
mapping := bleve.NewIndexMapping()
|
||||
u.messageIndex, err = bleve.New(indexPath, mapping)
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
} else if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
func (u *User) serverID(address string) []byte {
|
||||
id := make([]byte, 8+len(address))
|
||||
copy(id, u.id)
|
||||
copy(id[8:], address)
|
||||
return id
|
||||
}
|
||||
|
||||
func (u *User) channelID(server, channel string) []byte {
|
||||
id := make([]byte, 8+len(server)+1+len(channel))
|
||||
copy(id, u.id)
|
||||
copy(id[8:], server)
|
||||
copy(id[8+len(server)+1:], channel)
|
||||
return id
|
||||
}
|
||||
|
|
|
|||
|
|
@ -4,7 +4,6 @@ import (
|
|||
"crypto/tls"
|
||||
"errors"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
)
|
||||
|
||||
var (
|
||||
|
|
@ -29,17 +28,12 @@ func (u *User) SetCertificate(certPEM, keyPEM []byte) error {
|
|||
u.certificate = &cert
|
||||
u.lock.Unlock()
|
||||
|
||||
err = os.MkdirAll(Path.User(u.UUID), 0700)
|
||||
err = ioutil.WriteFile(Path.Certificate(u.Username), certPEM, 0600)
|
||||
if err != nil {
|
||||
return ErrCouldNotSaveCert
|
||||
}
|
||||
|
||||
err = ioutil.WriteFile(Path.Certificate(u.UUID), certPEM, 0600)
|
||||
if err != nil {
|
||||
return ErrCouldNotSaveCert
|
||||
}
|
||||
|
||||
err = ioutil.WriteFile(Path.Key(u.UUID), keyPEM, 0600)
|
||||
err = ioutil.WriteFile(Path.Key(u.Username), keyPEM, 0600)
|
||||
if err != nil {
|
||||
return ErrCouldNotSaveCert
|
||||
}
|
||||
|
|
@ -48,12 +42,12 @@ func (u *User) SetCertificate(certPEM, keyPEM []byte) error {
|
|||
}
|
||||
|
||||
func (u *User) loadCertificate() error {
|
||||
certPEM, err := ioutil.ReadFile(Path.Certificate(u.UUID))
|
||||
certPEM, err := ioutil.ReadFile(Path.Certificate(u.Username))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
keyPEM, err := ioutil.ReadFile(Path.Key(u.UUID))
|
||||
keyPEM, err := ioutil.ReadFile(Path.Key(u.Username))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
|
|
|||
170
storage/user_messages.go
Normal file
170
storage/user_messages.go
Normal file
|
|
@ -0,0 +1,170 @@
|
|||
package storage
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"os"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/khlieng/dispatch/Godeps/_workspace/src/github.com/blevesearch/bleve"
|
||||
"github.com/khlieng/dispatch/Godeps/_workspace/src/github.com/boltdb/bolt"
|
||||
)
|
||||
|
||||
type Message struct {
|
||||
ID uint64 `json:"id"`
|
||||
Server string `json:"server"`
|
||||
From string `json:"from"`
|
||||
To string `json:"to"`
|
||||
Content string `json:"content"`
|
||||
Time int64 `json:"time"`
|
||||
}
|
||||
|
||||
func (u *User) LogMessage(server, from, to, content string) {
|
||||
bucketKey := server + ":" + to
|
||||
var id uint64
|
||||
var idStr string
|
||||
var message Message
|
||||
|
||||
u.messageLog.Update(func(tx *bolt.Tx) error {
|
||||
b, _ := tx.Bucket(bucketMessages).CreateBucketIfNotExists([]byte(bucketKey))
|
||||
id, _ = b.NextSequence()
|
||||
idStr = strconv.FormatUint(id, 10)
|
||||
|
||||
message = Message{
|
||||
ID: id,
|
||||
Content: content,
|
||||
Server: server,
|
||||
From: from,
|
||||
To: to,
|
||||
Time: time.Now().Unix(),
|
||||
}
|
||||
|
||||
data, _ := json.Marshal(message)
|
||||
b.Put([]byte(idStr), data)
|
||||
|
||||
return nil
|
||||
})
|
||||
|
||||
u.messageIndex.Index(bucketKey+":"+idStr, message)
|
||||
}
|
||||
|
||||
func (u *User) GetLastMessages(server, channel string, count int) ([]Message, error) {
|
||||
messages := make([]Message, count)
|
||||
|
||||
u.messageLog.View(func(tx *bolt.Tx) error {
|
||||
b := tx.Bucket(bucketMessages).Bucket([]byte(server + ":" + channel))
|
||||
if b == nil {
|
||||
return nil
|
||||
}
|
||||
|
||||
c := b.Cursor()
|
||||
|
||||
for k, v := c.Last(); count > 0 && k != nil; k, v = c.Prev() {
|
||||
count--
|
||||
json.Unmarshal(v, &messages[count])
|
||||
}
|
||||
|
||||
return nil
|
||||
})
|
||||
|
||||
if count < len(messages) {
|
||||
return messages[count:], nil
|
||||
} else {
|
||||
return nil, nil
|
||||
}
|
||||
}
|
||||
|
||||
func (u *User) GetMessages(server, channel string, count int, fromID uint64) ([]Message, error) {
|
||||
messages := make([]Message, count)
|
||||
|
||||
u.messageLog.View(func(tx *bolt.Tx) error {
|
||||
b := tx.Bucket(bucketMessages).Bucket([]byte(server + ":" + channel))
|
||||
if b == nil {
|
||||
return nil
|
||||
}
|
||||
|
||||
c := b.Cursor()
|
||||
c.Seek([]byte(strconv.FormatUint(fromID, 10)))
|
||||
|
||||
for k, v := c.Prev(); count > 0 && k != nil; k, v = c.Prev() {
|
||||
count--
|
||||
json.Unmarshal(v, &messages[count])
|
||||
}
|
||||
|
||||
return nil
|
||||
})
|
||||
|
||||
if count < len(messages) {
|
||||
return messages[count:], nil
|
||||
}
|
||||
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
func (u *User) SearchMessages(server, channel, phrase string) ([]Message, error) {
|
||||
serverQuery := bleve.NewMatchQuery(server)
|
||||
serverQuery.SetField("server")
|
||||
channelQuery := bleve.NewMatchQuery(channel)
|
||||
channelQuery.SetField("to")
|
||||
contentQuery := bleve.NewMatchQuery(phrase)
|
||||
contentQuery.SetField("content")
|
||||
|
||||
query := bleve.NewBooleanQuery([]bleve.Query{serverQuery, channelQuery, contentQuery}, nil, nil)
|
||||
|
||||
search := bleve.NewSearchRequest(query)
|
||||
searchResults, err := u.messageIndex.Search(search)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
messages := []Message{}
|
||||
u.messageLog.View(func(tx *bolt.Tx) error {
|
||||
b := tx.Bucket(bucketMessages)
|
||||
|
||||
for _, hit := range searchResults.Hits {
|
||||
idx := strings.LastIndex(hit.ID, ":")
|
||||
bc := b.Bucket([]byte(hit.ID[:idx]))
|
||||
var message Message
|
||||
|
||||
json.Unmarshal(bc.Get([]byte(hit.ID[idx+1:])), &message)
|
||||
messages = append(messages, message)
|
||||
}
|
||||
|
||||
return nil
|
||||
})
|
||||
|
||||
return messages, nil
|
||||
}
|
||||
|
||||
func (u *User) openMessageLog() error {
|
||||
err := os.MkdirAll(Path.User(u.Username), 0700)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
u.messageLog, err = bolt.Open(Path.Log(u.Username), 0600, nil)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
u.messageLog.Update(func(tx *bolt.Tx) error {
|
||||
tx.CreateBucketIfNotExists(bucketMessages)
|
||||
|
||||
return nil
|
||||
})
|
||||
|
||||
indexPath := Path.Index(u.Username)
|
||||
u.messageIndex, err = bleve.Open(indexPath)
|
||||
if err == bleve.ErrorIndexPathDoesNotExist {
|
||||
mapping := bleve.NewIndexMapping()
|
||||
u.messageIndex, err = bleve.New(indexPath, mapping)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
} else if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
|
@ -13,6 +13,11 @@ func tempdir() string {
|
|||
}
|
||||
|
||||
func TestUser(t *testing.T) {
|
||||
defer func() {
|
||||
r := recover()
|
||||
assert.Nil(t, r)
|
||||
}()
|
||||
|
||||
Initialize(tempdir())
|
||||
Open()
|
||||
|
||||
|
|
@ -30,7 +35,7 @@ func TestUser(t *testing.T) {
|
|||
Name: "#testing",
|
||||
}
|
||||
|
||||
user := NewUser("unique")
|
||||
user := NewUser()
|
||||
user.AddServer(srv)
|
||||
user.AddChannel(chan1)
|
||||
user.AddChannel(chan2)
|
||||
|
|
@ -40,7 +45,7 @@ func TestUser(t *testing.T) {
|
|||
assert.Len(t, users, 1)
|
||||
|
||||
user = users[0]
|
||||
assert.Equal(t, "unique", user.UUID)
|
||||
assert.Equal(t, uint64(1), user.ID)
|
||||
|
||||
servers := user.GetServers()
|
||||
assert.Len(t, servers, 1)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue