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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
|
local jid_node = require "prosody.util.jid".node;
local flags = module:open_store("account_flags", "keyval+");
-- API
function add_flag(username, flag, comment)
local flag_data = {
when = os.time();
comment = comment;
};
local ok, err = flags:set_key(username, flag, flag_data);
if not ok then
return nil, err;
end
module:fire_event("user-flag-added/"..flag, {
user = username;
flag = flag;
data = flag_data;
});
return true;
end
function remove_flag(username, flag)
local ok, err = flags:set_key(username, flag, nil);
if not ok then
return nil, err;
end
module:fire_event("user-flag-removed/"..flag, {
user = username;
flag = flag;
});
return true;
end
function has_flag(username, flag) -- luacheck: ignore 131/has_flag
local ok, err = flags:get_key(username, flag);
if not ok and err then
error("Failed to check flags for user: "..err);
end
return not not ok;
end
function get_flag_info(username, flag) -- luacheck: ignore 131/get_flag_info
return flags:get_key(username, flag);
end
-- Shell commands
local function get_username(jid)
return (assert(jid_node(jid), "please supply a valid user JID"));
end
module:add_item("shell-command", {
section = "flags";
section_desc = "View and manage flags on user accounts";
name = "list";
desc = "List flags for the given user account";
args = {
{ name = "jid", type = "string" };
};
host_selector = "jid";
handler = function(self, jid) --luacheck: ignore 212/self
local c = 0;
local user_flags, err = flags:get(get_username(jid));
if not user_flags and err then
return false, "Unable to list flags: "..err;
end
if user_flags then
local print = self.session.print;
for flag_name, flag_data in pairs(user_flags) do
print(flag_name, os.date("%Y-%m-%d %R", flag_data.when), flag_data.comment);
c = c + 1;
end
end
return true, ("%d flags listed"):format(c);
end;
});
module:add_item("shell-command", {
section = "flags";
section_desc = "View and manage flags on user accounts";
name = "add";
desc = "Add a flag to the given user account, with optional comment";
args = {
{ name = "jid", type = "string" };
{ name = "flag", type = "string" };
{ name = "comment", type = "string" };
};
host_selector = "jid";
handler = function(self, jid, flag, comment) --luacheck: ignore 212/self
local username = get_username(jid);
local ok, err = add_flag(username, flag, comment);
if not ok then
return false, "Failed to add flag: "..err;
end
return true, "Flag added";
end;
});
module:add_item("shell-command", {
section = "flags";
section_desc = "View and manage flags on user accounts";
name = "remove";
desc = "Remove a flag from the given user account";
args = {
{ name = "jid", type = "string" };
{ name = "flag", type = "string" };
};
host_selector = "jid";
handler = function(self, jid, flag) --luacheck: ignore 212/self
local username = get_username(jid);
local ok, err = remove_flag(username, flag);
if not ok then
return false, "Failed to remove flag: "..err;
end
return true, "Flag removed";
end;
});
module:add_item("shell-command", {
section = "flags";
section_desc = "View and manage flags on user accounts";
name = "find";
desc = "Find all user accounts with a given flag on the specified host";
args = {
{ name = "host", type = "string" };
{ name = "flag", type = "string" };
};
host_selector = "host";
handler = function(self, host, flag) --luacheck: ignore 212/self 212/host
local users_with_flag = flags:get_key_from_all(flag);
local print = self.session.print;
local c = 0;
for user, flag_data in pairs(users_with_flag) do
print(user, os.date("%Y-%m-%d %R", flag_data.when), flag_data.comment);
c = c + 1;
end
return true, ("%d accounts listed"):format(c);
end;
});
|
