4dkankan-center-scene/src/main/java/com/fdkankan/scene/service/impl/ScenePlusServiceImpl.java

package com.fdkankan.scene.service.impl;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.io.FileUtil;
import cn.hutool.core.util.StrUtil;
import com.alibaba.fastjson.JSON;
import cn.hutool.extra.qrcode.QrCodeUtil;
import cn.hutool.extra.qrcode.QrConfig;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.fdkankan.common.constant.ErrorCode;
import com.fdkankan.common.constant.PayStatus;
import com.fdkankan.common.constant.SceneKind;
import com.fdkankan.common.constant.SceneSource;
import com.fdkankan.common.constant.SceneStatus;
import com.fdkankan.common.constant.SceneType;
import com.fdkankan.common.exception.BusinessException;
import com.fdkankan.common.util.FileUtils;
import com.fdkankan.model.constants.ConstantFilePath;
import com.fdkankan.model.constants.UploadFilePath;
import com.fdkankan.scene.bean.LaserSceneBean;
import com.fdkankan.scene.bean.SceneJsonBean;
import com.fdkankan.scene.config.FdkkLaserConfig;
import com.fdkankan.scene.entity.SceneEditControls;
import com.fdkankan.scene.entity.SceneEditInfoExt;
import com.fdkankan.scene.oss.OssUtil;
import com.fdkankan.scene.service.ISceneEditControlsService;
import com.fdkankan.scene.service.ISceneEditInfoExtService;
import com.fdkankan.scene.vo.SceneEditControlsVO;
import com.fdkankan.fyun.face.FYunFileServiceInterface;
import com.fdkankan.model.constants.ConstantFilePath;
import com.fdkankan.model.constants.UploadFilePath;
import com.fdkankan.scene.entity.SceneEditInfoExt;
import com.fdkankan.scene.entity.ScenePro;
import com.fdkankan.scene.service.ISceneEditInfoExtService;
import com.fdkankan.scene.service.ISceneProService;
import com.fdkankan.web.response.ResultData;
import com.fdkankan.redis.constant.RedisKey;
import com.fdkankan.redis.constant.RedisLockKey;
import com.fdkankan.redis.util.RedisLockUtil;
import com.fdkankan.redis.util.RedisUtil;
import com.fdkankan.scene.entity.SceneDataDownload;
import com.fdkankan.scene.entity.SceneEditInfo;
import com.fdkankan.scene.entity.ScenePlus;
import com.fdkankan.scene.entity.ScenePlusExt;
import com.fdkankan.scene.mapper.IScenePlusMapper;
import com.fdkankan.scene.service.ISceneDataDownloadService;
import com.fdkankan.scene.service.ISceneEditInfoService;
import com.fdkankan.scene.service.IScenePlusExtService;
import com.fdkankan.scene.service.IScenePlusService;
import com.fdkankan.scene.vo.BaseSceneParamVO;
import com.fdkankan.scene.vo.SceneCheckKeyParamVO;
import com.fdkankan.scene.vo.SceneDataDownloadVO;

import java.io.File;
import java.nio.charset.StandardCharsets;
import java.io.File;
import java.util.List;
import java.util.Objects;
import java.util.UUID;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

/**
 * <p>
 * 场景主表 服务实现类
 * </p>
 *
 * @author 
 * @since 2022-03-16
 */
@Slf4j
@Service
public class ScenePlusServiceImpl extends ServiceImpl<IScenePlusMapper, ScenePlus> implements IScenePlusService {

    @Value("${main.url}")
    private String mainUrl;
    @Value("${scene.pro.new.url}")
    private String sceneProNewUrl;

    @Autowired
    private IScenePlusExtService scenePlusExtService;
    @Autowired
    private IScenePlusService scenePlusService;
    @Autowired
    private ISceneEditInfoService sceneEditInfoService;
    @Autowired
    private ISceneDataDownloadService sceneDataDownloadService;
    @Autowired
    private RedisUtil redisUtil;
    @Autowired
    private RedisLockUtil redisLockUtil;
    @Autowired
    private ISceneEditInfoExtService sceneEditInfoExtService;
    @Autowired
    private ISceneEditControlsService sceneEditControlsService;
    @Autowired
    private OssUtil ossUtil;
    @Autowired
    FdkkLaserConfig fdkkLaserConfig;

    @Autowired
    private ISceneProService sceneProService;
    @Autowired
    private FYunFileServiceInterface fYunFileService;
    @Autowired
    private ISceneEditInfoExtService sceneEditInfoExtService;

    @Override
    public ScenePlus getScenePlusByNum(String num) {
        return this.getOne(new LambdaQueryWrapper<ScenePlus>().eq(ScenePlus::getNum, num));
    }


