package MainModel;

import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;

import org.jsoup.Jsoup;
import org.jsoup.nodes.Document;
import org.jsoup.select.Elements;

import jxl.Workbook;
import jxl.write.Label;
import jxl.write.WritableSheet;
import jxl.write.WritableWorkbook;
import jxl.write.WriteException;
import jxl.write.biff.RowsExceededException;

public class Test {
	private static int intCounted = 0;
	static String strOutput = null;
	static HashMap<String,ArrayList<String>> excelDataMap = new HashMap<String,ArrayList<String>>();
	static ArrayList<String> titleData = new ArrayList<String>();
	static ArrayList<String> linksData = new ArrayList<String>();
	
	//按照网页特征，判断页数，返回一个循环的次数(未知分页规律，暂时固定写死)
		private static int preGenerate() {
			return 8;
		}
		
	//处理的url，使其可以访问。
	private static String genURL(String url){
		intCounted++;
		return (intCounted > 0) ? url + new Integer(intCounted).toString() +".html" : url + ".html";
	}
	
	public static void getExcelData(String commonUrl, String urlTail) {
		String url = commonUrl + urlTail;
		Document doc;
		while(intCounted <preGenerate() ) {
			String html = genURL(url);
			System.out.println(html);
			try {
				doc = Jsoup.connect(html).get();
				//class=\"abl\" target=_blank
				Elements elementArray = doc.select("a.abl");
				for(int i = 0 ; i < elementArray.size(); i++) {
					titleData.add(elementArray.get(i).text().substring(9)) ;
					linksData.add(commonUrl +elementArray.get(i).attr("href")) ;
				}
				
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
		System.out.println(titleData.size());
		excelDataMap.put("Title", titleData);
		excelDataMap.put("Links",linksData);
		
	}
	
	public static void generateExcelFile() {
		File exlFile = new File("C://workfile_temp//jars//export_data.xls");
		WritableWorkbook writableWorkbook;
		try {
			writableWorkbook = Workbook.createWorkbook(exlFile);
			WritableSheet writableSheet = writableWorkbook.createSheet("Sheet1", 0);
			int i = 0;int j = 0;
			for(String str :excelDataMap.keySet()) {
				//写标题
				 writableSheet.addCell(new Label(i++, 0, str));
			}
			//行1：列（0，1）
			for(int k = 0 ; k < excelDataMap.get("Title").size(); k++){
				writableSheet.addCell(new Label(0, k+1, excelDataMap.get("Title").get(k)));
				writableSheet.addCell(new Label(1, k+1, excelDataMap.get("Links").get(k)));
			}
			
			writableWorkbook.write();
	        writableWorkbook.close();
				
		} catch (IOException e) {
			e.printStackTrace();
		} catch (RowsExceededException e) {
			e.printStackTrace();
		} catch (WriteException e) {
			e.printStackTrace();
		}
	}
	
	public static void main(String[] args) {
		String commonUrl = "http://opinion.people.com.cn";
		String urlTail  = "/GB/8213/49160/49219/index";
		
		getExcelData(commonUrl,urlTail);
		generateExcelFile();

	}
}