package com.xunlei;

import java.io.UnsupportedEncodingException;
import java.net.HttpURLConnection;

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

/**
 * @author Hank
 * @date 2015-4-22 下午4:25:22
 */
public class XunleihuiyuanNet {
	private String txtFile1 = "D:\\sourcefile_1.txt";
	private String txtFile2 = "D:\\sourcefile_2.txt";

	/*public static void main(String[] args) {
		XunleihuiyuanNet xunleihuiyuanNet = new XunleihuiyuanNet();
		String url = "http://www.xunleihuiyuan.net/";
		xunleihuiyuanNet.save1(url);
		String href = xunleihuiyuanNet.jx1(url);
		xunleihuiyuanNet.save2(href);
		String result = xunleihuiyuanNet.jx2();
		System.out.println(result);
	}*/

	//保存首页源码
	public void save1(String url) {
		HttpConnection httpConnection = new HttpConnection();
		httpConnection.createConnection(url, "GET", 7000);// 创建连接
		setConn(httpConnection.conn);// 设置连接
		httpConnection.outPut(httpConnection.getByte(), txtFile1);// 获得连接byte数组，并保存
	}
	
	//保存提取页源码
	public void save2(String href) {
		HttpConnection httpConnection = new HttpConnection();
		httpConnection.createConnection(href, "GET", 7000);// 创建连接
		setConn(httpConnection.conn);// 设置连接
		httpConnection.outPut(httpConnection.getByte(), txtFile2);// 获得连接byte数组，并保存
	}

	//解析首页提取想要的url地址
	public String jx1(String url) {
		HttpConnection httpConnection = new HttpConnection();
		try {
			Document document = Jsoup.parse(new String(httpConnection.inPut(txtFile1), "UTF-8"));
			Element element = document.getElementById("divMain");
			Elements elements = element.children();
			for (Element e : elements) {
				Elements es = e.getElementsByTag("h3");
				if (es != null && es.size() > 0) {
					element = e.getElementsByTag("h2").get(0).getElementsByTag("a").get(0);
					String href = element.attr("href");
					if (!href.startsWith("http://"))
						href = url + href;
					return href;
				}
			}
		} catch (UnsupportedEncodingException e) {
			e.printStackTrace();
		}
		return null;
	}

	//解析页面提前想要的信息
	public String jx2() {
		HttpConnection httpConnection = new HttpConnection();
		StringBuffer sb = new StringBuffer();
		try {
			Document document = Jsoup.parse(new String(httpConnection.inPut(txtFile2), "UTF-8"));
			Element element = document.getElementById("divMain");
			Elements elements = element.children().get(0).children();
			// String txt = element.children().get(0).children().get(5).text();
			String txt = null;
			for (Element e : elements) {
				if ("post-body formattext".equals(e.attr("class"))) {
					txt = e.text();
					break;
				}
			}
			//
			// System.out.println(element.getElementsByClass("post-body formattext").size());
			// System.out.println(element.children().get(0).children().get(5).attr("class"));
			String[] txts = txt.split(" ");
			for (String str : txts)
				sb.append(str + "\r\n");
		} catch (UnsupportedEncodingException e) {
			e.printStackTrace();
		}
		return sb.toString();
	}

	/**
	 * 设置连接
	 */
	public void setConn(HttpURLConnection conn) {
		conn.setRequestProperty("Accept", "*/*");
		conn.setRequestProperty("Connection", "keep-alive");
		conn.setRequestProperty("Host", "www.xunleihuiyuan.net");
		conn.setRequestProperty("User-Agent", "Mozilla/5.0 (Windows NT 6.1) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/33.0.1750.154 Safari/537.36");
	}
}