package kr.itn.itnhub.seed;

import kr.itn.itnhub.AbstractDbTest;
import kr.itn.itnhub.contact.Contact;
import kr.itn.itnhub.contact.ContactMapper;
import kr.itn.itnhub.org.OrgStatus;
import kr.itn.itnhub.org.Organization;
import kr.itn.itnhub.org.OrganizationMapper;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.jdbc.core.JdbcTemplate;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.util.List;

import static org.assertj.core.api.Assertions.assertThat;

class SeedServiceTest extends AbstractDbTest {

    @Autowired
    SeedService seedService;

    @Autowired
    OrganizationMapper mapper;

    @Autowired
    ContactMapper contactMapper;

    @Autowired
    JdbcTemplate jdbc;

    @BeforeEach
    void clean() {
        jdbc.update("delete from organization");
        jdbc.update("delete from contact");
    }

    private byte[] workbook(String[]... dataRows) throws Exception {
        try (XSSFWorkbook wb = new XSSFWorkbook();
             ByteArrayOutputStream out = new ByteArrayOutputStream()) {

            Sheet sheet = wb.createSheet(SeedParser.SHEET_NAME);
            sheet.createRow(0);
            sheet.createRow(1);

            int rowNum = 2;
            for (String[] data : dataRows) {
                Row row = sheet.createRow(rowNum++);
                row.createCell(1).setCellValue(data[0]);
                row.createCell(2).setCellValue(data[1]);
                row.createCell(5).setCellValue(data[2]);
                if (data[3] != null) row.createCell(6).setCellValue(data[3]);
                if (data[4] != null) row.createCell(7).setCellValue(data[4]);
                if (data[5] != null) row.createCell(8).setCellValue(data[5]);
                if (data[6] != null) row.createCell(9).setCellValue(data[6]);
                if (data[7] != null) row.createCell(10).setCellValue(data[7]);
            }
            wb.write(out);
            return out.toByteArray();
        }
    }

    @Test
    void 최초_시드는_전부_신규다() throws Exception {
        byte[] xlsx = workbook(
                new String[]{"001", "00", "국제방송교류재단", "팀", "송민지", "과장", "02-1", "a@b.kr"},
                new String[]{"002", "00", "세종학당재단", "팀", "권유나", "6급", "02-2", "c@d.kr"});

        SeedReport report = seedService.seed(new ByteArrayInputStream(xlsx));

        assertThat(report.created()).isEqualTo(2);
        assertThat(report.updated()).isZero();
        assertThat(report.total()).isEqualTo(2);
        assertThat(mapper.findAll()).hasSize(2);
    }

    @Test
    void 같은_파일을_다시_올려도_행이_늘지_않는다() throws Exception {
        byte[] xlsx = workbook(
                new String[]{"001", "00", "국제방송교류재단", "팀", "송민지", "과장", "02-1", "a@b.kr"});

        seedService.seed(new ByteArrayInputStream(xlsx));
        SeedReport second = seedService.seed(new ByteArrayInputStream(xlsx));

        assertThat(second.created()).isZero();
        assertThat(second.updated()).isEqualTo(1);
        assertThat(mapper.findAll()).hasSize(1);
    }

    @Test
    void 재시드는_이미_만든_채널ID를_보존한다() throws Exception {
        byte[] xlsx = workbook(
                new String[]{"001", "00", "국제방송교류재단", "팀", "송민지", "과장", "02-1", "a@b.kr"});

        seedService.seed(new ByteArrayInputStream(xlsx));
        Long id = mapper.findAll().get(0).getId();
        mapper.updateChannelIdMj(id, "chan-mj");
        mapper.updateChannelIdLaw(id, "chan-law");

        seedService.seed(new ByteArrayInputStream(xlsx));

        assertThat(mapper.findById(id).getStatus()).isEqualTo(OrgStatus.ACTIVE);
    }

    @Test
    void 같은_기관이_한_시트에_두번_들어있으면_한_행만_남고_두번째는_갱신으로_집계된다() throws Exception {
        byte[] xlsx = workbook(
                new String[]{"001", "00", "국제방송교류재단", "팀", "송민지", "과장", "02-1", "a@b.kr"},
                new String[]{"001", "00", "국제방송교류재단", "팀", "송민지", "과장", "02-1", "a@b.kr"});

        SeedReport report = seedService.seed(new ByteArrayInputStream(xlsx));

        assertThat(mapper.findAll()).hasSize(1);
        assertThat(report.total()).isEqualTo(2);
        assertThat(report.created()).isEqualTo(1);
        assertThat(report.updated()).isEqualTo(1);
    }

