CoreMigrationTo_6_0.java

package com.constellio.app.services.migrations.scripts;

import static com.constellio.data.conf.DigitSeparatorMode.THREE_LEVELS_OF_ONE_DIGITS;
import static com.constellio.data.conf.DigitSeparatorMode.TWO_DIGITS;
import static com.constellio.data.conf.HashingEncoding.BASE64;
import static com.constellio.data.conf.HashingEncoding.BASE64_URL_ENCODED;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;

import org.joda.time.LocalDateTime;

import com.constellio.app.entities.modules.MetadataSchemasAlterationHelper;
import com.constellio.app.entities.modules.MigrationResourcesProvider;
import com.constellio.app.entities.modules.MigrationScript;
import com.constellio.app.services.factories.AppLayerFactory;
import com.constellio.data.dao.services.factories.DataLayerFactory;
import com.constellio.model.entities.records.calculators.UserTitleCalculator;
import com.constellio.model.entities.records.wrappers.Collection;
import com.constellio.model.entities.records.wrappers.Event;
import com.constellio.model.entities.records.wrappers.Group;
import com.constellio.model.entities.records.wrappers.User;
import com.constellio.model.entities.records.wrappers.WorkflowTask;
import com.constellio.model.entities.schemas.MetadataValueType;
import com.constellio.model.entities.security.global.GlobalGroupStatus;
import com.constellio.model.entities.security.global.SolrGlobalGroup;
import com.constellio.model.entities.security.global.SolrUserCredential;
import com.constellio.model.entities.security.global.UserCredential;
import com.constellio.model.entities.security.global.UserCredentialStatus;
import com.constellio.model.services.factories.ModelLayerFactory;
import com.constellio.model.services.records.SchemasRecordsServices;
import com.constellio.model.services.records.cache.CacheConfig;
import com.constellio.model.services.records.cache.RecordsCache;
import com.constellio.model.services.schemas.MetadataSchemasManager;
import com.constellio.model.services.schemas.MetadataSchemasManagerException.OptimisticLocking;
import com.constellio.model.services.schemas.builders.CommonMetadataBuilder;
import com.constellio.model.services.schemas.builders.MetadataSchemaBuilder;
import com.constellio.model.services.schemas.builders.MetadataSchemaTypeBuilder;
import com.constellio.model.services.schemas.builders.MetadataSchemaTypesBuilder;
import com.constellio.model.services.schemas.validators.EmailValidator;
import com.constellio.model.services.security.authentification.AuthenticationService;
import com.constellio.model.services.users.UserCredentialAndGlobalGroupsMigration;
import com.constellio.model.services.users.UserServices;
import com.constellio.model.services.users.UserServicesRuntimeException.UserServicesRuntimeException_NoSuchUser;

public class CoreMigrationTo_6_0 implements MigrationScript {
	@Override
	public String getVersion() {
		return "6.0";
	}

	@Override
	public void migrate(String collection, MigrationResourcesProvider provider, AppLayerFactory appLayerFactory)
			throws Exception {

		if (Collection.SYSTEM_COLLECTION.equals(collection)) {
			ModelLayerFactory modelLayerFactory = appLayerFactory.getModelLayerFactory();

			if (!modelLayerFactory.getMetadataSchemasManager().getSchemaTypes(collection)
					.hasType(SolrUserCredential.SCHEMA_TYPE)) {

				new CoreSchemaAlterationFor6_0(collection, provider, appLayerFactory).migrate();

				SchemasRecordsServices schemas = new SchemasRecordsServices(Collection.SYSTEM_COLLECTION, modelLayerFactory);
				RecordsCache cache = modelLayerFactory.getRecordsCaches().getCache(Collection.SYSTEM_COLLECTION);

				cache.configureCache(CacheConfig.permanentCache(schemas.credentialSchemaType()));
				cache.configureCache(CacheConfig.permanentCache(schemas.globalGroupSchemaType()));
			}

			UserCredentialAndGlobalGroupsMigration migration = new UserCredentialAndGlobalGroupsMigration(modelLayerFactory);
			if (migration.isMigrationRequired()) {
				migration.migrateUserAndGroups();
			}
			try {
				modelLayerFactory.newUserServices().getUser("admin");
			} catch (UserServicesRuntimeException_NoSuchUser e) {
				createAdminUser(modelLayerFactory);
			}

		}
	}

	public void createAdminUser(ModelLayerFactory modelLayerFactory) {
		DataLayerFactory dataLayerFactory = modelLayerFactory.getDataLayerFactory();
		//String serviceKey = "adminkey";
		String password = "password";
		String username = "admin";
		String firstName = "System";
		String lastName = "Admin";
		String email = "admin@organization.com";
		UserCredentialStatus status = UserCredentialStatus.ACTIVE;
		String domain = "";
		List<String> globalGroups = new ArrayList<>();
		List<String> collections = new ArrayList<>();
		boolean isSystemAdmin = true;

		UserServices userServices = modelLayerFactory.newUserServices();
		UserCredential adminCredentials = userServices.createUserCredential(
				username, firstName, lastName, email, null, isSystemAdmin, globalGroups, collections,
				new HashMap<String, LocalDateTime>(), status, domain, Arrays.asList(""), null);
		userServices.addUpdateUserCredential(adminCredentials);
		AuthenticationService authenticationService = modelLayerFactory.newAuthenticationService();
		if (authenticationService.supportPasswordChange()) {
			authenticationService.changePassword("admin", password);
		}

		if (modelLayerFactory.getCollectionsListManager().getCollections().size() == 1) {
			dataLayerFactory.getDataLayerConfiguration().setHashingEncoding(BASE64_URL_ENCODED);
			dataLayerFactory.getDataLayerConfiguration().setContentDaoFileSystemDigitsSeparatorMode(THREE_LEVELS_OF_ONE_DIGITS);
		} else {
			dataLayerFactory.getDataLayerConfiguration().setHashingEncoding(BASE64);
			dataLayerFactory.getDataLayerConfiguration().setContentDaoFileSystemDigitsSeparatorMode(TWO_DIGITS);
		}
	}

