alfresco-community-repo/source/java/org/alfresco/rest/api/NetworkWebScriptGet.java

package org.alfresco.rest.api;

import java.io.IOException;

import org.alfresco.repo.security.authentication.AuthenticationUtil;
import org.alfresco.repo.tenant.TenantUtil;
import org.alfresco.rest.api.model.PersonNetwork;
import org.alfresco.rest.api.networks.NetworksEntityResource;
import org.alfresco.rest.framework.Api;
import org.alfresco.rest.framework.core.exceptions.EntityNotFoundException;
import org.alfresco.rest.framework.jacksonextensions.JacksonHelper;
import org.alfresco.rest.framework.jacksonextensions.JacksonHelper.Writer;
import org.alfresco.rest.framework.resource.parameters.Params;
import org.alfresco.rest.framework.webscripts.ResourceWebScriptHelper;
import org.codehaus.jackson.JsonGenerationException;
import org.codehaus.jackson.JsonGenerator;
import org.codehaus.jackson.map.JsonMappingException;
import org.codehaus.jackson.map.ObjectMapper;
import org.springframework.extensions.webscripts.AbstractWebScript;
import org.springframework.extensions.webscripts.Format;
import org.springframework.extensions.webscripts.WebScriptRequest;
import org.springframework.extensions.webscripts.WebScriptResponse;

public class NetworkWebScriptGet extends AbstractWebScript
{
	private Networks networks;
	private JacksonHelper jsonHelper;
    private ResourceWebScriptHelper helper;
    
    public void setJsonHelper(JacksonHelper jsonHelper)
    {
        this.jsonHelper = jsonHelper;
    }
    
	public void setHelper(ResourceWebScriptHelper helper)
	{
		this.helper = helper;
	}

	public void setNetworks(Networks networks)
	{
		this.networks = networks;
	}

	@Override
	public void execute(final WebScriptRequest req, WebScriptResponse res) throws IOException
	{
        // apply content type
        res.setContentType(Format.JSON.mimetype() + ";charset=UTF-8");

        jsonHelper.withWriter(res.getOutputStream(), new Writer()
        {
            @Override
            public void writeContents(JsonGenerator generator, ObjectMapper objectMapper)
                        throws JsonGenerationException, JsonMappingException, IOException
            {
                String personId = AuthenticationUtil.getFullyAuthenticatedUser();
                String networkId = TenantUtil.getCurrentDomain();

            	PersonNetwork networkMembership = networks.getNetwork(personId, networkId);
            	if(networkMembership != null)
            	{
	            	// TODO this is not ideal, but the only way to populate the embedded network entities (this would normally be
	            	// done automatically by the api framework).
	            	Object wrapped = helper.postProcessResponse(Api.ALFRESCO_PUBLIC, NetworksEntityResource.NAME, Params.valueOf(personId, null), networkMembership);
	
	                objectMapper.writeValue(generator, wrapped);
            	}
            	else
            	{
            		throw new EntityNotFoundException(networkId);
            	}
            }
        });		
	}

}