relay_pane_reuse.go RawHistoryFolder

1package usecase
2
3import (
4	"time"
5
6	"github.com/bnema/dumber/internal/domain/entity"
7	domainurl "github.com/bnema/dumber/internal/domain/url"
8)
9
10type relayPaneRecord struct {
11	paneID   entity.PaneID
12	openedAt time.Time
13}
14
15// RelayPaneReuse remembers panes opened for externally relayed URLs so a
16// repeated identical request (desktop apps retry OAuth URLs) reuses the pane
17// instead of creating duplicates. It is not safe for concurrent use; callers
18// confine it to the UI main thread. The zero value is ready to use.
19type RelayPaneReuse struct {
20	entries map[string]relayPaneRecord
21	// Now overrides the clock in tests; nil means time.Now.
22	Now func() time.Time
23}
24
25func (r *RelayPaneReuse) clock() time.Time {
26	if r.Now != nil {
27		return r.Now()
28	}
29	return time.Now()
30}
31
32// Lookup returns the pane to reuse for url. Entries that expired or whose pane
33// no longer exists (per exists) are forgotten first.
34func (r *RelayPaneReuse) Lookup(url string, exists func(entity.PaneID) bool) (entity.PaneID, bool) {
35	now := r.clock()
36	for key, rec := range r.entries {
37		if now.Sub(rec.openedAt) > domainurl.RelayReuseWindow || (exists != nil && !exists(rec.paneID)) {
38			delete(r.entries, key)
39		}
40	}
41	key := domainurl.RelayURLKey(url)
42	rec, ok := r.entries[key]
43	if !ok || !domainurl.ShouldReuseRelayPane(key, rec.openedAt, url, now, domainurl.RelayReuseWindow) {
44		return "", false
45	}
46	return rec.paneID, true
47}
48
49// Record remembers that paneID was created for url.
50func (r *RelayPaneReuse) Record(url string, paneID entity.PaneID) {
51	key := domainurl.RelayURLKey(url)
52	if key == "" || paneID == "" {
53		return
54	}
55	if r.entries == nil {
56		r.entries = make(map[string]relayPaneRecord)
57	}
58	r.entries[key] = relayPaneRecord{paneID: paneID, openedAt: r.clock()}
59}
60
61// Forget drops the record for url.
62func (r *RelayPaneReuse) Forget(url string) {
63	delete(r.entries, domainurl.RelayURLKey(url))
64}
65
66// Len reports how many relay panes are remembered.
67func (r *RelayPaneReuse) Len() int {
68	return len(r.entries)
69}