org.graylog.plugins.views.search.engine.PositionTrackingQuery Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of graylog2-server Show documentation
Graylog
There is a newer version: 6.1.4
/*
 * Copyright (C) 2020 Graylog, Inc.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the Server Side Public License, version 1,
 * as published by MongoDB, Inc.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * Server Side Public License for more details.
 *
 * You should have received a copy of the Server Side Public License
 * along with this program. If not, see
 * .
 */
package org.graylog.plugins.views.search.engine;

import java.util.Collections;
import java.util.List;
import java.util.Optional;
import java.util.function.Function;
import java.util.stream.Collectors;

public class PositionTrackingQuery {
    private final List fragments;

    public PositionTrackingQuery(List fragments) {
        this.fragments = fragments;
    }

    public static PositionTrackingQuery of(String queryString) {
        return new PositionTrackingQuery(Collections.singletonList(
                new QueryFragment(queryString, queryString, 1, 0, queryString.length())
        ));
    }

    public List getFragments() {
        return fragments;
    }

    public String getOriginalQuery() {
        return concatFragments(QueryFragment::getOriginalContent);
    }

    public String getInterpolatedQuery() {
        return concatFragments(QueryFragment::getInterpolatedContent);
    }

    private String concatFragments(Function supplier) {
        int line = 1;
        final StringBuilder stringBuilder = new StringBuilder();
        for (QueryFragment fragment : fragments) {
            if (line != fragment.getLine()) {
                stringBuilder.append("\n");
                line = fragment.getLine();
            }
            stringBuilder.append(supplier.apply(fragment));
        }
        return stringBuilder.toString();
    }

    public QueryPosition backtrackPosition(QueryPosition interpolated) {
        final List lineFragments = this.fragments.stream().filter(f -> f.getLine() == interpolated.beginLine()).collect(Collectors.toList());
        int linePosition = 0;
        for (QueryFragment fragment : lineFragments) {
            if (interpolated.beginColumn()  >= linePosition && interpolated.endColumn() <= linePosition + fragment.originalLength()) {
                if (fragment.isInterpolated()) { // we can't map 1:1 interpolated and original positions, let's use the whole fragment
                    return QueryPosition.create(fragment.getLine(), fragment.getOriginalBeginColumn(), fragment.getLine(), fragment.getOriginalEndColumn());
                } else { // we can map exactly the positions
                    final int offsetStart = interpolated.beginColumn() - linePosition;
                    final int offsetEnd = linePosition + fragment.interpolatedLength() - interpolated.endColumn();
                    final int globalOriginalStart = fragment.getOriginalBeginColumn() + offsetStart;
                    final int globalOriginalEnd = fragment.getOriginalBeginColumn() + fragment.originalLength() - offsetEnd;
                    return QueryPosition.create(interpolated.beginLine(), globalOriginalStart, interpolated.endLine(), globalOriginalEnd);
                }
            }
            linePosition = linePosition + fragment.interpolatedLength();
        }
        throw new IllegalArgumentException("Failed to backtrack position: " + interpolated);
    }
}