author | František Kučera <franta-hg@frantovo.cz> |
Wed, 16 Jan 2019 18:01:49 +0100 | |
branch | v_0 |
changeset 6 | 35607c973cf5 |
parent 5 | ec661baf433a |
child 7 | 8d73bff730a7 |
permissions | -rw-r--r-- |
0 | 1 |
/** |
2 |
* Relational pipes |
|
4 | 3 |
* Copyright © 2019 František Kučera (Frantovo.cz, GlobalCode.info) |
0 | 4 |
* |
5 |
* This program is free software: you can redistribute it and/or modify |
|
6 |
* it under the terms of the GNU General Public License as published by |
|
7 |
* the Free Software Foundation, either version 3 of the License, or |
|
8 |
* (at your option) any later version. |
|
9 |
* |
|
10 |
* This program is distributed in the hope that it will be useful, |
|
11 |
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
12 |
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
13 |
* GNU General Public License for more details. |
|
14 |
* |
|
15 |
* You should have received a copy of the GNU General Public License |
|
16 |
* along with this program. If not, see <http://www.gnu.org/licenses/>. |
|
17 |
*/ |
|
18 |
#pragma once |
|
19 |
||
20 |
#include <vector> |
|
2
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
21 |
#include <filesystem> |
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
22 |
|
4 | 23 |
#include <relpipe/writer/typedefs.h> |
24 |
#include <relpipe/writer/AttributeMetadata.h> |
|
25 |
#include <relpipe/writer/RelationalWriter.h> |
|
2
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
26 |
|
4 | 27 |
#include "RequestedField.h" |
0 | 28 |
|
29 |
namespace relpipe { |
|
30 |
namespace in { |
|
31 |
namespace filesystem { |
|
32 |
||
2
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
33 |
namespace fs = std::filesystem; |
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
34 |
using namespace relpipe::writer; |
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
35 |
|
4 | 36 |
class FileAttributeFinder : public AttributeFinder { |
37 |
private: |
|
2
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
38 |
std::wstring_convert<codecvt_utf8<wchar_t>> convertor; // TODO: support also other encodings. |
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
39 |
|
4 | 40 |
fs::path currentFile; |
41 |
string_t currentOwner; |
|
42 |
string_t currentGroup; |
|
2
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
43 |
|
6
35607c973cf5
distinguish link type and link target type
František Kučera <franta-hg@frantovo.cz>
parents:
5
diff
changeset
|
44 |
string_t getType(const fs::path& file, bool x) { |
2
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
45 |
// TODO: Use whole words? (letters are compatible with find -type) |
6
35607c973cf5
distinguish link type and link target type
František Kučera <franta-hg@frantovo.cz>
parents:
5
diff
changeset
|
46 |
if (fs::is_symlink(file) && x) return L"l"; // symlinks are both symlinks and files/directories/etc. |
35607c973cf5
distinguish link type and link target type
František Kučera <franta-hg@frantovo.cz>
parents:
5
diff
changeset
|
47 |
else if (fs::is_regular_file(file)) return L"f"; |
2
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
48 |
else if (fs::is_directory(file)) return L"d"; |
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
49 |
else if (fs::is_fifo(file)) return L"p"; |
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
50 |
else if (fs::is_socket(file)) return L"s"; |
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
51 |
else if (fs::is_block_file(file)) return L"b"; |
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
52 |
else if (fs::is_character_file(file)) return L"c"; |
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
53 |
else return L"o"; |
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
54 |
} |
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
55 |
|
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
56 |
void fetchOwner(const fs::path& file, string_t& owner, string_t& group) { |
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
57 |
// TODO: throw exception on error |
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
58 |
// TODO: get user and group in C++ way? |
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
59 |
struct stat info; |
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
60 |
stat(file.c_str(), &info); |
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
61 |
/** |
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
62 |
* The return value may point to a static area, and may be |
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
63 |
* overwritten by subsequent calls to getpwent(3), getpw‐ |
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
64 |
* nam(), or getpwuid(). (Do not pass the returned pointer |
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
65 |
* to free(3).) |
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
66 |
*/ |
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
67 |
struct passwd* pw = getpwuid(info.st_uid); |
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
68 |
struct group* gr = getgrgid(info.st_gid); |
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
69 |
owner = convertor.from_bytes(pw->pw_name); |
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
70 |
group = convertor.from_bytes(gr->gr_name); |
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
71 |
} |
4 | 72 |
public: |
2
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
73 |
|
4 | 74 |
static const string_t FIELD_PATH_ORIGINAL; |
75 |
static const string_t FIELD_NAME; |
|
76 |
static const string_t FIELD_SIZE; |
|
77 |
static const string_t FIELD_PATH_ABSOLUTE; |
|
78 |
static const string_t FIELD_PATH_CANONICAL; |
|
79 |
static const string_t FIELD_TYPE; |
|
6
35607c973cf5
distinguish link type and link target type
František Kučera <franta-hg@frantovo.cz>
parents:
5
diff
changeset
|
80 |
static const string_t FIELD_SYMLINK_TARGET_TYPE; |
4 | 81 |
static const string_t FIELD_OWNER; |
82 |
static const string_t FIELD_GROUP; |
|
83 |
||
84 |
virtual vector<AttributeMetadata> toMetadata(const RequestedField& field) override { |
|
85 |
if (field.group == RequestedField::GROUP_FILE) { |
|
5
ec661baf433a
support field aliases
František Kučera <franta-hg@frantovo.cz>
parents:
4
diff
changeset
|
86 |
vector<AttributeMetadata> metadata; |
ec661baf433a
support field aliases
František Kučera <franta-hg@frantovo.cz>
parents:
4
diff
changeset
|
87 |
for (string_t alias : field.getAliases()) { |
ec661baf433a
support field aliases
František Kučera <franta-hg@frantovo.cz>
parents:
4
diff
changeset
|
88 |
if (field.name == FIELD_SIZE) metadata.push_back(AttributeMetadata{alias, TypeId::INTEGER}); |
ec661baf433a
support field aliases
František Kučera <franta-hg@frantovo.cz>
parents:
4
diff
changeset
|
89 |
else metadata.push_back(AttributeMetadata{alias, TypeId::STRING}); |
ec661baf433a
support field aliases
František Kučera <franta-hg@frantovo.cz>
parents:
4
diff
changeset
|
90 |
} |
ec661baf433a
support field aliases
František Kučera <franta-hg@frantovo.cz>
parents:
4
diff
changeset
|
91 |
return metadata; |
2
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
92 |
} else { |
4 | 93 |
return {}; |
2
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
94 |
} |
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
95 |
} |
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
96 |
|
4 | 97 |
void startFile(const fs::path& file) override { |
98 |
currentFile = file; |
|
99 |
}; |
|
2
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
100 |
|
4 | 101 |
void endFile() override { |
102 |
currentOwner.clear(); |
|
103 |
currentGroup.clear(); |
|
104 |
}; |
|
2
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
105 |
|
4 | 106 |
virtual void writeField(RelationalWriter* writer, const RequestedField& field) override { |
107 |
if (field.group == RequestedField::GROUP_FILE) { |
|
5
ec661baf433a
support field aliases
František Kučera <franta-hg@frantovo.cz>
parents:
4
diff
changeset
|
108 |
for (string_t alias : field.getAliases()) { |
ec661baf433a
support field aliases
František Kučera <franta-hg@frantovo.cz>
parents:
4
diff
changeset
|
109 |
if (field.name == FIELD_NAME) { |
ec661baf433a
support field aliases
František Kučera <franta-hg@frantovo.cz>
parents:
4
diff
changeset
|
110 |
writer->writeAttribute(currentFile.filename().wstring()); |
ec661baf433a
support field aliases
František Kučera <franta-hg@frantovo.cz>
parents:
4
diff
changeset
|
111 |
} else if (field.name == FIELD_PATH_ORIGINAL) { |
ec661baf433a
support field aliases
František Kučera <franta-hg@frantovo.cz>
parents:
4
diff
changeset
|
112 |
writer->writeAttribute(currentFile.wstring()); |
ec661baf433a
support field aliases
František Kučera <franta-hg@frantovo.cz>
parents:
4
diff
changeset
|
113 |
} else if (field.name == FIELD_PATH_ABSOLUTE) { |
ec661baf433a
support field aliases
František Kučera <franta-hg@frantovo.cz>
parents:
4
diff
changeset
|
114 |
writer->writeAttribute(fs::absolute(currentFile).wstring()); |
ec661baf433a
support field aliases
František Kučera <franta-hg@frantovo.cz>
parents:
4
diff
changeset
|
115 |
} else if (field.name == FIELD_PATH_CANONICAL) { |
ec661baf433a
support field aliases
František Kučera <franta-hg@frantovo.cz>
parents:
4
diff
changeset
|
116 |
writer->writeAttribute(fs::canonical(currentFile).wstring()); |
ec661baf433a
support field aliases
František Kučera <franta-hg@frantovo.cz>
parents:
4
diff
changeset
|
117 |
} else if (field.name == FIELD_TYPE) { |
6
35607c973cf5
distinguish link type and link target type
František Kučera <franta-hg@frantovo.cz>
parents:
5
diff
changeset
|
118 |
writer->writeAttribute(getType(currentFile, true)); |
35607c973cf5
distinguish link type and link target type
František Kučera <franta-hg@frantovo.cz>
parents:
5
diff
changeset
|
119 |
} else if (field.name == FIELD_SYMLINK_TARGET_TYPE) { |
35607c973cf5
distinguish link type and link target type
František Kučera <franta-hg@frantovo.cz>
parents:
5
diff
changeset
|
120 |
writer->writeAttribute(getType(currentFile, false)); |
5
ec661baf433a
support field aliases
František Kučera <franta-hg@frantovo.cz>
parents:
4
diff
changeset
|
121 |
} else if (field.name == FIELD_SIZE) { |
ec661baf433a
support field aliases
František Kučera <franta-hg@frantovo.cz>
parents:
4
diff
changeset
|
122 |
integer_t size = fs::is_regular_file(currentFile) ? fs::file_size(currentFile) : 0; |
ec661baf433a
support field aliases
František Kučera <franta-hg@frantovo.cz>
parents:
4
diff
changeset
|
123 |
writer->writeAttribute(&size, typeid (size)); |
ec661baf433a
support field aliases
František Kučera <franta-hg@frantovo.cz>
parents:
4
diff
changeset
|
124 |
} else if (field.name == FIELD_OWNER) { |
ec661baf433a
support field aliases
František Kučera <franta-hg@frantovo.cz>
parents:
4
diff
changeset
|
125 |
if (currentOwner.empty()) fetchOwner(currentFile, currentOwner, currentGroup); |
ec661baf433a
support field aliases
František Kučera <franta-hg@frantovo.cz>
parents:
4
diff
changeset
|
126 |
writer->writeAttribute(currentOwner); |
ec661baf433a
support field aliases
František Kučera <franta-hg@frantovo.cz>
parents:
4
diff
changeset
|
127 |
} else if (field.name == FIELD_GROUP) { |
ec661baf433a
support field aliases
František Kučera <franta-hg@frantovo.cz>
parents:
4
diff
changeset
|
128 |
if (currentOwner.empty()) fetchOwner(currentFile, currentOwner, currentGroup); |
ec661baf433a
support field aliases
František Kučera <franta-hg@frantovo.cz>
parents:
4
diff
changeset
|
129 |
writer->writeAttribute(currentGroup); |
ec661baf433a
support field aliases
František Kučera <franta-hg@frantovo.cz>
parents:
4
diff
changeset
|
130 |
} else { |
ec661baf433a
support field aliases
František Kučera <franta-hg@frantovo.cz>
parents:
4
diff
changeset
|
131 |
// TODO: should not happend; check supported attributes in toMetadata()? |
ec661baf433a
support field aliases
František Kučera <franta-hg@frantovo.cz>
parents:
4
diff
changeset
|
132 |
writer->writeAttribute(L""); |
ec661baf433a
support field aliases
František Kučera <franta-hg@frantovo.cz>
parents:
4
diff
changeset
|
133 |
} |
2
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
134 |
} |
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
135 |
} |
4 | 136 |
} |
2
f07ed604a0ab
read filenames separated by a null-byte; fetch their size, owner, group… and some xattr (extended attribute); a preview version
František Kučera <franta-hg@frantovo.cz>
parents:
0
diff
changeset
|
137 |
|
4 | 138 |
virtual ~FileAttributeFinder() override { |
0 | 139 |
} |
140 |
}; |
|
141 |
||
4 | 142 |
const string_t FileAttributeFinder::FIELD_PATH_ORIGINAL = L"path"; |
143 |
const string_t FileAttributeFinder::FIELD_NAME = L"name"; |
|
144 |
const string_t FileAttributeFinder::FIELD_SIZE = L"size"; |
|
145 |
const string_t FileAttributeFinder::FIELD_PATH_ABSOLUTE = L"path_absolute"; |
|
146 |
const string_t FileAttributeFinder::FIELD_PATH_CANONICAL = L"path_canonical"; |
|
147 |
const string_t FileAttributeFinder::FIELD_TYPE = L"type"; |
|
6
35607c973cf5
distinguish link type and link target type
František Kučera <franta-hg@frantovo.cz>
parents:
5
diff
changeset
|
148 |
const string_t FileAttributeFinder::FIELD_SYMLINK_TARGET_TYPE = L"symlink_target_type"; |
4 | 149 |
const string_t FileAttributeFinder::FIELD_OWNER = L"owner"; |
150 |
const string_t FileAttributeFinder::FIELD_GROUP = L"group"; |
|
151 |
||
0 | 152 |
} |
153 |
} |
|
154 |
} |