    @Override
    public ResultData checkKey(SceneCheckKeyParamVO param) throws Exception {
        ScenePlus scenePlus = scenePlusService.getOne(new LambdaQueryWrapper<ScenePlus>()
            .eq(ScenePlus::getNum, param.getNum())
            .eq(ScenePlus::getPayStatus, PayStatus.PAY.code()));
        if (Objects.isNull(scenePlus)) {
            throw new BusinessException(ErrorCode.FAILURE_CODE_5005);
        }
        SceneEditInfo sceneEditInfo = sceneEditInfoService.getByScenePlusId(scenePlus.getId());

        if (Objects.isNull(sceneEditInfo)) {
            throw new BusinessException(ErrorCode.FAILURE_CODE_5005);
        }

        if(!param.getPassword().equals(sceneEditInfo.getScenePassword())){
            throw new BusinessException(ErrorCode.FAILURE_CODE_5021);
        }

        return ResultData.ok();
    }

    @Override
    public ResultData updateScenePlusUserIdByCameraId(List<Long> cameraIdList, Long userId) {
        LambdaUpdateWrapper<ScenePlus> updateWrapper = new LambdaUpdateWrapper<ScenePlus>()
            .set(ScenePlus::getUserId, userId).in(ScenePlus::getCameraId, cameraIdList);
        this.update(updateWrapper);
        return ResultData.ok();
    }

    @Override
    public ResultData moveScene(List<String> numList, Long cameraId, Long userId) {
        LambdaUpdateWrapper<ScenePlus> updateWrapper =
            new LambdaUpdateWrapper<ScenePlus>()
                .set(ScenePlus::getUserId, userId)
                .set(ScenePlus::getCameraId, cameraId)
                .in(ScenePlus::getNum, numList);
        this.update(updateWrapper);
        return ResultData.ok();

    }

    @Override
    public ResultData downLoadZSData(String num) throws Exception {

        if(StrUtil.isEmpty(num)){
            throw new BusinessException(ErrorCode.FAILURE_CODE_3001);
        }

        String version = "v4";
        ScenePlus scenePlus = this.getScenePlusByNum(num);
        ScenePro scenePro = null;
        if(Objects.isNull(scenePlus)){
            scenePro = sceneProService.getByNum(num);
            version = "v3";
            if(Objects.isNull(scenePro)){
                throw new BusinessException(ErrorCode.FAILURE_CODE_5005);
            }
        }

        SceneDataDownload sceneDataDownload = sceneDataDownloadService.findBySceneNum(num);
        if(sceneDataDownload == null){
            throw new BusinessException(ErrorCode.FAILURE_CODE_5025);
        }
        SceneDataDownloadVO sceneDataDownloadVO = BeanUtil.copyProperties(sceneDataDownload, SceneDataDownloadVO.class);
        sceneDataDownloadVO.setVersion(version);
        return ResultData.ok(sceneDataDownloadVO);
    }

    @Override
    public ScenePlus checkSceneAvail(String num) {
        ScenePlus scenePlus = this.getScenePlusByNum(num);
        if(Objects.isNull(scenePlus)){
            throw new BusinessException(ErrorCode.FAILURE_CODE_5005);
        }
        if(!SceneStatus.SUCCESS.code().equals(scenePlus.getSceneStatus())
            && !SceneStatus.NO_DISPLAY.code().equals(scenePlus.getSceneStatus())){
            throw new BusinessException(ErrorCode.FAILURE_CODE_5033);
        }
        if(!PayStatus.PAY.code().equals(scenePlus.getPayStatus())){
            throw new BusinessException(ErrorCode.FAILURE_CODE_5034);
        }

        return scenePlus;
    }

    @Override
    public ResultData updateViewCount(String sceneNum) {
        String key = RedisKey.SCENE_VISIT_CNT;
        if(!redisUtil.hHasKey(key, sceneNum)){
            String lockKey = String.format(RedisLockKey.LOCK_SCENE_VISIT_CNT, sceneNum);
            String lockVal = cn.hutool.core.lang.UUID.randomUUID().toString();
            boolean lock = redisLockUtil.lock(lockKey, lockVal, RedisKey.EXPIRE_TIME_10_MINUTE);
            if(!lock){
                return ResultData.ok();
            }
            try {
                String cnt = redisUtil.hget(key, sceneNum);
                if(StrUtil.isEmpty(cnt)){
                    //标记是否找到了场景，如果找到就不往下找
                    Integer viewCount = null;
                    ScenePlus scenePlus = scenePlusService.getScenePlusByNum(sceneNum);
                    ScenePlusExt scenePlusExt = scenePlusExtService.getScenePlusExtByPlusId(scenePlus.getId());
                    viewCount = scenePlusExt.getViewCount();
                    redisUtil.hset(key,  sceneNum, Objects.isNull(viewCount) ? "0" : String.valueOf(viewCount));
                }
            }finally {
                redisLockUtil.unlockLua(lockKey, lockVal);
            }
        }
        redisUtil.hincr(key, sceneNum, 1);
        return ResultData.ok();
    }

