fix after test
Test Workflow / test (push) Successful in 3s

This commit is contained in:
vladp
2025-12-24 14:20:49 +07:00
parent f6d4ebfe0e
commit c5ef58ddd4
2 changed files with 38 additions and 0 deletions
@@ -0,0 +1,17 @@
package ru.soune.nocopy.dto.file;
import com.fasterxml.jackson.annotation.JsonProperty;
import lombok.Builder;
import lombok.Data;
import java.util.List;
@Data
@Builder
public class FileExtensionResponse {
@JsonProperty("file_extension")
private List<String> extension;
@JsonProperty("count")
private Integer count;
}
@@ -44,6 +44,8 @@ public class FileUploadHandler implements RequestHandler {
return handleInitUpload(request, fileRequest); return handleInitUpload(request, fileRequest);
case "file_types": case "file_types":
return handleGetFileTypes(request); return handleGetFileTypes(request);
case "file_extension":
return handleGetFileExtensions(fileRequest, request);
case "chunks": case "chunks":
return handleGetChunkStatus(request, fileRequest); return handleGetChunkStatus(request, fileRequest);
case "cancel": case "cancel":
@@ -243,4 +245,23 @@ public class FileUploadHandler implements RequestHandler {
"Failed to get file types", null); "Failed to get file types", null);
} }
} }
private BaseResponse handleGetFileExtensions(FileUploadRequest fileRequest, BaseRequest request) {
try {
FileType fileType = FileType.valueOf(fileRequest.getFileType().toUpperCase());
List<String> allowedExtensions = fileType.getAllowedExtensions();
FileExtensionResponse response = FileExtensionResponse.builder()
.extension(allowedExtensions)
.count(allowedExtensions.size())
.build();
return new BaseResponse(request.getMsgId(), MessageCode.SUCCESS.getCode(),
MessageCode.SUCCESS.getDescription(), response);
} catch (Exception e) {
log.error("Error getting file extensions", e);
return new BaseResponse(request.getMsgId(), MessageCode.FILE_UPLOAD_ERROR.getCode(),
"Failed to get file extensions", null);
}
}
} }