package com.zhujizheng.IHome.remark.service;

import com.zhujizheng.IHome.generator.dao.PYRemark;
import com.zhujizheng.IHome.generator.mapper.PYRemarkMapper;
import com.zhujizheng.IHome.remark.dto.PYRemarkUpdateDTO;
import com.zhujizheng.IHome.remark.vo.PYRemarkVO;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.ArrayList;
import java.util.List;

@Slf4j
@Service
@Transactional(rollbackFor = RuntimeException.class)
public class PYRemarkServiceImpl implements PYRemarkService {

    @Autowired
    private PYRemarkMapper remarkMapper;

    @Override
    public List<PYRemarkVO> getAllRemark(int userId) {
        List<PYRemark> list = remarkMapper.selectAllRemark(userId);
        if (list == null) {
            return null;
        }

        List<PYRemarkVO> voList = new ArrayList<>();
        for (PYRemark remark : list) {
            PYRemarkVO vo = PYRemarkVO.createVOWithRemark(remark);
            voList.add(vo);
        }
        return voList;
    }

    @Override
    public PYRemarkVO getRemark(int userId, int friendId) {
        PYRemark remark = remarkMapper.selectRemark(userId, friendId);
        return PYRemarkVO.createVOWithRemark(remark);
    }

    @Override
    public int changeRemark(PYRemarkUpdateDTO updateDTO) {
        PYRemark oldRemark = remarkMapper.selectRemark(updateDTO.getUserId(), updateDTO.getFriendId());
        if (oldRemark == null) {
            PYRemark remark = PYRemark.createWithUpdateDTO(updateDTO);
            return remarkMapper.insertRemark(remark);
        } else {
            return remarkMapper.updateRemark(oldRemark.getUserId(), oldRemark.getFriendId(), updateDTO.getRemark());
        }
    }
}