Users.java example

Explorer
constellio-master
package com.constellio.sdk.tests.setups;

import java.io.InputStream;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

import com.constellio.model.entities.records.Record;
import com.constellio.model.entities.records.wrappers.Collection;
import org.apache.commons.io.IOUtils;

import com.constellio.model.entities.records.wrappers.Group;
import com.constellio.model.entities.records.wrappers.User;
import com.constellio.model.entities.security.global.GlobalGroup;
import com.constellio.model.entities.security.global.GlobalGroupStatus;
import com.constellio.model.entities.security.global.UserCredential;
import com.constellio.model.entities.security.global.UserCredentialStatus;
import com.constellio.model.services.security.authentification.AuthenticationService;
import com.constellio.model.services.users.UserPhotosServices;
import com.constellio.model.services.users.UserServices;

public class Users {

	String admin = "admin";
	String heroes = "heroes";
	String legends = "legends";
	String sidekicks = "sidekicks";
	String rumors = "rumors";
	String chuckNorrisUsername = "chuck";
	String aliceUsername = "alice";
	String bobGrattonUsername = "bob";
	String charlesFrancoisXavierUsername = "charles";
	String dakotaLindienUsername = "dakota";
	String edouardLechatUsername = "edouard";
	String gandalfLeblancUsername = "gandalf";
	String robinUsername = "robin";
	String sasquatchUsername = "sasquatch";

	UserServices userServices;

	public UserCredential chuckNorris() {
		return getUser(chuckNorrisUsername);
	}

	public User chuckNorrisIn(String collection) {
		return getUser(chuckNorrisUsername, collection);
	}

	public UserCredential admin() {
		return userServices.getUser(admin);
	}

	public User adminIn(String collection) {
		return userServices.getUserInCollection(admin, collection);
	}

	public UserCredential alice() {
		return getUser(aliceUsername);
	}

	public User aliceIn(String collection) {
		return getUser(aliceUsername, collection);
	}

	public UserCredential bob() {
		return getUser(bobGrattonUsername);
	}

	public User bobIn(String collection) {
		return getUser(bobGrattonUsername, collection);
	}

	public UserCredential charles() {
		return getUser(charlesFrancoisXavierUsername);
	}

	public User charlesIn(String collection) {
		return getUser(charlesFrancoisXavierUsername, collection);
	}

	public UserCredential dakotaLIndien() {
		return getUser(dakotaLindienUsername);
	}

	public User dakotaIn(String collection) {
		return dakotaLIndienIn(collection);
	}

	public User dakotaLIndienIn(String collection) {
		return getUser(dakotaLindienUsername, collection);
	}

	public UserCredential edouardLechat() {
		return getUser(edouardLechatUsername);
	}

	public User edouardIn(String collection) {
		return edouardLechatIn(collection);
	}

	public User edouardLechatIn(String collection) {
		return getUser(edouardLechatUsername, collection);
	}

	public UserCredential gandalfLeblanc() {
		return getUser(gandalfLeblancUsername);
	}

	public UserCredential robin() {
		return getUser(robinUsername);
	}

	public UserCredential sasquatch() {
		return getUser(sasquatchUsername);
	}

	public User gandalfIn(String collection) {
		return gandalfLeblancIn(collection);
	}

	public User gandalfLeblancIn(String collection) {
		return getUser(gandalfLeblancUsername, collection);
	}

	public User robinIn(String collection) {
		return getUser(robinUsername, collection);
	}

	public User sasquatchIn(String collection) {
		return getUser(sasquatchUsername, collection);
	}

	public List<UserCredential> getAllUsers() {
		return Arrays.asList(
				chuckNorris(), admin(), alice(), bob(), charles(), dakotaLIndien(), edouardLechat(), gandalfLeblanc(), robin(),
				sasquatch());
	}

	public GlobalGroup heroes() {
		return getGroup(heroes);
	}

	public Group heroesIn(String collection) {
		return getGroup(heroes, collection);
	}

	public GlobalGroup legends() {
		return getGroup(legends);
	}

	public Group legendsIn(String collection) {
		return getGroup(legends, collection);
	}

	public GlobalGroup sidekicks() {
		return getGroup(sidekicks);
	}

	public Group sidekicksIn(String collection) {
		return getGroup(sidekicks, collection);
	}

