core/src/main/java/de/ids_mannheim/korap/resource/rewrite/FoundryInject.java

package de.ids_mannheim.korap.resource.rewrite;

import java.util.Collection;

import com.fasterxml.jackson.databind.JsonNode;

import de.ids_mannheim.korap.config.BeanInjectable;
import de.ids_mannheim.korap.config.ContextHolder;
import de.ids_mannheim.korap.config.KustvaktConfiguration;
import de.ids_mannheim.korap.exceptions.KustvaktException;
import de.ids_mannheim.korap.resource.LayerMapper;
import de.ids_mannheim.korap.resource.rewrite.KoralNode.RewriteIdentifier;
import de.ids_mannheim.korap.user.User;
import edu.emory.mathcs.backport.java.util.Collections;

/**
 * @author hanl
 * @date 30/06/2015
 */
public class FoundryInject implements RewriteTask.IterableRewritePath,
        BeanInjectable {

    private Collection userdaos;


    public FoundryInject () {
        this.userdaos = Collections.emptyList();
    }


    @Override
    public KoralNode rewriteQuery (KoralNode node, KustvaktConfiguration config,
            User user) throws KustvaktException {
        LayerMapper mapper;
        // EM: do not use DB
//        if (user != null && !userdaos.isEmpty()) {
//            UserDataDbIface dao = BeansFactory.getTypeFactory()
//                    .getTypeInterfaceBean(userdaos, UserSettings.class);
//            mapper = new LayerMapper(config, dao.get(user));
//        }
//        else
            mapper = new LayerMapper(config);
            
        if (node.get("@type").equals("koral:span")) {
            if (!node.isMissingNode("/wrap")){
                node = node.at("/wrap");
                JsonNode term = rewriteQuery(node, config, user).rawNode();
                node.replaceAt("/wrap", term, new RewriteIdentifier("koral:term", "replace"));
            }
        }
        else if (node.get("@type").equals("koral:term") && !node.has("foundry")) {
            String layer;
            if (node.has("layer"))
                layer = node.get("layer");
            else
                layer = node.get("key");
            String foundry = mapper.findFoundry(layer);
            if (foundry != null)
                node.put("foundry", foundry);
        }
        return node;
    }


    @Override
    public String path () {
        return "query";
    }


    @Override
    public JsonNode rewriteResult (KoralNode node) {
        return null;
    }


    @Override
    public <T extends ContextHolder> void insertBeans (T beans) {
        this.userdaos = beans.getUserDataProviders();
    }
}