whereami/observation_index.go at main · rubiojr/whereami · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
package main

import (
	"errors"
	"path/filepath"

	"github.com/rubiojr/whereami/internal/observations"
)

var (
	observationRepo          *observations.Repository
	observationDataRoot      string
	observationBookmarksPath string
	observationIndexRebuilt  func()
)

func openObservationIndex(dataRoot, bookmarksPath string) error {
	canonicalDataRoot, err := filepath.EvalSymlinks(dataRoot)
	if err != nil {
		return err
	}
	canonicalDataRoot, err = filepath.Abs(canonicalDataRoot)
	if err != nil {
		return err
	}
	canonicalBookmarksPath := bookmarksPath
	if bookmarksPath != "" {
		if resolved, resolveErr := filepath.EvalSymlinks(bookmarksPath); resolveErr == nil {
			canonicalBookmarksPath = resolved
		}
		canonicalBookmarksPath, err = filepath.Abs(canonicalBookmarksPath)
		if err != nil {
			return err
		}
	}
	dbPath := filepath.Join(effectiveCacheDir(), "observations", "observations.sqlite")
	repo, err := observations.Open(dbPath)
	if err != nil {
		return err
	}

	observationRepo = repo
	observationDataRoot = canonicalDataRoot
	observationBookmarksPath = canonicalBookmarksPath
	if err := rebuildObservationIndex(); err != nil {
		_ = repo.Close()
		observationRepo = nil
		return err
	}
	return nil
}

func rebuildObservationIndex() error {
	if observationRepo == nil {
		return errors.New("observation repository is unavailable")
	}
	if err := observationRepo.Rebuild(observationDataRoot, observationBookmarksPath); err != nil {
		return err
	}
	if observationIndexRebuilt != nil {
		observationIndexRebuilt()
	}
	return nil
}

func addObservationSources(paths []string) error {
	if len(paths) == 0 {
		return nil
	}
	if observationRepo == nil {
		return errors.New("observation repository is unavailable")
	}
	if err := observationRepo.AddSources(observationDataRoot, observationBookmarksPath, paths); err != nil {
		return err
	}
	if observationIndexRebuilt != nil {
		observationIndexRebuilt()
	}
	return nil
}

func closeObservationIndex() error {
	if observationRepo == nil {
		return nil
	}
	return observationRepo.Close()
}