argos/dmp-backend/core/src/main/java/eu/eudat/query/UserSettingsQuery.java

package eu.eudat.query;

import eu.eudat.authorization.AuthorizationFlags;
import eu.eudat.authorization.Permission;
import eu.eudat.commons.enums.UserSettingsType;
import eu.eudat.commons.scope.user.UserScope;
import eu.eudat.data.UserSettingsEntity;
import eu.eudat.model.UserSettings;
import eu.eudat.query.utils.QueryUtilsService;
import gr.cite.commons.web.authz.service.AuthorizationService;
import gr.cite.tools.data.query.FieldResolver;
import gr.cite.tools.data.query.QueryBase;
import gr.cite.tools.data.query.QueryContext;
import jakarta.persistence.Tuple;
import jakarta.persistence.criteria.CriteriaBuilder;
import jakarta.persistence.criteria.Predicate;
import org.springframework.beans.factory.config.ConfigurableBeanFactory;
import org.springframework.context.annotation.Scope;
import org.springframework.stereotype.Component;

import java.time.Instant;
import java.util.*;

@Component
@Scope(value = ConfigurableBeanFactory.SCOPE_PROTOTYPE)
public class UserSettingsQuery extends QueryBase<UserSettingsEntity> {

	private String like;
	private Collection<UUID> ids;
	private Collection<UserSettingsType> userSettingsTypes;
	private Collection<String> keys;
	private Collection<UserSettingsType> types;
	private Collection<UUID> entityIds;

	private EnumSet<AuthorizationFlags> authorize = EnumSet.of(AuthorizationFlags.None);

	private final UserScope userScope;
	private final AuthorizationService authService;
	private final QueryUtilsService queryUtilsService;
	public UserSettingsQuery(
			UserScope userScope,
			AuthorizationService authService, QueryUtilsService queryUtilsService
	) {
		this.userScope = userScope;
		this.authService = authService;
		this.queryUtilsService = queryUtilsService;
	}

	public UserSettingsQuery like(String like) {
		this.like = like;
		return this;
	}

	public UserSettingsQuery ids(UUID value) {
		this.ids = List.of(value);
		return this;
	}

	public UserSettingsQuery ids(UUID... value) {
		this.ids = Arrays.asList(value);
		return this;
	}

	public UserSettingsQuery ids(Collection<UUID> values) {
		this.ids = values;
		return this;
	}

	public UserSettingsQuery userSettingsTypes(UserSettingsType value) {
		this.userSettingsTypes = List.of(value);
		return this;
	}

	public UserSettingsQuery userSettingsTypes(UserSettingsType... value) {
		this.userSettingsTypes = Arrays.asList(value);
		return this;
	}

	public UserSettingsQuery userSettingsTypes(Collection<UserSettingsType> values) {
		this.userSettingsTypes = values;
		return this;
	}

	public UserSettingsQuery keys(String value) {
		this.keys = List.of(value);
		return this;
	}

	public UserSettingsQuery keys(String... value) {
		this.keys = Arrays.asList(value);
		return this;
	}

	public UserSettingsQuery keys(Collection<String> values) {
		this.keys = values;
		return this;
	}


	public UserSettingsQuery types(UserSettingsType value) {
		this.types = List.of(value);
		return this;
	}

	public UserSettingsQuery types(UserSettingsType... value) {
		this.types = Arrays.asList(value);
		return this;
	}

	public UserSettingsQuery types(Collection<UserSettingsType> values) {
		this.types = values;
		return this;
	}

	public UserSettingsQuery entityIds(UUID value) {
		this.entityIds = List.of(value);
		return this;
	}

	public UserSettingsQuery entityIds(UUID... value) {
		this.entityIds = Arrays.asList(value);
		return this;
	}

	public UserSettingsQuery entityIds(Collection<UUID> values) {
		this.entityIds = values;
		return this;
	}

	public UserSettingsQuery authorize(EnumSet<AuthorizationFlags> values) {
		this.authorize = values;
		return this;
	}

	@Override
	protected Boolean isFalseQuery() {
		return this.isEmpty(this.ids) || this.isEmpty(this.keys) || this.isEmpty(this.types) || this.isEmpty(this.entityIds);
	}

	@Override
	protected Class<UserSettingsEntity> entityClass() {
		return UserSettingsEntity.class;
	}

