IndexFromXWiki.java

package org.jbehave.core.io.rest.xwiki;

import static java.text.MessageFormat.format;

import java.util.Collection;
import java.util.HashMap;
import java.util.Map;

import com.google.gson.Gson;
import com.google.gson.JsonParser;
import com.google.gson.reflect.TypeToken;

import org.jbehave.core.io.rest.IndexWithBreadcrumbs;
import org.jbehave.core.io.rest.RESTClient;
import org.jbehave.core.io.rest.RESTClient.Type;
import org.jbehave.core.io.rest.Resource;
import org.jbehave.core.io.rest.ResourceNameResolver;

/**
 * Indexes resources from XWiki using the REST API
 */
public class IndexFromXWiki extends IndexWithBreadcrumbs {

    private static final String INDEX_URI = "{0}?media=json";
    private static final String PAGE_URI = "{0}/{1}";

    public IndexFromXWiki() {
        this(null, null);
    }

    public IndexFromXWiki(String username, String password) {
        this(username, password, new ToLowerCase());
    }

    public IndexFromXWiki(String username, String password, ResourceNameResolver nameResolver) {
        super(new RESTClient(Type.JSON, username, password), nameResolver);
    }

    @Override
    protected Map<String, Resource> createIndexFromEntity(String rootURI, String entity) {
        Collection<Page> pages = parse(entity);
        Map<String, Resource> index = new HashMap<>();
        for (Page page : pages) {
            String parentName = (page.parent != null ? resolveName(page.parent) : null);
            String uri = format(PAGE_URI, rootURI, page.name);
            Resource resource = new Resource(uri, resolveName(page.name), parentName);
            index.put(resource.getName(), resource);
        }
        return index;
    }

    @Override
    protected String uri(String rootPath) {
        return format(INDEX_URI, rootPath);
    }

    private Collection<Page> parse(String entity) {
        Gson gson = new Gson();
        return gson.fromJson(
                jsonMember(entity, "pageSummaries"),
                new TypeToken<Collection<Page>>() {
                }.getType());
    }

    private String jsonMember(String entity, String memberName) {
        return new JsonParser().parse(entity).getAsJsonObject().get(memberName)
                .toString();
    }

    private static class Page {
        private String name;
        private String parent;
    }

}