RoomPicker/app/src/main/java/ru/dragonestia/picker/controller/BucketController.java

package ru.dragonestia.picker.controller;

import lombok.RequiredArgsConstructor;
import lombok.extern.log4j.Log4j2;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*;
import ru.dragonestia.picker.controller.response.BucketInfoResponse;
import ru.dragonestia.picker.controller.response.BucketListResponse;
import ru.dragonestia.picker.controller.response.BucketRegisterResponse;
import ru.dragonestia.picker.model.Bucket;
import ru.dragonestia.picker.model.type.SlotLimit;
import ru.dragonestia.picker.service.BucketService;
import ru.dragonestia.picker.service.NodeService;
import ru.dragonestia.picker.util.NamingValidator;

import java.util.Objects;

@Log4j2
@RestController
@RequestMapping("/nodes/{nodeIdentifier}/buckets")
@RequiredArgsConstructor
public class BucketController {

    private final NodeService nodeService;
    private final BucketService bucketService;

    @GetMapping
    ResponseEntity<BucketListResponse> allBuckets(@PathVariable(name = "nodeIdentifier") String nodeId) {
        var nodeOpt = nodeService.findNode(nodeId);
        return nodeOpt.map(node -> ResponseEntity.ok(new BucketListResponse(nodeId,
                bucketService.allBuckets(node).stream()
                        .map(bucket -> new BucketListResponse.BucketDTO(bucket.getIdentifier(), bucket.getSlots().getSlots(), bucket.isLocked()))
                        .toList()
        ))).orElseGet(() -> ResponseEntity.notFound().build());
    }

    @PostMapping
    ResponseEntity<BucketRegisterResponse> registerBucket(@PathVariable(name = "nodeIdentifier") String nodeId,
                                                          @RequestParam(name = "identifier") String bucketId,
                                                          @RequestParam(name = "slots") int slots,
                                                          @RequestParam(name = "payload") String payload,
                                                          @RequestParam(name = "locked", defaultValue = "false") boolean locked) {

        var nodeOpt = nodeService.findNode(nodeId);

        if (nodeOpt.isEmpty()) {
            return ResponseEntity.status(404)
                    .body(new BucketRegisterResponse(false, "Node does not exist"));
        }

        var bucket = Bucket.create(bucketId, nodeOpt.get(), SlotLimit.of(slots), payload);
        bucket.setLocked(locked);
        try {
            bucketService.createBucket(bucket);
            return ResponseEntity.ok(new BucketRegisterResponse(true, ""));
        } catch (Error error) {
            return ResponseEntity.status(400).body(new BucketRegisterResponse(false, error.getMessage()));
        } catch (Exception ex) {
            return ResponseEntity.status(500).body(new BucketRegisterResponse(false, ex.getMessage()));
        }
    }

    @DeleteMapping("/{identifier}")
    ResponseEntity<?> removeBucket(@PathVariable("nodeIdentifier") String nodeId,
                                   @PathVariable("identifier") String bucketId) {
        if (!NamingValidator.validateNodeIdentifier(nodeId) || !NamingValidator.validateBucketIdentifier(bucketId)) {
            return ResponseEntity.ok().build();
        }

        var nodeOpt = nodeService.findNode(nodeId);
        nodeOpt.flatMap(node -> bucketService.findBucket(node, bucketId))
                .ifPresent(bucketService::removeBucket);

        return ResponseEntity.ok().build();
    }

    @GetMapping("/{identifier}")
    ResponseEntity<BucketInfoResponse> info(@PathVariable("nodeIdentifier") String nodeId,
                                            @PathVariable("identifier") String bucketId) {
        if (!NamingValidator.validateNodeIdentifier(nodeId) || !NamingValidator.validateBucketIdentifier(bucketId)) {
            return ResponseEntity.ok().build();
        }

        var nodeOpt = nodeService.findNode(nodeId);
        if (nodeOpt.isEmpty()) {
            return ResponseEntity.notFound().build();
        }

        var bucketOpt = bucketService.findBucket(Objects.requireNonNull(nodeOpt.get()), bucketId);
        return bucketOpt.map(bucket -> ResponseEntity.ok(new BucketInfoResponse(bucket)))
                .orElseGet(() -> ResponseEntity.notFound().build());
    }

    @PostMapping("/{identifier}/lock")
    ResponseEntity<Boolean> lockBucket(@PathVariable("nodeIdentifier") String nodeId,
                                 @PathVariable("identifier") String bucketId,
                                 @RequestParam(name = "state") boolean value) {

        if (!NamingValidator.validateNodeIdentifier(nodeId) || !NamingValidator.validateBucketIdentifier(bucketId)) {
            return ResponseEntity.notFound().build();
        }

        var nodeOpt = nodeService.findNode(nodeId);
        if (nodeOpt.isEmpty()) {
            return ResponseEntity.notFound().build();
        }

        var bucketOpt = bucketService.findBucket(Objects.requireNonNull(nodeOpt.get()), bucketId);
        if (bucketOpt.isEmpty()) {
            return ResponseEntity.notFound().build();
        }

        var bucket = bucketOpt.get();
        bucket.setLocked(value);
        return ResponseEntity.ok(true);
    }
}