170 lines
8.8 KiB
Java
170 lines
8.8 KiB
Java
package eu.eudat.logic.managers;
|
|
|
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
|
import eu.eudat.data.dao.criteria.DataManagementPlanCriteria;
|
|
import eu.eudat.data.dao.entities.UserInfoDao;
|
|
import eu.eudat.data.entities.DMP;
|
|
import eu.eudat.data.entities.UserInfo;
|
|
import eu.eudat.data.entities.UserRole;
|
|
import eu.eudat.data.query.items.table.userinfo.UserInfoTableRequestItem;
|
|
import eu.eudat.exceptions.security.ExpiredTokenException;
|
|
import eu.eudat.exceptions.security.NonValidTokenException;
|
|
import eu.eudat.exceptions.security.NullEmailException;
|
|
import eu.eudat.exceptions.security.UnauthorisedException;
|
|
import eu.eudat.logic.builders.entity.UserRoleBuilder;
|
|
import eu.eudat.logic.builders.model.models.DataTableDataBuilder;
|
|
import eu.eudat.logic.security.customproviders.Zenodo.ZenodoAccessType;
|
|
import eu.eudat.logic.security.customproviders.Zenodo.ZenodoCustomProvider;
|
|
import eu.eudat.logic.security.validators.zenodo.helpers.ZenodoResponseToken;
|
|
import eu.eudat.logic.services.ApiContext;
|
|
import eu.eudat.logic.services.operations.authentication.AuthenticationService;
|
|
import eu.eudat.logic.utilities.builders.XmlBuilder;
|
|
import eu.eudat.models.HintedModelFactory;
|
|
import eu.eudat.models.data.dmp.DataManagementPlan;
|
|
import eu.eudat.models.data.doi.DOIRequest;
|
|
import eu.eudat.models.data.helpers.common.DataTableData;
|
|
import eu.eudat.models.data.login.Credentials;
|
|
import eu.eudat.models.data.security.Principal;
|
|
import eu.eudat.models.data.userinfo.UserListingModel;
|
|
import eu.eudat.models.data.userinfo.UserProfile;
|
|
import eu.eudat.queryable.QueryableList;
|
|
import org.json.JSONObject;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.core.env.Environment;
|
|
import org.springframework.stereotype.Component;
|
|
import org.w3c.dom.Document;
|
|
import org.w3c.dom.Element;
|
|
|
|
import java.io.IOException;
|
|
import java.time.Instant;
|
|
import java.util.*;
|
|
import java.util.stream.Collectors;
|
|
|
|
@Component
|
|
public class UserManager {
|
|
|
|
private ApiContext apiContext;
|
|
private ZenodoCustomProvider zenodoCustomProvider;
|
|
private Environment environment;
|
|
|
|
@Autowired
|
|
public UserManager(ApiContext apiContext, ZenodoCustomProvider zenodoCustomProvider, Environment environment) {
|
|
this.apiContext = apiContext;
|
|
this.zenodoCustomProvider = zenodoCustomProvider;
|
|
this.environment = environment;
|
|
}
|
|
|
|
public eu.eudat.models.data.user.composite.DatasetProfile generateDatasetProfileModel(eu.eudat.data.entities.DatasetProfile profile) {
|
|
Document viewStyleDoc = XmlBuilder.fromXml(profile.getDefinition());
|
|
Element root = (Element) viewStyleDoc.getDocumentElement();
|
|
eu.eudat.models.data.entities.xmlmodels.datasetprofiledefinition.ViewStyleModel viewstyle = new eu.eudat.models.data.entities.xmlmodels.datasetprofiledefinition.ViewStyleModel().fromXml(root);
|
|
|
|
eu.eudat.models.data.user.composite.DatasetProfile datasetprofile = new eu.eudat.models.data.user.composite.DatasetProfile();
|
|
datasetprofile.buildProfile(viewstyle);
|
|
|
|
return datasetprofile;
|
|
}
|
|
|
|
public DataTableData<UserListingModel> getPaged(UserInfoTableRequestItem userInfoTableRequestItem) throws Exception {
|
|
QueryableList<eu.eudat.data.entities.UserInfo> users = apiContext.getOperationsContext().getDatabaseRepository().getUserInfoDao().getWithCriteria(userInfoTableRequestItem.getCriteria()).withHint(HintedModelFactory.getHint(UserListingModel.class));
|
|
QueryableList<eu.eudat.data.entities.UserInfo> pagedUsers = PaginationManager.applyPaging(users, userInfoTableRequestItem);
|
|
|
|
List<UserListingModel> modelUsers = pagedUsers.select(item -> new UserListingModel().fromDataModel(item));
|
|
return apiContext.getOperationsContext().getBuilderFactory().getBuilder(DataTableDataBuilder.class).totalCount(users.count()).data(modelUsers).build();
|
|
}
|
|
|
|
public UserProfile getSingle(UUID userId) throws Exception {
|
|
eu.eudat.data.entities.UserInfo user = apiContext.getOperationsContext().getDatabaseRepository().getUserInfoDao().find(userId);
|
|
UserProfile profile = new UserProfile().fromDataModel(user);
|
|
List<Integer> roles = new LinkedList<>();
|
|
DataManagementPlanCriteria criteria = new DataManagementPlanCriteria();
|
|
criteria.setAllVersions(false);
|
|
QueryableList<DMP> items = apiContext.getOperationsContext().getDatabaseRepository().getDmpDao().getWithCriteria(criteria);
|
|
List<DMP> dmps = apiContext.getOperationsContext().getDatabaseRepository().getDmpDao().getAuthenticated(items, userId, roles).take(5).toList();
|
|
profile.setAssociatedDmps(dmps.stream().map(x -> new DataManagementPlan().fromDataModel(x)).collect(Collectors.toList()));
|
|
return profile;
|
|
}
|
|
|
|
public void editRoles(UserListingModel user) {
|
|
eu.eudat.data.entities.UserInfo userInfo = apiContext.getOperationsContext().getDatabaseRepository().getUserInfoDao().find(user.getId());
|
|
userInfo.getUserRoles().stream().forEach(item -> apiContext.getOperationsContext().getDatabaseRepository().getUserRoleDao().delete(item));
|
|
for (Integer role : user.getAppRoles()) {
|
|
UserRole userRole = apiContext.getOperationsContext().getBuilderFactory().getBuilder(UserRoleBuilder.class).role(role).userInfo(userInfo).build();
|
|
apiContext.getOperationsContext().getDatabaseRepository().getUserRoleDao().createOrUpdate(userRole);
|
|
}
|
|
}
|
|
|
|
public void updateSettings(Map<String, Object> settings, Principal principal) throws IOException {
|
|
eu.eudat.data.entities.UserInfo userInfo = apiContext.getOperationsContext().getDatabaseRepository().getUserInfoDao().find(principal.getId());
|
|
apiContext.getOperationsContext().getDatabaseRepository().detachEntity(userInfo);
|
|
HashMap<String, Object> result =
|
|
new ObjectMapper().readValue(userInfo.getAdditionalinfo(), HashMap.class);
|
|
result.putAll(settings);
|
|
userInfo.setAdditionalinfo(new JSONObject(result).toString());
|
|
apiContext.getOperationsContext().getDatabaseRepository().getUserInfoDao()
|
|
.createOrUpdate(userInfo);
|
|
}
|
|
|
|
public Principal authenticate(AuthenticationService authenticationServiceImpl, Credentials credentials) throws NullEmailException {
|
|
Principal principal = authenticationServiceImpl.Touch(credentials);
|
|
if (principal == null) throw new UnauthorisedException("Could not Sign In User");
|
|
return principal;
|
|
}
|
|
|
|
public DataTableData<UserListingModel> getCollaboratorsPaged(UserInfoTableRequestItem userInfoTableRequestItem, Principal principal) throws Exception {
|
|
UserInfoDao userInfoDao = apiContext.getOperationsContext().getDatabaseRepository().getUserInfoDao();
|
|
QueryableList<UserInfo> users = userInfoDao.getWithCriteria(userInfoTableRequestItem.getCriteria());
|
|
|
|
List<UserListingModel> colaborators = userInfoDao.getAuthenticated(users, principal.getId())
|
|
.withHint(HintedModelFactory.getHint(UserListingModel.class))
|
|
.select(colaborator -> new UserListingModel().fromDataModel(colaborator));
|
|
|
|
DataTableData<UserListingModel> dataTableData = new DataTableData<>();
|
|
dataTableData.setData(colaborators);
|
|
dataTableData.setTotalCount((long) colaborators.size());
|
|
return dataTableData;
|
|
}
|
|
|
|
public Boolean isDOITokenValid(Principal principal) throws NonValidTokenException, ExpiredTokenException, IOException {
|
|
if (principal.getZenodoToken() != null && !principal.getZenodoToken().isEmpty()) {
|
|
if (Instant.now().isBefore(principal.getZenodoDuration())) {
|
|
return true;
|
|
}
|
|
try {
|
|
this.updateDOIToken(ZenodoAccessType.REFRESH_TOKEN, principal.getZenodoRefresh(), this.environment.getProperty("zenodo.login.redirect_uri"), principal);
|
|
return true;
|
|
}catch (IOException e) {
|
|
this.deleteDOIToken(principal);
|
|
throw new ExpiredTokenException("Zenodo Token is expired.");
|
|
}
|
|
}
|
|
throw new NonValidTokenException("This account has no Zenodo Token");
|
|
}
|
|
|
|
public void registerDOIToken(DOIRequest doiRequest, Principal principal) throws IOException {
|
|
this.updateDOIToken(ZenodoAccessType.AUTHORIZATION_CODE, doiRequest.getZenodoRequest().getCode(), doiRequest.getRedirectUri(), principal);
|
|
}
|
|
|
|
private void updateDOIToken(ZenodoAccessType accessType, String code, String redirectUri, Principal principal) throws IOException {
|
|
ZenodoResponseToken responseToken = this.zenodoCustomProvider.getAccessToken(accessType, code
|
|
, this.environment.getProperty("zenodo.login.client_id")
|
|
, this.environment.getProperty("zenodo.login.client_secret")
|
|
, redirectUri);
|
|
Map<String, Object> settings = new HashMap<>();
|
|
settings.put("zenodoEmail", responseToken.getEmail());
|
|
settings.put("zenodoRefresh", responseToken.getRefreshToken());
|
|
settings.put("zenodoToken", responseToken.getAccessToken());
|
|
settings.put("expirationDate", Instant.now().plusSeconds(responseToken.getExpiresIn()).toEpochMilli());
|
|
this.updateSettings(settings, principal);
|
|
}
|
|
|
|
public void deleteDOIToken(Principal principal) throws IOException {
|
|
Map<String, Object> settings = new HashMap<>();
|
|
settings.put("zenodoEmail", "");
|
|
settings.put("zenodoRefresh", "");
|
|
settings.put("zenodoToken", "");
|
|
settings.put("expirationDate", 0);
|
|
this.updateSettings(settings, principal);
|
|
}
|
|
}
|