// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.search.grouping.vespa;

import com.yahoo.fs4.QueryPacketData;
import com.yahoo.prelude.fastsearch.DocsumDefinitionSet;
import com.yahoo.prelude.fastsearch.FastHit;
import com.yahoo.prelude.fastsearch.GroupingListHit;
import com.yahoo.search.Query;
import com.yahoo.search.Searcher;
import com.yahoo.search.result.Hit;
import com.yahoo.searchlib.aggregation.FS4Hit;
import com.yahoo.searchlib.aggregation.VdsHit;

/**
 * Implementation of the {@link ResultBuilder.HitConverter} interface for {@link GroupingExecutor}.
 *
 * @author Simon Thoresen
 */
class HitConverter implements ResultBuilder.HitConverter {

    private final Searcher searcher;
    private final Query query;

    /**
     * Creates a new instance of this class.
     *
     * @param searcher The searcher that owns this converter.
     * @param query    The query that returned the hits.
     */
    public HitConverter(Searcher searcher, Query query) {
        this.searcher = searcher;
        this.query = query;
    }

    @Override
    public com.yahoo.search.result.Hit toSearchHit(String summaryClass, com.yahoo.searchlib.aggregation.Hit hit) {
        if (hit instanceof FS4Hit) {
            return convertFs4Hit(summaryClass, (FS4Hit)hit);
        } else if (hit instanceof VdsHit) {
            return convertVdsHit(summaryClass, (VdsHit)hit);
        } else {
            throw new UnsupportedOperationException("Hit type '" + hit.getClass().getName() + "' not supported.");
        }
    }

    private Hit convertFs4Hit(String summaryClass, FS4Hit groupHit) {
        FastHit hit = new FastHit();
        hit.setRelevance(groupHit.getRank());
        hit.setGlobalId(groupHit.getGlobalId());
        hit.setPartId(groupHit.getPath());
        hit.setDistributionKey(groupHit.getDistributionKey());
        hit.setFillable();
        hit.setSearcherSpecificMetaData(searcher, summaryClass);

        GroupingListHit hitContext = (GroupingListHit)groupHit.getContext();
        if (hitContext == null)
            throw new NullPointerException("Hit has no context.");
        hit.setSource(hitContext.getSource());
        hit.setQuery(hitContext.getQuery());
        if (hitContext.getQueryPacketData() != null)
            hit.setQueryPacketData(hitContext.getQueryPacketData());
        return hit;
    }

    private Hit convertVdsHit(String summaryClass, VdsHit grpHit) {
        FastHit ret = new FastHit();
        ret.setRelevance(grpHit.getRank());
        if (grpHit.getSummary().getData().length > 0) {
            GroupingListHit ctxHit = (GroupingListHit)grpHit.getContext();
            if (ctxHit == null) {
                throw new NullPointerException("Hit has no context.");
            }
            DocsumDefinitionSet defs = ctxHit.getDocsumDefinitionSet();
            defs.lazyDecode(summaryClass, grpHit.getSummary().getData(), ret);
            ret.setFilled(summaryClass);
            ret.setFilled(query.getPresentation().getSummary());
        }
        return ret;
    }
}

@@ -158,46 +158,8 @@
 a;%0A%0A
-import com.yahoo.fs4.QueryPacketData;%0A
 impo

Optimize imports