/***
 * Copyright (c) 2009 Caelum - www.caelum.com.br/opensource - guilherme.silveira@caelum.com.br
 * All rights reserved.
 *
 * 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 br.com.caelum.vraptor.util.extjs;

import java.io.IOException;
import java.io.Writer;

import javax.servlet.http.HttpServletResponse;

import br.com.caelum.vraptor.interceptor.TypeNameExtractor;
import br.com.caelum.vraptor.ioc.Component;
import br.com.caelum.vraptor.serialization.ProxyInitializer;
import br.com.caelum.vraptor.serialization.xstream.XStreamSerializer;

import com.thoughtworks.xstream.XStream;
import com.thoughtworks.xstream.io.HierarchicalStreamWriter;
import com.thoughtworks.xstream.io.json.JsonHierarchicalStreamDriver;
import com.thoughtworks.xstream.io.json.JsonWriter;

/**
 * Implementation for {@link ExtJSJson}.
 * @author Daniel Kist
 * @author Lucas Cavalcanti
 *
 */
@Component
public class DefaultExtJson implements ExtJSJson {

	private final XStreamSerializer serializer;
	private final XStream xstream;
	private ExtJSWrapper wrapper;

	public DefaultExtJson(HttpServletResponse response, TypeNameExtractor extractor, ProxyInitializer initializer)
	throws IOException {
	xstream = new XStream(new JsonHierarchicalStreamDriver() {
		@Override
		public HierarchicalStreamWriter createWriter(Writer writer) {
		return new JsonWriter(writer, new char[0], "", JsonWriter.DROP_ROOT_MODE) {
			@Override
			public void addAttribute(String key, String value) {
			if (!key.equals("class")) {
				super.addAttribute(key, value);
			}
			}
		};
		}
	});
	xstream.setMode(XStream.NO_REFERENCES);
	xstream.aliasField("data", ExtJSWrapper.class, "list");
	serializer = new XStreamSerializer(xstream, response.getWriter(), extractor, initializer);
	}

	public ExtJSJson from(Object object) {
	wrapper = new ExtJSWrapper(object);
	serializer.from(object);
	return this;
	}

	public ExtJSJson exclude(String... names) {
	serializer.exclude(names);
	return this;
	}

	public ExtJSJson include(String... fields) {
	serializer.include(fields);
	return this;
	}

	public ExtJSJson selected(Object value) {
	wrapper.setSelected(value);
	return this;
	}

	public ExtJSJson serialize() {
	serializer.from(wrapper).recursive().serialize();
	return this;
	}

	public ExtJSJson success() {
	wrapper.setSuccess(true);
	return this;
	}

	public ExtJSJson success(boolean success) {
	wrapper.setSuccess(success);
	return this;
	}

	public ExtJSJson total(Integer total) {
	wrapper.setTotal(total);
	return this;
	}
}