abba/guacamole-ext-wallix-sync
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

Add target group synchronization (without member connections yet).

Browse Source
This commit is contained in:
Abba Soungui YOUNOUSS
2025-11-08 17:02:27 +01:00
parent 569858f9ca
commit cebef4f2dd
4 changed files with 183 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
src/main/java/cm/soungui/guacamole/ext/wallix/sync/TargetGroup.java Normal file
View File

@@ -0,0 +1,46 @@
package cm.soungui.guacamole.ext.wallix.sync;
import cm.soungui.guacamole.ext.wallix.sync.db.Entity;
public class TargetGroup {
private int id;
private String name;
public int getId() {
return id;
}
public void setId(int id) {
this.id = id;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
@Override
public int hashCode() {
return getName() == null ? super.hashCode() : getName().hashCode();
}
@Override
public boolean equals(Object obj) {
if(super.equals(obj)) {
return true;
}
if (obj instanceof TargetGroup) {
TargetGroup otherObject = (TargetGroup) obj;
boolean namesAreTheSame = otherObject.getName() != null && getName() != null && getName().equals(otherObject.getName());
return namesAreTheSame;
}
return false;
}
}

31
src/main/java/cm/soungui/guacamole/ext/wallix/sync/Wallix.java
View File

@@ -125,7 +125,7 @@ public class Wallix {
return output;
}
public Set<UserGroup> getGroups() throws Exception {
public Set<UserGroup> getUserGroups() throws Exception {
String output = get("/usergroups");
ObjectMapper objectMapper = new ObjectMapper();
@@ -188,6 +188,35 @@ public class Wallix {
}
}
public Set<TargetGroup> getTargetGroups() throws Exception {
String output = get("/targetgroups?fields=group_name,session");
ObjectMapper objectMapper = new ObjectMapper();
JsonNode jsonNode = objectMapper.readTree(output);
Set<TargetGroup> list = new HashSet<>();
jsonNode.elements().forEachRemaining((e) -> {
ObjectNode node = (ObjectNode) e;
TargetGroup group = new TargetGroup();
group.setName(node.findValue(Wallix.API.ATTRIBUTE_GROUP_NAME).asText());
// List<JsonNode> membersList = node.findValues(Wallix.API.ATTRIBUTE_GROUP_USERS);
// if (! membersList.isEmpty()) {
// JsonNode usernames = membersList.get(0);
// usernames.forEach(userNode -> {
// User user = new User();
// user.setName(getNameWithoutDomain(userNode.textValue()));
// group.getMembers().add(user);
// });
// }
list.add(group);
});
return list;
}
public final class API {
public static final String ATTRIBUTE_USER_DISABLED = "is_disabled";

44
src/main/java/cm/soungui/guacamole/ext/wallix/sync/WallixConfigSynchronizer.java
View File

@@ -28,7 +28,8 @@ public class WallixConfigSynchronizer implements Runnable {
try {
System.out.println("Fetching config");
synchronizeUsers();
synchronizeGroups();
synchronizeUserGroups();
synchronizeTargetGroups();
Thread.sleep(sleepDuration);
} catch (Exception ex) {
ex.printStackTrace();
@@ -75,10 +76,10 @@ public class WallixConfigSynchronizer implements Runnable {
return dbUser.isDisabled() != wallixUser.isDisabled();
}
private void synchronizeGroups() throws Exception {
private void synchronizeUserGroups() throws Exception {
System.out.println("Synchronizing groups");
System.out.println("Getting Wallix groups");
Set<UserGroup> wallixGroups = Wallix.getInstance().getGroups();
Set<UserGroup> wallixGroups = Wallix.getInstance().getUserGroups();
DB db = DB.getInstance();
Set<UserGroup> dbGroups = db.getUserGroups();
@@ -104,12 +105,45 @@ public class WallixConfigSynchronizer implements Runnable {
}
}
} else {
System.out.println("Deleting group : " + group.getName());
db.deleteGroup(group);
System.out.println("Deleting user group : " + group.getName());
db.deleteUserGroup(group);
}
}
}
private void synchronizeTargetGroups() throws Exception {
System.out.println("Synchronizing target groups");
System.out.println("Getting Wallix target groups");
DB db = DB.getInstance();
Set<TargetGroup> dbGroups = db.getTargetGroups();
Set<TargetGroup> wallixGroups = Wallix.getInstance().getTargetGroups();
for (TargetGroup group : wallixGroups) {
if (! dbGroups.contains(group)) {
System.out.println("Adding target group " + group.getName());
db.addTargetGroup(group);
}
}
Set<TargetGroup> newDbGroups = db.getTargetGroups();
for (TargetGroup group : newDbGroups) {
if (wallixGroups.contains(group)) {
for (TargetGroup wallixGroup : wallixGroups) {
if (group.equals(wallixGroup)) {
// db.updateGroupMembers(group, wallixGroup.getMembers());
}
}
} else {
System.out.println("Deleting target group : " + group.getName());
db.deleteTargetGroup(group);
}
}
}
private void printWallixVersion() throws Exception {
Wallix wallix = Wallix.getInstance();
System.out.println(wallix.getVersion());

70
src/main/java/cm/soungui/guacamole/ext/wallix/sync/db/DB.java
View File

@@ -14,6 +14,7 @@ import org.apache.guacamole.environment.Environment;
import org.apache.guacamole.environment.LocalEnvironment;
import cm.soungui.guacamole.ext.wallix.sync.Configuration;
import cm.soungui.guacamole.ext.wallix.sync.TargetGroup;
import cm.soungui.guacamole.ext.wallix.sync.db.Entity.EntityType;
public class DB {
@@ -127,7 +128,7 @@ public class DB {
return users;
}
public void deleteGroup(UserGroup group) throws SQLException, GuacamoleException {
public void deleteUserGroup(UserGroup group) throws SQLException, GuacamoleException {
if (group == null || group.getName() == null) {
return;
}
@@ -139,7 +140,7 @@ public class DB {
stmt.executeUpdate();
}
public UserGroup getGroup(String name) throws SQLException, GuacamoleException {
public UserGroup getUserGroup(String name) throws SQLException, GuacamoleException {
DB db = DB.getInstance();
Connection connection = db.getMySQLConnection();
PreparedStatement stmt = connection
@@ -348,4 +349,69 @@ public class DB {
stmt.executeUpdate();
}
public void addTargetGroup(TargetGroup group) throws SQLException, GuacamoleException {
if (group == null || group.getName() == null) {
throw new NullPointerException();
}
DB db = DB.getInstance();
Connection connection = db.getMySQLConnection();
PreparedStatement stmt = connection
.prepareStatement("insert into guacamole_connection_group (connection_group_name, type) values (?, 'ORGANIZATIONAL')",
Statement.RETURN_GENERATED_KEYS);
stmt.setString(1, group.getName());
stmt.executeUpdate();
}
public TargetGroup getTargetGroup(String name) throws SQLException, GuacamoleException {
DB db = DB.getInstance();
Connection connection = db.getMySQLConnection();
PreparedStatement stmt = connection
.prepareStatement("select connection_group_id from guacamole_connection_group where type='ORGANIZATIONAL' and name=?");
stmt.setString(1, name);
ResultSet rs = stmt.executeQuery();
while (rs.next()) {
TargetGroup group = new TargetGroup();
group.setId(rs.getInt("connection_group_id"));
group.setName(name);
return group;
}
return null;
}
public Set<TargetGroup> getTargetGroups() throws GuacamoleException, SQLException {
DB db = DB.getInstance();
Connection connection = db.getMySQLConnection();
ResultSet rs = connection
.prepareStatement("select g.connection_group_id, g.connection_group_name from guacamole_connection_group g where type='ORGANIZATIONAL'")
.executeQuery();
HashSet<TargetGroup> groups = new HashSet<>();
while (rs.next()) {
TargetGroup group = new TargetGroup();
group.setId(rs.getInt("connection_group_id"));
group.setName(rs.getString("connection_group_name"));
groups.add(group);
}
return groups;
}
public void deleteTargetGroup(TargetGroup group) throws SQLException, GuacamoleException {
if (group == null || group.getName() == null) {
return;
}
DB db = DB.getInstance();
Connection connection = db.getMySQLConnection();
PreparedStatement stmt = connection
.prepareStatement("delete from guacamole_connection_group where type='ORGANIZATIONAL' and connection_group_name=?");
stmt.setString(1, group.getName());
stmt.executeUpdate();
}
}