71 lines
2.8 KiB
Java
71 lines
2.8 KiB
Java
package com.sshmanager.controller;
|
|
|
|
import com.sshmanager.dto.ConnectionCreateRequest;
|
|
import com.sshmanager.dto.ConnectionDto;
|
|
import com.sshmanager.entity.User;
|
|
import com.sshmanager.repository.UserRepository;
|
|
import com.sshmanager.service.ConnectionService;
|
|
import org.springframework.http.ResponseEntity;
|
|
import org.springframework.security.core.Authentication;
|
|
import org.springframework.web.bind.annotation.*;
|
|
|
|
import java.util.HashMap;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
|
|
@RestController
|
|
@RequestMapping("/api/connections")
|
|
public class ConnectionController {
|
|
|
|
private final ConnectionService connectionService;
|
|
private final UserRepository userRepository;
|
|
|
|
public ConnectionController(ConnectionService connectionService,
|
|
UserRepository userRepository) {
|
|
this.connectionService = connectionService;
|
|
this.userRepository = userRepository;
|
|
}
|
|
|
|
private Long getCurrentUserId(Authentication auth) {
|
|
User user = userRepository.findByUsername(auth.getName()).orElseThrow(() -> new IllegalStateException("User not found"));
|
|
return user.getId();
|
|
}
|
|
|
|
@GetMapping
|
|
public ResponseEntity<List<ConnectionDto>> list(Authentication authentication) {
|
|
Long userId = getCurrentUserId(authentication);
|
|
return ResponseEntity.ok(connectionService.listByUserId(userId));
|
|
}
|
|
|
|
@GetMapping("/{id}")
|
|
public ResponseEntity<ConnectionDto> get(@PathVariable Long id, Authentication authentication) {
|
|
Long userId = getCurrentUserId(authentication);
|
|
return ResponseEntity.ok(connectionService.getById(id, userId));
|
|
}
|
|
|
|
@PostMapping
|
|
public ResponseEntity<ConnectionDto> create(@RequestBody ConnectionCreateRequest request,
|
|
Authentication authentication) {
|
|
Long userId = getCurrentUserId(authentication);
|
|
return ResponseEntity.ok(connectionService.create(request, userId));
|
|
}
|
|
|
|
@PutMapping("/{id}")
|
|
public ResponseEntity<ConnectionDto> update(@PathVariable Long id,
|
|
@RequestBody ConnectionCreateRequest request,
|
|
Authentication authentication) {
|
|
Long userId = getCurrentUserId(authentication);
|
|
return ResponseEntity.ok(connectionService.update(id, request, userId));
|
|
}
|
|
|
|
@DeleteMapping("/{id}")
|
|
public ResponseEntity<Map<String, String>> delete(@PathVariable Long id,
|
|
Authentication authentication) {
|
|
Long userId = getCurrentUserId(authentication);
|
|
connectionService.delete(id, userId);
|
|
Map<String, String> result = new HashMap<>();
|
|
result.put("message", "Deleted");
|
|
return ResponseEntity.ok(result);
|
|
}
|
|
}
|