ready to switch to peer DB

client/peerstorage.go

@@ -6,7 +6,9 @@ package client
 import (
 	"crypto/sha256"
 	"encoding/json"
+	"errors"
 	"path/filepath"
+	"sort"
 
 	"forge.redroom.link/yves/meowlib"
 	"github.com/dgraph-io/badger"
@@ -18,7 +20,7 @@ type PeerStorage struct {
 	cache  map[string]*Peer
 }
 
-// Open a badger database from struct PeerStorage
+// Open the badger database from struct PeerStorage
 func (ps *PeerStorage) open() error {
 
 	opts := badger.DefaultOptions(filepath.Join(GetConfig().StoragePath, GetConfig().GetIdentity().Uuid, ps.DbFile))
@@ -31,7 +33,7 @@ func (ps *PeerStorage) open() error {
 	return nil
 }
 
-// Store function StorePeer stores a peer in a badger database with Peer.Uid as key
+// Store function StorePeer stores a peer in the badger database with Peer.Uid as key
 func (ps *PeerStorage) StorePeer(peer *Peer) error {
 	err := ps.open()
 	if err != nil {
@@ -60,7 +62,7 @@ func (ps *PeerStorage) StorePeer(peer *Peer) error {
 
 }
 
-// LoadPeer function loads a Peer from a badger database with Peer.GetUid() as key
+// LoadPeer function loads a Peer from the badger database with Peer.GetUid() as key
 func (ps *PeerStorage) LoadPeer(uid string, password string) (*Peer, error) {
 	var peer Peer
 	err := ps.open()
@@ -86,7 +88,7 @@ func (ps *PeerStorage) LoadPeer(uid string, password string) (*Peer, error) {
 	return &peer, err
 }
 
-// DeletePeer function deletes a Peer from a badger database with Peer.GetUid() as key
+// DeletePeer function deletes a Peer from the badger database with Peer.GetUid() as key
 func (ps *PeerStorage) DeletePeer(uid string) error {
 	err := ps.open()
 	if err != nil {
@@ -100,8 +102,8 @@ func (ps *PeerStorage) DeletePeer(uid string) error {
 	})
 }
 
-// LoadAllPeers function loads all Peers from a badger database
-func (ps *PeerStorage) LoadAllPeers(password string) ([]*Peer, error) {
+// LoadPeers function loads Peers from the badger database with a specific password
+func (ps *PeerStorage) LoadPeers(password string) ([]*Peer, error) {
 	var peers []*Peer
 	err := ps.open()
 	if err != nil {
@@ -125,48 +127,104 @@ func (ps *PeerStorage) LoadAllPeers(password string) ([]*Peer, error) {
 			})
 			if err != nil {
 				peers = append(peers, &sc)
+				ps.cache[sc.Uid] = &sc
 			}
 		}
 		return nil
 	})
+	// Sort peers based on peer.Name
+	sort.Slice(peers, func(i, j int) bool {
+		return peers[i].Name < peers[j].Name
+	})
 	return peers, err
 }
 
-// LoadPeersFromUids function loads Peers with id in []Uid parameter from a badger database
-func (ps *PeerStorage) LoadPeersFromUids(uids []string, password string) ([]*Peer, error) {
-	var peers []*Peer
-	err := ps.open()
-	if err != nil {
-		return nil, err
+// GetPeers function returns all peers from the cache as a sorted array
+func (ps *PeerStorage) GetPeers() ([]*Peer, error) {
+	peers := make([]*Peer, 0, len(ps.cache))
+	for _, peer := range ps.cache {
+		peers = append(peers, peer)
 	}
-	defer ps.close()
-	err = ps.db.View(func(txn *badger.Txn) error {
-		for _, uid := range uids {
-			shakey := sha256.Sum256([]byte(uid))
-			key := shakey[:]
-			item, err := txn.Get(key)
-			if err != nil {
-				return err
-			}
-			var sc Peer
-			err = item.Value(func(val []byte) error {
-				jsonsrv, err := meowlib.SymDecrypt(password, val)
-				if err != nil {
-					return err
-				}
-				return json.Unmarshal(jsonsrv, &sc)
-			})
-			if err == nil {
-				peers = append(peers, &sc)
-			}
-
-		}
-		return nil
+	// Sort peers based on peer.Name
+	sort.Slice(peers, func(i, j int) bool {
+		return peers[i].Name < peers[j].Name
 	})
-	return peers, err
+	return peers, nil
 }
 
-// close a badger database
+// close the badger database
 func (ps *PeerStorage) close() {
 	ps.db.Close()
 }
+
+func (ps *PeerStorage) GetFromPublicKey(publickey string) *Peer {
+	for _, peer := range ps.cache {
+		if peer.ContactPublicKey == publickey {
+			return peer
+		}
+	}
+	return nil
+}
+
+func (ps *PeerStorage) GetFromInvitationId(invitationId string) *Peer {
+	for _, peer := range ps.cache {
+		if peer.InvitationId == invitationId {
+			return peer
+		}
+	}
+	return nil
+}
+
+func (ps *PeerStorage) GetFromMyLookupKey(publickey string) *Peer {
+	for _, peer := range ps.cache {
+		if peer.MyLookupKp.Public == publickey {
+			return peer
+		}
+	}
+	return nil
+}
+
+func (ps *PeerStorage) GetFromName(name string) *Peer {
+	for _, peer := range ps.cache {
+		if peer.Name == name {
+			return peer
+		}
+	}
+	return nil
+}
+
+func (ps *PeerStorage) GetFromUid(uid string) *Peer {
+	return ps.cache[uid]
+}
+
+// Checks if the received contact card is an answer to an invitation, returns true if it is, and the proposed and received nicknames
+func (ps *PeerStorage) CheckInvitation(ReceivedContact *meowlib.ContactCard) (isAnswer bool, proposedNick string, receivedNick string, invitationMessage string) {
+	// invitation Id found, this is an answer to an invitation
+	for _, p := range ps.cache {
+		if p.InvitationId == ReceivedContact.InvitationId {
+			return true, p.Name, ReceivedContact.Name, ReceivedContact.InvitationMessage
+		}
+	}
+	// it's an invitation
+	return false, "", ReceivedContact.Name, ReceivedContact.InvitationMessage
+}
+
+// Finalizes an invitation, returns nil if successful
+func (ps *PeerStorage) FinalizeInvitation(ReceivedContact *meowlib.ContactCard) error {
+	for i, p := range ps.cache {
+		if p.InvitationId == ReceivedContact.InvitationId {
+			//id.Peers[i].Name = ReceivedContact.Name
+			ps.cache[i].ContactEncryption = ReceivedContact.EncryptionPublicKey
+			ps.cache[i].ContactLookupKey = ReceivedContact.LookupPublicKey
+			ps.cache[i].ContactPublicKey = ReceivedContact.ContactPublicKey
+			srvs := []string{}
+			for srv := range ReceivedContact.PullServers {
+				srvs = append(srvs, ReceivedContact.PullServers[srv].GetUid())
+			}
+			ps.cache[i].ContactPullServers = srvs
+			ps.StorePeer(ps.cache[i])
+			return nil
+		}
+	}
+	return errors.New("no matching contact found for invitationId " + ReceivedContact.InvitationId)
+}