package com.joke.util;

import java.io.*;
import java.net.*;
import java.util.ArrayList;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import org.apache.commons.httpclient.DefaultHttpMethodRetryHandler;
import org.apache.commons.httpclient.HttpClient;
import org.apache.commons.httpclient.HttpException;
import org.apache.commons.httpclient.HttpStatus;
import org.apache.commons.httpclient.methods.GetMethod;
import org.apache.commons.httpclient.params.HttpMethodParams;
import net.sf.json.JSONArray;
import net.sf.json.JSONException;
import net.sf.json.JSONObject;
import com.joke.vo.AutoVo;

/**
 * 百度接口提供的热词
 * 
 * @author yangzhongying mail:j@seejoke.com
 * @date 2014-8-13 下午01:13:05
 * @package com.joke.util
 */
public class BaiduService {

	private static JSONObject jsonData;
	private static JSONArray info;

	/**
	 * 获取10个关键字匹配
	 * 
	 * @param keyword
	 * @param limit
	 * @return
	 * @throws HttpException
	 * @throws IOException
	 * @throws JSONException
	 */
	public static List<AutoVo> getSuggestionList(String keyword)
			throws HttpException, IOException, JSONException {
		List<AutoVo> list = new ArrayList<AutoVo>();
		HttpClient httpClient = new HttpClient();
		GetMethod getMethod = new GetMethod(
				"http://suggestion.baidu.com/su?wd="
						+ URLEncoder.encode(keyword, "utf-8")
						+ "&p=3&cb=window.bdsug.sug&t=1302750139281");
		// Execute the method
		httpClient.executeMethod(getMethod);
		getMethod.getParams().setParameter(HttpMethodParams.RETRY_HANDLER,
				new DefaultHttpMethodRetryHandler());

		// If failed, return empty list
		int statusCode = httpClient.executeMethod(getMethod);
		if (statusCode != HttpStatus.SC_OK) {
			return list;
		}
		// Get the response body
		byte[] responseBody = getMethod.getResponseBody();
		String response = new String(responseBody, "gbk");
		// Pick the json part
		Pattern pattern = Pattern.compile("\\{.*\\}");
		Matcher matcher = pattern.matcher(response);
		if (matcher.find()) {
			String jsonStr = matcher.group();
			if (jsonStr != null && !"".equals(jsonStr)) {
				jsonData = JSONObject.fromObject(jsonStr);
				info = jsonData.getJSONArray("s");
				AutoVo bo = null;
				for (int i = 0; i < info.size(); i++) {
					bo = new AutoVo();
					String hotword = (String) info.get(i);
					if (hotword != null && !"".equals(hotword)) {
						bo.setId(i);
						bo.setKeyword(hotword);
						list.add(bo);
					}
				}
			}
		}
		return list;
	}
}