-
-
Notifications
You must be signed in to change notification settings - Fork 106
/
CompRepoItemAdd.groovy
109 lines (93 loc) · 3.53 KB
/
CompRepoItemAdd.groovy
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
/*
* Tigase XMPP Server - The instant messaging server
* Copyright (C) 2004 Tigase, Inc. (office@tigase.com)
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as published by
* the Free Software Foundation, version 3 of the License.
*
* 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 Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. Look for COPYING file in the top folder.
* If not, see http://www.gnu.org/licenses/.
*/
/*
Add an item in a component repository:
tigase.db.ComponentRepository
Works only for some components which actually use the repository that way.
AS:Description: Add new item
AS:CommandId: comp-repo-item-add
AS:Component: vhost-man,ext,ext-disco
AS:ComponentClass: tigase.server.ext.ComponentProtocol,tigase.server.ext.ComponentProtocolManager
AS:Group: Configuration
*/
package tigase.admin
import tigase.db.comp.ComponentRepository
import tigase.server.Command
import tigase.server.Iq
import tigase.server.Permissions
def MARKER = "command-marker"
def repo = (ComponentRepository) comp_repo
def p = (Iq) packet
def admins = (Set) adminsSet
def stanzaFromBare = p.getStanzaFrom().getBareJID()
def isServiceAdmin = admins.contains(stanzaFromBare)
def item = repo.getItemInstance()
def marker = Command.getFieldValue(p, MARKER)
def supportedComponents = [ "vhost-man" ]
def NOTIFY_CLUSTER = "notify-cluster"
boolean clusterMode = Boolean.valueOf(System.getProperty("cluster-mode", false.toString()));
boolean notifyCluster = Boolean.valueOf(Command.getFieldValue(packet, NOTIFY_CLUSTER))
Queue results = new LinkedList()
if (marker == null) {
def result = p.commandResult(Command.DataType.form)
item.addCommandFields(result)
Command.addHiddenField(result, MARKER, MARKER)
if (clusterMode) {
Command.addHiddenField(result, NOTIFY_CLUSTER, true.toString())
}
return result
}
if (clusterMode && notifyCluster && supportedComponents.contains(componentName)) {
def nodes = (List) connectedNodes
if (nodes && nodes.size() > 0) {
nodes.each { node ->
def forward = p.copyElementOnly();
Command.removeFieldValue(forward, NOTIFY_CLUSTER)
Command.addHiddenField(forward, NOTIFY_CLUSTER, false.toString())
forward.setPacketTo(node);
forward.setPermissions(Permissions.ADMIN);
results.offer(forward)
}
}
}
def oldItem = item.getKey() != null ? repo.getItem(item.getKey()) : null;
def result = p.commandResult(Command.DataType.result)
try {
item.initFromCommand(p)
if (oldItem == null) {
def validateResult = repo.validateItem(item)
if (validateResult == null || (isServiceAdmin && !item.getKey().isEmpty())) {
repo.addItem(item)
Command.addTextField(result, "Note", "Operation successful.")
if (validateResult != null) {
Command.addTextField(result, "Note", " ")
Command.addTextField(result, "Warning", validateResult)
}
} else {
Command.addTextField(result, "Error", "The item did not pass validation checking.")
Command.addTextField(result, "Note", " ")
Command.addTextField(result, "Warning", validateResult)
}
} else {
Command.addTextField(result, "Error", "The item is already added, you can't add it twice.")
}
} catch (IllegalArgumentException ex) {
Command.addTextField(result, "Error", ex.getMessage());
}
results.add(result);
return results;