	@Override
	protected <X, Y> Predicate applyAuthZ(QueryContext<X, Y> queryContext) {
		if (this.authorize.contains(AuthorizationFlags.None)) return null;
		if (this.authorize.contains(AuthorizationFlags.Permission) && this.authService.authorize(Permission.BrowseUserSettings)) return null;
		UUID ownerId = null;
		if (this.authorize.contains(AuthorizationFlags.DmpAssociated)) ownerId = this.userScope.getUserIdSafe();

		List<Predicate> predicates = new ArrayList<>();
		if (ownerId != null) {
			predicates.add(queryContext.CriteriaBuilder.equal(queryContext.Root.get(UserSettingsEntity._entityId), ownerId));
		}
		if (!predicates.isEmpty()) {
			Predicate[] predicatesArray = predicates.toArray(new Predicate[0]);
			return queryContext.CriteriaBuilder.and(predicatesArray);
		} else {
			return queryContext.CriteriaBuilder.or(); //Creates a false query
		}
	}

	@Override
	protected <X, Y> Predicate applyFilters(QueryContext<X, Y> queryContext) {
		List<Predicate> predicates = new ArrayList<>();
		if (this.like != null && !this.like.isEmpty()) {
			predicates.add(queryContext.CriteriaBuilder.or(queryUtilsService.ilike(queryContext.CriteriaBuilder, queryContext.Root.get(UserSettingsEntity._key), this.like),
					queryUtilsService.ilike(queryContext.CriteriaBuilder, queryContext.Root.get(UserSettingsEntity._value), this.like)
					));
		}
		if (this.ids != null) {
			CriteriaBuilder.In<UUID> inClause = queryContext.CriteriaBuilder.in(queryContext.Root.get(UserSettingsEntity._id));
			for (UUID item : this.ids) inClause.value(item);
			predicates.add(inClause);
		}

		if (this.userSettingsTypes != null) {
			CriteriaBuilder.In<UserSettingsType> inClause = queryContext.CriteriaBuilder.in(queryContext.Root.get(UserSettingsEntity._type));
			for (UserSettingsType item : this.userSettingsTypes) inClause.value(item);
			predicates.add(inClause);
		}

		if (this.keys != null) {
			CriteriaBuilder.In<String> inClause = queryContext.CriteriaBuilder.in(queryContext.Root.get(UserSettingsEntity._key));
			for (String item : this.keys) inClause.value(item);
			predicates.add(inClause);
		}

		if (this.entityIds != null) {
			CriteriaBuilder.In<UUID> inClause = queryContext.CriteriaBuilder.in(queryContext.Root.get(UserSettingsEntity._entityId));
			for (UUID item : this.entityIds) inClause.value(item);
			predicates.add(inClause);
		}

		if (this.types != null) {
			CriteriaBuilder.In<UserSettingsType> inClause = queryContext.CriteriaBuilder.in(queryContext.Root.get(UserSettingsEntity._type));
			for (UserSettingsType item : this.types) inClause.value(item);
			predicates.add(inClause);
		}


		if (!predicates.isEmpty()) {
			Predicate[] predicatesArray = predicates.toArray(new Predicate[0]);
			return queryContext.CriteriaBuilder.and(predicatesArray);
		} else {
			return null;
		}
	}

	@Override
	protected String fieldNameOf(FieldResolver item) {
		if (item.match(UserSettings._id)) return UserSettingsEntity._id;
		else if (item.match(UserSettings._entityId)) return UserSettingsEntity._entityId;
		else if (item.match(UserSettings._value)) return UserSettingsEntity._value;
		else if (item.match(UserSettings._key)) return UserSettingsEntity._key;
		else if (item.match(UserSettings._type)) return UserSettingsEntity._type;
		else if (item.match(UserSettings._createdAt)) return UserSettingsEntity._createdAt;
		else if (item.match(UserSettings._updatedAt)) return UserSettingsEntity._updatedAt;
		else if (item.match(UserSettings._hash)) return UserSettingsEntity._updatedAt;
		else if (item.match(UserSettings._belongsToCurrentTenant)) return UserSettingsEntity._tenantId;
		else return null;
	}

	@Override
	protected UserSettingsEntity convert(Tuple tuple, Set<String> columns) {
		UserSettingsEntity item = new UserSettingsEntity();
		item.setId(QueryBase.convertSafe(tuple, columns, UserSettingsEntity._id, UUID.class));
		item.setTenantId(QueryBase.convertSafe(tuple, columns, UserSettingsEntity._tenantId, UUID.class));
		item.setEntityId(QueryBase.convertSafe(tuple, columns, UserSettingsEntity._entityId, UUID.class));
		item.setKey(QueryBase.convertSafe(tuple, columns, UserSettingsEntity._key, String.class));
		item.setValue(QueryBase.convertSafe(tuple, columns, UserSettingsEntity._value, String.class));
		item.setType(QueryBase.convertSafe(tuple, columns, UserSettingsEntity._type, UserSettingsType.class));
		item.setCreatedAt(QueryBase.convertSafe(tuple, columns, UserSettingsEntity._createdAt, Instant.class));
		item.setUpdatedAt(QueryBase.convertSafe(tuple, columns, UserSettingsEntity._updatedAt, Instant.class));
		return item;
	}
}