	public GlobalGroup rumors() {
		return getGroup(rumors);
	}

	public Group rumorsIn(String collection) {
		return getGroup(rumors, collection);
	}

	public Users using(UserServices userServices) {
		this.userServices = userServices;
		return this;
	}

	public Users setUp(UserServices userServices) {
		this.userServices = userServices;
		if (userServices.getUserCredential(dakotaLindienUsername) == null) {
			addGroup(heroes, "The heroes", null);
			addGroup(legends, "The legends", null);
			addGroup(sidekicks, "The sidekicks", heroes);
			addGroup(rumors, "The rumors", legends);

			addUser(chuckNorrisUsername, "Chuck", "Norris");
			addUser(aliceUsername, "Alice", "Wonderland", "legends");
			addUser(bobGrattonUsername, "Bob 'Elvis'", "Gratton");
			addUser(charlesFrancoisXavierUsername, "Charles-François", "Xavier", "heroes");
			addUser(dakotaLindienUsername, "Dakota", "L'Indien", "heroes");
			addUser(edouardLechatUsername, "Edouard", "Lechat", "legends");
			addUser(gandalfLeblancUsername, "Gandalf", "Leblanc", "legends", "heroes");
			addUser(robinUsername, "Good Guy", "Robin", "sidekicks");
			addUser(sasquatchUsername, "Big", "Foot", "rumors");
		}
		return this;
	}

	public Users withPhotos(UserPhotosServices userPhotosServices) {
		try {
			//addUserPhoto(userPhotosServices, chuckNorrisUsername);
			//addUserPhoto(userPhotosServices, aliceUsername);
			//addUserPhoto(userPhotosServices, bobGrattonUsername);
			//addUserPhoto(userPhotosServices, charlesFrancoisXavierUsername);
			//addUserPhoto(userPhotosServices, dakotaLindienUsername);
			//addUserPhoto(userPhotosServices, edouardLechatUsername);
			//addUserPhoto(userPhotosServices, gandalfLeblancUsername);
			//addUserPhoto(userPhotosServices, sasquatchUsername);
			//addUserPhoto(userPhotosServices, robinUsername);
		} catch (Exception e) {
			e.printStackTrace();
		}
		return this;
	}

	private void addUserPhoto(UserPhotosServices userPhotosServices, String userName) {
		InputStream inputStream = Users.class.getResourceAsStream("Users-" + userName + ".jpg");
		try {
			userPhotosServices.changePhoto(inputStream, userName);
		} finally {
			IOUtils.closeQuietly(inputStream);
		}

	}

	private UserCredential getUser(String username) {
		return userServices.getUser(username);
	}

	private User getUser(String username, String collection) {
		return userServices.getUserInCollection(username, collection);
	}

	private GlobalGroup getGroup(String code) {
		return userServices.getGroup(code);
	}

	private Group getGroup(String code, String collection) {
		return userServices.getGroupInCollection(code, collection);
	}

	private void addUser(String username, String firstName, String lastName, String... groups) {
		String email = (username + "@doculibre.com").toLowerCase();
		List<String> globalGroups = Arrays.asList(groups);
		List<String> collections = new ArrayList<>();
		UserCredential credential = userServices.createUserCredential(
				username, firstName, lastName, email, globalGroups, collections, UserCredentialStatus.ACTIVE, "domain",
				Arrays.asList(""), null);
		userServices.addUpdateUserCredential(credential);
	}

	private void addGroup(String code, String title, String parent) {
		GlobalGroup group = userServices.createGlobalGroup(
				code, title, new ArrayList<String>(), parent, GlobalGroupStatus.ACTIVE, true);
		userServices.addUpdateGlobalGroup(group);
	}

	public Users withPasswords(AuthenticationService authenticationService) {
		authenticationService.changePassword(chuckNorrisUsername, "password");
		authenticationService.changePassword(aliceUsername, "password");
		authenticationService.changePassword(bobGrattonUsername, "password");
		authenticationService.changePassword(charlesFrancoisXavierUsername, "password");
		authenticationService.changePassword(dakotaLindienUsername, "password");
		authenticationService.changePassword(edouardLechatUsername, "password");
		authenticationService.changePassword(gandalfLeblancUsername, "password");
		authenticationService.changePassword(robinUsername, "password");
		authenticationService.changePassword(sasquatchUsername, "password");

		return this;
	}

}