raex_back/src/main/java/com/izouma/nineth/web/MintActivityController.java

package com.izouma.nineth.web;

import com.izouma.nineth.domain.Collection;
import com.izouma.nineth.domain.MintActivity;
import com.izouma.nineth.domain.Tag;
import com.izouma.nineth.dto.MintActivityRule;
import com.izouma.nineth.dto.nftdomain.CreateCollection;
import com.izouma.nineth.dto.nftdomain.CreateCollectionDTO;
import com.izouma.nineth.enums.CollectionType;
import com.izouma.nineth.repo.CollectionRepo;
import com.izouma.nineth.service.MintActivityService;
import com.izouma.nineth.dto.PageQuery;
import com.izouma.nineth.exception.BusinessException;
import com.izouma.nineth.repo.MintActivityRepo;
import com.izouma.nineth.service.SysConfigService;
import com.izouma.nineth.utils.ObjUtils;
import com.izouma.nineth.utils.excel.ExcelUtils;
import freemarker.ext.beans.BeansWrapper;
import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.cache.annotation.Cacheable;
import org.springframework.data.domain.Page;
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.Set;
import java.util.stream.Collectors;

@Slf4j
@RestController
@RequestMapping("/mintActivity")
@AllArgsConstructor
public class MintActivityController extends BaseController {
    private MintActivityService mintActivityService;
    private MintActivityRepo    mintActivityRepo;
    private CollectionRepo      collectionRepo;
    private SysConfigService    sysConfigService;

    @PreAuthorize("hasAnyRole('ADMIN','SAAS')")
    @PostMapping("/save")
    public MintActivity save(@RequestBody MintActivity record) {
        if (record.getRule() != null) mintActivityService.checkRule(record.getRule());
//        List<CreateCollection> createCollections = new ArrayList<>();
//        record.getTargetCollectionIds().forEach(collection -> {
//            double hour = collection.getDelayHours();
//            int minutes = (int) (hour * 60);
////            collection.setMinutes(minutes);
//            createCollections.add(collection);
//        });
//        record.setTargetCollectionIds(createCollections);
        if (record.getId() != null) {
            MintActivity orig = mintActivityRepo.findById(record.getId()).orElseThrow(new BusinessException("无记录"));
            orig.setName(record.getName());
            orig.setCover(record.getCover());
            orig.setMinterId(record.getMinterId());
            orig.setMinter(record.getMinter());
            orig.setMinterAvatar(record.getMinterAvatar());
            orig.setDetail(record.getDetail());
            orig.setAudit(record.isAudit());
            orig.setNum(record.getNum());
            orig.setCollectionName(record.getCollectionName());
            orig.setRule(record.getRule());
            orig.setConsume(record.isConsume());
            orig.setGasPrice(record.getGasPrice());
            orig.setAirDrop(record.isAirDrop());
            orig.setAutoDrop(record.isAutoDrop());
            orig.setOnShelf(record.isOnShelf());
            orig.setScheduleSale(record.isScheduleSale());
            orig.setStartTime(record.getStartTime());
            orig.setAutoDrop(record.isAutoDrop());
            orig.setAirDropCollectionId(record.getAirDropCollectionId());
            orig.setScheduleEnd(record.isScheduleEnd());
            orig.setEndTime(record.getEndTime());
            orig.setTargetCollectionIds(record.getTargetCollectionIds());
            orig = mintActivityRepo.save(orig);
            mintActivityService.syncStock(record.getId());
            return orig;
        } else {
            record.setStock(record.getTotal());
            record = mintActivityRepo.save(record);
            mintActivityService.syncStock(record.getId());
            return record;
        }
    }


    //@PreAuthorize("hasRole('ADMIN')")
    @PostMapping("/all")
    public Page<MintActivity> all(@RequestBody PageQuery pageQuery) {
        return mintActivityService.all(pageQuery);
    }

    @GetMapping("/get/{id}")
//    @Cacheable(cacheNames = "mintActivity", key = "#id")
    public MintActivity get(@PathVariable Long id) {
        MintActivity mintActivity = mintActivityRepo.findById(id).orElseThrow(new BusinessException("无记录"));
//        String mintCountCollection = sysConfigService.getString("mint_countCollection");
//        String regex = ",";
//        String[] split = mintCountCollection.split(regex);
//        if (Arrays.asList(split).contains(String.valueOf(id))) {
//            // Id在split数组中
//            return mintActivity;
//        } else {
//            // Id不在split数组中
//        List<Long> tags = mintActivity.getHoldingTags();
//        tags.forEach(tagId -> {
//            List<Collection> collections = collectionRepo.findByTag(tagId);
//            if (collections.stream().anyMatch(collection -> collection.getType().equals(CollectionType.BLIND_BOX)))
//        });
        try {
            MintActivityRule mintActivityRule = mintActivity.getRule();
            List<MintActivityRule> and = mintActivityRule.getAnd();
            mintActivityRule.setAnd(mintActivityService.mintActivityRules(and, mintActivity.getId()));
            mintActivity.setRule(mintActivityRule);
            if (mintActivity.getTargetCollectionIds() != null) {
                if (!mintActivity.getTargetCollectionIds().isEmpty()) {
                    List<CreateCollectionDTO> dtos = new ArrayList<>();
                    mintActivity.getTargetCollectionIds().forEach(createCollection -> {
                        CreateCollectionDTO dto = new CreateCollectionDTO();
                        dto.setCollection(collectionRepo.findById(createCollection.getId())
                                                        .orElse(null));
                        dto.setNum(createCollection.getNum());
                        dtos.add(dto);
                    });
                    mintActivity.setAirDropCollection(dtos);
                }
            }
        } catch (Exception e) {
            log.error("生成铸造内容出错" + id, e);
        }

        mintActivity.setVTotal(mintActivity.getTotal());
        mintActivity.setTotal(0);
        return mintActivity;
//        }
    }

    @PreAuthorize("hasAnyRole('ADMIN','SAAS')")
    @PostMapping("/del/{id}")
    public void del(@PathVariable Long id) {
        mintActivityRepo.softDelete(id);
    }

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