    @Override
    public ResultData uploadShareLogo(String num, MultipartFile file) throws Exception {

        ScenePlus scenePlus = this.getScenePlusByNum(num);
        if (Objects.isNull(scenePlus)) {
            return ResultData.error(ErrorCode.FAILURE_CODE_5005);
        }
        ScenePlusExt scenePlusExt = scenePlusExtService.getScenePlusExtByPlusId(scenePlus.getId());
        SceneEditInfoExt sceneEditInfoExt = sceneEditInfoExtService.getByScenePlusId(scenePlus.getId());

        String fileName = file.getOriginalFilename();
        String extName = fileName.substring(fileName.lastIndexOf("."));
        File newFile = File.createTempFile(UUID.randomUUID().toString() ,extName);
        file.transferTo(newFile);

        String logoPath = newFile.getPath();

        String outPathZh = ConstantFilePath.BASE_PATH + File.separator + "sceneQRcode/"+num+".png";
        String outPathEn = ConstantFilePath.BASE_PATH + File.separator + "sceneQRcode/"+num+"_en.png";

        String webSize = mainUrl + "/" + sceneProNewUrl + num;
        QrCodeUtil.generate(webSize, QrConfig.create().setImg(logoPath).setWidth(512).setHeight(512), FileUtil.file(outPathZh));
        QrCodeUtil.generate(webSize + "&lang=en", QrConfig.create().setImg(logoPath).setWidth(512).setHeight(512), FileUtil.file(outPathEn));
        //上传logo图片
        String ossLogoPath = String.format(UploadFilePath.DOWNLOADS_QRCODE, num) + "shareLogo" + extName;
        fYunFileService.uploadFile(scenePlusExt.getYunFileBucket(), logoPath, ossLogoPath);

        //上传二维码
        fYunFileService.uploadFile(scenePlusExt.getYunFileBucket(), outPathZh, String.format(UploadFilePath.DOWNLOADS_QRCODE, num) + num + ".png");
        fYunFileService.uploadFile(scenePlusExt.getYunFileBucket(), outPathEn, String.format(UploadFilePath.DOWNLOADS_QRCODE, num) + num + "_en.png");

        sceneEditInfoExt.setShareLogoImg(ossLogoPath);
        sceneEditInfoExtService.updateById(sceneEditInfoExt);

        FileUtil.del(newFile);
        FileUtil.del(new File(outPathEn));
        FileUtil.del(new File(outPathZh));


        return ResultData.ok();
    }

