@@ -6,10 +6,13 @@ import lombok.*;
|
||||
import org.springframework.data.annotation.CreatedDate;
|
||||
import org.springframework.data.annotation.LastModifiedDate;
|
||||
import org.springframework.data.jpa.domain.support.AuditingEntityListener;
|
||||
import ru.soune.nocopy.entity.file.permission.PermissionType;
|
||||
import ru.soune.nocopy.entity.monitoring.FileMonitoringEntity;
|
||||
|
||||
import java.time.LocalDateTime;
|
||||
import java.util.HashMap;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.concurrent.atomic.AtomicInteger;
|
||||
|
||||
@Data
|
||||
@@ -92,6 +95,14 @@ public class FileEntity {
|
||||
@Column
|
||||
private String newExtension;
|
||||
|
||||
@ElementCollection
|
||||
@CollectionTable(name = "file_permissions",
|
||||
joinColumns = @JoinColumn(name = "file_id"))
|
||||
@Column(name = "allowed")
|
||||
@MapKeyEnumerated(EnumType.STRING)
|
||||
@MapKeyColumn(name = "permission_type")
|
||||
private Map<PermissionType, Boolean> permissions = new HashMap<>();
|
||||
|
||||
@OneToOne(mappedBy = "file", cascade = CascadeType.ALL, orphanRemoval = true)
|
||||
@JsonIgnore
|
||||
private ImageHashEntity imageHash;
|
||||
@@ -113,6 +124,15 @@ public class FileEntity {
|
||||
if (this.createdAt == null) {
|
||||
this.createdAt = LocalDateTime.now();
|
||||
}
|
||||
|
||||
if (this.permissions == null) {
|
||||
this.permissions = new HashMap<>();
|
||||
this.permissions.put(PermissionType.DOWNLOAD, true);
|
||||
// this.permissions.put(PermissionType.VIEW, true);
|
||||
// this.permissions.put(PermissionType.EDIT, true);
|
||||
}
|
||||
|
||||
|
||||
this.updatedAt = LocalDateTime.now();
|
||||
}
|
||||
|
||||
@@ -124,4 +144,29 @@ public class FileEntity {
|
||||
public static void initializeCounter(int maxId) {
|
||||
supportIdCounter = new AtomicInteger(maxId);
|
||||
}
|
||||
|
||||
@Transient
|
||||
public boolean hasPermission(PermissionType type) {
|
||||
return permissions.getOrDefault(type, false);
|
||||
}
|
||||
|
||||
@Transient
|
||||
public void setPermission(PermissionType type, boolean allowed) {
|
||||
permissions.put(type, allowed);
|
||||
}
|
||||
|
||||
@Transient
|
||||
public boolean canDownload() {
|
||||
return hasPermission(PermissionType.DOWNLOAD);
|
||||
}
|
||||
|
||||
// @Transient
|
||||
// public boolean canView() {
|
||||
// return hasPermission(PermissionType.VIEW);
|
||||
// }
|
||||
//
|
||||
// @Transient
|
||||
// public boolean canEdit() {
|
||||
// return hasPermission(PermissionType.EDIT);
|
||||
// }
|
||||
}
|
||||
|
||||
@@ -0,0 +1,27 @@
|
||||
package ru.soune.nocopy.entity.file.permission;
|
||||
|
||||
import java.util.Arrays;
|
||||
import java.util.Map;
|
||||
import java.util.Set;
|
||||
import java.util.stream.Collectors;
|
||||
|
||||
public enum PermissionType {
|
||||
DOWNLOAD;
|
||||
|
||||
public static Set<String> getValidPermissionNames() {
|
||||
return Arrays.stream(PermissionType.values())
|
||||
.map(Enum::name)
|
||||
.collect(Collectors.toSet());
|
||||
}
|
||||
|
||||
public static Set<String> getInvalidPermissions(Map<String, Boolean> permissions) {
|
||||
Set<String> validNames = getValidPermissionNames();
|
||||
return permissions.keySet().stream()
|
||||
.filter(key -> !validNames.contains(key))
|
||||
.collect(Collectors.toSet());
|
||||
}
|
||||
|
||||
public static boolean areAllValid(Map<String, Boolean> permissions) {
|
||||
return getInvalidPermissions(permissions).isEmpty();
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user