/*
 * $Header: /home/cvspublic/jakarta-slide/src/share/org/apache/slide/extractor/MSExcelExtractor.java,v 1.3 2005/01/14 18:34:13 luetzkendorf Exp $
 * $Revision: 1.3 $
 * $Date: 2005/01/14 18:34:13 $
 *
 * ====================================================================
 *
 * Copyright 2004 The Apache Software Foundation 
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */

package org.apache.slide.extractor;

import java.io.CharArrayReader;
import java.io.CharArrayWriter;
import java.io.FileInputStream;
import java.io.InputStream;
import java.io.Reader;
import java.util.Iterator;
import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.poifs.filesystem.POIFSFileSystem;

/**
 * Content extractor for Microsoft Excel documents.
 */
public class MSExcelExtractor extends AbstractContentExtractor {
	
	static final String CONTENT_TYPE_EXCEL_1 = "application/msexcel";
	static final String CONTENT_TYPE_EXCEL_2 = "application/vnd.ms-excel";
	static final String CONTENT_TYPE_EXCEL_ALL_CSV = CONTENT_TYPE_EXCEL_1+","+CONTENT_TYPE_EXCEL_2;
	
    public MSExcelExtractor(String uri, String contentType, String namespace) {
      super(uri, contentType, namespace);
    }

    public Reader extract(InputStream content)  throws ExtractorException
    {
        try
        {
            CharArrayWriter writer = new CharArrayWriter();

            POIFSFileSystem fs = new POIFSFileSystem(content);
            HSSFWorkbook workbook = new HSSFWorkbook(fs);

            for (int i = 0; i < workbook.getNumberOfSheets(); i++ )
            {
                HSSFSheet sheet = workbook.getSheetAt(i);

             Iterator rows = sheet.rowIterator();
                while( rows.hasNext() ) {
                    HSSFRow row = (HSSFRow) rows.next();

                    Iterator cells = row.cellIterator();
                    while( cells.hasNext() ) {
                        HSSFCell cell = (HSSFCell) cells.next();
                        switch ( cell.getCellType() ) {
                            case HSSFCell.CELL_TYPE_NUMERIC:
                                String num = Double.toString(cell.getNumericCellValue()).trim();
                                if(num.length() > 0)
                                    writer.write(num + " ");
                                break;
                            case HSSFCell.CELL_TYPE_STRING:
                                String text = cell.getStringCellValue().trim();
                                if(text.length() > 0)
                                    writer.write(text + " ");
                                break;
                        }
                    }
                }
            }

            return new CharArrayReader(writer.toCharArray());
        }
        catch(Exception e )
        {
            throw new ExtractorException(e.getMessage());
        }
    }

    public static void main(String[] args) throws Exception
    {
        FileInputStream in = new FileInputStream(args[0]);

        MSExcelExtractor ex = new MSExcelExtractor(null, null, null);

        Reader reader = ex.extract(in);

        int c = 0;
        do
        {
            c = reader.read();
            System.out.print((char)c);
        }
        while(c != -1);
    }
    
	/* (non-Javadoc)
	 * @see org.apache.slide.extractor.Extractor#getContentType()
	 */
	public String getContentType() {
		if(super.getContentType()==null){
			return CONTENT_TYPE_EXCEL_ALL_CSV;
		}
		return super.getContentType();
	}
}