mirror of
https://github.com/AvengeMedia/DankMaterialShell.git
synced 2026-06-13 14:36:32 -04:00
Fix/clipboard pinned recents dedupe (#2605)
* fix(clipboard): unpin pinned duplicates from history entries * fix(clipboard): dedupe recents when using pinned entries
This commit is contained in:
@@ -935,7 +935,7 @@ func (m *Manager) CreateHistoryEntryFromPinned(pinnedEntry *Entry) error {
|
||||
Pinned: false,
|
||||
}
|
||||
|
||||
if err := m.storeEntryWithoutDedup(newEntry); err != nil {
|
||||
if err := m.storeEntry(newEntry); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
@@ -945,36 +945,6 @@ func (m *Manager) CreateHistoryEntryFromPinned(pinnedEntry *Entry) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *Manager) storeEntryWithoutDedup(entry Entry) error {
|
||||
if m.db == nil {
|
||||
return fmt.Errorf("database not available")
|
||||
}
|
||||
|
||||
entry.Hash = computeHash(entry.Data)
|
||||
|
||||
return m.db.Update(func(tx *bolt.Tx) error {
|
||||
b := tx.Bucket([]byte("clipboard"))
|
||||
|
||||
id, err := b.NextSequence()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
entry.ID = id
|
||||
|
||||
encoded, err := encodeEntry(entry)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if err := b.Put(itob(id), encoded); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return m.trimLengthInTx(b)
|
||||
})
|
||||
}
|
||||
|
||||
func (m *Manager) ClearHistory() {
|
||||
if m.db == nil {
|
||||
return
|
||||
@@ -1653,6 +1623,37 @@ func (m *Manager) UnpinEntry(id uint64) error {
|
||||
return err
|
||||
}
|
||||
|
||||
if entry.Pinned {
|
||||
currentKey := itob(id)
|
||||
var keepKey []byte
|
||||
var deleteKeys [][]byte
|
||||
|
||||
c := b.Cursor()
|
||||
for k, v := c.Last(); k != nil; k, v = c.Prev() {
|
||||
if bytes.Equal(k, currentKey) || extractHash(v) != entry.Hash {
|
||||
continue
|
||||
}
|
||||
duplicate, err := decodeEntryMeta(v)
|
||||
if err == nil && !duplicate.Pinned {
|
||||
key := append([]byte(nil), k...)
|
||||
if keepKey == nil {
|
||||
keepKey = key
|
||||
} else {
|
||||
deleteKeys = append(deleteKeys, key)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if keepKey != nil {
|
||||
for _, key := range deleteKeys {
|
||||
if err := b.Delete(key); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
return b.Delete(currentKey)
|
||||
}
|
||||
}
|
||||
|
||||
entry.Pinned = false
|
||||
encoded, err := encodeEntry(entry)
|
||||
if err != nil {
|
||||
|
||||
@@ -14,6 +14,7 @@ import (
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/mock"
|
||||
"github.com/stretchr/testify/require"
|
||||
bolt "go.etcd.io/bbolt"
|
||||
|
||||
mocks_wlcontext "github.com/AvengeMedia/DankMaterialShell/core/internal/mocks/wlcontext"
|
||||
)
|
||||
@@ -273,6 +274,110 @@ func TestHandleGetEntry_MissingIDReturnsNullResult(t *testing.T) {
|
||||
assert.Nil(t, resp.Result)
|
||||
}
|
||||
|
||||
func TestUnpinEntry_KeepsTopUnpinnedDuplicate(t *testing.T) {
|
||||
m := newTestManagerWithDB(t)
|
||||
|
||||
require.NoError(t, m.storeEntry(Entry{
|
||||
Data: []byte("saved content"),
|
||||
MimeType: "text/plain;charset=utf-8",
|
||||
Preview: "saved content",
|
||||
Size: len("saved content"),
|
||||
Timestamp: time.Now().Add(-time.Minute).Truncate(time.Second),
|
||||
IsImage: false,
|
||||
}))
|
||||
|
||||
history := m.GetHistory()
|
||||
require.Len(t, history, 1)
|
||||
pinnedID := history[0].ID
|
||||
require.NoError(t, m.PinEntry(pinnedID))
|
||||
|
||||
pinnedEntry, err := m.GetEntry(pinnedID)
|
||||
require.NoError(t, err)
|
||||
require.True(t, pinnedEntry.Pinned)
|
||||
|
||||
// Bypass storeEntry to simulate legacy duplicate ordinary history entries.
|
||||
insertLegacyUnpinnedDuplicate := func(timestamp time.Time) Entry {
|
||||
duplicate := Entry{
|
||||
Data: pinnedEntry.Data,
|
||||
MimeType: pinnedEntry.MimeType,
|
||||
Preview: pinnedEntry.Preview,
|
||||
Size: pinnedEntry.Size,
|
||||
Timestamp: timestamp,
|
||||
IsImage: pinnedEntry.IsImage,
|
||||
Pinned: false,
|
||||
}
|
||||
duplicate.Hash = computeHash(duplicate.Data)
|
||||
|
||||
require.NoError(t, m.db.Update(func(tx *bolt.Tx) error {
|
||||
b := tx.Bucket([]byte("clipboard"))
|
||||
id, err := b.NextSequence()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
duplicate.ID = id
|
||||
|
||||
encoded, err := encodeEntry(duplicate)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return b.Put(itob(id), encoded)
|
||||
}))
|
||||
|
||||
return duplicate
|
||||
}
|
||||
|
||||
olderHistoryDuplicate := insertLegacyUnpinnedDuplicate(time.Now().Add(time.Hour))
|
||||
topHistoryDuplicate := insertLegacyUnpinnedDuplicate(time.Now().Add(-time.Hour))
|
||||
require.Greater(t, topHistoryDuplicate.ID, olderHistoryDuplicate.ID)
|
||||
require.True(t, olderHistoryDuplicate.Timestamp.After(topHistoryDuplicate.Timestamp))
|
||||
|
||||
history = m.GetHistory()
|
||||
require.Len(t, history, 3)
|
||||
require.Equal(t, topHistoryDuplicate.ID, history[0].ID)
|
||||
require.NoError(t, m.UnpinEntry(pinnedID))
|
||||
|
||||
history = m.GetHistory()
|
||||
require.Len(t, history, 1)
|
||||
assert.False(t, history[0].Pinned)
|
||||
assert.Equal(t, pinnedEntry.Hash, history[0].Hash)
|
||||
assert.Equal(t, topHistoryDuplicate.ID, history[0].ID)
|
||||
}
|
||||
|
||||
func TestCreateHistoryEntryFromPinned_KeepsLatestUnpinnedDuplicate(t *testing.T) {
|
||||
m := newTestManagerWithDB(t)
|
||||
|
||||
require.NoError(t, m.storeEntry(Entry{
|
||||
Data: []byte("saved content"),
|
||||
MimeType: "text/plain;charset=utf-8",
|
||||
Preview: "saved content",
|
||||
Size: len("saved content"),
|
||||
Timestamp: time.Now().Add(-time.Minute).Truncate(time.Second),
|
||||
IsImage: false,
|
||||
}))
|
||||
|
||||
history := m.GetHistory()
|
||||
require.Len(t, history, 1)
|
||||
pinnedID := history[0].ID
|
||||
require.NoError(t, m.PinEntry(pinnedID))
|
||||
|
||||
pinnedEntry, err := m.GetEntry(pinnedID)
|
||||
require.NoError(t, err)
|
||||
require.True(t, pinnedEntry.Pinned)
|
||||
require.NoError(t, m.CreateHistoryEntryFromPinned(pinnedEntry))
|
||||
firstDuplicate := m.GetHistory()[0]
|
||||
require.NotEqual(t, pinnedID, firstDuplicate.ID)
|
||||
require.NoError(t, m.CreateHistoryEntryFromPinned(pinnedEntry))
|
||||
latestDuplicate := m.GetHistory()[0]
|
||||
|
||||
history = m.GetHistory()
|
||||
require.Len(t, history, 2)
|
||||
assert.Equal(t, latestDuplicate.ID, history[0].ID)
|
||||
assert.False(t, history[0].Pinned)
|
||||
assert.Equal(t, pinnedID, history[1].ID)
|
||||
assert.True(t, history[1].Pinned)
|
||||
assert.NotEqual(t, firstDuplicate.ID, latestDuplicate.ID)
|
||||
}
|
||||
|
||||
func TestManager_ConcurrentSubscriberAccess(t *testing.T) {
|
||||
m := &Manager{
|
||||
subscribers: make(map[string]chan State),
|
||||
|
||||
Reference in New Issue
Block a user