NCP-3 Update entity like helper.php

src/main/java/ru/soune/no_copy/entity/UserContent.java

@@ -3,52 +3,51 @@ package ru.soune.no_copy.entity;
 import jakarta.persistence.*;
 import lombok.*;
 import org.hibernate.annotations.CreationTimestamp;
+import org.springframework.data.annotation.CreatedDate;
+import org.springframework.data.jpa.domain.support.AuditingEntityListener;
 
 import java.time.LocalDateTime;
 
 @Entity
+@Table(name = "user_content")
+@Getter
+@Setter
 @NoArgsConstructor
 @AllArgsConstructor
-@Builder
-@Getter @Setter
-@Table(name = "user_content",
-indexes = {
-        @Index(name = "idx_user_id", columnList = "user_id"),
-        @Index(name = "idx_file_type", columnList = "file_type"),
-        @Index(name = "idx_upload_date", columnList = "upload_date")
-})
+@ToString
+@EntityListeners(AuditingEntityListener.class)
 public class UserContent {
 
     @Id
     @GeneratedValue(strategy = GenerationType.IDENTITY)
-    private Long id;
+    private Long contentId;
 
     @ManyToOne(fetch = FetchType.LAZY)
-    @JoinColumn(name = "user_id", nullable = false,
-            foreignKey = @ForeignKey(name = "fk_usercontent_user"))
+    @JoinColumn(name = "user_id", nullable = false)
+    @ToString.Exclude
     private User user;
 
-    @Column(nullable = false)
+    @Column(name = "filename", nullable = false, length = 255)
     private String filename;
 
-    @Column(name = "original_filename", nullable = false)
+    @Column(name = "original_filename", nullable = false, length = 255)
     private String originalFilename;
 
     @Enumerated(EnumType.STRING)
-    @Column(name = "file_type", nullable = false, length = 16)
+    @Column(name = "file_type", nullable = false, length = 20)
     private FileType fileType;
 
-    @Column(name = "file_extension", length = 16)
+    @Column(name = "file_extension", nullable = false, length = 10)
     private String fileExtension;
 
     @Column(name = "file_size", nullable = false)
     private Long fileSize;
 
-    @Column(name = "file_path", columnDefinition = "TEXT", nullable = false)
+    @Column(name = "file_path", nullable = false, columnDefinition = "TEXT")
     private String filePath;
 
     @Column(name = "protection_level")
-    private Integer protectionLevel;
+    private Integer protectionLevel = 0;
 
     @Column(name = "geometric_resistant")
     private Boolean geometricResistant = false;
@@ -63,11 +62,19 @@ public class UserContent {
     private String watermarkId;
 
     @Enumerated(EnumType.STRING)
-    @Column(length = 16)
+    @Column(name = "status", nullable = false, length = 20)
     private ContentStatus status = ContentStatus.PROCESSING;
 
-    @CreationTimestamp
+    @CreatedDate
     @Column(name = "upload_date", nullable = false, updatable = false)
     private LocalDateTime uploadDate;
+
+    public Double getFileSizeInGB() {
+        return fileSize != null ? fileSize / (1024.0 * 1024.0 * 1024.0) : 0.0;
+    }
+
+    public Boolean isProtected() {
+        return protectionLevel != null && protectionLevel > 0;
+    }
 }