	private class CoreSchemaAlterationFor6_0 extends MetadataSchemasAlterationHelper {
		public CoreSchemaAlterationFor6_0(String collection, MigrationResourcesProvider migrationResourcesProvider,
				AppLayerFactory appLayerFactory) {
			super(collection, migrationResourcesProvider, appLayerFactory);
		}

		@Override
		protected void migrate(MetadataSchemaTypesBuilder builder) {
			MetadataSchemasManager manager = modelLayerFactory.getMetadataSchemasManager();
			createUserCredentialSchema(builder);
			createGlobalGroupSchema(builder);
		}

		private void createUserCredentialSchema(MetadataSchemaTypesBuilder builder) {
			MetadataSchemaTypeBuilder credentialsTypeBuilder = builder.createNewSchemaType(SolrUserCredential.SCHEMA_TYPE);
			credentialsTypeBuilder.setSecurity(false);
			MetadataSchemaBuilder credentials = credentialsTypeBuilder.getDefaultSchema();

			credentials.getMetadata(CommonMetadataBuilder.TITLE).defineDataEntry().asCalculated(UserTitleCalculator.class);

			credentials.createUndeletable(SolrUserCredential.USERNAME).setType(MetadataValueType.STRING)
					.setDefaultRequirement(true).setUniqueValue(true).setUnmodifiable(true);
			credentials.createUndeletable(SolrUserCredential.FIRST_NAME).setType(MetadataValueType.STRING);
			credentials.createUndeletable(SolrUserCredential.LAST_NAME).setType(MetadataValueType.STRING);
			credentials.createUndeletable(SolrUserCredential.EMAIL).setType(MetadataValueType.STRING)
					.setUniqueValue(false).addValidator(EmailValidator.class);
			credentials.createUndeletable(SolrUserCredential.PERSONAL_EMAILS).setType(MetadataValueType.STRING)
					.setMultivalue(true);
			credentials.createUndeletable(SolrUserCredential.SERVICE_KEY).setType(MetadataValueType.STRING).setEncrypted(true);
			credentials.createUndeletable(SolrUserCredential.TOKEN_KEYS).setType(MetadataValueType.STRING).setMultivalue(true)
					.setEncrypted(true);
			credentials.createUndeletable(SolrUserCredential.TOKEN_EXPIRATIONS).setType(MetadataValueType.DATE_TIME)
					.setMultivalue(true);
			credentials.createUndeletable(SolrUserCredential.SYSTEM_ADMIN).setType(MetadataValueType.BOOLEAN)
					.setDefaultRequirement(true).setDefaultValue(false);
			credentials.createUndeletable(SolrUserCredential.COLLECTIONS).setType(MetadataValueType.STRING).setMultivalue(true);
			credentials.createUndeletable(SolrUserCredential.GLOBAL_GROUPS).setType(MetadataValueType.STRING).setMultivalue(true);
			credentials.createUndeletable(SolrUserCredential.STATUS).defineAsEnum(UserCredentialStatus.class)
					.setDefaultRequirement(true);
			credentials.createUndeletable(SolrUserCredential.DOMAIN).setType(MetadataValueType.STRING);
			credentials.createUndeletable(SolrUserCredential.MS_EXCHANGE_DELEGATE_LIST).setType(MetadataValueType.STRING)
					.setMultivalue(true);
			credentials.createUndeletable(SolrUserCredential.DN).setType(MetadataValueType.STRING);
		}

		private void createGlobalGroupSchema(MetadataSchemaTypesBuilder builder) {
			MetadataSchemaTypeBuilder credentialsTypeBuilder = builder.createNewSchemaType(SolrGlobalGroup.SCHEMA_TYPE);
			credentialsTypeBuilder.setSecurity(false);
			MetadataSchemaBuilder groups = credentialsTypeBuilder.getDefaultSchema();

			groups.createUniqueCodeMetadata();
			groups.createUndeletable(SolrGlobalGroup.NAME).setType(MetadataValueType.STRING).setDefaultRequirement(true);
			groups.createUndeletable(SolrGlobalGroup.COLLECTIONS).setType(MetadataValueType.STRING).setMultivalue(true);
			groups.createUndeletable(SolrGlobalGroup.PARENT).setType(MetadataValueType.STRING);
			groups.createUndeletable(SolrGlobalGroup.STATUS).defineAsEnum(GlobalGroupStatus.class).setDefaultRequirement(true);
			groups.createUndeletable(SolrGlobalGroup.HIERARCHY).setType(MetadataValueType.STRING);
			groups.createUndeletable(SolrGlobalGroup.LOCALLY_CREATED).setType(MetadataValueType.BOOLEAN);
		}
	}
}