package text_category;

import java.io.BufferedWriter;
import java.io.FileWriter;
import java.io.PrintWriter;
import java.text.NumberFormat;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

import com.opensymphony.xwork2.ActionContext;
import com.opensymphony.xwork2.ActionSupport;


public class Category extends ActionSupport {
	
	public String execute() throws Exception
	{
		Collection actionMessage = new ArrayList(1);

		Map attributes = ActionContext.getContext().getSession();
		if (attributes.get("url") == null)
		{
			actionMessage.add("url is null");
			setActionMessages(actionMessage);
			return INPUT;
		}
		if (attributes.get("content") == null)
		{
			actionMessage.add("content is null");
			setActionMessages(actionMessage);
			return INPUT;
		}
		
		setUrl(attributes.get("url").toString());
		setContent(attributes.get("content").toString());
		
		if (getUrl() == null || getUrl().length() <= 0)
			return INPUT;
		if (getContent() == null || getContent().length() <= 0)
			return INPUT;
		
		WVTManager manager = new WVTManager();
				
		List res = manager.category(getUrl(), getContent());
        
		if (res == null || res.isEmpty())						
		{
			actionMessage.add("category failed");
			setActionMessages(actionMessage);
			return INPUT;
		}
		
		setResult(res);
		if (categoryResult == null || categoryResult.isEmpty())
		{
			actionMessage.add("cann't set the result");
			setActionMessages(actionMessage);
			return INPUT;
		}
		return SUCCESS;
	}
	
	public void setResult(List res)
	{
		categoryResult = new ArrayList();
		int i = 1;
		for (Iterator it = res.iterator(); it.hasNext(); )
		{
			Result r = new Result();
			CategoryResult cr = (CategoryResult)it.next();
			if (cr.getSimilarity() != 0)
			{
				r.setCategoryName(cr.getCategoryName());
				
				String str = getText("bar" + i);
				r.setImgBar(str);
				
				str = getText("bg" + i);
				r.setBackground(str);
				
				NumberFormat percentFormat = NumberFormat.getPercentInstance();				
				str = percentFormat.format(cr.getSimilarity());
				r.setSimilarity(str);
				categoryResult.add(r);
				
				i++;
			}			
		}
	}

	private String url;
	private String content;
	private List categoryResult;

	public String getUrl() {
		return url;
	}

	public void setUrl(String url) {
		this.url = url;
	}
	
	public void setModel(Object url)
	{
		this.url = url.toString();
	}

	public String getContent() {
		return content;
	}

	public void setContent(String content) {
		this.content = content;
	}

	public List getCategoryResult() {
		return categoryResult;
	}

	public void setCategoryresult(List categoryResult) {
		this.categoryResult = categoryResult;
	}
}