summaryrefslogtreecommitdiff
path: root/lib/mu-labels-cache.cc
blob: 0e4fa8353fd76e9d64eb831f99a0103a660e7177 (plain)
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
/*
** Copyright (C) 2025 Dirk-Jan C. Binnema <djcb@djcbsoftware.nl>
**
** This program is free software; you can redistribute it and/or modify it
** under the terms of the GNU General Public License as published by the
** Free Software Foundation; either version 3, or (at your option) any
** later version.
**
** This program is distributed in the hope that it will be useful,
** but WITHOUT ANY WARRANTY; without even the implied warranty of
** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
** GNU General Public License for more details.
**
** You should have received a copy of the GNU General Public License
** along with this program; if not, write to the Free Software Foundation,
** Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
**
*/

#include "mu-labels-cache.hh"
#include "mu-store.hh"
#include "message/mu-labels.hh"

using namespace Mu;

void
Mu::LabelsCache::serialize() const
{
	std::string s;
	for (const auto&[label, n]: label_map_)
		s += mu_format("{}{}{}\n", label, SepaChar2, n);

	config_.set<Config::Id::Labels>(s);
	mu_debug("labels: serialized {} change(s)", dirty_);
	dirty_ = 0;
}

Mu::LabelsCache::Map
Mu::LabelsCache::deserialize(const std::string& serialized) const
{
	Map  map;

	std::stringstream ss{serialized, std::ios_base::in};
	std::string       line;

	while (std::getline(ss, line)) {
		if (const auto parts =
		    Mu::split(line, SepaChar2); parts.size() != 2)
			mu_warning("error: '{}'", line);
		else
			map.emplace(std::move(parts[0]),
				    static_cast<std::size_t>(
					    g_ascii_strtoll(parts[1].c_str(),{}, 10)));
	}
	return map;
}

Result<void>
Mu::LabelsCache::restore(const Store& store)
{
	const auto res{store.run_query("")};
	if (!res)
		return Err(Error{Error::Code::Query,
				"failed to run query: {}",
				 *res.error().what()});
	label_map_.clear();
	++dirty_;

	for (auto&& item: *res) {
		if (auto &&msg{item.message()}; msg) {
			for (const auto& label: msg->labels())
				increase(label);
		}
	}

	serialize();

	return Ok();
}