add vendoring

vendor/gitlab.com/etke.cc/linkpearl/accountdata.go

@@ -0,0 +1,129 @@
+package linkpearl
+
+import (
+	"strings"
+
+	"maunium.net/go/mautrix/id"
+)
+
+// GetAccountData of the user (from cache and API, with encryption support)
+func (l *Linkpearl) GetAccountData(name string) (map[string]string, error) {
+	cached, ok := l.acc.Get(name)
+	if ok {
+		l.log.Debug("GetAccountData(%s) from cache (data): %+v", name, cached)
+		if cached == nil {
+			return map[string]string{}, nil
+		}
+		return cached, nil
+	}
+
+	l.log.Debug("GetAccountData(%s) from API", name)
+	var data map[string]string
+	err := l.GetClient().GetAccountData(name, &data)
+	if err != nil {
+		l.log.Debug("GetAccountData(%s) from API (error): %v", name, err)
+		data = map[string]string{}
+		if strings.Contains(err.Error(), "M_NOT_FOUND") {
+			l.acc.Add(name, data)
+			return data, nil
+		}
+		return data, err
+	}
+	data = l.decryptAccountData(data)
+	l.log.Debug("GetAccountData(%s) from API (data): %+v", name, data)
+
+	l.acc.Add(name, data)
+	return data, err
+}
+
+// SetAccountData of the user (to cache and API, with encryption support)
+func (l *Linkpearl) SetAccountData(name string, data map[string]string) error {
+	l.acc.Add(name, data)
+
+	l.log.Debug("SetAccountData(%s) to API (data): %+v", name, data)
+	data = l.encryptAccountData(data)
+	return l.GetClient().SetAccountData(name, data)
+}
+
+// GetRoomAccountData of the room (from cache and API, with encryption support)
+func (l *Linkpearl) GetRoomAccountData(roomID id.RoomID, name string) (map[string]string, error) {
+	key := roomID.String() + name
+	cached, ok := l.acc.Get(key)
+	if ok {
+		l.log.Debug("GetRoomAccountData(%s, %s) from cache (data): %+v", roomID, name, cached)
+		if cached == nil {
+			return map[string]string{}, nil
+		}
+		return cached, nil
+	}
+
+	l.log.Debug("GetRoomAccountData(%s, %s) from API", roomID, name)
+	var data map[string]string
+	err := l.GetClient().GetRoomAccountData(roomID, name, &data)
+	if err != nil {
+		l.log.Debug("GetRoomAccountData(%s, %s) from API (error): %v", roomID, name, err)
+		data = map[string]string{}
+		if strings.Contains(err.Error(), "M_NOT_FOUND") {
+			l.acc.Add(key, data)
+			return data, nil
+		}
+		return data, err
+	}
+	data = l.decryptAccountData(data)
+	l.log.Debug("GetRoomAccountData(%s,%s) from API (data): %+v", roomID, name, data)
+
+	l.acc.Add(key, data)
+	return data, err
+}
+
+// SetRoomAccountData of the room (to cache and API, with encryption support)
+func (l *Linkpearl) SetRoomAccountData(roomID id.RoomID, name string, data map[string]string) error {
+	key := roomID.String() + name
+	l.acc.Add(key, data)
+
+	l.log.Debug("SetRoomAccountData(%s, %s) to API (data): %+v", roomID, name, data)
+	data = l.encryptAccountData(data)
+	return l.GetClient().SetRoomAccountData(roomID, name, data)
+}
+
+func (l *Linkpearl) encryptAccountData(data map[string]string) map[string]string {
+	if l.acr == nil {
+		return data
+	}
+
+	encrypted := make(map[string]string, len(data))
+	for k, v := range data {
+		ek, err := l.acr.Encrypt(k)
+		if err != nil {
+			l.log.Error("cannot encrypt account data (key=%s): %v", k, err)
+		}
+		ev, err := l.acr.Encrypt(v)
+		if err != nil {
+			l.log.Error("cannot encrypt account data (key=%s): %v", k, err)
+		}
+		encrypted[ek] = ev // worst case: plaintext value
+	}
+
+	return encrypted
+}
+
+func (l *Linkpearl) decryptAccountData(data map[string]string) map[string]string {
+	if l.acr == nil {
+		return data
+	}
+
+	decrypted := make(map[string]string, len(data))
+	for ek, ev := range data {
+		k, err := l.acr.Decrypt(ek)
+		if err != nil {
+			l.log.Error("cannot decrypt account data (key=%s): %v", k, err)
+		}
+		v, err := l.acr.Decrypt(ev)
+		if err != nil {
+			l.log.Error("cannot decrypt account data (key=%s): %v", k, err)
+		}
+		decrypted[k] = v // worst case: encrypted value, usual case: migration from plaintext to encrypted account data
+	}
+
+	return decrypted
+}