UserXMLFormat.java

/* Copyright (c) 2001 - 2007 TOPP - www.openplans.org.  All rights reserved.
 * This code is licensed under the GPL 2.0 license, availible at the root
 * application directory.
 */
package org.geoserver.usermanagement;

import org.geoserver.rest.FreemarkerFormat;
import org.jdom.Attribute;
import org.jdom.Document;
import org.jdom.Element;
import org.jdom.JDOMException;
import org.jdom.input.SAXBuilder;
import org.jdom.xpath.XPath;
import org.restlet.data.MediaType;
import org.restlet.resource.Representation;
import org.restlet.resource.StringRepresentation;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;


public class UserXMLFormat extends FreemarkerFormat {
    public UserXMLFormat(String templ) {
        super(templ, UserXMLFormat.class, MediaType.TEXT_XML);
    }

    public Map readRepresentation(Representation rep) {
        try {
            SAXBuilder builder = new SAXBuilder();
            Document doc = builder.build(rep.getStream());
            Map map = new HashMap();
            XPath passwordPath = XPath.newInstance("/user/password");
            XPath rolePath = XPath.newInstance("/user/roles/role/@name");
            Element passwordObj = (Element) passwordPath.selectSingleNode(doc);
            List roleObjs = rolePath.selectNodes(doc);
            List roles = new ArrayList();

            Iterator it = roleObjs.iterator();

            while (it.hasNext()) {
                Attribute att = (Attribute) it.next();
                roles.add(att.getValue());
            }

            map.put("password", passwordObj.getText());
            map.put("roles", roles);

            return map;
        } catch (Exception e) {
            e.printStackTrace();

            return new HashMap();
        }
    }
}