PeopleConverter.java

/*******************************************************************************
 * Copyright (c) 2010-2014 SAP AG and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     SAP AG - initial API and implementation
 *******************************************************************************/
package org.eclipse.skalli.core.extension.people;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

import org.eclipse.skalli.model.Member;
import org.eclipse.skalli.model.ext.commons.PeopleExtension;
import org.eclipse.skalli.services.extension.rest.RestConverterBase;
import org.eclipse.skalli.services.extension.rest.RestException;
import org.eclipse.skalli.services.extension.rest.RestUtils;

import com.thoughtworks.xstream.converters.MarshallingContext;
import com.thoughtworks.xstream.converters.UnmarshallingContext;
import com.thoughtworks.xstream.io.HierarchicalStreamReader;
import com.thoughtworks.xstream.io.HierarchicalStreamWriter;

public class PeopleConverter extends RestConverterBase<PeopleExtension> {

    public static final String API_VERSION = "1.0"; //$NON-NLS-1$
    public static final String NAMESPACE = "http://www.eclipse.org/skalli/2010/API/Extension-People"; //$NON-NLS-1$

    private static final String URL_USERS = RestUtils.URL_API + "users/"; //$NON-NLS-1$

    public PeopleConverter() {
        super(PeopleExtension.class);
    }

    @SuppressWarnings("nls")
    @Override
    public void marshal(PeopleExtension extension) throws IOException {
        writer.array("leads", "lead");
        for (Member member : extension.getLeads()) {
            writer
                .object()
                .pair("userId", member.getUserID())
                .link(USER_RELATION, URL_USERS, member.getUserID())
                .end();
        }
        writer.end();
        writer.array("members", "member");
        for (Member member : extension.getMembers()) {
            writer
                .object()
                .pair("userId", member.getUserID())
                .link(USER_RELATION, URL_USERS, member.getUserID())
                .end();
        }
        writer.end();
    }

    @Override
    protected PeopleExtension unmarshal() throws RestException, IOException {
        return unmarshal(new PeopleExtension());
    }

    @SuppressWarnings("nls")
    private PeopleExtension unmarshal(PeopleExtension ext) throws RestException, IOException {
        while (reader.hasMore()) {
            if (reader.isKey("leads")) {
                ext.getLeads().addAll(readMembers("lead"));
            } else if (reader.isKey("members")) {
                ext.getMembers().addAll(readMembers("member"));
            } else {
                reader.skip();
            }
        }
        return ext;
    }

    private List<Member> readMembers(String itemKey) throws RestException, IOException {
        List<Member> members = new ArrayList<Member>();
        reader.array(itemKey);
        while (reader.hasMore()) {
            Member member = readMember();
            if (member != null) {
                members.add(member);
            }
        }
        reader.end();
        return members;
    }

    private Member readMember() throws RestException, IOException {
        Member member = null;
        reader.object();
        while (reader.hasMore()) {
            if (reader.isKey("userId")) { //$NON-NLS-1$
                member = new Member(reader.valueString());
            } else {
                reader.skip();
            }
        }
        reader.end();
        return member;
    }

    @Deprecated
    public PeopleConverter(String host) {
        super(PeopleExtension.class, "people", host); //$NON-NLS-1$
    }

    @Deprecated
    @Override
    public void marshal(Object source, HierarchicalStreamWriter writer, MarshallingContext context) {
        PeopleExtension ext = (PeopleExtension) source;
        writer.startNode("leads"); //$NON-NLS-1$
        for (Member member : ext.getLeads()) {
            writer.startNode("lead"); //$NON-NLS-1$
            writeNode(writer, "userId", member.getUserID()); //$NON-NLS-1$
            writeUserLink(writer, USER_RELATION, member.getUserID());
            writer.endNode();
        }
        writer.endNode();
        writer.startNode("members"); //$NON-NLS-1$
        for (Member member : ext.getMembers()) {
            writer.startNode("member"); //$NON-NLS-1$
            writeNode(writer, "userId", member.getUserID()); //$NON-NLS-1$
            writeUserLink(writer, USER_RELATION, member.getUserID());
            writer.endNode();
        }
        writer.endNode();

    }

    @Override
    public Object unmarshal(HierarchicalStreamReader reader, UnmarshallingContext context) {
        return iterateNodes(null, reader, context);
    }

    private PeopleExtension iterateNodes(PeopleExtension ext, HierarchicalStreamReader reader,
            UnmarshallingContext context) {
        if (ext == null) {
            ext = new PeopleExtension();
        }

        while (reader.hasMoreChildren()) {
            reader.moveDown();

            String field = reader.getNodeName();

            if ("leads".equals(field) && reader.hasMoreChildren()) { //$NON-NLS-1$
                iterateNodes(ext, reader, context);
            }
            else if ("members".equals(field) && reader.hasMoreChildren()) { //$NON-NLS-1$
                iterateNodes(ext, reader, context);
            }
            else if ("lead".equals(field) && reader.hasMoreChildren()) { //$NON-NLS-1$
                ext.addLead(unmarshalProjectMember(reader));
            }
            else if ("member".equals(field) && reader.hasMoreChildren()) { //$NON-NLS-1$
                ext.addMember(unmarshalProjectMember(reader));
            }

            reader.moveUp();
        }

        return ext;
    }

    private Member unmarshalProjectMember(HierarchicalStreamReader reader) {
        Member result = null;
        while (reader.hasMoreChildren()) {
            reader.moveDown();
            String field = reader.getNodeName();
            String value = reader.getValue();
            if ("userId".equals(field)) { //$NON-NLS-1$
                result = new Member(value);
            }
            reader.moveUp();
        }
        return result;
    }

    @Override
    public String getApiVersion() {
        return API_VERSION;
    }

    @Override
    public String getNamespace() {
        return NAMESPACE;
    }

    @Override
    public String getXsdFileName() {
        return "extension-people.xsd"; //$NON-NLS-1$
    }

}