package com.baeldung.jexcel;

import jxl.*;
import java.util.Map;
import java.util.HashMap;
import java.util.ArrayList;
import java.util.List;
import jxl.read.biff.BiffException;
import java.io.File;
import java.io.IOException;
import jxl.write.*;
import jxl.write.Number;
import jxl.format.Colour;

public class JExcelHelper {

    public Map<Integer, List<String>> readJExcel(String fileLocation) throws IOException, BiffException {
        Map<Integer, List<String>> data = new HashMap<>();

        Workbook workbook = Workbook.getWorkbook(new File(fileLocation));
        Sheet sheet = workbook.getSheet(0);
        int rows = sheet.getRows();
        int columns = sheet.getColumns();

        for (int i = 0; i < rows; i++) {
            data.put(i, new ArrayList<String>());
            for (int j = 0; j < columns; j++) {
                data.get(i).add(sheet.getCell(j, i).getContents());
            }
        }
        return data;
    }

    public void writeJExcel() throws IOException, WriteException {
        WritableWorkbook workbook = null;
        try {
            File currDir = new File(".");
            String path = currDir.getAbsolutePath();
            String fileLocation = path.substring(0, path.length() - 1) + "temp.xls";

            workbook = Workbook.createWorkbook(new File(fileLocation));

            WritableSheet sheet = workbook.createSheet("Sheet 1", 0);

            WritableCellFormat headerFormat = new WritableCellFormat();
            WritableFont font = new WritableFont(WritableFont.ARIAL, 16, WritableFont.BOLD);
            headerFormat.setFont(font);
            headerFormat.setBackground(Colour.LIGHT_BLUE);
            headerFormat.setWrap(true);
            Label headerLabel = new Label(0, 0, "Name", headerFormat);
            sheet.setColumnView(0, 60);
            sheet.addCell(headerLabel);

            headerLabel = new Label(1, 0, "Age", headerFormat);
            sheet.setColumnView(0, 40);
            sheet.addCell(headerLabel);

            WritableCellFormat cellFormat = new WritableCellFormat();
            cellFormat.setWrap(true);

            Label cellLabel = new Label(0, 2, "John Smith", cellFormat);
            sheet.addCell(cellLabel);
            Number cellNumber = new Number(1, 2, 20, cellFormat);
            sheet.addCell(cellNumber);

            cellLabel = new Label(0, 3, "Ana Johnson", cellFormat);
            sheet.addCell(cellLabel);
            cellNumber = new Number(1, 3, 30, cellFormat);
            sheet.addCell(cellNumber);

            workbook.write();
        } finally {
            if (workbook != null) {
                workbook.close();
            }
        }

    }
}