org.xacml4j.v30.marshal.json.JsonResponseContextMarshaller Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of xacml-json Show documentation
There is a newer version: 1.4.0
package org.xacml4j.v30.marshal.json;

/*
 * #%L
 * Xacml4J Gson Integration
 * %%
 * Copyright (C) 2009 - 2014 Xacml4J.org
 * %%
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Lesser Public License for more details.
 * 
 * You should have received a copy of the GNU General Lesser Public
 * License along with this program.  If not, see
 * .
 * #L%
 */

import java.io.IOException;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.io.Writer;

import org.xacml4j.v30.Advice;
import org.xacml4j.v30.Attribute;
import org.xacml4j.v30.AttributeAssignment;
import org.xacml4j.v30.Category;
import org.xacml4j.v30.Obligation;
import org.xacml4j.v30.ResponseContext;
import org.xacml4j.v30.Result;
import org.xacml4j.v30.Status;
import org.xacml4j.v30.StatusCode;
import org.xacml4j.v30.marshal.Marshaller;
import org.xacml4j.v30.pdp.PolicyIDReference;
import org.xacml4j.v30.pdp.PolicySetIDReference;

import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import com.google.gson.JsonObject;
import com.google.gson.stream.JsonWriter;

public class JsonResponseContextMarshaller implements Marshaller {

	private final Gson json;

	public JsonResponseContextMarshaller() {
		json = new GsonBuilder().registerTypeAdapter(ResponseContext.class, new ResponseContextAdapter())
				.registerTypeAdapter(Result.class, new ResultAdapter())
				.registerTypeAdapter(Status.class, new StatusAdapter())
				.registerTypeAdapter(StatusCode.class, new StatusCodeAdapter())
                .registerTypeAdapter(Category.class, new CategoryAdapter())
                .registerTypeAdapter(Obligation.class, new ObligationOrAdviceAdapter())
				.registerTypeAdapter(AttributeAssignment.class, new AttributeAssignmentSerializer())
				.registerTypeAdapter(Advice.class, new ObligationOrAdviceAdapter())
				.registerTypeAdapter(Attribute.class, new AttributeSerializer())
				.registerTypeAdapter(PolicyIDReference.class, new IdReferenceAdapter())
				.registerTypeAdapter(PolicySetIDReference.class, new IdReferenceAdapter()).create();
	}

	@Override
	public Object marshal(ResponseContext source) throws IOException {
		return json.toJsonTree(source);
	}

	@Override
	public void marshal(ResponseContext source, Object target) throws IOException {
		if (target instanceof Writer) {
			json.toJson(source, ResponseContext.class, new JsonWriter((Writer) target));
			return;
		}
		if (target instanceof OutputStream) {
			json.toJson(source, ResponseContext.class, new JsonWriter(new OutputStreamWriter((OutputStream) target)));
			return;
		}
		if (target instanceof JsonObject) {
			JsonObject o = (JsonObject) target;
			o.add("Response", json.toJsonTree(source));
			return;
		}
		throw new IllegalArgumentException("Unsupported marshalling target");
	}

}