blob: 8ca724c432e2c32e5e920aad9c8517875a663868 (
plain) (
blame)
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
|
#include "resource.h"
#include <yql/essentials/utils/yql_panic.h>
#include <library/cpp/resource/resource.h>
namespace NYql::NDocs {
bool IsMatching(const TResourceFilter& filter, TStringBuf key) {
return key.Contains(filter.BaseDirectorySuffix) &&
key.EndsWith(filter.CutSuffix);
}
TStringBuf RelativePath(const TResourceFilter& filter, TStringBuf key Y_LIFETIME_BOUND) {
size_t pos = key.find(filter.BaseDirectorySuffix);
YQL_ENSURE(pos != TString::npos);
pos += filter.BaseDirectorySuffix.size();
TStringBuf tail = TStringBuf(key).SubStr(pos);
tail.remove_suffix(filter.CutSuffix.size());
return tail;
}
TResourcesByRelativePath FindResources(const TResourceFilter& filter) {
YQL_ENSURE(
filter.BaseDirectorySuffix.EndsWith('/'),
"BaseDirectory should end with '/', but got '" << filter.BaseDirectorySuffix << "'");
TResourcesByRelativePath resources;
for (TStringBuf key : NResource::ListAllKeys()) {
if (!key.StartsWith("resfs/file/") || !IsMatching(filter, key)) {
continue;
}
TStringBuf path = RelativePath(filter, key);
YQL_ENSURE(!resources.contains(path));
resources[path] = NResource::Find(key);
}
return resources;
}
} // namespace NYql::NDocs
|
