raex_back/src/main/java/com/izouma/nineth/web/nftdomain/CartController.java

package com.izouma.nineth.web.nftdomain;

import com.alipay.api.domain.Car;
import com.izouma.nineth.domain.Collection;
import com.izouma.nineth.domain.DomainOrder;
import com.izouma.nineth.enums.AuthorityName;
import com.izouma.nineth.enums.CollectionType;
import com.izouma.nineth.enums.OrderStatus;
import com.izouma.nineth.repo.CollectionRepo;
import com.izouma.nineth.repo.DomainOrderRepo;
import com.izouma.nineth.repo.OrderRepo;
import com.izouma.nineth.utils.SecurityUtils;
import com.izouma.nineth.web.BaseController;
import com.izouma.nineth.domain.nftdomain.Cart;
import com.izouma.nineth.service.nftdomain.CartService;
import com.izouma.nineth.dto.PageQuery;
import com.izouma.nineth.exception.BusinessException;
import com.izouma.nineth.repo.nftdomain.CartRepo;
import com.izouma.nineth.utils.ObjUtils;
import com.izouma.nineth.utils.excel.ExcelUtils;

import lombok.AllArgsConstructor;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageImpl;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Map;

@RestController
@RequestMapping("/cart")
@AllArgsConstructor
public class CartController extends BaseController {
    private CartService     cartService;
    private CartRepo        cartRepo;
    private CollectionRepo  collectionRepo;
    private DomainOrderRepo domainOrderRepo;
    private OrderRepo       orderRepo;

    @PreAuthorize("hasRole('ADMIN')")
    @PostMapping("/save")
    public Cart save(@RequestBody Cart record) {
        if (record.getId() != null) {
            Cart orig = cartRepo.findById(record.getId()).orElseThrow(new BusinessException("无记录"));
            ObjUtils.merge(orig, record);
            return cartRepo.save(orig);
        }
        return cartRepo.save(record);
    }

    @PostMapping("/create")
    public Cart save(Long collectionId) {
        return cartService.create(collectionId);
    }

    @PostMapping("/status")
    public Map<String, Object> status(Long collectionId) {
        return cartService.collectionCartStatus(collectionId);
    }

    @PostMapping("/myStatus")
    public Map<String, Object> status() {
        return cartService.collectionCartStatus();
    }


    //@PreAuthorize("hasRole('ADMIN')")
    @PostMapping("/all")
    public Page<Cart> all(@RequestBody PageQuery pageQuery) {
        if (!SecurityUtils.hasRole(AuthorityName.ROLE_ADMIN) & !SecurityUtils.hasRole(AuthorityName.ROLE_ORDERINFO)) {
            pageQuery.getQuery().put("userId", SecurityUtils.getAuthenticatedUser().getId());
        }
        Page<Cart> carts = cartService.all(pageQuery);
        List<Cart> content = carts.getContent();
        List<Cart> newResult = new ArrayList<>();
        content.forEach(cart -> {
            if (cart.isEnabled()) {
                Collection collection = collectionRepo.findById(cart.getCollectionId()).orElse(null);
                if (collection != null) {
                    if (!collection.isSalable() || collection.getStock() < 0 || collection.isInPaying()) {
                        if (!collection.isInPaying()) {
                            cart.setEnabled(false);
                            cartRepo.save(cart);
                        } else {
                            if (orderRepo.countByCollectionIdAndUserIdAndStatus(cart.getCollectionId(), cart
                                    .getUserId(), OrderStatus.NOT_PAID) > 0) {
                                cart.setInPaying(true);
                            }
                        }
                    } else {
                        if (collection.getType().equals(CollectionType.DOMAIN)) {
                            String domainName = collection.getName().substring(9);
                            DomainOrder domainOrder = domainOrderRepo
                                    .findFirstByDomainNameAndOrderStatus(domainName, OrderStatus.FINISH);
                            if (domainOrder != null)
                                cart.setEndTime(domainOrder.getEndTime());
                        }
                    }
                } else {
                    cart.setEnabled(false);
                    cartRepo.save(cart);
                }
            }
            newResult.add(cart);
        });
        return new PageImpl<>(newResult, carts.getPageable(), carts.getTotalElements());
    }

    @GetMapping("/get/{id}")
    public Cart get(@PathVariable Long id) {
        return cartRepo.findById(id).orElseThrow(new BusinessException("无记录"));
    }

//    @PostMapping("/del/{id}")
//    public void del(@PathVariable Long id) {
//        cartRepo.softDelete(id);
//    }

    @PostMapping("/del")
    public void del(String cartIds) {
        List<String> ids = Arrays.asList(cartIds.split(","));
        ids.forEach(id -> {
            cartRepo.findById(Long.valueOf(id)).ifPresent(deleteCart -> {
                cartRepo.delete(deleteCart);
            });
        });
    }

    @GetMapping("/excel")
    @PreAuthorize("hasRole('ADMIN')")
    @ResponseBody
    public void excel(HttpServletResponse response, PageQuery pageQuery) throws IOException {
        List<Cart> data = all(pageQuery).getContent();
        ExcelUtils.export(response, data);
    }
}