    @Test
    void 담당자정보가_없는_기관은_정보대기로_들어온다() throws Exception {
        byte[] xlsx = workbook(
                new String[]{"008", "00", "경찰청", "데이터정책계", "박진우", "경위", "02-3", "e@f.kr"},
                new String[]{"008", "01", "경찰청_치안정책연구소", null, null, null, null, null});

        seedService.seed(new ByteArrayInputStream(xlsx));

        List<Organization> all = mapper.findAll();
        assertThat(all).hasSize(2);
        assertThat(all).extracting(Organization::getStatus)
                .containsExactly(OrgStatus.READY, OrgStatus.INFO_PENDING);
    }

    @Test
    void 시트에_담당자명이_있으면_신청기관_담당자_연락처를_만들어_연결한다() throws Exception {
        byte[] xlsx = workbook(
                new String[]{"001", "00", "국제방송교류재단", "데이터정보화팀", "송민지", "과장", "02-1", "a@b.kr"});

        seedService.seed(new ByteArrayInputStream(xlsx));

        Organization org = mapper.findAll().get(0);
        assertThat(org.getApplicantContactId()).isNotNull();

        Contact applicant = org.getApplicant();
        assertThat(applicant.getCategory()).isEqualTo("APPLICANT");
        assertThat(applicant.getName()).isEqualTo("송민지");
        assertThat(applicant.getAffiliation()).isEqualTo("국제방송교류재단");
        assertThat(applicant.getDeptName()).isEqualTo("데이터정보화팀");
        assertThat(applicant.getTitle()).isEqualTo("과장");
        assertThat(applicant.getPhone()).isEqualTo("02-1");
        assertThat(applicant.getEmail()).isEqualTo("a@b.kr");
    }

    @Test
    void 재시드는_이미_연결된_신청기관_담당자를_중복생성하거나_다시_연결하지_않는다() throws Exception {
        byte[] xlsx = workbook(
                new String[]{"001", "00", "국제방송교류재단", "팀", "송민지", "과장", "02-1", "a@b.kr"});

        seedService.seed(new ByteArrayInputStream(xlsx));
        Long firstContactId = mapper.findAll().get(0).getApplicantContactId();

        seedService.seed(new ByteArrayInputStream(xlsx));

        Organization after = mapper.findAll().get(0);
        assertThat(after.getApplicantContactId()).isEqualTo(firstContactId);
        assertThat(jdbc.queryForObject("select count(*) from contact", Integer.class)).isEqualTo(1);
    }

    @Test
    void 같은_담당자를_공유하는_하위기관_여러곳은_담당자를_한_행만_만들어_모두_연결한다() throws Exception {
        byte[] xlsx = workbook(
                new String[]{"008", "00", "경찰청", "데이터정책계", "박진우", "경위", "02-3150-3205", "hi@police.go.kr"},
                new String[]{"008", "02", "서울청", "데이터정책계", "박진우", "경위", "02-3150-3205", "hi@police.go.kr"},
                new String[]{"008", "03", "부산청", "데이터정책계", "박진우", "경위", "02-3150-3205", "hi@police.go.kr"});

        SeedReport report = seedService.seed(new ByteArrayInputStream(xlsx));

        assertThat(report.created()).isEqualTo(3);
        assertThat(jdbc.queryForObject("select count(*) from contact", Integer.class)).isEqualTo(1);

        List<Organization> all = mapper.findAll();
        assertThat(all).hasSize(3);
        assertThat(all).extracting(Organization::getStatus)
                .containsOnly(OrgStatus.READY);

        Long contactId = all.get(0).getApplicantContactId();
        assertThat(contactId).isNotNull();
        assertThat(all).extracting(Organization::getApplicantContactId)
                .containsOnly(contactId);
    }

    @Test
    void 이미_다른_담당자로_배정된_기관은_재시드가_배정을_바꾸지_않는다() throws Exception {
        byte[] xlsx = workbook(
                new String[]{"001", "00", "국제방송교류재단", "팀", "송민지", "과장", "02-1", "a@b.kr"});
        seedService.seed(new ByteArrayInputStream(xlsx));
        Long id = mapper.findAll().get(0).getId();

        Contact manual = new Contact();
        manual.setCategory("APPLICANT");
        manual.setName("사람이 직접 배정한 담당자");
        contactMapper.insert(manual);
        mapper.updateAssignments(id, manual.getId(), null, null, null, null);

        seedService.seed(new ByteArrayInputStream(xlsx));

        Organization after = mapper.findById(id);
        assertThat(after.getApplicantContactId()).isEqualTo(manual.getId());
        assertThat(after.getApplicant().getName()).isEqualTo("사람이 직접 배정한 담당자");
    }
}