Compare commits
1
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
f4e5ea2976 |
@@ -13,13 +13,6 @@ java {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
//dependencyManagement {
|
|
||||||
// imports {
|
|
||||||
// mavenBom "org.jetbrains.kotlinx:kotlinx-coroutines-bom:1.6.4"
|
|
||||||
// }
|
|
||||||
//}
|
|
||||||
|
|
||||||
|
|
||||||
configurations {
|
configurations {
|
||||||
compileOnly {
|
compileOnly {
|
||||||
extendsFrom annotationProcessor
|
extendsFrom annotationProcessor
|
||||||
@@ -28,9 +21,6 @@ configurations {
|
|||||||
|
|
||||||
repositories {
|
repositories {
|
||||||
mavenCentral()
|
mavenCentral()
|
||||||
flatDir {
|
|
||||||
dirs 'libs'
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
dependencies {
|
dependencies {
|
||||||
@@ -55,11 +45,6 @@ dependencies {
|
|||||||
testImplementation 'org.springframework.boot:spring-boot-starter-test'
|
testImplementation 'org.springframework.boot:spring-boot-starter-test'
|
||||||
testImplementation 'org.mockito:mockito-core:5.3.1'
|
testImplementation 'org.mockito:mockito-core:5.3.1'
|
||||||
testRuntimeOnly 'org.junit.platform:junit-platform-launcher'
|
testRuntimeOnly 'org.junit.platform:junit-platform-launcher'
|
||||||
|
|
||||||
// implementation 'org.jetbrains.kotlin:kotlin-stdlib:1.8.21'
|
|
||||||
// implementation 'org.jetbrains.kotlinx:kotlinx-coroutines-core:1.6.4'
|
|
||||||
// implementation 'io.insert-koin:koin-core:3.5.0'
|
|
||||||
implementation name: 'testlib-fat-0.1.2-all (1)'
|
|
||||||
}
|
}
|
||||||
|
|
||||||
tasks.named('test') {
|
tasks.named('test') {
|
||||||
|
|||||||
+3
-3
@@ -120,7 +120,7 @@ services:
|
|||||||
prometheus:
|
prometheus:
|
||||||
image: prom/prometheus:latest
|
image: prom/prometheus:latest
|
||||||
container_name: prometheus
|
container_name: prometheus
|
||||||
deploy:
|
deploy: # ← ДОБАВЛЕНО
|
||||||
resources:
|
resources:
|
||||||
limits:
|
limits:
|
||||||
cpus: '0.5'
|
cpus: '0.5'
|
||||||
@@ -148,7 +148,7 @@ services:
|
|||||||
loki:
|
loki:
|
||||||
image: grafana/loki:2.9.2
|
image: grafana/loki:2.9.2
|
||||||
container_name: loki
|
container_name: loki
|
||||||
deploy:
|
deploy: # ← ДОБАВЛЕНО
|
||||||
resources:
|
resources:
|
||||||
limits:
|
limits:
|
||||||
cpus: '1.0'
|
cpus: '1.0'
|
||||||
@@ -173,7 +173,7 @@ services:
|
|||||||
tempo:
|
tempo:
|
||||||
image: grafana/tempo:2.4.1
|
image: grafana/tempo:2.4.1
|
||||||
container_name: tempo
|
container_name: tempo
|
||||||
deploy:
|
deploy: # ← ДОБАВЛЕНО
|
||||||
resources:
|
resources:
|
||||||
limits:
|
limits:
|
||||||
cpus: '0.5'
|
cpus: '0.5'
|
||||||
|
|||||||
@@ -7,6 +7,7 @@ pipeline {
|
|||||||
defaultValue: 'dev',
|
defaultValue: 'dev',
|
||||||
description: 'Ветка для деплоя'
|
description: 'Ветка для деплоя'
|
||||||
)
|
)
|
||||||
|
|
||||||
string(
|
string(
|
||||||
name: 'SERVER',
|
name: 'SERVER',
|
||||||
defaultValue: '92.242.61.23',
|
defaultValue: '92.242.61.23',
|
||||||
@@ -50,57 +51,156 @@ pipeline {
|
|||||||
sshpass -p '$SSH_PASS' ssh -o StrictHostKeyChecking=no $SSH_USER@$SERVER "
|
sshpass -p '$SSH_PASS' ssh -o StrictHostKeyChecking=no $SSH_USER@$SERVER "
|
||||||
cd /opt/deployments/${params.BRANCH}
|
cd /opt/deployments/${params.BRANCH}
|
||||||
|
|
||||||
echo '1. Остановка старого приложения...'
|
echo '========================================'
|
||||||
docker stop app-backend 2>/dev/null || echo 'Контейнер не найден'
|
echo 'Step 1: Checking current state...'
|
||||||
docker rm app-backend 2>/dev/null || echo 'Контейнер не найден'
|
echo '========================================'
|
||||||
|
|
||||||
echo '2. Удаление старых образов...'
|
docker ps --filter 'name=postgres|app-backend|storage|grafana|prometheus|loki|tempo|alloy' --format 'table {{.Names}}\\t{{.Image}}\\t{{.Status}}'
|
||||||
docker images --filter 'reference=*app*' -q | xargs -r docker rmi -f 2>/dev/null || echo 'Нет образов для удаления'
|
|
||||||
|
|
||||||
echo '3. Создание сети если нужно...'
|
echo '========================================'
|
||||||
docker network create app-network 2>/dev/null || echo 'Сеть уже существует'
|
echo 'Step 2: Setup environment for compatibility...'
|
||||||
|
echo '========================================'
|
||||||
|
|
||||||
echo '4. Запуск инфраструктуры...'
|
docker network create app-network 2>/dev/null || echo 'Network app-network already exists'
|
||||||
docker-compose up -d db storage
|
|
||||||
|
|
||||||
echo '5. Ожидание PostgreSQL...'
|
echo 'COMPOSE_COMPATIBILITY=true' > .env
|
||||||
sleep 10
|
|
||||||
|
|
||||||
echo '6. Сборка нового образа приложения...'
|
echo '========================================'
|
||||||
docker build --no-cache -t app-backend:latest .
|
echo 'Step 3: Force cleanup old application...'
|
||||||
|
echo '========================================'
|
||||||
|
|
||||||
echo '7. Запуск приложения...'
|
docker-compose --compatibility down 2>/dev/null || echo 'No previous compose stack'
|
||||||
docker run -d \\
|
|
||||||
--name app-backend \\
|
|
||||||
--network app-network \\
|
|
||||||
--network-alias app \\
|
|
||||||
-p 80:8080 \\
|
|
||||||
-v uploads_data:/data/uploads:rw \\
|
|
||||||
-e POSTGRES_DB=no_copy_ \\
|
|
||||||
-e POSTGRES_USER=postgres \\
|
|
||||||
-e POSTGRES_PASSWORD=postgres \\
|
|
||||||
-e POSTGRES_PORT=5432 \\
|
|
||||||
-e POSTGRES_HOST=db \\
|
|
||||||
--restart unless-stopped \\
|
|
||||||
app-backend:latest
|
|
||||||
|
|
||||||
echo '8. Запуск мониторинга...'
|
docker images --filter 'reference=*app-backend*' -q | xargs -r docker rmi -f 2>/dev/null || echo 'No images to remove'
|
||||||
docker-compose up -d grafana prometheus loki tempo alloy
|
|
||||||
|
|
||||||
echo '9. Проверка...'
|
echo '========================================'
|
||||||
sleep 5
|
echo 'Step 4: Verify files and fix compose...'
|
||||||
|
echo '========================================'
|
||||||
|
|
||||||
echo 'Статус контейнеров:'
|
if [ ! -f docker-compose.yaml ]; then
|
||||||
docker ps --format 'table {{.Names}}\\t{{.Image}}\\t{{.Status}}'
|
echo 'ERROR: docker-compose.yaml not found!'
|
||||||
|
ls -la
|
||||||
echo '10. Проверка health...'
|
exit 1
|
||||||
if curl -s -f http://localhost:80/health > /dev/null 2>&1; then
|
|
||||||
echo 'Приложение работает'
|
|
||||||
echo 'URL: http://${params.SERVER}:80'
|
|
||||||
else
|
|
||||||
echo 'Проверка health не удалась'
|
|
||||||
docker logs app-backend --tail=20
|
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
echo 'Checking docker-compose.yaml syntax...'
|
||||||
|
docker-compose --compatibility config 2>&1 | head -20 || echo 'Config check output'
|
||||||
|
|
||||||
|
echo '========================================'
|
||||||
|
echo 'Step 5: Start infrastructure services...'
|
||||||
|
echo '========================================'
|
||||||
|
|
||||||
|
echo 'Starting PostgreSQL with resource limits...'
|
||||||
|
docker-compose --compatibility up -d db
|
||||||
|
|
||||||
|
echo 'Starting storage service...'
|
||||||
|
docker-compose --compatibility up -d storage
|
||||||
|
|
||||||
|
echo 'Waiting for PostgreSQL to be ready...'
|
||||||
|
for i in {1..30}; do
|
||||||
|
if docker-compose --compatibility exec db pg_isready -U postgres 2>/dev/null; then
|
||||||
|
echo 'PostgreSQL is ready'
|
||||||
|
break
|
||||||
|
fi
|
||||||
|
echo 'Waiting... attempt ' \$i
|
||||||
|
sleep 2
|
||||||
|
done
|
||||||
|
|
||||||
|
echo '========================================'
|
||||||
|
echo 'Step 6: Build application image...'
|
||||||
|
echo '========================================'
|
||||||
|
|
||||||
|
if [ ! -f Dockerfile ]; then
|
||||||
|
echo 'ERROR: Dockerfile not found!'
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo 'Building Docker image...'
|
||||||
|
docker build --no-cache -t app-backend:latest . 2>&1 | tail -30
|
||||||
|
|
||||||
|
echo 'Image created successfully'
|
||||||
|
docker images app-backend:latest --format 'table {{.Repository}}\\t{{.Tag}}\\t{{.Size}}\\t{{.CreatedAt}}'
|
||||||
|
|
||||||
|
echo '========================================'
|
||||||
|
echo 'Step 7: Start application with resource limits...'
|
||||||
|
echo '========================================'
|
||||||
|
|
||||||
|
docker volume create uploads_data 2>/dev/null || echo 'Volume uploads_data already exists'
|
||||||
|
|
||||||
|
echo 'Starting main application...'
|
||||||
|
docker-compose --compatibility up -d app
|
||||||
|
|
||||||
|
echo 'Waiting for application to start...'
|
||||||
|
sleep 15
|
||||||
|
|
||||||
|
echo '========================================'
|
||||||
|
echo 'Step 8: Start monitoring stack...'
|
||||||
|
echo '========================================'
|
||||||
|
|
||||||
|
docker-compose --compatibility up -d grafana prometheus loki tempo alloy
|
||||||
|
|
||||||
|
echo '========================================'
|
||||||
|
echo 'Step 9: Verify deployment...'
|
||||||
|
echo '========================================'
|
||||||
|
|
||||||
|
echo 'All containers status:'
|
||||||
|
docker-compose --compatibility ps
|
||||||
|
|
||||||
|
echo ''
|
||||||
|
echo 'Resource limits check:'
|
||||||
|
echo 'App backend limits:'
|
||||||
|
docker inspect app-backend --format='{{.HostConfig.Memory}} {{.HostConfig.CpuQuota}} {{.HostConfig.CpuPeriod}}' 2>/dev/null || echo 'Cannot inspect'
|
||||||
|
|
||||||
|
echo ''
|
||||||
|
echo 'PostgreSQL limits:'
|
||||||
|
docker inspect postgres --format='{{.HostConfig.Memory}} {{.HostConfig.CpuQuota}} {{.HostConfig.CpuPeriod}}' 2>/dev/null || echo 'Cannot inspect'
|
||||||
|
|
||||||
|
echo ''
|
||||||
|
echo 'Container resource usage:'
|
||||||
|
docker stats --no-stream --format 'table {{.Name}}\\t{{.CPUPerc}}\\t{{.MemUsage}}\\t{{.MemPerc}}' postgres app-backend storage 2>/dev/null || echo 'Stats not available'
|
||||||
|
|
||||||
|
echo '========================================'
|
||||||
|
echo 'Step 10: Health checks...'
|
||||||
|
echo '========================================'
|
||||||
|
|
||||||
|
echo 'Application health check...'
|
||||||
|
for i in {1..10}; do
|
||||||
|
if curl -s -f http://localhost:80/health 2>/dev/null; then
|
||||||
|
echo '✓ Application health check PASSED'
|
||||||
|
break
|
||||||
|
elif [ \$i -eq 10 ]; then
|
||||||
|
echo '✗ Application health check FAILED'
|
||||||
|
echo 'Last logs:'
|
||||||
|
docker-compose --compatibility logs --tail=20 app
|
||||||
|
else
|
||||||
|
echo 'Waiting for application... attempt ' \$i
|
||||||
|
sleep 5
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
|
||||||
|
echo 'Checking monitoring services...'
|
||||||
|
if curl -s -f http://localhost:3000/api/health 2>/dev/null; then
|
||||||
|
echo '✓ Grafana is healthy'
|
||||||
|
else
|
||||||
|
echo '⚠ Grafana may be starting'
|
||||||
|
fi
|
||||||
|
|
||||||
|
if curl -s -f http://localhost:9090/-/healthy 2>/dev/null; then
|
||||||
|
echo '✓ Prometheus is healthy'
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo '========================================'
|
||||||
|
echo 'Deployment Summary:'
|
||||||
|
echo '========================================'
|
||||||
|
echo 'Application URL: http://${params.SERVER}:80'
|
||||||
|
echo 'Grafana: http://${params.SERVER}:3000 (admin/admin)'
|
||||||
|
echo 'Prometheus: http://${params.SERVER}:9090'
|
||||||
|
echo 'Loki: http://${params.SERVER}:3100'
|
||||||
|
echo ''
|
||||||
|
echo 'To view logs: docker-compose logs -f app'
|
||||||
|
echo 'To view all containers: docker-compose ps'
|
||||||
|
echo 'To stop: docker-compose down'
|
||||||
|
echo '========================================'
|
||||||
"
|
"
|
||||||
"""
|
"""
|
||||||
}
|
}
|
||||||
@@ -113,6 +213,8 @@ pipeline {
|
|||||||
success {
|
success {
|
||||||
echo "Deployment successful"
|
echo "Deployment successful"
|
||||||
echo "Application URL: http://${params.SERVER}:80"
|
echo "Application URL: http://${params.SERVER}:80"
|
||||||
|
echo "Grafana: http://${params.SERVER}:3000"
|
||||||
|
echo "Prometheus: http://${params.SERVER}:9090"
|
||||||
}
|
}
|
||||||
failure {
|
failure {
|
||||||
echo "Deployment failed for branch ${params.BRANCH}"
|
echo "Deployment failed for branch ${params.BRANCH}"
|
||||||
|
|||||||
Binary file not shown.
Binary file not shown.
@@ -1,35 +1,17 @@
|
|||||||
package ru.soune.nocopy.configuration;
|
package ru.soune.nocopy.configuration;
|
||||||
|
|
||||||
import com.vrt.fileprotection.FileProtector;
|
|
||||||
import lombok.AllArgsConstructor;
|
|
||||||
import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
|
import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
|
||||||
import org.springframework.context.annotation.Bean;
|
import org.springframework.context.annotation.Bean;
|
||||||
import org.springframework.context.annotation.Configuration;
|
import org.springframework.context.annotation.Configuration;
|
||||||
import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder;
|
import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder;
|
||||||
import org.springframework.security.crypto.password.PasswordEncoder;
|
import org.springframework.security.crypto.password.PasswordEncoder;
|
||||||
|
|
||||||
import java.util.Collections;
|
|
||||||
|
|
||||||
@Configuration
|
@Configuration
|
||||||
@EnableAutoConfiguration
|
@EnableAutoConfiguration
|
||||||
@AllArgsConstructor
|
|
||||||
public class ApplicationConfig {
|
public class ApplicationConfig {
|
||||||
|
|
||||||
@Bean
|
@Bean
|
||||||
PasswordEncoder passwordEncoder() {
|
PasswordEncoder passwordEncoder() {
|
||||||
return new BCryptPasswordEncoder();
|
return new BCryptPasswordEncoder();
|
||||||
}
|
}
|
||||||
|
|
||||||
@Bean
|
|
||||||
public com.vrt.NoCopyFileService noCopyFileService(
|
|
||||||
FileProtector.FileProvider fileProvider,
|
|
||||||
FileProtector.ProcessingListener processingListener) {
|
|
||||||
|
|
||||||
return new com.vrt.NoCopyFileService(
|
|
||||||
Collections.emptyList(),
|
|
||||||
fileProvider,
|
|
||||||
processingListener
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -1,19 +0,0 @@
|
|||||||
package ru.soune.nocopy.configuration.file;
|
|
||||||
|
|
||||||
import lombok.AllArgsConstructor;
|
|
||||||
import org.springframework.boot.context.event.ApplicationReadyEvent;
|
|
||||||
import org.springframework.context.event.EventListener;
|
|
||||||
import org.springframework.stereotype.Component;
|
|
||||||
import ru.soune.nocopy.service.file.FileProcessingOrchestrator;
|
|
||||||
|
|
||||||
@Component
|
|
||||||
@AllArgsConstructor
|
|
||||||
public class NoCopyInitializer {
|
|
||||||
|
|
||||||
private final FileProcessingOrchestrator orchestrator;
|
|
||||||
|
|
||||||
@EventListener(ApplicationReadyEvent.class)
|
|
||||||
public void initializeOnStartup() {
|
|
||||||
orchestrator.initializeProcessingQueue();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,7 +1,5 @@
|
|||||||
package ru.soune.nocopy.controller;
|
package ru.soune.nocopy.controller;
|
||||||
|
|
||||||
import com.vrt.NoCopyFileService;
|
|
||||||
import com.vrt.fileprotection.FileProtector;
|
|
||||||
import lombok.RequiredArgsConstructor;
|
import lombok.RequiredArgsConstructor;
|
||||||
import lombok.extern.slf4j.Slf4j;
|
import lombok.extern.slf4j.Slf4j;
|
||||||
import org.springframework.core.io.Resource;
|
import org.springframework.core.io.Resource;
|
||||||
@@ -21,15 +19,12 @@ import ru.soune.nocopy.dto.file.ChunkUploadResponse;
|
|||||||
import ru.soune.nocopy.dto.file.CompleteUploadResponse;
|
import ru.soune.nocopy.dto.file.CompleteUploadResponse;
|
||||||
import ru.soune.nocopy.dto.file.FileEntityResponse;
|
import ru.soune.nocopy.dto.file.FileEntityResponse;
|
||||||
import ru.soune.nocopy.dto.file.UploadProgress;
|
import ru.soune.nocopy.dto.file.UploadProgress;
|
||||||
import ru.soune.nocopy.entity.AuthToken;
|
import ru.soune.nocopy.entity.auth.AuthToken;
|
||||||
import ru.soune.nocopy.entity.file.FileEntity;
|
|
||||||
import ru.soune.nocopy.entity.file.FileStatus;
|
import ru.soune.nocopy.entity.file.FileStatus;
|
||||||
import ru.soune.nocopy.entity.file.ProtectionStatus;
|
|
||||||
import ru.soune.nocopy.entity.file.UploadStatus;
|
import ru.soune.nocopy.entity.file.UploadStatus;
|
||||||
import ru.soune.nocopy.exception.*;
|
import ru.soune.nocopy.exception.*;
|
||||||
import ru.soune.nocopy.handler.*;
|
import ru.soune.nocopy.handler.*;
|
||||||
import ru.soune.nocopy.repository.AuthTokenRepository;
|
import ru.soune.nocopy.repository.AuthTokenRepository;
|
||||||
import ru.soune.nocopy.repository.FileEntityRepository;
|
|
||||||
import ru.soune.nocopy.service.file.FileEntityService;
|
import ru.soune.nocopy.service.file.FileEntityService;
|
||||||
import ru.soune.nocopy.service.file.FileUploadService;
|
import ru.soune.nocopy.service.file.FileUploadService;
|
||||||
|
|
||||||
@@ -40,7 +35,6 @@ import java.nio.file.Paths;
|
|||||||
import java.util.HashMap;
|
import java.util.HashMap;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.Map;
|
import java.util.Map;
|
||||||
import java.util.Optional;
|
|
||||||
import java.util.stream.Collectors;
|
import java.util.stream.Collectors;
|
||||||
|
|
||||||
@Slf4j
|
@Slf4j
|
||||||
@@ -56,10 +50,6 @@ public class ApiController {
|
|||||||
|
|
||||||
private final AuthTokenRepository authTokenRepository;
|
private final AuthTokenRepository authTokenRepository;
|
||||||
|
|
||||||
private final FileEntityRepository fileEntityRepository;
|
|
||||||
|
|
||||||
private final NoCopyFileService noCopyFileService;
|
|
||||||
|
|
||||||
@PostMapping("/v{version}/data")
|
@PostMapping("/v{version}/data")
|
||||||
public ResponseEntity<?> handlePostRequest(@RequestBody BaseRequest request,
|
public ResponseEntity<?> handlePostRequest(@RequestBody BaseRequest request,
|
||||||
@PathVariable("version") int version) {
|
@PathVariable("version") int version) {
|
||||||
@@ -342,47 +332,6 @@ public class ApiController {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@GetMapping("/protect/{fileId}")
|
|
||||||
public ResponseEntity<?> protect( @PathVariable(required = false) String fileId) {
|
|
||||||
Optional<FileEntity> fileEntity = fileEntityRepository.findById(fileId);
|
|
||||||
|
|
||||||
if (!fileEntity.isPresent()) {
|
|
||||||
return ResponseEntity.notFound().build();
|
|
||||||
}
|
|
||||||
|
|
||||||
FileProtector.FileInfo fileInfo = createFileInfo(fileEntity.get());
|
|
||||||
|
|
||||||
noCopyFileService.addFile(fileInfo);
|
|
||||||
|
|
||||||
fileEntity.get().setProtectionStatus(ProtectionStatus.PROCESSING);
|
|
||||||
|
|
||||||
fileEntityRepository.save(fileEntity.get());
|
|
||||||
|
|
||||||
return ResponseEntity.ok().build();
|
|
||||||
}
|
|
||||||
|
|
||||||
private FileProtector.FileInfo createFileInfo(FileEntity fileEntity) {
|
|
||||||
FileProtector.Type type = determineFileType(fileEntity.getMimeType());
|
|
||||||
|
|
||||||
return new FileProtector.FileInfo(type, fileEntity.getId(), String.valueOf(fileEntity.getUserId()));
|
|
||||||
}
|
|
||||||
|
|
||||||
private FileProtector.Type determineFileType(String mimeType) {
|
|
||||||
if (mimeType == null) {
|
|
||||||
return FileProtector.Type.IMAGE;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (mimeType.startsWith("image")) {
|
|
||||||
return FileProtector.Type.IMAGE;
|
|
||||||
} else if (mimeType.startsWith("video")) {
|
|
||||||
return FileProtector.Type.VIDEO;
|
|
||||||
} else if (mimeType.startsWith("audio")) {
|
|
||||||
return FileProtector.Type.AUDIO;
|
|
||||||
} else {
|
|
||||||
return FileProtector.Type.IMAGE;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private ResponseEntity<BaseResponse> createValidationErrorResponse(BindingResult bindingResult, Integer msgId) {
|
private ResponseEntity<BaseResponse> createValidationErrorResponse(BindingResult bindingResult, Integer msgId) {
|
||||||
List<Map<String, String>> fieldErrors = bindingResult.getFieldErrors()
|
List<Map<String, String>> fieldErrors = bindingResult.getFieldErrors()
|
||||||
.stream()
|
.stream()
|
||||||
@@ -419,6 +368,18 @@ public class ApiController {
|
|||||||
return contentType;
|
return contentType;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private boolean isPreviewSupported(String mimeType) {
|
||||||
|
if (mimeType == null) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
return mimeType.startsWith("image") ||
|
||||||
|
mimeType.startsWith("text") ||
|
||||||
|
mimeType.equals("pdf") ||
|
||||||
|
mimeType.startsWith("video") ||
|
||||||
|
mimeType.startsWith("audio");
|
||||||
|
}
|
||||||
|
|
||||||
private Long getUserIdFromToken(String tokenHeader) {
|
private Long getUserIdFromToken(String tokenHeader) {
|
||||||
String token = tokenHeader.replace("Bearer ", "");
|
String token = tokenHeader.replace("Bearer ", "");
|
||||||
AuthToken authToken = authTokenRepository.findByToken(token)
|
AuthToken authToken = authTokenRepository.findByToken(token)
|
||||||
|
|||||||
@@ -6,7 +6,7 @@ import org.springframework.web.bind.annotation.*;
|
|||||||
import ru.soune.nocopy.dto.ChangePasswordRequest;
|
import ru.soune.nocopy.dto.ChangePasswordRequest;
|
||||||
import ru.soune.nocopy.dto.UserDTO;
|
import ru.soune.nocopy.dto.UserDTO;
|
||||||
import ru.soune.nocopy.dto.UserRequest;
|
import ru.soune.nocopy.dto.UserRequest;
|
||||||
import ru.soune.nocopy.entity.AuthToken;
|
import ru.soune.nocopy.entity.auth.AuthToken;
|
||||||
import ru.soune.nocopy.entity.User;
|
import ru.soune.nocopy.entity.User;
|
||||||
import ru.soune.nocopy.exception.InvalidUserEmail;
|
import ru.soune.nocopy.exception.InvalidUserEmail;
|
||||||
import ru.soune.nocopy.exception.NotFoundAuthToken;
|
import ru.soune.nocopy.exception.NotFoundAuthToken;
|
||||||
|
|||||||
@@ -0,0 +1,93 @@
|
|||||||
|
package ru.soune.nocopy.controller;
|
||||||
|
|
||||||
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
||||||
|
import lombok.RequiredArgsConstructor;
|
||||||
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
import org.springframework.http.HttpStatus;
|
||||||
|
import org.springframework.http.ResponseEntity;
|
||||||
|
import org.springframework.web.bind.annotation.*;
|
||||||
|
import ru.soune.nocopy.dto.BaseResponse;
|
||||||
|
import ru.soune.nocopy.dto.LoginAnswer;
|
||||||
|
import ru.soune.nocopy.dto.MessageCode;
|
||||||
|
import ru.soune.nocopy.entity.User;
|
||||||
|
import ru.soune.nocopy.entity.auth.AuthToken;
|
||||||
|
import ru.soune.nocopy.service.AuthService;
|
||||||
|
import ru.soune.nocopy.service.VkOAuthService;
|
||||||
|
import ru.soune.nocopy.service.auth.ExternalAuthUserService;
|
||||||
|
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
|
|
||||||
|
@RestController
|
||||||
|
@RequestMapping("/api")
|
||||||
|
@RequiredArgsConstructor
|
||||||
|
@Slf4j
|
||||||
|
public class VkAuthController {
|
||||||
|
|
||||||
|
private final VkOAuthService vkOAuthService;
|
||||||
|
|
||||||
|
private final AuthService authService;
|
||||||
|
|
||||||
|
private final ExternalAuthUserService externalAuthUserService;
|
||||||
|
|
||||||
|
@GetMapping("/v{version}/vk/authorization/{code}/{state}/{codeVerifier}/{deviceId}")
|
||||||
|
public BaseResponse vkAuth(@PathVariable("code") String code,
|
||||||
|
@PathVariable("state") String state,
|
||||||
|
@PathVariable("codeVerifier") String codeVerifier,
|
||||||
|
@PathVariable("deviceId") String deviceId) {
|
||||||
|
try {
|
||||||
|
Map<String, Object> authMap = vkOAuthService.authTokenMap(code, state, codeVerifier, deviceId);
|
||||||
|
ObjectMapper mapper = new ObjectMapper();
|
||||||
|
|
||||||
|
String bodyJson = (String) authMap.get("body");
|
||||||
|
Map<String, Object> body = mapper.readValue(bodyJson, Map.class);
|
||||||
|
|
||||||
|
String accessToken = (String) body.get("access_token");
|
||||||
|
long vkUserId = ((Number) body.get("user_id")).longValue();
|
||||||
|
Integer expiresIn = (Integer) body.get("expires_in");
|
||||||
|
String refreshToken = (String) body.get("refresh_token");
|
||||||
|
|
||||||
|
Map<String, Object> vkUserInfo = vkOAuthService.userVkInfo(accessToken);
|
||||||
|
log.info("vkUserInfo {}", vkUserInfo);
|
||||||
|
|
||||||
|
String userInfoBodyJson = (String) vkUserInfo.get("body");
|
||||||
|
Map<String, Object> userInfoBody = mapper.readValue(userInfoBodyJson, Map.class);
|
||||||
|
Map<String, Object> user = (Map<String, Object>) userInfoBody.get("user");
|
||||||
|
|
||||||
|
String firstName = (String) user.get("first_name");
|
||||||
|
String lastName = (String) user.get("last_name");
|
||||||
|
String fullName = firstName + " " + lastName;
|
||||||
|
|
||||||
|
User userEntity = externalAuthUserService.findOrCreateUserFromVK(Long.toString(vkUserId), fullName,
|
||||||
|
null, accessToken, refreshToken, expiresIn);
|
||||||
|
|
||||||
|
AuthToken ourToken = authService.createTokenForUser(userEntity);
|
||||||
|
|
||||||
|
LoginAnswer loginAnswer = new LoginAnswer();
|
||||||
|
loginAnswer.setToken(ourToken.getToken());
|
||||||
|
loginAnswer.setFieldErrors(new ArrayList<>());
|
||||||
|
|
||||||
|
return new BaseResponse(20009, MessageCode.SUCCESS.getCode(),
|
||||||
|
MessageCode.SUCCESS.getDescription(), loginAnswer);
|
||||||
|
} catch (Exception e) {
|
||||||
|
log.error("VK auth error: ", e);
|
||||||
|
|
||||||
|
LoginAnswer loginAnswer = new LoginAnswer();
|
||||||
|
loginAnswer.setToken(null);
|
||||||
|
|
||||||
|
List<Map<String, String>> errors = new ArrayList<>();
|
||||||
|
Map<String, String> error = new HashMap<>();
|
||||||
|
error.put("field", "vk_auth");
|
||||||
|
error.put("message", e.getMessage());
|
||||||
|
errors.add(error);
|
||||||
|
|
||||||
|
loginAnswer.setFieldErrors(errors);
|
||||||
|
|
||||||
|
return new BaseResponse(20009, MessageCode.SUCCESS.getCode(),
|
||||||
|
MessageCode.SUCCESS.getDescription(), loginAnswer);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
@@ -1,10 +0,0 @@
|
|||||||
package ru.soune.nocopy.dto;
|
|
||||||
|
|
||||||
import java.time.LocalDateTime;
|
|
||||||
|
|
||||||
public record AuthResponse (
|
|
||||||
boolean success,
|
|
||||||
String message,
|
|
||||||
String token,
|
|
||||||
LocalDateTime expiresAt
|
|
||||||
) {}
|
|
||||||
@@ -6,11 +6,14 @@ import jakarta.validation.constraints.Size;
|
|||||||
import lombok.*;
|
import lombok.*;
|
||||||
import org.springframework.data.annotation.CreatedDate;
|
import org.springframework.data.annotation.CreatedDate;
|
||||||
import org.springframework.data.jpa.domain.support.AuditingEntityListener;
|
import org.springframework.data.jpa.domain.support.AuditingEntityListener;
|
||||||
|
import ru.soune.nocopy.entity.auth.AuthToken;
|
||||||
|
import ru.soune.nocopy.entity.auth.ExternalAuthProvider;
|
||||||
|
|
||||||
import java.time.LocalDate;
|
import java.time.LocalDate;
|
||||||
import java.time.LocalDateTime;
|
import java.time.LocalDateTime;
|
||||||
import java.util.ArrayList;
|
import java.util.ArrayList;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
import java.util.Optional;
|
||||||
|
|
||||||
@Entity
|
@Entity
|
||||||
@Table(name = "users")
|
@Table(name = "users")
|
||||||
@@ -19,18 +22,19 @@ import java.util.List;
|
|||||||
@NoArgsConstructor
|
@NoArgsConstructor
|
||||||
@AllArgsConstructor
|
@AllArgsConstructor
|
||||||
@ToString
|
@ToString
|
||||||
|
@Builder
|
||||||
@EntityListeners(AuditingEntityListener.class)
|
@EntityListeners(AuditingEntityListener.class)
|
||||||
public class User {
|
public class User {
|
||||||
|
|
||||||
@Id
|
@Id
|
||||||
@GeneratedValue(strategy = GenerationType.IDENTITY)
|
@GeneratedValue(strategy = GenerationType.IDENTITY)
|
||||||
private Long Id;
|
private Long id;
|
||||||
|
|
||||||
@Column(name = "full_name", nullable = false)
|
@Column(name = "full_name", nullable = false)
|
||||||
private String fullName;
|
private String fullName;
|
||||||
|
|
||||||
@Size(max = 1024)
|
@Size(max = 255)
|
||||||
@Column(name = "email", nullable = false, length = 1024, unique = true)
|
@Column(name = "email", nullable = false, length = 255, unique = true)
|
||||||
private String email;
|
private String email;
|
||||||
|
|
||||||
@Column(name = "company")
|
@Column(name = "company")
|
||||||
@@ -90,4 +94,14 @@ public class User {
|
|||||||
@JsonIgnore
|
@JsonIgnore
|
||||||
@ToString.Exclude
|
@ToString.Exclude
|
||||||
private List<ImageProtection> imageProtections = new ArrayList<>();
|
private List<ImageProtection> imageProtections = new ArrayList<>();
|
||||||
|
|
||||||
|
@OneToMany(mappedBy = "user", cascade = CascadeType.ALL, fetch = FetchType.LAZY)
|
||||||
|
@ToString.Exclude
|
||||||
|
private List<ExternalAuthProvider> externalAuthProviders = new ArrayList<>();
|
||||||
|
|
||||||
|
public Optional<ExternalAuthProvider> getProvider(ExternalAuthProvider.ProviderType providerType) {
|
||||||
|
return externalAuthProviders.stream()
|
||||||
|
.filter(p -> p.getProviderType() == providerType)
|
||||||
|
.findFirst();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
+3
-2
@@ -1,9 +1,10 @@
|
|||||||
package ru.soune.nocopy.entity;
|
package ru.soune.nocopy.entity.auth;
|
||||||
|
|
||||||
import jakarta.persistence.*;
|
import jakarta.persistence.*;
|
||||||
import lombok.*;
|
import lombok.*;
|
||||||
import org.springframework.data.annotation.CreatedDate;
|
import org.springframework.data.annotation.CreatedDate;
|
||||||
import org.springframework.data.jpa.domain.support.AuditingEntityListener;
|
import org.springframework.data.jpa.domain.support.AuditingEntityListener;
|
||||||
|
import ru.soune.nocopy.entity.User;
|
||||||
|
|
||||||
import java.time.LocalDateTime;
|
import java.time.LocalDateTime;
|
||||||
|
|
||||||
@@ -30,7 +31,7 @@ public class AuthToken {
|
|||||||
private String token;
|
private String token;
|
||||||
|
|
||||||
@Column(name = "expires_at", nullable = false)
|
@Column(name = "expires_at", nullable = false)
|
||||||
private LocalDateTime expiresAt = LocalDateTime.now().plusDays(30);
|
private LocalDateTime expiresAt = LocalDateTime.now().plusHours(1);
|
||||||
|
|
||||||
@CreatedDate
|
@CreatedDate
|
||||||
@Column(name = "created_at", updatable = false, nullable = false)
|
@Column(name = "created_at", updatable = false, nullable = false)
|
||||||
@@ -0,0 +1,89 @@
|
|||||||
|
package ru.soune.nocopy.entity.auth;
|
||||||
|
|
||||||
|
import jakarta.persistence.*;
|
||||||
|
import lombok.*;
|
||||||
|
import org.springframework.data.annotation.CreatedDate;
|
||||||
|
import org.springframework.data.jpa.domain.support.AuditingEntityListener;
|
||||||
|
import ru.soune.nocopy.entity.User;
|
||||||
|
|
||||||
|
import java.time.LocalDateTime;
|
||||||
|
|
||||||
|
@Entity
|
||||||
|
@Table(name = "external_auth_providers",
|
||||||
|
uniqueConstraints = @UniqueConstraint(
|
||||||
|
columnNames = {"provider_type", "provider_user_id"}
|
||||||
|
),
|
||||||
|
indexes = {
|
||||||
|
@Index(name = "idx_provider_type_user_id",
|
||||||
|
columnList = "provider_type, provider_user_id"),
|
||||||
|
@Index(name = "idx_user_id_provider_type",
|
||||||
|
columnList = "user_id, provider_type")
|
||||||
|
})
|
||||||
|
@Getter @Setter
|
||||||
|
@NoArgsConstructor
|
||||||
|
@AllArgsConstructor
|
||||||
|
@Builder
|
||||||
|
@EntityListeners(AuditingEntityListener.class)
|
||||||
|
public class ExternalAuthProvider {
|
||||||
|
|
||||||
|
public enum ProviderType {
|
||||||
|
VK, YANDEX
|
||||||
|
}
|
||||||
|
|
||||||
|
@Id
|
||||||
|
@GeneratedValue(strategy = GenerationType.IDENTITY)
|
||||||
|
private Long id;
|
||||||
|
|
||||||
|
@ManyToOne(fetch = FetchType.LAZY)
|
||||||
|
@JoinColumn(name = "user_id", nullable = false, foreignKey = @ForeignKey(name = "fk_extauth_user"))
|
||||||
|
@ToString.Exclude
|
||||||
|
private User user;
|
||||||
|
|
||||||
|
@Enumerated(EnumType.STRING)
|
||||||
|
@Column(name = "provider_type", nullable = false, length = 20)
|
||||||
|
private ProviderType providerType;
|
||||||
|
|
||||||
|
@Column(name = "provider_user_id", nullable = false, length = 255)
|
||||||
|
private String providerUserId;
|
||||||
|
|
||||||
|
@Column(name = "access_token", length = 2048)
|
||||||
|
private String accessToken;
|
||||||
|
|
||||||
|
@Column(name = "refresh_token", length = 2048)
|
||||||
|
private String refreshToken;
|
||||||
|
|
||||||
|
@Column(name = "token_expires_at")
|
||||||
|
private LocalDateTime tokenExpiresAt;
|
||||||
|
|
||||||
|
@Column(name = "email", length = 255)
|
||||||
|
private String email;
|
||||||
|
|
||||||
|
@Column(name = "phone", length = 20)
|
||||||
|
private String phoneNumber;
|
||||||
|
|
||||||
|
@CreatedDate
|
||||||
|
@Column(name = "linked_at", updatable = false)
|
||||||
|
private LocalDateTime linkedAt;
|
||||||
|
|
||||||
|
@Column(name = "last_used_at")
|
||||||
|
private LocalDateTime lastUsedAt;
|
||||||
|
|
||||||
|
@Column(name = "is_active")
|
||||||
|
@Builder.Default
|
||||||
|
private Boolean isActive = true;
|
||||||
|
|
||||||
|
@PrePersist
|
||||||
|
public void prePersist() {
|
||||||
|
if (this.linkedAt == null) {
|
||||||
|
this.linkedAt = LocalDateTime.now();
|
||||||
|
}
|
||||||
|
if (this.lastUsedAt == null) {
|
||||||
|
this.lastUsedAt = LocalDateTime.now();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@PreUpdate
|
||||||
|
public void preUpdate() {
|
||||||
|
this.lastUsedAt = LocalDateTime.now();
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -60,26 +60,11 @@ public class FileEntity {
|
|||||||
@Column(name = "updated_at")
|
@Column(name = "updated_at")
|
||||||
private LocalDateTime updatedAt;
|
private LocalDateTime updatedAt;
|
||||||
|
|
||||||
@Column(name = "protected_file_path")
|
|
||||||
private String protectedFilePath;
|
|
||||||
|
|
||||||
@Column(name = "protection_status")
|
|
||||||
@Enumerated(EnumType.STRING)
|
|
||||||
private ProtectionStatus protectionStatus;
|
|
||||||
|
|
||||||
@Column(name = "protected_at")
|
|
||||||
private LocalDateTime protectedAt;
|
|
||||||
|
|
||||||
@PrePersist
|
@PrePersist
|
||||||
public void prePersist() {
|
public void prePersist() {
|
||||||
if (this.status == null) {
|
if (this.status == null) {
|
||||||
this.status = FileStatus.ACTIVE;
|
this.status = FileStatus.ACTIVE;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (this.protectionStatus == null) {
|
|
||||||
this.protectionStatus = ProtectionStatus.NOT_PROTECTED;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (this.createdAt == null) {
|
if (this.createdAt == null) {
|
||||||
this.createdAt = LocalDateTime.now();
|
this.createdAt = LocalDateTime.now();
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -7,10 +7,21 @@ import java.util.List;
|
|||||||
|
|
||||||
@Getter
|
@Getter
|
||||||
public enum FileType {
|
public enum FileType {
|
||||||
IMAGE("image", Arrays.asList("jpg", "jpeg", "png", "gif", "bmp", "webp", "jfif")),
|
IMAGE("image", Arrays.asList("jpg", "jpeg", "png", "gif", "bmp", "webp")),
|
||||||
VIDEO("video", Arrays.asList("mp4", "avi", "mov", "wmv", "flv", "mkv", "webm", "m4v", "mpg", "mpeg",
|
VIDEO("video", Arrays.asList("mp4", "avi", "mov", "wmv", "flv", "mkv", "webm", "m4v", "mpg", "mpeg",
|
||||||
"3gp", "3g2", "f4v", "m2ts", "mts", "vob", "ogv", "divx")),
|
"3gp", "3g2", "f4v", "m2ts", "mts", "vob", "ogv", "divx")),
|
||||||
AUDIO("audio", Arrays.asList("mp3", "wav", "flac"));
|
AUDIO("audio", Arrays.asList("mp3", "wav", "flac")),
|
||||||
|
DOCUMENT("document", Arrays.asList(
|
||||||
|
"pdf", "txt", "rtf",
|
||||||
|
"doc", "docx", "xls", "xlsx", "ppt", "pptx", "pps", "ppsx", "dot", "dotx", "xlt", "xltx", "pot", "potx",
|
||||||
|
"odt", "ods", "odp", "odg", "odf", "odb", "odc", "odi", "odm", "ott", "ots", "otp", "otg", "oth",
|
||||||
|
"sxw", "sxc", "sxi", "sxd", "sxg", "stc", "sti", "stw", "sxm",
|
||||||
|
"pages", "numbers", "key",
|
||||||
|
"csv", "tsv", "xml", "html", "htm", "tex", "md", "markdown",
|
||||||
|
"epub", "mobi", "azw", "azw3", "fb2",
|
||||||
|
"wps", "wpt", "et", "dps", "vsd", "vsdx",
|
||||||
|
"java", "py", "cpp", "c", "h", "js", "css", "php", "sql", "json", "yaml", "yml", "sh", "bat",
|
||||||
|
"one", "note"));
|
||||||
|
|
||||||
private final String displayName;
|
private final String displayName;
|
||||||
private final List<String> allowedExtensions;
|
private final List<String> allowedExtensions;
|
||||||
|
|||||||
@@ -1,9 +0,0 @@
|
|||||||
package ru.soune.nocopy.entity.file;
|
|
||||||
|
|
||||||
public enum ProtectionStatus {
|
|
||||||
NOT_PROTECTED,
|
|
||||||
PROCESSING,
|
|
||||||
PROTECTED,
|
|
||||||
FAILED,
|
|
||||||
FAILED_SAVE
|
|
||||||
}
|
|
||||||
@@ -6,7 +6,7 @@ import lombok.extern.slf4j.Slf4j;
|
|||||||
import org.springframework.stereotype.Component;
|
import org.springframework.stereotype.Component;
|
||||||
import ru.soune.nocopy.dto.*;
|
import ru.soune.nocopy.dto.*;
|
||||||
import ru.soune.nocopy.dto.file.*;
|
import ru.soune.nocopy.dto.file.*;
|
||||||
import ru.soune.nocopy.entity.AuthToken;
|
import ru.soune.nocopy.entity.auth.AuthToken;
|
||||||
import ru.soune.nocopy.entity.file.FileEntity;
|
import ru.soune.nocopy.entity.file.FileEntity;
|
||||||
import ru.soune.nocopy.entity.file.FileStatus;
|
import ru.soune.nocopy.entity.file.FileStatus;
|
||||||
import ru.soune.nocopy.exception.FileEntityNotFoundException;
|
import ru.soune.nocopy.exception.FileEntityNotFoundException;
|
||||||
|
|||||||
@@ -10,7 +10,7 @@ import org.springframework.validation.BindingResult;
|
|||||||
import org.springframework.validation.FieldError;
|
import org.springframework.validation.FieldError;
|
||||||
import ru.soune.nocopy.dto.*;
|
import ru.soune.nocopy.dto.*;
|
||||||
import ru.soune.nocopy.dto.file.*;
|
import ru.soune.nocopy.dto.file.*;
|
||||||
import ru.soune.nocopy.entity.AuthToken;
|
import ru.soune.nocopy.entity.auth.AuthToken;
|
||||||
import ru.soune.nocopy.entity.file.FileType;
|
import ru.soune.nocopy.entity.file.FileType;
|
||||||
import ru.soune.nocopy.entity.file.FileUploadSession;
|
import ru.soune.nocopy.entity.file.FileUploadSession;
|
||||||
import ru.soune.nocopy.exception.*;
|
import ru.soune.nocopy.exception.*;
|
||||||
|
|||||||
@@ -4,7 +4,7 @@ import com.fasterxml.jackson.databind.ObjectMapper;
|
|||||||
import lombok.RequiredArgsConstructor;
|
import lombok.RequiredArgsConstructor;
|
||||||
import org.springframework.stereotype.Component;
|
import org.springframework.stereotype.Component;
|
||||||
import ru.soune.nocopy.dto.*;
|
import ru.soune.nocopy.dto.*;
|
||||||
import ru.soune.nocopy.entity.AuthToken;
|
import ru.soune.nocopy.entity.auth.AuthToken;
|
||||||
import ru.soune.nocopy.exception.NotValidFieldException;
|
import ru.soune.nocopy.exception.NotValidFieldException;
|
||||||
import ru.soune.nocopy.repository.UserRepository;
|
import ru.soune.nocopy.repository.UserRepository;
|
||||||
import ru.soune.nocopy.service.AuthService;
|
import ru.soune.nocopy.service.AuthService;
|
||||||
|
|||||||
@@ -4,7 +4,7 @@ import com.fasterxml.jackson.databind.ObjectMapper;
|
|||||||
import lombok.RequiredArgsConstructor;
|
import lombok.RequiredArgsConstructor;
|
||||||
import org.springframework.stereotype.Component;
|
import org.springframework.stereotype.Component;
|
||||||
import ru.soune.nocopy.dto.*;
|
import ru.soune.nocopy.dto.*;
|
||||||
import ru.soune.nocopy.entity.AuthToken;
|
import ru.soune.nocopy.entity.auth.AuthToken;
|
||||||
import ru.soune.nocopy.entity.User;
|
import ru.soune.nocopy.entity.User;
|
||||||
import ru.soune.nocopy.exception.NotValidFieldException;
|
import ru.soune.nocopy.exception.NotValidFieldException;
|
||||||
import ru.soune.nocopy.repository.AuthTokenRepository;
|
import ru.soune.nocopy.repository.AuthTokenRepository;
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ import org.springframework.stereotype.Component;
|
|||||||
import org.springframework.validation.BeanPropertyBindingResult;
|
import org.springframework.validation.BeanPropertyBindingResult;
|
||||||
import org.springframework.validation.BindingResult;
|
import org.springframework.validation.BindingResult;
|
||||||
import ru.soune.nocopy.dto.*;
|
import ru.soune.nocopy.dto.*;
|
||||||
import ru.soune.nocopy.entity.AuthToken;
|
import ru.soune.nocopy.entity.auth.AuthToken;
|
||||||
import ru.soune.nocopy.exception.NotValidFieldException;
|
import ru.soune.nocopy.exception.NotValidFieldException;
|
||||||
import ru.soune.nocopy.exception.ValidationException;
|
import ru.soune.nocopy.exception.ValidationException;
|
||||||
import ru.soune.nocopy.handler.validator.RegRequestValidator;
|
import ru.soune.nocopy.handler.validator.RegRequestValidator;
|
||||||
|
|||||||
@@ -1,11 +1,14 @@
|
|||||||
package ru.soune.nocopy.handler.validator;
|
package ru.soune.nocopy.handler.validator;
|
||||||
|
|
||||||
|
import org.apache.commons.validator.routines.DomainValidator;
|
||||||
import org.apache.commons.validator.routines.EmailValidator;
|
import org.apache.commons.validator.routines.EmailValidator;
|
||||||
import org.springframework.stereotype.Component;
|
import org.springframework.stereotype.Component;
|
||||||
import org.springframework.validation.Errors;
|
import org.springframework.validation.Errors;
|
||||||
import org.springframework.validation.Validator;
|
import org.springframework.validation.Validator;
|
||||||
import ru.soune.nocopy.dto.RegRequest;
|
import ru.soune.nocopy.dto.RegRequest;
|
||||||
|
|
||||||
|
import java.net.InetAddress;
|
||||||
|
import java.net.UnknownHostException;
|
||||||
import java.util.*;
|
import java.util.*;
|
||||||
|
|
||||||
@Component
|
@Component
|
||||||
@@ -153,6 +156,15 @@ public class RegRequestValidator implements Validator {
|
|||||||
errors.rejectValue("password", "password.contains.spaces",
|
errors.rejectValue("password", "password.contains.spaces",
|
||||||
"Password cannot contain spaces");
|
"Password cannot contain spaces");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// recomment if need complexity
|
||||||
|
// checkPasswordComplexity(password, errors);
|
||||||
|
|
||||||
|
// reccoment if need check simply standart password
|
||||||
|
// checkCommonPasswords(password, errors);
|
||||||
|
|
||||||
|
// reccoment if need check simply standart password
|
||||||
|
// checkForSequences(password, errors);
|
||||||
}
|
}
|
||||||
|
|
||||||
private void checkPasswordComplexity(String password, Errors errors) {
|
private void checkPasswordComplexity(String password, Errors errors) {
|
||||||
|
|||||||
@@ -2,7 +2,7 @@ package ru.soune.nocopy.repository;
|
|||||||
|
|
||||||
import org.springframework.data.jpa.repository.JpaRepository;
|
import org.springframework.data.jpa.repository.JpaRepository;
|
||||||
import org.springframework.stereotype.Repository;
|
import org.springframework.stereotype.Repository;
|
||||||
import ru.soune.nocopy.entity.AuthToken;
|
import ru.soune.nocopy.entity.auth.AuthToken;
|
||||||
|
|
||||||
import java.time.LocalDate;
|
import java.time.LocalDate;
|
||||||
import java.util.Optional;
|
import java.util.Optional;
|
||||||
|
|||||||
@@ -0,0 +1,25 @@
|
|||||||
|
package ru.soune.nocopy.repository;
|
||||||
|
|
||||||
|
import org.springframework.data.jpa.repository.JpaRepository;
|
||||||
|
import org.springframework.stereotype.Repository;
|
||||||
|
import ru.soune.nocopy.entity.User;
|
||||||
|
import ru.soune.nocopy.entity.auth.ExternalAuthProvider;
|
||||||
|
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Optional;
|
||||||
|
|
||||||
|
@Repository
|
||||||
|
public interface ExternalAuthProviderRepository extends JpaRepository<ExternalAuthProvider, Long> {
|
||||||
|
|
||||||
|
Optional<ExternalAuthProvider> findByProviderTypeAndProviderUserId(
|
||||||
|
ExternalAuthProvider.ProviderType providerType,
|
||||||
|
String providerUserId
|
||||||
|
);
|
||||||
|
|
||||||
|
List<ExternalAuthProvider> findByUser(User user);
|
||||||
|
|
||||||
|
boolean existsByProviderTypeAndProviderUserId(
|
||||||
|
ExternalAuthProvider.ProviderType providerType,
|
||||||
|
String providerUserId
|
||||||
|
);
|
||||||
|
}
|
||||||
@@ -6,7 +6,6 @@ import org.springframework.data.repository.query.Param;
|
|||||||
import org.springframework.stereotype.Repository;
|
import org.springframework.stereotype.Repository;
|
||||||
import ru.soune.nocopy.entity.file.FileEntity;
|
import ru.soune.nocopy.entity.file.FileEntity;
|
||||||
import ru.soune.nocopy.entity.file.FileStatus;
|
import ru.soune.nocopy.entity.file.FileStatus;
|
||||||
import ru.soune.nocopy.entity.file.ProtectionStatus;
|
|
||||||
|
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.Optional;
|
import java.util.Optional;
|
||||||
@@ -22,17 +21,11 @@ public interface FileEntityRepository extends JpaRepository<FileEntity, String>
|
|||||||
|
|
||||||
Optional<FileEntity> findByUploadSessionId(String uploadSessionId);
|
Optional<FileEntity> findByUploadSessionId(String uploadSessionId);
|
||||||
|
|
||||||
List<FileEntity> findByProtectionStatus(ProtectionStatus protectionStatus);
|
|
||||||
|
|
||||||
boolean existsByFilePath(String filePath);
|
boolean existsByFilePath(String filePath);
|
||||||
|
|
||||||
@Query("SELECT SUM(f.fileSize) FROM FileEntity f WHERE f.userId = :userId AND f.status = 'ACTIVE'")
|
@Query("SELECT SUM(f.fileSize) FROM FileEntity f WHERE f.userId = :userId AND f.status = 'ACTIVE'")
|
||||||
Long getTotalSizeByUserId(@Param("userId") Long userId);
|
Long getTotalSizeByUserId(@Param("userId") Long userId);
|
||||||
|
|
||||||
@Query("SELECT f FROM FileEntity f WHERE f.protectionStatus = null AND f.status = 'ACTIVE' OR " +
|
|
||||||
"f.protectionStatus = 'NOT_PROTECTED' AND f.status = 'ACTIVE'")
|
|
||||||
List<FileEntity> findAllActiveFilesAndNotProtected();
|
|
||||||
|
|
||||||
@Query("SELECT f FROM FileEntity f WHERE f.userId = :userId AND f.originalFileName LIKE %:keyword%")
|
@Query("SELECT f FROM FileEntity f WHERE f.userId = :userId AND f.originalFileName LIKE %:keyword%")
|
||||||
List<FileEntity> searchByFileName(@Param("userId") Long userId, @Param("keyword") String keyword);
|
List<FileEntity> searchByFileName(@Param("userId") Long userId, @Param("keyword") String keyword);
|
||||||
|
|
||||||
|
|||||||
@@ -2,7 +2,10 @@ package ru.soune.nocopy.repository;
|
|||||||
|
|
||||||
import jakarta.validation.constraints.Size;
|
import jakarta.validation.constraints.Size;
|
||||||
import org.springframework.data.jpa.repository.JpaRepository;
|
import org.springframework.data.jpa.repository.JpaRepository;
|
||||||
|
import org.springframework.data.jpa.repository.Query;
|
||||||
|
import org.springframework.data.repository.query.Param;
|
||||||
import ru.soune.nocopy.entity.User;
|
import ru.soune.nocopy.entity.User;
|
||||||
|
import ru.soune.nocopy.entity.auth.ExternalAuthProvider;
|
||||||
|
|
||||||
import java.util.Optional;
|
import java.util.Optional;
|
||||||
|
|
||||||
@@ -10,4 +13,10 @@ public interface UserRepository extends JpaRepository<User, Long> {
|
|||||||
Optional<User> findByEmail(String email);
|
Optional<User> findByEmail(String email);
|
||||||
boolean existsByEmail(String email);
|
boolean existsByEmail(String email);
|
||||||
boolean existsByPhone(@Size(min = 11, max = 14) String phone);
|
boolean existsByPhone(@Size(min = 11, max = 14) String phone);
|
||||||
|
@Query("SELECT u FROM User u JOIN u.externalAuthProviders p " +
|
||||||
|
"WHERE p.providerType = :providerType AND p.providerUserId = :providerUserId")
|
||||||
|
Optional<User> findByExternalProvider(
|
||||||
|
@Param("providerType") ExternalAuthProvider.ProviderType providerType,
|
||||||
|
@Param("providerUserId") String providerUserId
|
||||||
|
);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -6,7 +6,7 @@ import org.springframework.security.crypto.password.PasswordEncoder;
|
|||||||
import org.springframework.stereotype.Service;
|
import org.springframework.stereotype.Service;
|
||||||
import org.springframework.transaction.annotation.Transactional;
|
import org.springframework.transaction.annotation.Transactional;
|
||||||
import ru.soune.nocopy.dto.*;
|
import ru.soune.nocopy.dto.*;
|
||||||
import ru.soune.nocopy.entity.AuthToken;
|
import ru.soune.nocopy.entity.auth.AuthToken;
|
||||||
import ru.soune.nocopy.entity.User;
|
import ru.soune.nocopy.entity.User;
|
||||||
import ru.soune.nocopy.exception.NotValidFieldException;
|
import ru.soune.nocopy.exception.NotValidFieldException;
|
||||||
import ru.soune.nocopy.repository.AuthTokenRepository;
|
import ru.soune.nocopy.repository.AuthTokenRepository;
|
||||||
@@ -87,6 +87,20 @@ public class AuthService {
|
|||||||
.ifPresent(authTokenRepository::delete);
|
.ifPresent(authTokenRepository::delete);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Transactional
|
||||||
|
public AuthToken createTokenForUser(User user) {
|
||||||
|
AuthToken authToken = new AuthToken();
|
||||||
|
authToken.setToken(generateAuthToken());
|
||||||
|
authToken.setUser(user);
|
||||||
|
authToken.setExpiresAt(LocalDateTime.now().plusHours(1));
|
||||||
|
authToken.setIsActive(true);
|
||||||
|
|
||||||
|
user.setLastLoginAt(LocalDateTime.now());
|
||||||
|
userRepository.save(user);
|
||||||
|
|
||||||
|
return authTokenRepository.save(authToken);
|
||||||
|
}
|
||||||
|
|
||||||
private String generateAuthToken() {
|
private String generateAuthToken() {
|
||||||
byte[] bytes = new byte[32];
|
byte[] bytes = new byte[32];
|
||||||
secureRandom.nextBytes(bytes);
|
secureRandom.nextBytes(bytes);
|
||||||
|
|||||||
@@ -0,0 +1,69 @@
|
|||||||
|
package ru.soune.nocopy.service;
|
||||||
|
|
||||||
|
import org.springframework.beans.factory.annotation.Value;
|
||||||
|
import org.springframework.http.*;
|
||||||
|
import org.springframework.stereotype.Service;
|
||||||
|
import org.springframework.util.LinkedMultiValueMap;
|
||||||
|
import org.springframework.util.MultiValueMap;
|
||||||
|
import org.springframework.web.client.RestTemplate;
|
||||||
|
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.Map;
|
||||||
|
|
||||||
|
@Service
|
||||||
|
public class VkOAuthService {
|
||||||
|
@Value("${auth.vk.oauth2}")
|
||||||
|
String oauthVkPath;
|
||||||
|
|
||||||
|
@Value("${auth.vk.user_info}")
|
||||||
|
String vkUserInfo;
|
||||||
|
|
||||||
|
@Value("${auth.vk.client_id}")
|
||||||
|
String clientId;
|
||||||
|
|
||||||
|
@Value("${auth.vk.redirect_uri}")
|
||||||
|
String redirectUri;
|
||||||
|
|
||||||
|
public Map<String, Object> authTokenMap(String code, String state, String codeVerifier, String deviceId) {
|
||||||
|
MultiValueMap<String, String> params = new LinkedMultiValueMap<>();
|
||||||
|
params.add("grant_type", "authorization_code");
|
||||||
|
params.add("code_verifier", codeVerifier);
|
||||||
|
params.add("redirect_uri", redirectUri);
|
||||||
|
params.add("code", code);
|
||||||
|
params.add("client_id", clientId);
|
||||||
|
params.add("device_id", deviceId);
|
||||||
|
params.add("state", state);
|
||||||
|
params.add("scope", "email,phone");
|
||||||
|
|
||||||
|
HttpHeaders headers = new HttpHeaders();
|
||||||
|
headers.setContentType(MediaType.APPLICATION_FORM_URLENCODED);
|
||||||
|
|
||||||
|
return sendRequest(params, headers,oauthVkPath);
|
||||||
|
}
|
||||||
|
|
||||||
|
public Map<String, Object> userVkInfo(String accessToken) {
|
||||||
|
MultiValueMap<String, String> params = new LinkedMultiValueMap<>();
|
||||||
|
params.add("access_token", accessToken);
|
||||||
|
params.add("client_id", clientId);
|
||||||
|
|
||||||
|
HttpHeaders headers = new HttpHeaders();
|
||||||
|
headers.setContentType(MediaType.APPLICATION_FORM_URLENCODED);
|
||||||
|
|
||||||
|
|
||||||
|
return sendRequest(params, headers, vkUserInfo);
|
||||||
|
}
|
||||||
|
|
||||||
|
private Map<String, Object> sendRequest(MultiValueMap<String, String> params, HttpHeaders headers, String url) {
|
||||||
|
Map<String, Object> info = new HashMap<>();
|
||||||
|
RestTemplate restTemplate = new RestTemplate();
|
||||||
|
ResponseEntity<String> response = restTemplate.exchange(url, HttpMethod.POST,
|
||||||
|
new HttpEntity<>(params, headers), String.class);
|
||||||
|
|
||||||
|
info.put("status", response.getStatusCodeValue());
|
||||||
|
info.put("body", response.getBody());
|
||||||
|
info.put("headers", response.getHeaders());
|
||||||
|
info.put("success", response.getStatusCode().is2xxSuccessful());
|
||||||
|
|
||||||
|
return info;
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,102 @@
|
|||||||
|
package ru.soune.nocopy.service.auth;
|
||||||
|
|
||||||
|
import lombok.RequiredArgsConstructor;
|
||||||
|
import org.springframework.stereotype.Service;
|
||||||
|
import org.springframework.transaction.annotation.Transactional;
|
||||||
|
import ru.soune.nocopy.entity.User;
|
||||||
|
import ru.soune.nocopy.entity.auth.ExternalAuthProvider;
|
||||||
|
import ru.soune.nocopy.repository.ExternalAuthProviderRepository;
|
||||||
|
import ru.soune.nocopy.repository.UserRepository;
|
||||||
|
|
||||||
|
import java.time.LocalDateTime;
|
||||||
|
import java.util.Optional;
|
||||||
|
|
||||||
|
@Service
|
||||||
|
@RequiredArgsConstructor
|
||||||
|
public class ExternalAuthUserService {
|
||||||
|
private final UserRepository userRepository;
|
||||||
|
|
||||||
|
private final ExternalAuthProviderRepository externalAuthProviderRepository;
|
||||||
|
|
||||||
|
@Transactional
|
||||||
|
public User findOrCreateUserFromVK(String vkUserId, String fullName, String email,
|
||||||
|
String vkAccessToken, String vkRefreshToken, Integer expiresIn) {
|
||||||
|
|
||||||
|
Optional<User> existingByVkId = userRepository.findByExternalProvider(ExternalAuthProvider.ProviderType.VK,
|
||||||
|
vkUserId);
|
||||||
|
|
||||||
|
if (existingByVkId.isPresent()) {
|
||||||
|
User user = existingByVkId.get();
|
||||||
|
updateVKProvider(user, vkAccessToken, vkRefreshToken, expiresIn, email);
|
||||||
|
|
||||||
|
return user;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (email != null && !email.isBlank()) {
|
||||||
|
Optional<User> existingByEmail = userRepository.findByEmail(email);
|
||||||
|
|
||||||
|
if (existingByEmail.isPresent()) {
|
||||||
|
User user = existingByEmail.get();
|
||||||
|
linkVKProvider(user, vkUserId, vkAccessToken, vkRefreshToken, expiresIn, email);
|
||||||
|
|
||||||
|
return user;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return createNewUserWithVK(vkUserId, fullName, email, vkAccessToken, vkRefreshToken, expiresIn);
|
||||||
|
}
|
||||||
|
|
||||||
|
private User createNewUserWithVK(String vkUserId, String fullName, String email,
|
||||||
|
String vkAccessToken, String vkRefreshToken, Integer expiresIn) {
|
||||||
|
User newUser = User.builder()
|
||||||
|
.fullName(fullName)
|
||||||
|
.email(email)
|
||||||
|
.password("generateRandomPassword()")
|
||||||
|
.isActive(true)
|
||||||
|
.build();
|
||||||
|
|
||||||
|
User savedUser = userRepository.save(newUser);
|
||||||
|
|
||||||
|
linkVKProvider(savedUser, vkUserId, vkAccessToken, vkRefreshToken, expiresIn, email);
|
||||||
|
|
||||||
|
return savedUser;
|
||||||
|
}
|
||||||
|
|
||||||
|
private void linkVKProvider(User user, String vkUserId, String vkAccessToken, String vkRefreshToken,
|
||||||
|
Integer expiresIn, String email) {
|
||||||
|
ExternalAuthProvider provider = ExternalAuthProvider.builder()
|
||||||
|
.user(user)
|
||||||
|
.providerType(ExternalAuthProvider.ProviderType.VK)
|
||||||
|
.providerUserId(vkUserId)
|
||||||
|
.accessToken(vkAccessToken)
|
||||||
|
.refreshToken(vkRefreshToken)
|
||||||
|
.tokenExpiresAt(LocalDateTime.now().plusSeconds(expiresIn))
|
||||||
|
.email(email)
|
||||||
|
.isActive(true)
|
||||||
|
.build();
|
||||||
|
|
||||||
|
externalAuthProviderRepository.save(provider);
|
||||||
|
|
||||||
|
user.getExternalAuthProviders().add(provider);
|
||||||
|
}
|
||||||
|
|
||||||
|
private void updateVKProvider(User user, String vkAccessToken, String vkRefreshToken,
|
||||||
|
Integer expiresIn, String email) {
|
||||||
|
Optional<ExternalAuthProvider> existingProvider = user.getProvider(ExternalAuthProvider.ProviderType.VK);
|
||||||
|
|
||||||
|
if (existingProvider.isPresent()) {
|
||||||
|
ExternalAuthProvider provider = existingProvider.get();
|
||||||
|
provider.setAccessToken(vkAccessToken);
|
||||||
|
provider.setTokenExpiresAt(LocalDateTime.now().plusSeconds(expiresIn));
|
||||||
|
provider.setEmail(email);
|
||||||
|
provider.setRefreshToken(vkRefreshToken);
|
||||||
|
provider.setLastUsedAt(LocalDateTime.now());
|
||||||
|
|
||||||
|
externalAuthProviderRepository.save(provider);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private String generateRandomPassword() {
|
||||||
|
return java.util.UUID.randomUUID().toString();
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,72 +0,0 @@
|
|||||||
package ru.soune.nocopy.service.file;
|
|
||||||
|
|
||||||
import com.vrt.NoCopyFileService;
|
|
||||||
import com.vrt.fileprotection.FileProtector;
|
|
||||||
import lombok.RequiredArgsConstructor;
|
|
||||||
import lombok.extern.slf4j.Slf4j;
|
|
||||||
import org.springframework.scheduling.annotation.Scheduled;
|
|
||||||
import org.springframework.stereotype.Service;
|
|
||||||
import ru.soune.nocopy.entity.file.FileEntity;
|
|
||||||
import ru.soune.nocopy.entity.file.ProtectionStatus;
|
|
||||||
import ru.soune.nocopy.repository.FileEntityRepository;
|
|
||||||
|
|
||||||
import java.util.List;
|
|
||||||
|
|
||||||
@Slf4j
|
|
||||||
@Service
|
|
||||||
@RequiredArgsConstructor
|
|
||||||
public class FileProcessingOrchestrator {
|
|
||||||
|
|
||||||
private final NoCopyFileService noCopyFileService;
|
|
||||||
|
|
||||||
private final FileEntityRepository fileRepository;
|
|
||||||
|
|
||||||
public void initializeProcessingQueue() {
|
|
||||||
List<FileEntity> filesToProtect = fileRepository.findByProtectionStatus(ProtectionStatus.NOT_PROTECTED);
|
|
||||||
|
|
||||||
for (FileEntity fileEntity : filesToProtect) {
|
|
||||||
try {
|
|
||||||
FileProtector.FileInfo fileInfo = createFileInfo(fileEntity);
|
|
||||||
|
|
||||||
noCopyFileService.addFile(fileInfo);
|
|
||||||
|
|
||||||
log.info("Add to query: {}", fileEntity.getOriginalFileName());
|
|
||||||
} catch (Exception e) {
|
|
||||||
log.error("Fail add to query: {}", fileEntity.getId(), e);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@Scheduled(fixedDelay = 120000)
|
|
||||||
public void checkNewFilesForProtection() {
|
|
||||||
List<FileEntity> newFiles = fileRepository.findAllActiveFilesAndNotProtected();
|
|
||||||
|
|
||||||
for (FileEntity fileEntity : newFiles) {
|
|
||||||
FileProtector.FileInfo fileInfo = createFileInfo(fileEntity);
|
|
||||||
|
|
||||||
noCopyFileService.addFile(fileInfo);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private FileProtector.FileInfo createFileInfo(FileEntity fileEntity) {
|
|
||||||
FileProtector.Type type = determineFileType(fileEntity.getMimeType());
|
|
||||||
|
|
||||||
return new FileProtector.FileInfo(type, fileEntity.getId(), String.valueOf(fileEntity.getUserId()));
|
|
||||||
}
|
|
||||||
|
|
||||||
private FileProtector.Type determineFileType(String mimeType) {
|
|
||||||
if (mimeType == null) {
|
|
||||||
return FileProtector.Type.IMAGE;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (mimeType.startsWith("image")) {
|
|
||||||
return FileProtector.Type.IMAGE;
|
|
||||||
} else if (mimeType.startsWith("video")) {
|
|
||||||
return FileProtector.Type.VIDEO;
|
|
||||||
} else if (mimeType.startsWith("audio")) {
|
|
||||||
return FileProtector.Type.AUDIO;
|
|
||||||
} else {
|
|
||||||
return FileProtector.Type.IMAGE;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,46 +0,0 @@
|
|||||||
package ru.soune.nocopy.service.file;
|
|
||||||
|
|
||||||
import com.vrt.fileprotection.FileProtector;
|
|
||||||
import lombok.AllArgsConstructor;
|
|
||||||
import lombok.extern.slf4j.Slf4j;
|
|
||||||
import org.springframework.stereotype.Component;
|
|
||||||
import ru.soune.nocopy.entity.file.FileEntity;
|
|
||||||
import ru.soune.nocopy.entity.file.ProtectionStatus;
|
|
||||||
import ru.soune.nocopy.exception.FileEntityNotFoundException;
|
|
||||||
import ru.soune.nocopy.repository.FileEntityRepository;
|
|
||||||
|
|
||||||
@Slf4j
|
|
||||||
@Component
|
|
||||||
@AllArgsConstructor
|
|
||||||
public class NoCopyProcessingListener implements FileProtector.ProcessingListener {
|
|
||||||
|
|
||||||
private final FileEntityRepository fileEntityRepository;
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void onStartProcessing(FileProtector.FileInfo fileInfo) {
|
|
||||||
changeStatus(ProtectionStatus.PROCESSING, fileInfo.getId());
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void onProcessingFailed(FileProtector.FileInfo fileInfo) {
|
|
||||||
changeStatus(ProtectionStatus.FAILED, fileInfo.getId());
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void onSavingFailed(FileProtector.FileInfo fileInfo) {
|
|
||||||
changeStatus(ProtectionStatus.FAILED_SAVE, fileInfo.getId());
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public void onFinish(FileProtector.FileInfo fileInfo) {
|
|
||||||
changeStatus(ProtectionStatus.PROTECTED, fileInfo.getId());;
|
|
||||||
}
|
|
||||||
|
|
||||||
private void changeStatus(ProtectionStatus newStatus, String fileId) {
|
|
||||||
FileEntity fileEntity = fileEntityRepository.findById(fileId)
|
|
||||||
.orElseThrow(() -> new FileEntityNotFoundException(fileId));
|
|
||||||
fileEntity.setProtectionStatus(newStatus);
|
|
||||||
|
|
||||||
fileEntityRepository.save(fileEntity);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,166 +0,0 @@
|
|||||||
package ru.soune.nocopy.service.file;
|
|
||||||
|
|
||||||
import com.vrt.fileprotection.FileProtector;
|
|
||||||
import com.vrt.fileprotection.OperationResult;
|
|
||||||
import lombok.RequiredArgsConstructor;
|
|
||||||
import lombok.extern.slf4j.Slf4j;
|
|
||||||
import org.jetbrains.annotations.NotNull;
|
|
||||||
import org.jetbrains.annotations.Nullable;
|
|
||||||
import org.springframework.stereotype.Component;
|
|
||||||
import ru.soune.nocopy.entity.file.FileEntity;
|
|
||||||
import ru.soune.nocopy.repository.FileEntityRepository;
|
|
||||||
|
|
||||||
import java.io.File;
|
|
||||||
import java.io.IOException;
|
|
||||||
import java.nio.file.Files;
|
|
||||||
import java.nio.file.Path;
|
|
||||||
import java.nio.file.Paths;
|
|
||||||
import java.time.LocalDateTime;
|
|
||||||
|
|
||||||
@Slf4j
|
|
||||||
@Component
|
|
||||||
@RequiredArgsConstructor
|
|
||||||
public class ProtectionFileProvider implements FileProtector.FileProvider {
|
|
||||||
|
|
||||||
|
|
||||||
private final FileEntityRepository fileRepository;
|
|
||||||
|
|
||||||
@Nullable
|
|
||||||
@Override
|
|
||||||
public File getImageFile(@NotNull String id) {
|
|
||||||
return getFileById(id);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Nullable
|
|
||||||
@Override
|
|
||||||
public File getVideoFile(@NotNull String id) {
|
|
||||||
return getFileById(id);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Nullable
|
|
||||||
@Override
|
|
||||||
public File getAudioFile(@NotNull String id) {
|
|
||||||
return getFileById(id);
|
|
||||||
}
|
|
||||||
|
|
||||||
@NotNull
|
|
||||||
@Override
|
|
||||||
public com.vrt.fileprotection.OperationResult writeSignature(@NotNull String s, @NotNull byte[] bytes) {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
@NotNull
|
|
||||||
@Override
|
|
||||||
public com.vrt.fileprotection.OperationResult writeAudioFile(@NotNull String id, @NotNull byte[] data, @Nullable String fileExt) {
|
|
||||||
try {
|
|
||||||
writeProtectedFile(id, data, fileExt);
|
|
||||||
return OperationResult.Companion.success();
|
|
||||||
} catch (Exception e) {
|
|
||||||
log.error("Failed to create protected file: {}", id, e);
|
|
||||||
return OperationResult.Companion.failure("Failed to create protected file");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@NotNull
|
|
||||||
@Override
|
|
||||||
public com.vrt.fileprotection.OperationResult writeVideoFile(@NotNull String id, @NotNull byte[] data, @Nullable String fileExt) {
|
|
||||||
try {
|
|
||||||
writeProtectedFile(id, data, fileExt);
|
|
||||||
return OperationResult.Companion.success();
|
|
||||||
} catch (Exception e) {
|
|
||||||
log.error("Failed to create protected file: {}", id, e);
|
|
||||||
return OperationResult.Companion.failure("Failed to create protected file");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@NotNull
|
|
||||||
@Override
|
|
||||||
public com.vrt.fileprotection.OperationResult writeImageFile(@NotNull String id, @NotNull byte[] data, @Nullable String fileExt) {
|
|
||||||
try {
|
|
||||||
writeProtectedFile(id, data, fileExt);
|
|
||||||
return OperationResult.Companion.success();
|
|
||||||
} catch (Exception e) {
|
|
||||||
log.error("Failed to create protected file: {}", id, e);
|
|
||||||
return OperationResult.Companion.failure("Failed to create protected file");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@Nullable
|
|
||||||
@Override
|
|
||||||
public File getSignature(@NotNull String s) {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
private void writeProtectedFile(String id, byte[] data, String fileExt) throws IOException {
|
|
||||||
FileEntity fileEntity = fileRepository.findById(id)
|
|
||||||
.orElseThrow(() -> new RuntimeException("File not found: " + id));
|
|
||||||
|
|
||||||
String extension = determineFileExtension(fileExt, fileEntity);
|
|
||||||
Path protectedFilePath = prepareProtectedPath(fileEntity, extension);
|
|
||||||
|
|
||||||
if (Files.exists(protectedFilePath)) {
|
|
||||||
Files.delete(protectedFilePath);
|
|
||||||
}
|
|
||||||
|
|
||||||
Files.write(protectedFilePath, data);
|
|
||||||
|
|
||||||
fileEntity.setProtectedFilePath(protectedFilePath.toString());
|
|
||||||
fileEntity.setProtectedAt(LocalDateTime.now());
|
|
||||||
fileEntity.setUpdatedAt(LocalDateTime.now());
|
|
||||||
fileEntity.setFileExtension(extension);
|
|
||||||
|
|
||||||
fileRepository.save(fileEntity);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
private File getFileById(String id) {
|
|
||||||
try {
|
|
||||||
FileEntity fileEntity = fileRepository.findById(id).orElseThrow(() ->
|
|
||||||
new RuntimeException("File not found: " + id));
|
|
||||||
|
|
||||||
File file = new File(fileEntity.getFilePath());
|
|
||||||
|
|
||||||
if (!file.exists()) {
|
|
||||||
throw new RuntimeException("File not found on disk: " + fileEntity.getFilePath());
|
|
||||||
}
|
|
||||||
|
|
||||||
return file;
|
|
||||||
} catch (Exception e) {
|
|
||||||
log.error("Error getting file: {}", id, e);
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private String determineFileExtension(String fileExt, FileEntity fileEntity) {
|
|
||||||
if (fileExt != null && !fileExt.trim().isEmpty()) {
|
|
||||||
return fileExt;
|
|
||||||
} else {
|
|
||||||
return fileEntity.getFileExtension();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private Path prepareProtectedPath(FileEntity fileEntity, String extension) throws IOException {
|
|
||||||
Path originalPath = Paths.get(fileEntity.getFilePath());
|
|
||||||
|
|
||||||
String pathStr = originalPath.toString();
|
|
||||||
pathStr = pathStr.replaceFirst("/uploads/uploads/", "/uploads/protected/");
|
|
||||||
|
|
||||||
Path protectedPath = Paths.get(pathStr);
|
|
||||||
String fileName = protectedPath.getFileName().toString();
|
|
||||||
|
|
||||||
if (extension != null) {
|
|
||||||
String nameWithoutExt = fileName;
|
|
||||||
int lastDotIndex = fileName.lastIndexOf('.');
|
|
||||||
if (lastDotIndex > 0) {
|
|
||||||
nameWithoutExt = fileName.substring(0, lastDotIndex);
|
|
||||||
}
|
|
||||||
fileName = nameWithoutExt + "." + extension;
|
|
||||||
|
|
||||||
protectedPath = protectedPath.getParent().resolve(fileName);
|
|
||||||
}
|
|
||||||
|
|
||||||
Files.createDirectories(protectedPath.getParent());
|
|
||||||
|
|
||||||
return protectedPath;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -25,13 +25,20 @@ file:
|
|||||||
storage:
|
storage:
|
||||||
base-path: ${FILE_STORAGE_PATH:/data/uploads}
|
base-path: ${FILE_STORAGE_PATH:/data/uploads}
|
||||||
# chunk-size: ${FILE_CHUNK_SIZE:5242880} # 5MB
|
# chunk-size: ${FILE_CHUNK_SIZE:5242880} # 5MB
|
||||||
chunk-size: ${FILE_CHUNK_SIZE:1048576} # 1MB
|
chunk-size: ${FILE_CHUNK_SIZE:1048576} # 15MB
|
||||||
max-file-size: ${MAX_FILE_SIZE:10737418240} # 10GB
|
max-file-size: ${MAX_FILE_SIZE:10737418240} # 10GB
|
||||||
max-retry-attempts: ${MAX_RETRY_ATTEMPTS:3}
|
max-retry-attempts: ${MAX_RETRY_ATTEMPTS:3}
|
||||||
chunk-timeout-ms: ${CHUNK_TIMEOUT_MS:300000} # 5 минут
|
chunk-timeout-ms: ${CHUNK_TIMEOUT_MS:300000} # 5 минут
|
||||||
temp-ttl-hours: ${TEMP_TTL_HOURS:72} # 3 дня
|
temp-ttl-hours: ${TEMP_TTL_HOURS:72} # 3 дня
|
||||||
session-expiry-hours: ${SESSION_EXPIRY_HOURS:24}
|
session-expiry-hours: ${SESSION_EXPIRY_HOURS:24}
|
||||||
|
|
||||||
|
auth:
|
||||||
|
vk:
|
||||||
|
oauth2: "https://id.vk.ru/oauth2/auth"
|
||||||
|
user_info: "https://id.vk.ru/oauth2/user_info"
|
||||||
|
client_id: "54389559"
|
||||||
|
redirect_uri: "http://localhost"
|
||||||
|
|
||||||
security:
|
security:
|
||||||
cors:
|
cors:
|
||||||
allowed-origins: "*"
|
allowed-origins: "*"
|
||||||
|
|||||||
@@ -9,7 +9,7 @@ import org.mockito.MockitoAnnotations;
|
|||||||
import org.springframework.context.MessageSource;
|
import org.springframework.context.MessageSource;
|
||||||
import org.springframework.security.crypto.password.PasswordEncoder;
|
import org.springframework.security.crypto.password.PasswordEncoder;
|
||||||
import ru.soune.nocopy.dto.LoginRequest;
|
import ru.soune.nocopy.dto.LoginRequest;
|
||||||
import ru.soune.nocopy.entity.AuthToken;
|
import ru.soune.nocopy.entity.auth.AuthToken;
|
||||||
import ru.soune.nocopy.entity.User;
|
import ru.soune.nocopy.entity.User;
|
||||||
import ru.soune.nocopy.exception.NotValidationPasswordException;
|
import ru.soune.nocopy.exception.NotValidationPasswordException;
|
||||||
import ru.soune.nocopy.exception.UserNotFoundException;
|
import ru.soune.nocopy.exception.UserNotFoundException;
|
||||||
|
|||||||
Reference in New Issue
Block a user