    /**
     * <p>
     *     初始化场景数据
     * </p>
     * @author dengsixing
     * @date 2022/8/22
     * @param num 场景码
     * @param title 场景标题
     * @param des 场景描述
     * @param sceneSource   场景来源：相机拍摄10以内表示，1表示八目，2双目，3转台，4 激光 其他来源10以上，11：一键换装，12：123看房，13文通虚拟场景
     * @param dataSource    场景计算数据目录
     * @param sceneScheme   方案：1-双目，2-转台，3-六目,4-八目,10-获取4k图，11-获取2k，12-获取1k
     * @param sceneResolution  分辨率（2k,4k）
     * @param sceneFrom     场景来源，lite：双目lite相机，pro：八目相机，minion：双面转台相机，laser：激光相机，virtual：虚拟场景，sketch：图片建模场景
     * @return com.fdkankan.web.response.ResultData
     **/
    @Override
    public ResultData initScene(String num, String title, String des, Integer sceneSource,
        String dataSource, Integer sceneScheme, String sceneResolution, String sceneFrom) {

        ScenePlus scenePlusByNum = this.getScenePlusByNum(num);
        if(Objects.nonNull(scenePlusByNum))
            return ResultData.ok();

        //生成场景主表
        ScenePlus scenePlus = new ScenePlus();
        scenePlus.setNum(num);
        scenePlus.setSceneStatus(SceneStatus.SUCCESS.code());
        scenePlus.setSceneSource(sceneSource);
        scenePlus.setPayStatus(PayStatus.PAY.code());
        scenePlus.setSceneType(SceneType.OTHER.code());
        this.save(scenePlus);

        //生成场景拓展表
        ScenePlusExt scenePlusExt = new ScenePlusExt();
        scenePlusExt.setPlusId(scenePlus.getId());
        scenePlusExt.setDataSource(dataSource);
        scenePlusExt.setSceneScheme(sceneScheme);
        scenePlusExt.setSceneResolution(sceneResolution);
        scenePlusExt.setSceneFrom(sceneFrom);
        scenePlusExt.setSceneKind(SceneKind.FACE.code());
        scenePlusExtService.save(scenePlusExt);

        //生成编辑主表
        SceneEditInfo sceneEditInfo = new SceneEditInfo();
        sceneEditInfo.setScenePlusId(scenePlus.getId());
        sceneEditInfo.setTitle(title);
        sceneEditInfo.setDescription(des);
        sceneEditInfoService.save(sceneEditInfo);
        sceneEditInfo = sceneEditInfoService.getByScenePlusId(scenePlus.getId());

        //生成编辑拓展表
        SceneEditInfoExt sceneEditInfoExt = new SceneEditInfoExt();
        sceneEditInfoExt.setScenePlusId(scenePlus.getId());
        sceneEditInfoExt.setEditInfoId(sceneEditInfo.getId());
        sceneEditInfoExtService.save(sceneEditInfoExt);
        sceneEditInfoExt = sceneEditInfoExtService.getByEditInfoId(sceneEditInfo.getId());

        //生成编辑显示控制表
        SceneEditControls sceneEditControls = new SceneEditControls();
        sceneEditControls.setEditInfoId(sceneEditInfo.getId());
        sceneEditControlsService.save(sceneEditControls);
        sceneEditControls = sceneEditControlsService.getBySceneEditId(sceneEditInfo.getId());

        //生成sceneJson
        SceneJsonBean sceneJson = new SceneJsonBean();
        BeanUtil.copyProperties(sceneEditInfoExt, sceneJson);
        BeanUtil.copyProperties(sceneEditInfo, sceneJson);
        SceneEditControlsVO sceneEditControlsVO = BeanUtil.copyProperties(sceneEditControls, SceneEditControlsVO.class);
        sceneJson.setControls(sceneEditControlsVO);
        sceneJson.setNum(num);
        sceneJson.setCreateTime(scenePlus.getCreateTime());

        sceneJson.setSceneResolution(scenePlusExt.getSceneResolution());
        sceneJson.setSceneFrom(scenePlusExt.getSceneFrom());
        sceneJson.setSceneKind(scenePlusExt.getSceneKind());
        if(StrUtil.isNotEmpty(scenePlusExt.getVideos())){
            sceneJson.setVideos(scenePlusExt.getVideos());
        }

        //本地写sceneJson文件
        String sceneJsonPath = String.format(UploadFilePath.DATA_VIEW_PATH+"scene.json", num);
        ossUtil.uploadFileBytes(sceneJsonPath, JSON.toJSONString(sceneJson).getBytes(StandardCharsets.UTF_8));

        redisUtil.del(String.format(RedisKey.SCENE_JSON, num));

        return ResultData.ok();
    }

    @Override
    public ResultData editScene(LaserSceneBean param) {

        ScenePlus scenePlus = this.getScenePlusByNum(param.getNum());
        if(Objects.isNull(scenePlus))
            throw new BusinessException(ErrorCode.FAILURE_CODE_5005);

        SceneEditInfo sceneEditInfo = sceneEditInfoService.getByScenePlusId(scenePlus.getId());
        sceneEditInfo.setTitle(param.getTitle());
        if(StrUtil.isEmpty(sceneEditInfo.getEntry())){
            sceneEditInfo.setEntry("{}");
        }
        sceneEditInfoService.updateById(sceneEditInfo);

        if (StrUtil.isNotBlank(param.getThumb())){
            ScenePlusExt scenePlusExt = scenePlusExtService.getScenePlusExtByPlusId(scenePlus.getId());
            scenePlusExt.setThumb(param.getThumb());
            scenePlusExtService.updateById(scenePlusExt);
            File sourceFile=FileUtil.file(FdkkLaserConfig.getProfile()+param.getThumb());
            log.info("sourceFile,-{}",sourceFile.getAbsoluteFile());
            File targetFile=FileUtil.file(FdkkLaserConfig.getProfile()+String.format(UploadFilePath.USER_EDIT_PATH, param.getNum())+ File.separator+"thumb-1k.jpg");
            log.info("targetFile,-{}",targetFile.getAbsoluteFile());
            FileUtils.copyFile(sourceFile,targetFile);
        }


        return ResultData.ok();
    }
}